deug

app.py

@@ -157,6 +157,11 @@ def tool_node(state: AgentState) -> AgentState:
     # We expect exactly one of these keys to be non‐empty:
     #   "web_search_query", "ocr_path", "excel_path"/"excel_sheet_name", "audio_path"
     # Whichever is present, call the corresponding tool and return its result.
+    
+    if state.get("wiki_query"):
+        out = wikipedia_search_tool(state)
+        return out
+    
     if state.get("web_search_query"):
         # print(f">>> tools_node dispatching web_search_tool with query: {state['web_search_query']!r}")
         out = web_search_tool(state)
@@ -178,9 +183,7 @@ def tool_node(state: AgentState) -> AgentState:
         out = audio_transcriber_tool(state)
         return out
 
-    if state.get("wiki_query"):
-        out = wikipedia_search_tool(state)
-        return out
+
 
     # If we somehow reach here, no recognized tool key was set:
     # print(">>> tools_node: no valid tool key found in state!")

tools.py

@@ -19,6 +19,7 @@ def _download_file_for_task(task_id: str, ext: str) -> str:
     Saves under ./hf_files/{task_id}.{ext}. Returns the local path if successful,
     or an empty string if no file / download failed.
     """
+    print("reached _download_file_for_task")
     os.makedirs("hf_files", exist_ok=True)
     local_path = os.path.join("hf_files", f"{task_id}.{ext}")
     url = f"{DEFAULT_API_URL}/files/{task_id}"
@@ -43,6 +44,7 @@ def web_search_tool(state: AgentState) -> AgentState:
     Returns: {"web_search_query": None, "web_search_result": <string>}.
     Retries up to 5 times on either a DuckDuckGo “202 Ratelimit” response or any exception (e.g. timeout).
     """
+    print("reached web_search_tool")
     query = state.get("web_search_query", "")
     if not query:
         return {}  # nothing to do
@@ -95,6 +97,7 @@ def ocr_image_tool(state: AgentState) -> AgentState:
       { "ocr_path": None, "ocr_result": "<OCRed text or error string>" }
     Always attempts to download the file for the given path or task ID.
     """
+    print("reached ocr_image_tool")
     path_or_id = state.get("ocr_path", "")
     if not path_or_id:
         return {}
@@ -139,6 +142,7 @@ def parse_excel_tool(state: AgentState) -> AgentState:
       }
     Always attempts to download the file for the given path or task ID.
     """
+    print("reached parse_excel_tool")
     path_or_id = state.get("excel_path", "")
     sheet = state.get("excel_sheet_name", "")
     if not path_or_id:
@@ -200,15 +204,7 @@ def parse_excel_tool(state: AgentState) -> AgentState:
         "excel_result": table_block
     }
 
-def run_tools(state: AgentState, tool_out: AgentState) -> AgentState:
-    """
-    Merges whatever partial state the tool wrapper returned (tool_out)
-    into the main state. That is, combine previous keys with new keys:
-      new_state = { **state, **tool_out }.
-    This node should be wired as its own graph node, not as a transition function.
-    """
-    new_state = {**state, **tool_out}
-    return new_state
+
 
 
 import os
@@ -235,6 +231,7 @@ def audio_transcriber_tool(state: AgentState) -> AgentState:
       }
     Always attempts to download the file for the given path or task ID.
     """
+    print("reached audio_transcriber_tool")
     path_or_id = state.get("audio_path", "")
     if not path_or_id:
         return {}
@@ -335,3 +332,17 @@ def wikipedia_search_tool(state: AgentState) -> AgentState:
         return {"wiki_query": None, "wiki_result": f"Wikipedia search error: {e}"}
     except Exception as e:
         return {"wiki_query": None, "wiki_result": f"Unexpected error in wikipedia_search_tool: {e}"}
+
+
+
+
+
+def run_tools(state: AgentState, tool_out: AgentState) -> AgentState:
+    """
+    Merges whatever partial state the tool wrapper returned (tool_out)
+    into the main state. That is, combine previous keys with new keys:
+      new_state = { **state, **tool_out }.
+    This node should be wired as its own graph node, not as a transition function.
+    """
+    new_state = {**state, **tool_out}
+    return new_state