Update app.py

app.py

@@ -16,25 +16,6 @@ from langchain.schema import HumanMessage, AIMessage, SystemMessage
 # Create a ToolNode that knows about your web_search function
 import json
 
-def parse_tool_json(text: str) -> dict | None:
-    """
-    Given a string like '{"tool":"web_search","query":"..."}'
-    or '"{\"tool\":\"web_search\",\"query\":\"...\"}"', return
-    the parsed dict. Otherwise, return None.
-    """
-    t = text.strip()
-    # If it’s wrapped in single or double quotes, remove them:
-    if (t.startswith('"') and t.endswith('"')) or (t.startswith("'") and t.endswith("'")):
-        t = t[1:-1]
-    try:
-        obj = json.loads(t)
-        if isinstance(obj, dict) and "tool" in obj:
-            return obj
-    except Exception:
-        return None
-    return None
-
-
 # (Keep Constan
 # 
 # 
@@ -65,7 +46,7 @@ def respond_to_input(user_input: str) -> str:
             "⚠️ **MANDATORY** ⚠️: If (and only if) you need to call a tool, your entire response MUST be exactly ONE JSON OBJECT and NOTHING ELSE.  \n"
             "For example, if you want to call web_search, you must respond with exactly:\n"
             "```json\n"
-            "{\"tool\":\"web_search\",\"query\":\"Mercedes Sosa studio albums 2000-2009\"}\n"
+            '{"tool":"web_search","query":"Mercedes Sosa studio albums 2000-2009"}'\n"
             "```\n"
             "That JSON string must start at the very first character of your response and end at the very last character—"
             "no surrounding quotes, no markdown fences, no explanatory text.  \n\n"
@@ -81,12 +62,11 @@ def respond_to_input(user_input: str) -> str:
         ]
     }
 
-        # C) FIRST PASS: invoke with only initial_state (no second argument!)
+    # C) FIRST PASS: invoke with only initial_state (no second argument!)
     try:
         first_pass = compiled_graph.invoke(initial_state)
     except Exception as e:
         print("‼️ ERROR during first invoke:", repr(e))
-        # If you want extra debug, you can try printing first_pass["messages"] if defined
         return ""  # return fallback
 
     # D) Log the AIMessage(s) from first_pass
@@ -103,15 +83,25 @@ def respond_to_input(user_input: str) -> str:
             last_msg = msg.content
             break
 
-    # F) Attempt to parse last_msg as JSON for a tool call
-    tool_dict = parse_tool_json(last_msg or "")
+    # F) Attempt to parse last_msg as JSON for a tool call (inline, no parse_tool_json)
+    tool_dict = None
+    t = (last_msg or "").strip()
+    if (t.startswith('"') and t.endswith('"')) or (t.startswith("'") and t.endswith("'")):
+        t = t[1:-1]
+    try:
+        obj = json.loads(t)
+        if isinstance(obj, dict) and "tool" in obj:
+            tool_dict = obj
+    except Exception:
+        tool_dict = None
+
     if tool_dict:
         # G) If valid JSON, run the tool
         print(">> Parsed tool call:", tool_dict)
         tool_result = tool_node.run(tool_dict)
         print(f">> Tool '{tool_dict['tool']}' returned: {repr(tool_result)}")
 
-        # H) SECOND PASS: feed the tool’s output back in as an AIMessage,
+        # H) SECOND PASS: feed the tool's output back in as an AIMessage,
         #    with no new human input
         continuation_state = {
             "messages": [