From 36a641b5f4c3535611e44105ac8e7cf4cfe5a611 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Thu, 23 Nov 2023 23:44:13 -0800
Subject: [PATCH] requirements.txt clean up

---
 .pre-commit-config.yaml               |  2 +-
 requirements.txt                      | 10 +---------
 swarms/structs/flow.py                |  6 +++---
 swarms/structs/sequential_workflow.py |  1 +
 4 files changed, 6 insertions(+), 13 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index ae0a4fc0c..0c9367058 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -7,7 +7,7 @@ repos:
     rev: 'v0.0.255'
     hooks:
       - id: ruff
-        args: [--fix]
+        args: [----unsafe-fixes]
   - repo: https://github.com/nbQA-dev/nbQA
     rev: 1.6.3
     hooks:
diff --git a/requirements.txt b/requirements.txt
index d75d5d1a2..067356df7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,9 +3,6 @@ transformers
 pandas
 langchain
 nest_asyncio
-pegasusx
-google-generativeai 
-EdgeGPT
 langchain-experimental
 playwright
 wget==3.2
@@ -24,9 +21,7 @@ soundfile
 huggingface-hub
 google-generativeai
 sentencepiece
-duckduckgo-search
 PyPDF2
-agent-protocol
 accelerate
 chromadb
 tiktoken
@@ -54,7 +49,6 @@ openai
 opencv-python
 prettytable
 safetensors
-streamlit
 test-tube
 timm
 torchmetrics
@@ -62,7 +56,6 @@ webdataset
 marshmallow
 yapf
 autopep8
-dalle3
 cohere
 torchvision
 rich
@@ -71,5 +64,4 @@ rich
 mkdocs
 mkdocs-material
 mkdocs-glightbox
-
-pre-commit
+pre-commit
\ No newline at end of file
diff --git a/swarms/structs/flow.py b/swarms/structs/flow.py
index 166d619e3..18a141a3f 100644
--- a/swarms/structs/flow.py
+++ b/swarms/structs/flow.py
@@ -135,9 +135,9 @@ def parse_done_token(response: str) -> bool:
 
 class Flow:
     """
-    Flow is a chain like structure from langchain that provides the autonomy to language models
-    to generate sequential responses.
-
+    Flow is the structure that provides autonomy to any llm in a reliable and effective fashion.
+    The flow structure is designed to be used with any llm and provides the following features:
+    
     Features:
     * Interactive, AI generates, then user input
     * Message history and performance history fed -> into context -> truncate if too long
diff --git a/swarms/structs/sequential_workflow.py b/swarms/structs/sequential_workflow.py
index 1d7f411db..753ada151 100644
--- a/swarms/structs/sequential_workflow.py
+++ b/swarms/structs/sequential_workflow.py
@@ -149,6 +149,7 @@ def get_task_results(self) -> Dict[str, Any]:
         return {task.description: task.result for task in self.tasks}
 
     def remove_task(self, task_description: str) -> None:
+        """Remove tasks from sequential workflow"""
         self.tasks = [
             task for task in self.tasks if task.description != task_description
         ]