Update app.py

app.py

@@ -48,16 +48,16 @@ graph.add_node("tools", tool_node)
 #    Wrap the user_input into state["messages"]
 graph.add_edge(
     START,
-    "agent",
-    transition=lambda state, user_input: {"messages": [user_input]},
+    "agent"
+    # transition=lambda state, user_input: {"messages": [user_input]},
 )
 
 # Edge C: "tools" → "agent"
 #    Whatever string the tool returns becomes the next prompt to the LLM
 graph.add_edge(
     "tools",
-    "agent",
-    transition=lambda state, tool_output: tool_output,
+    "agent"
+    # transition=lambda state, tool_output: tool_output,
 )
 
 # 7) Use add_conditional_edges out of "agent" instead of two separate edges
@@ -77,7 +77,7 @@ graph.add_conditional_edges(
     {
         "tools": "tools",  # if route_agent(...) == "tools", go to the "tools" node
         "final": END,      # if route_agent(...) == "final", transition to END
-    },
+    }
 )
 
 def respond_to_input(user_input: str) -> str: