Update app.py

app.py

@@ -5,7 +5,6 @@ import requests
 import pandas as pd
 
 from smolagents import Tool, CodeAgent, HfApiModel
-#from smolagents.models import HfApiModel
 
 from audio_transcriber import AudioTranscriptionTool
 from image_analyzer import ImageAnalysisTool
@@ -34,13 +33,19 @@ Only return the exact answer."""
 # Define agent tools
 audio_tool = AudioTranscriptionTool()
 image_tool = ImageAnalysisTool()
-wiki_tool = Tool.from_function(
-    name="wikipedia_search",
-    description="Search for facts using Wikipedia.",
-    input_schema={"query": {"type": "string", "description": "Search query"}},
-    output_type="string",
-    forward=lambda query: WikipediaSearcher().search(query)
-)
+
+class WikipediaTool(Tool):
+    name = "wikipedia_search"
+    description = "Search for facts using Wikipedia."
+    inputs = {
+        "query": {"type": "string", "description": "Search query"}
+    }
+    output_type = "string"
+
+    def forward(self, query: str) -> str:
+        return WikipediaSearcher().search(query)
+
+wiki_tool = WikipediaTool()
 
 tools = [audio_tool, image_tool, wiki_tool]