clean up agents

swarms/workers/developer_agent.py

@@ -1,25 +1,21 @@
+import argparse
+import asyncio
 import enum
+import logging
 import os
-from pathlib import Path
+import re
+import shutil
 import sys
 import time
-import shutil
-import argparse
-import asyncio
-import re
-from typing import List, Optional, Callable, Any
+from pathlib import Path
+from typing import Any, Callable, List, Optional
 
 import openai
+from agent_protocol import Agent, Step, Task
 from openai_function_call import openai_function
-from tenacity import retry, stop_after_attempt, wait_random_exponential
-import logging
-
-from smol_dev.prompts import plan, specify_file_paths, generate_code_sync
+from smol_dev.prompts import generate_code_sync, plan, specify_file_paths
 from smol_dev.utils import generate_folder, write_file
-
-from agent_protocol import Agent, Step, Task
-
-
+from tenacity import retry, stop_after_attempt, wait_random_exponential
 
 
 class DeveloperAgent:

swarms/workers/multi_modal_worker.py

@@ -1,11 +1,7 @@
 # from swarms.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent
 from swarms.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent
-from langchain.tools import BaseTool
-
-class MultiModalVisualAgentTool(BaseTool):
-    name = "multi_visual_agent"
-    description = "Multi-Modal Visual agent tool"
 
+class MultiModalVisualAgentTool:
     def __init__(self, agent: MultiModalVisualAgent):
         self.agent = agent
     
@@ -13,6 +9,3 @@ class MultiModalVisualAgentTool(BaseTool):
         #run the multi-modal visual agent with the give task
         return self.agent.run_text(text)
 
-
-multimodal_agent = MultiModalVisualAgent()
-multimodal_agent_tool = MultiModalVisualAgentTool(multimodal_agent)