Update app.py

app.py

@@ -30,7 +30,13 @@ graph = StateGraph(AgentState)
 
 def invoke_model(state):
     question = state['messages'][-1]
-    return {"messages": [model_with_tools.invoke(question)]}
+    response = model_with_tools.invoke(question)
+    # If no tool calls are found, return the raw response content
+    if not response.additional_kwargs.get("tool_calls", []):
+        return {"messages": [response.content]}
+    # Otherwise, return the response object as before
+    return {"messages": [response]}
+
 
 graph.add_node("agent", invoke_model)