Update workflows/sales_analysis.py

workflows/sales_analysis.py

@@ -118,45 +118,15 @@ class SalesAnalysisWorkflow:
             # Add error handling
             workflow.add_node("handle_error", self.handle_error_node)
             
-            # 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"
-                
-            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)
+            # Define a simple linear workflow without error handling for now
+            # This should work with any version of LangGraph
+            workflow.add_edge("planning", "data_collection")
+            workflow.add_edge("data_collection", "analysis")
+            workflow.add_edge("analysis", "validation")
+            workflow.add_edge("validation", "insights")
+            workflow.add_edge("insights", END)
+            
+            # We'll handle errors within each node function instead of using graph branching
             workflow.add_edge("handle_error", END)
             
             # Set the entry point