Update workflows/sales_analysis.py

workflows/sales_analysis.py

@@ -118,16 +118,45 @@ class SalesAnalysisWorkflow:
             # Add error handling
             workflow.add_node("handle_error", self.handle_error_node)
             
-            # Add conditional edges for error handling
-            # Using add_conditional_edges API instead of add_edge_condition
-            def route_on_error(state):
-                return "handle_error" if state["status"] == "error" else None
+            # Add conditional edges for error handling using LangGraph's newer API
+            def route_planning(state):
+                if state["status"] == "error":
+                    return "handle_error"
+                return "data_collection"
                 
-            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)
+            def route_data_collection(state):
+                if state["status"] == "error":
+                    return "handle_error"
+                return "analysis"
+                
+            def route_analysis(state):
+                if state["status"] == "error":
+                    return "handle_error"
+                return "validation"
+                
+            def route_validation(state):
+                if state["status"] == "error":
+                    return "handle_error"
+                return "insights"
+                
+            def route_insights(state):
+                if state["status"] == "error":
+                    return "handle_error"
+                return END
+                
+            # Replace linear edges with conditional ones
+            workflow.remove_edge("planning", "data_collection")
+            workflow.remove_edge("data_collection", "analysis")
+            workflow.remove_edge("analysis", "validation")
+            workflow.remove_edge("validation", "insights")
+            workflow.remove_edge("insights", END)
+            
+            # Add the conditional routing
+            workflow.add_conditional_edges("planning", route_planning)
+            workflow.add_conditional_edges("data_collection", route_data_collection)
+            workflow.add_conditional_edges("analysis", route_analysis)
+            workflow.add_conditional_edges("validation", route_validation)
+            workflow.add_conditional_edges("insights", route_insights)
             workflow.add_edge("handle_error", END)
             
             # Set the entry point