Update agent.py

agent.py

@@ -646,14 +646,13 @@ def task_classifier(question: str) -> str:
 
     return "default"
 
-def select_tool_and_run(question: str, tools: list) -> any:
-    """
-    Select the best tool based on question intent and run it.
-    """
-    # Step 1: Use task_classifier to identify the intent of the question
+
+
+def select_tool_and_run(question: str, tools: list):
+    # Classify intent
     intent = task_classifier(question)
 
-    # Step 2: Use the intent_tool_map to map the intent to the tool
+    # Map intent to expected tool names
     intent_tool_map = {
         "math": "calculator",
         "wiki_search": "wiki_tool",
@@ -661,35 +660,31 @@ def select_tool_and_run(question: str, tools: list) -> any:
         "arxiv": "arxiv_tool",
         "get_youtube_transcript": "youtube_tool",
         "video_analysis": "video_tool",
-        "analyze_tool": "analyze_tool",
+        "data_analysis": "analyze_tool",         # FIXED key from 'analyze_tool'
         "wikidata_query": "wikidata_query",
         "default": "default_tool"
     }
 
+    # Resolve the tool name for the detected intent
     tool_name = intent_tool_map.get(intent, "default_tool")
 
-    # Step 3: Get the selected tool
-    tool_func = tools.get(tool_name)
+    # Try to find the matching tool object from the list
+    tool_func = next((tool for tool in tools if getattr(tool, "name", "") == tool_name), None)
 
     if not tool_func:
-        print(f"No matching tool found for {intent}")
+        print(f"No matching tool found for intent: {intent} (tool_name: {tool_name})")
         return None
 
-    # If question was transformed into JSON for tool input
+    # Try to parse question into JSON for tools expecting structured input
     try:
         parsed_input = json.loads(question)
     except json.JSONDecodeError:
-        parsed_input = question  # fallback if question is not JSON
+        parsed_input = question  # Fallback to raw string if not JSON
 
+    # Run the tool and return result
     return tool_func.run(parsed_input)
 
 
-# Run the function to select and execute the tool
-result = select_tool_and_run(question, tools)
-
-# Output the result
-print(result)
-
 
 # Function to extract math operation from the question