Update workflows/sales_analysis.py

workflows/sales_analysis.py

@@ -92,30 +92,31 @@ class SalesAnalysisWorkflow:
         workflow.add_node("handle_error", self.handle_error_node)
         
         # Add conditional edges for error handling
-        workflow.add_edge(
+        # Using add_edge_condition instead of add_edge with condition parameter
+        workflow.add_edge_condition(
             "planning", 
             "handle_error", 
-            condition=lambda x: x["status"] == "error"
+            lambda x: x["status"] == "error"
         )
-        workflow.add_edge(
+        workflow.add_edge_condition(
             "data_collection", 
             "handle_error", 
-            condition=lambda x: x["status"] == "error"
+            lambda x: x["status"] == "error"
         )
-        workflow.add_edge(
+        workflow.add_edge_condition(
             "analysis", 
             "handle_error", 
-            condition=lambda x: x["status"] == "error"
+            lambda x: x["status"] == "error"
         )
-        workflow.add_edge(
+        workflow.add_edge_condition(
             "validation", 
             "handle_error", 
-            condition=lambda x: x["status"] == "error"
+            lambda x: x["status"] == "error"
         )
-        workflow.add_edge(
+        workflow.add_edge_condition(
             "insights", 
             "handle_error", 
-            condition=lambda x: x["status"] == "error"
+            lambda x: x["status"] == "error"
         )
         workflow.add_edge("handle_error", END)