Update app.py

app.py

@@ -56,55 +56,77 @@ compiled_graph = graph.compile()
 
 # 3) The corrected respond_to_input:
 def respond_to_input(user_input: str) -> str:
-    # 1) The tightened system prompt
+    # A) Describe your tools exactly—no extra text allowed around JSON
     system_msg = SystemMessage(
         content=(
             "You are an assistant with access to exactly these tools:\n"
             "  1) web_search(query:str)\n"
             "  2) parse_excel(path:str,sheet_name:str)\n"
             "  3) ocr_image(path:str)\n\n"
-            "⚠️ **IMPORTANT** ⚠️: If (and only if) you need to call one of these tools, "
-            "output exactly one JSON object and nothing else. For example:\n"
+            "⚠️ If (and only if) you need to call a tool, output exactly one JSON object "
+            "and nothing else. For example:\n"
             "```json\n"
             '{"tool":"web_search","query":"Mercedes Sosa albums 2000-2009"}\n'
             "```\n"
-            "That JSON must start at the very first character of your response and end at the very last character—"
-            "no quotes, no code fences, no extra explanation.\n\n"
-            "If you do NOT need any tool, simply reply with your final answer as plain text (no JSON)."
+            "That JSON must start at the very first character of your response and end at the last character—"
+            "no quotes, no code fences, and no extra explanation.\n\n"
+            "If you do not need any tool, simply reply with your final answer as plain text."
         )
     )
 
-    # 2) Kick off the graph with only that system prompt
+    # B) Start the graph with only the SystemMessage in state["messages"]
     initial_state = { "messages": [system_msg] }
+
+    # C) Invoke the graph, passing user_input separately
     final_state = compiled_graph.invoke(initial_state, user_input)
 
-    # 3) Find the last AIMessage the LLM returned
+    # D) Log out everything the agent wrote, so you can diagnose
+    print("===== AGENT MESSAGES =====")
+    for i, msg in enumerate(final_state["messages"]):
+        if isinstance(msg, AIMessage):
+            print(f"  [AIMessage #{i}]: {repr(msg.content)}")
+    print("==========================")
+
+    # E) Take the very last AIMessage that the agent produced
     last_msg = None
     for msg in reversed(final_state["messages"]):
         if isinstance(msg, AIMessage):
             last_msg = msg.content
             break
 
-    # 4) Try to parse that as a tool‐call dict
+    # F) Try to parse that last message as a JSON‐dict for a tool call
     tool_dict = parse_tool_json(last_msg or "")
-    if tool_dict is not None:
-        # 4.a) If it’s valid, run the tool
+    if tool_dict:
+        # G) If valid JSON, run the tool
+        print(">> Parsed tool call:", tool_dict)
         result = tool_node.run(tool_dict)
-        # 4.b) Feed the tool’s output back into the agent a second time
-        new_state = {
-            "messages": [*final_state["messages"], AIMessage(content=result)]
+
+        # H) Feed back the tool’s output as the next AIMessage, then re‐invoke
+        # (Note: we pass an empty string as user_input because we’re continuing the same turn)
+        continuation_state = {
+            "messages": [
+                *final_state["messages"],
+                AIMessage(content=result)
+            ]
         }
-        second_pass = compiled_graph.invoke(new_state, "")
-        # 4.c) Return the last AIMessage from that second pass
+        second_pass = compiled_graph.invoke(continuation_state, "")
+
+        # I) Log that second pass
+        print("===== AGENT SECOND PASS =====")
+        for i, msg in enumerate(second_pass["messages"]):
+            if isinstance(msg, AIMessage):
+                print(f"  [AIMessage2 #{i}]: {repr(msg.content)}")
+        print("==============================")
+
+        # J) Return the final AIMessage from that second pass
         for msg in reversed(second_pass["messages"]):
             if isinstance(msg, AIMessage):
-                return msg.content
+                return msg.content or ""
         return ""
     else:
-        # 5) If it wasn’t valid JSON → just return the plain last_msg
+        # K) If it wasn’t valid JSON, just return the plain last_msg
         return last_msg or ""
 
-
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")