re-architecture

swarms/agents/boss/babyagi_agent.py

@@ -17,7 +17,7 @@ from langchain.agents import ZeroShotAgent, Tool, AgentExecutor
 from langchain import OpenAI, SerpAPIWrapper, LLMChain
 
 
-from swarms.agents.workers.auto_agent import agent
+from swarms.agents.workers.auto_worker import agent
 
 # Define your embedding model
 embeddings_model = OpenAIEmbeddings()

swarms/agents/workers/auto_worker.py

@@ -32,7 +32,7 @@ from langchain.embeddings import OpenAIEmbeddings
 
 from langchain.tools.human.tool import HumanInputRun
 # from swarms.agents.workers.auto_agent import 
-from swarms.agents.workers.visual_agent import multimodal_agent_tool
+from swarms.agents.workers.visual_worker import multimodal_agent_tool
 from swarms.tools.main import Terminal, CodeWriter, CodeEditor, process_csv, WebpageQATool
 
 

swarms/agents/workers/multi_modal_agents/__init__.py

@@ -1 +0,0 @@
-from swarms.agents.workers.multi_modal_agents.omni_agent.omni_agent import chat_huggingface

swarms/agents/workers/multi_modal_workers/__init__.py

@@ -0,0 +1 @@
+from swarms.agents.workers.multi_modal_workers.omni_agent.omni_agent import chat_huggingface

swarms/agents/workers/multi_modal_workers/omni_agent/omni_chat.py

@@ -22,7 +22,7 @@ import flask
 from flask import request, jsonify
 import waitress
 from flask_cors import CORS, cross_origin
-from swarms.agents.workers.multi_modal_agents.omni_agent.get_token_ids import get_token_ids_for_task_parsing, get_token_ids_for_choose_model, count_tokens, get_max_context_length
+from swarms.agents.workers.multi_modal_workers.omni_agent.get_token_ids import get_token_ids_for_task_parsing, get_token_ids_for_choose_model, count_tokens, get_max_context_length
 from huggingface_hub.inference_api import InferenceApi
 from huggingface_hub.inference_api import ALL_TASKS
 

swarms/agents/workers/omni_worker.py

@@ -1,6 +1,6 @@
 #boss node -> worker agent -> omni agent [worker of the worker]
 from langchain.tools import tool
-from swarms.agents.workers.multi_modal_agents.omni_agent.omni_chat import chat_huggingface
+from swarms.agents.workers.multi_modal_workers.omni_agent.omni_chat import chat_huggingface
 
 class OmniWorkerAgent:
     def __init__(self, api_key, api_endpoint, api_type):

swarms/agents/workers/visual_worker.py

@@ -1,4 +1,4 @@
-from swarms.agents.workers.multi_modal_agents.multi_modal_agent import MultiModalVisualAgent
+from swarms.agents.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent
 from langchain.tools import BaseTool
 
 class MultiModalVisualAgentTool(BaseTool):

swarms/swarms.py

@@ -1,7 +1,7 @@
 from swarms.tools.agent_tools import *
-from swarms.agents.workers.worker_agent import WorkerNode
+from swarms.agents.workers.worker import WorkerNode
 from swarms.agents.boss.boss_agent import BossNode
-from swarms.agents.workers.omni_agent import OmniWorkerAgent
+from swarms.agents.workers.omni_worker import OmniWorkerAgent