error handling for bossnode

swarms/agents/boss/boss_agent.py

@@ -1,4 +1,5 @@
 from swarms.tools.agent_tools import *
+from pydantic import ValidationError
 
 # ---------- Boss Node ----------
 class BossNode:
@@ -9,14 +10,31 @@ class BossNode:
         self.verbose = verbose
         self.max_iterations = max_iterations
 
+        try:
+            # Ensure llm is a dictionary before passing it to BabyAGI
+            assert isinstance(llm, dict), "llm should be a dictionary."
+
             self.baby_agi = BabyAGI.from_llm(
                 llm=self.llm,
                 vectorstore=self.vectorstore,
-            task_execution_chain=self.task_execution_chain
+                task_execution_chain=self.task_execution_chain,
+                verbose=self.verbose,
+                max_iterations=self.max_iterations,
             )
+        except ValidationError as e:
+            print(f"Validation Error while initializing BabyAGI: {e}")
+        except Exception as e:
+            print(f"Unexpected Error while initializing BabyAGI: {e}")
 
     def create_task(self, objective):
-        return {"objective": objective}
+        try:
+            task = {"objective": objective}
+            return task
+        except Exception as e:
+            print(f"Unexpected Error while creating a task: {e}")
 
     def execute_task(self, task):
+        try:
             self.baby_agi(task)
+        except Exception as e:
+            print(f"Unexpected Error while executing a task: {e}")