Update app.py

app.py

@@ -72,12 +72,13 @@ def route_agent(state: AgentState, agent_out):
     return "final"
 
 graph.add_conditional_edges(
-    "agent",
-    route_agent,
-    {
-        "tools": "tools",  # if route_agent(...) == "tools", go to the "tools" node
-        "final": END,      # if route_agent(...) == "final", transition to END
+    source="agent",
+    path=route_agent,
+    path_map={          # This keys must match route_agent(...) returns
+        "tools": "tools",  # if route_agent returns "tools", go to node "tools"
+        "final": END     # if route_agent returns "final", go to END
     }
+    # (We omit `then=` here, so there's no post-processing step.)
 )
 compiled_graph = graph.compile()
 def respond_to_input(user_input: str) -> str: