tool call improvements

app.py

@@ -40,7 +40,12 @@ def execute_tool(tool_name: str, tool_args: Dict[str, Any], tools: Dict[str, Cal
     """Execute a tool with the given arguments."""
     if tool_name not in tools:
         raise ValueError(f"Tool {tool_name} not found")
-    return tools[tool_name](**tool_args)
+    tool_func = tools[tool_name]
+    # If the tool is decorated with @tool, use the run method
+    if hasattr(tool_func, 'run'):
+        return tool_func.run(**tool_args)
+    # Otherwise, call the function directly
+    return tool_func(**tool_args)
 
 # Convert existing functions to tools
 @tool
@@ -105,7 +110,6 @@ def open_youtube_video(url: str, query: str) -> str:
     )
     return response.text
 
-@tool
 def google_search(query: str) -> str:
     '''Performs a web search for the given query using DuckDuckGo.'''
     try: