diff --git a/swarms/agents/boss/babyagi_agent.py b/swarms/agents/boss/babyagi_agent.py index ad1a4772..2e4fd067 100644 --- a/swarms/agents/boss/babyagi_agent.py +++ b/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() diff --git a/swarms/agents/workers/auto_agent.py b/swarms/agents/workers/auto_worker.py similarity index 97% rename from swarms/agents/workers/auto_agent.py rename to swarms/agents/workers/auto_worker.py index 199d0d08..98104ea8 100644 --- a/swarms/agents/workers/auto_agent.py +++ b/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 diff --git a/swarms/agents/workers/metaprompt_agent.py b/swarms/agents/workers/metaprompt_worker.py similarity index 100% rename from swarms/agents/workers/metaprompt_agent.py rename to swarms/agents/workers/metaprompt_worker.py diff --git a/swarms/agents/workers/multi_modal_agents/__init__.py b/swarms/agents/workers/multi_modal_agents/__init__.py deleted file mode 100644 index 57888f93..00000000 --- a/swarms/agents/workers/multi_modal_agents/__init__.py +++ /dev/null @@ -1 +0,0 @@ -from swarms.agents.workers.multi_modal_agents.omni_agent.omni_agent import chat_huggingface \ No newline at end of file diff --git a/swarms/agents/workers/multi_modal_workers/__init__.py b/swarms/agents/workers/multi_modal_workers/__init__.py new file mode 100644 index 00000000..766f5e26 --- /dev/null +++ b/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 \ No newline at end of file diff --git a/swarms/agents/workers/multi_modal_agents/multi_modal_agent.py b/swarms/agents/workers/multi_modal_workers/multi_modal_agent.py similarity index 100% rename from swarms/agents/workers/multi_modal_agents/multi_modal_agent.py rename to swarms/agents/workers/multi_modal_workers/multi_modal_agent.py diff --git a/swarms/agents/workers/multi_modal_agents/omni_agent/__init__.py b/swarms/agents/workers/multi_modal_workers/omni_agent/__init__.py similarity index 100% rename from swarms/agents/workers/multi_modal_agents/omni_agent/__init__.py rename to swarms/agents/workers/multi_modal_workers/omni_agent/__init__.py diff --git a/swarms/agents/workers/multi_modal_agents/omni_agent/get_token_ids.py b/swarms/agents/workers/multi_modal_workers/omni_agent/get_token_ids.py similarity index 100% rename from swarms/agents/workers/multi_modal_agents/omni_agent/get_token_ids.py rename to swarms/agents/workers/multi_modal_workers/omni_agent/get_token_ids.py diff --git a/swarms/agents/workers/multi_modal_agents/omni_agent/model_server.py b/swarms/agents/workers/multi_modal_workers/omni_agent/model_server.py similarity index 100% rename from swarms/agents/workers/multi_modal_agents/omni_agent/model_server.py rename to swarms/agents/workers/multi_modal_workers/omni_agent/model_server.py diff --git a/swarms/agents/workers/multi_modal_agents/omni_agent/omni_chat.py b/swarms/agents/workers/multi_modal_workers/omni_agent/omni_chat.py similarity index 99% rename from swarms/agents/workers/multi_modal_agents/omni_agent/omni_chat.py rename to swarms/agents/workers/multi_modal_workers/omni_agent/omni_chat.py index d9351e56..3fd081e5 100644 --- a/swarms/agents/workers/multi_modal_agents/omni_agent/omni_chat.py +++ b/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 diff --git a/swarms/agents/workers/omni_agent.py b/swarms/agents/workers/omni_worker.py similarity index 93% rename from swarms/agents/workers/omni_agent.py rename to swarms/agents/workers/omni_worker.py index fc19346c..aec411d4 100644 --- a/swarms/agents/workers/omni_agent.py +++ b/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): diff --git a/swarms/agents/workers/visual_agent.py b/swarms/agents/workers/visual_worker.py similarity index 83% rename from swarms/agents/workers/visual_agent.py rename to swarms/agents/workers/visual_worker.py index a87d7d28..c4a74e32 100644 --- a/swarms/agents/workers/visual_agent.py +++ b/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): diff --git a/swarms/agents/workers/worker_agent.py b/swarms/agents/workers/worker.py similarity index 100% rename from swarms/agents/workers/worker_agent.py rename to swarms/agents/workers/worker.py diff --git a/swarms/swarms.py b/swarms/swarms.py index 92c74ddf..70a5a1a1 100644 --- a/swarms/swarms.py +++ b/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