Update app.py

app.py

@@ -83,14 +83,14 @@ def route_agent(state: AgentState, agent_out):
     return "final"
 
 graph.add_conditional_edges(
-    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
+    "agent",
+    route_agent,
+    {
+        "tools": "tools",  # if route_agent(...) == "tools", go to node "tools"
+        "final": END       # if route_agent(...) == "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:
     print(f"Reached: User input: {user_input}")