Update workflows/sales_analysis.py

workflows/sales_analysis.py

@@ -92,32 +92,15 @@ class SalesAnalysisWorkflow:
         workflow.add_node("handle_error", self.handle_error_node)
         
         # Add conditional edges for error handling
-        # Using add_edge_condition instead of add_edge with condition parameter
-        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 the newer add_conditional_edges method
+        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