error handlingg

swarms/workers/autobot.py

@@ -32,9 +32,13 @@ class AutoBot:
                  temperature=0.5):
         self.openai_api_key = openai_api_key
         self.temperature = temperature
+        
+        try:
             self.llm = ChatOpenAI(model_name=model_name, 
                                 openai_api_key=self.openai_api_key, 
                                 temperature=self.temperature)
+        except Exception as error:
+            raise RuntimeError(f"Error Initializing ChatOpenAI: {error}")    
         
         self.ai_name = ai_name
         self.ai_role = ai_role
@@ -55,12 +59,17 @@ class AutoBot:
         ]
 
     def setup_memory(self):
+        try:
             embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key)
             embedding_size = 1536
             index = faiss.IndexFlatL2(embedding_size)
             self.vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
+        except Exception as error:
+            raise RuntimeError(f"Error setting up memory perhaps try try tuning the embedding size: {error}")
+        
     
     def setup_agent(self):
+        try: 
             self.agent = AutoGPT.from_llm_and_tools(
                 ai_name=self.ai_name,
                 ai_role=self.ai_role,
@@ -69,7 +78,12 @@ class AutoBot:
                 memory=self.vectorstore.as_retriever(search_kwargs={"k": 8}),
             )
         
+        except Exception as error:
+            raise RuntimeError(f"Error setting up agent: {error}")
+    
     def run(self, task):
+        try:
             result = self.agent.run([task])
             return result
-    
+        except Exception as error:
+            raise RuntimeError(f"Error while running agent: {error}")