clean up accidentally switched workernames

Former-commit-id: 5194209802d9ff7a2b3d4c71dd860bc2abbefe37

swarms/agents/workers/worker_agent.py

@@ -17,6 +17,8 @@ class WorkerNode:
     """Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on"""
 
     def __init__(self, llm, tools, vectorstore):
+        if not llm or not tools or not vectorstore:
+            raise ValueError("llm, tools, and vectorstore cannot be None")
         self.llm = llm
         self.tools = tools
         self.vectorstore = vectorstore
@@ -24,6 +26,8 @@ class WorkerNode:
 
     def create_agent(self, ai_name, ai_role, human_in_the_loop, search_kwargs):
         logging.info("Creating agent in WorkerNode")
+        try:
+                
             self.agent = AutoGPT.from_llm_and_tools(
                 ai_name=ai_name,
                 ai_role=ai_role,
@@ -35,7 +39,13 @@ class WorkerNode:
             )
             self.agent.chain.verbose = True
 
+        except Exception as e:
+            logging.error(f"Error while creating agent: {str(e)}")
+            raise e
+
     def add_tool(self, tool: Tool):
+        if not isinstance(tool, Tool):
+            raise TypeError("Tool must be an instance of Tool")
         self.tools.append(tool)
 
     def run(self, prompt: str) -> str:
@@ -45,25 +55,39 @@ class WorkerNode:
         if not prompt:
             raise ValueError("Prompt is empty")
         
+        try:
+                
+
             self.agent.run([f"{prompt}"])
             return "Task completed by WorkerNode"
+        except Exception as e:
+            logging.error(f"While running the agent: {str(e)}")
+            raise e
         
     
-worker_tool = Tool(
+
-    name="WorkerNode AI Agent",
+# worker_tool = Tool(
-    func=WorkerNode.run,
+#     name="WorkerNode AI Agent",
-    description="Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on"
+#     func=WorkerNode.run,
-)
+#     description="Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on"
+# )
 
 
 class WorkerNodeInitializer:
     def __init__(self, openai_api_key):
+        if not openai_api_key:
+            raise ValueError("openai_api_key cannot be None")
         self.openai_api_key = openai_api_key
 
     def initialize_llm(self, llm_class, temperature=0.5):
+        if not llm_class:
+            raise ValueError("llm_class cannot be None")
         return llm_class(openai_api_key=self.openai_api_key, temperature=temperature)
 
     def initialize_tools(self, llm_class):
+        if not llm_class:
+            raise ValueError("llm_class cannot be none")
+        logging.info('Creating WorkerNode')
         llm = self.initialize_llm(llm_class)
         web_search = DuckDuckGoSearchRun()
         tools = [
@@ -82,6 +106,8 @@ class WorkerNodeInitializer:
         return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
 
     def create_worker_node(self, llm_class=ChatOpenAI):
+        if not llm_class:
+            raise ValueError("llm_class cannot be None")
         worker_tools = self.initialize_tools(llm_class)
         vectorstore = self.initialize_vectorstore()
         worker_node = WorkerNode(llm=self.initialize_llm(llm_class), tools=worker_tools, vectorstore=vectorstore)
@@ -89,6 +115,8 @@ class WorkerNodeInitializer:
         return worker_node
 
 def worker_node(openai_api_key):
+    if not openai_api_key:
+        raise ValueError("openai_api_key cannot be none")
     initializer = WorkerNodeInitializer(openai_api_key)
     worker_node = initializer.create_worker_node()
     return worker_node