requirements.txt clean up

.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:

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,15 +49,13 @@ openai
 opencv-python
 prettytable
 safetensors
-streamlit
 test-tube
 timm
 torchmetrics
 webdataset
 marshmallow
 yapf
 autopep8
-dalle3
 cohere
 torchvision
 rich
@@ -71,5 +64,4 @@ rich
 mkdocs
 mkdocs-material
 mkdocs-glightbox
-
-pre-commit
+pre-commit

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

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
         ]