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
 
-        self.baby_agi = BabyAGI.from_llm(
-            llm=self.llm,
-            vectorstore=self.vectorstore,
-            task_execution_chain=self.task_execution_chain
-        )
+        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,
+                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):
-        self.baby_agi(task)
+        try:
+            self.baby_agi(task)
+        except Exception as e:
+            print(f"Unexpected Error while executing a task: {e}")