extensive logging for swarms and workernode

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,

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,6 +61,7 @@ class Swarms:
         return BossNode(self.openai_api_key, llm, vectorstore, agent_executor, verbose=True, max_iterations=5)
 
     def run_swarms(self, 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"
@@ -70,6 +73,9 @@ class Swarms:
             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