diff --git a/swarms/agents/workers/worker.py b/swarms/agents/workers/worker.py index 9748a96e..3cf4d653 100644 --- a/swarms/agents/workers/worker.py +++ b/swarms/agents/workers/worker.py @@ -10,6 +10,9 @@ from typing import List, Any # from langchain.memory.chat_message_histories import FileChatMessageHistory +import logging + +logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s') class WorkerNode(BaseTool): name = "WorkerNode" @@ -24,13 +27,16 @@ class WorkerNode(BaseTool): def __init__(self, llm, tools, vectorstore): + logging.info("Initializing WorkerNode") super().__init__() self.llm = llm self.tools = tools self.vectorstore = vectorstore self.agent = None + def create_agent(self, ai_name, ai_role, human_in_the_loop, search_kwargs): + logging.info("Creating agent in WorkerNode") self.agent = AutoGPT.from_llm_and_tools( ai_name=ai_name, ai_role=ai_role, diff --git a/swarms/swarms.py b/swarms/swarms.py index accee8ff..753ba1db 100644 --- a/swarms/swarms.py +++ b/swarms/swarms.py @@ -3,7 +3,9 @@ from swarms.agents.workers.worker import WorkerNode from swarms.agents.boss.boss_agent import BossNode # from swarms.agents.workers.omni_worker import OmniWorkerAgent +import logging +logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s') class Swarms: def __init__(self, openai_api_key): @@ -59,17 +61,21 @@ class Swarms: return BossNode(self.openai_api_key, llm, vectorstore, agent_executor, verbose=True, max_iterations=5) def run_swarms(self, objective): - # Run the swarm with the given objective - worker_tools = self.initialize_tools(OpenAI) - assert worker_tools is not None, "worker_tools is not initialized" - - vectorstore = self.initialize_vectorstore() - worker_node = self.initialize_worker_node(worker_tools, vectorstore) - boss_node = self.initialize_boss_node(vectorstore, worker_node) - - task = boss_node.create_task(objective) - boss_node.execute_task(task) - worker_node.run_agent(objective) + try: + # Run the swarm with the given objective + worker_tools = self.initialize_tools(OpenAI) + assert worker_tools is not None, "worker_tools is not initialized" + + vectorstore = self.initialize_vectorstore() + worker_node = self.initialize_worker_node(worker_tools, vectorstore) + boss_node = self.initialize_boss_node(vectorstore, worker_node) + + task = boss_node.create_task(objective) + boss_node.execute_task(task) + worker_node.run_agent(objective) + except Exception as e: + logging.error(f"An error occurred in run_swarms: {e}") + raise