Update app.py

app.py

@@ -64,11 +64,10 @@ class AdvancedAgent:
         sys_msg = SystemMessage(content=system_prompt)
         llm_with_tools = llm.bind_tools(tools)
 
-        # Assistant node must be able to access llm_with_tools.
         def assistant(state: MessagesState):
             """Assistant node that processes messages and generates responses"""
             messages = state["messages"]
-            response = llm_with_tools.invoke(messages)
+            response = llm_with_tools.invoke(messages)  # <-- messages, not dict!
             return {"messages": messages + [response]}
 
         tools_node = ToolNode(tools)
@@ -77,13 +76,10 @@ class AdvancedAgent:
         builder.add_node("assistant", assistant)
         builder.add_node("tools", tools_node)
 
-        # The correct flow is: assistant -> tools (if tool needed) -> assistant (after tool used) -> END (if done)
         builder.set_entry_point("assistant")
         builder.add_edge("assistant", "tools")
         builder.add_edge("tools", "assistant")
         builder.add_edge("assistant", END)
-
-        # Conditional: if tools_condition is met, go to tools, else END
         builder.add_conditional_edges(
             "assistant",
             tools_condition,
@@ -99,7 +95,7 @@ class AdvancedAgent:
             HumanMessage(content=question)
         ]
         try:
-            result = self.graph.invoke({"messages": messages})
+            result = self.graph.invoke(messages)  # <-- messages, not dict!
             final_messages = result["messages"]
             ai_messages = [msg for msg in final_messages if isinstance(msg, AIMessage)]
             if not ai_messages:
@@ -262,4 +258,4 @@ if __name__ == "__main__":
         print("ℹ️  SPACE_ID environment variable not found (running locally?). Repo URL cannot be determined.")
     print("-"*(60 + len(" App Starting ")) + "\n")
     print("Launching Gradio Interface for Advanced Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)