Update workflows/sales_analysis.py

workflows/sales_analysis.py

@@ -119,31 +119,15 @@ class SalesAnalysisWorkflow:
             workflow.add_node("handle_error", self.handle_error_node)
             
             # Add conditional edges for error handling
-            workflow.add_edge_condition(
-                "planning", 
-                "handle_error", 
-                lambda x: x["status"] == "error"
-            )
-            workflow.add_edge_condition(
-                "data_collection", 
-                "handle_error", 
-                lambda x: x["status"] == "error"
-            )
-            workflow.add_edge_condition(
-                "analysis", 
-                "handle_error", 
-                lambda x: x["status"] == "error"
-            )
-            workflow.add_edge_condition(
-                "validation", 
-                "handle_error", 
-                lambda x: x["status"] == "error"
-            )
-            workflow.add_edge_condition(
-                "insights", 
-                "handle_error", 
-                lambda x: x["status"] == "error"
-            )
+            # Using add_conditional_edges API instead of add_edge_condition
+            def route_on_error(state):
+                return "handle_error" if state["status"] == "error" else None
+                
+            workflow.add_conditional_edges("planning", route_on_error)
+            workflow.add_conditional_edges("data_collection", route_on_error)
+            workflow.add_conditional_edges("analysis", route_on_error)
+            workflow.add_conditional_edges("validation", route_on_error)
+            workflow.add_conditional_edges("insights", route_on_error)
             workflow.add_edge("handle_error", END)
             
             # Set the entry point