diff --git a/swarms/workers/developer_agent.py b/swarms/workers/developer_agent.py index 0ac8fd9b..3e6550cc 100644 --- a/swarms/workers/developer_agent.py +++ b/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: diff --git a/swarms/workers/multi_modal_worker.py b/swarms/workers/multi_modal_worker.py index fc6d36b9..ce2fe0f7 100644 --- a/swarms/workers/multi_modal_worker.py +++ b/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) \ No newline at end of file