Update agent.py

agent.py

@@ -2,7 +2,6 @@ from __future__ import annotations
 import os
 from langchain_openai import ChatOpenAI
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
-from langgraph.prebuilt import ToolExecutor
 from typing import Any, Dict, List, Optional
 import json
 
@@ -23,10 +22,47 @@ from tools import (
 # ─────────────────────────── Configuration ───────────────────────────────
 MAX_TOOL_CALLS = 5
 
+class SimpleToolExecutor:
+    """Simple tool executor that maps tool names to functions."""
+    
+    def __init__(self, tools: List):
+        self.tools_map = {tool.name: tool for tool in tools}
+    
+    def run(self, tool_name: str, tool_input: str) -> str:
+        """Execute a tool with the given input."""
+        if tool_name not in self.tools_map:
+            return f"Tool '{tool_name}' not found. Available tools: {list(self.tools_map.keys())}"
+        
+        tool = self.tools_map[tool_name]
+        try:
+            # Handle different input formats
+            if tool_name in ['add_tool', 'subtract_tool', 'multiply_tool', 'divide_tool']:
+                # Math tools expect two numbers
+                # Try to parse the input as two numbers
+                import re
+                numbers = re.findall(r'-?\d+(?:\.\d+)?', tool_input)
+                if len(numbers) >= 2:
+                    a, b = float(numbers[0]), float(numbers[1])
+                    return tool.run(a=a, b=b)
+                else:
+                    return f"Math tool requires two numbers. Got: {tool_input}"
+            else:
+                # Other tools expect a string input
+                # Clean the input - remove quotes if present
+                clean_input = tool_input.strip().strip('"').strip("'")
+                if "search" in tool_name:
+                    return tool.run(clean_input)
+                elif tool_name in ['audio_transcriber_tool', 'excel_tool', 'analyze_code_tool', 'image_tool']:
+                    return tool.run(task_id=clean_input)
+                else:
+                    return tool.run(clean_input)
+        except Exception as e:
+            return f"Error executing {tool_name}: {str(e)}"
+
 class CustomReActAgent:
     def __init__(self, tools: List, model_name="gpt-4o-mini"):
         self.llm = ChatOpenAI(model_name=model_name, temperature=0.3)
-        self.tool_executor = ToolExecutor(tools)
+        self.tool_executor = SimpleToolExecutor(tools)
 
     def run(self, question: str, task_id: Optional[str] = None, max_turns: int = 15, system_prompt: str = "") -> str:
         messages = [SystemMessage(content=system_prompt)]