Update tools.py

tools.py

@@ -9,12 +9,9 @@ import imageio
 
 from PIL import Image
 from dotenv import load_dotenv
+from gaia_benchmark.tools import tool
 import whisper
 
-def tool(func):
-    func.is_tool = True
-    return func
-
 load_dotenv()
 
 @tool
@@ -37,7 +34,7 @@ def youtube_frames_to_images(url: str, sample_interval_seconds: int = 5) -> List
             'force_ipv4': True,
         }
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.extract_info(url, download=True)
+            info = ydl.extract_info(url, download=True)
 
         video_path = next((os.path.join(tmpdir, f) for f in os.listdir(tmpdir) if f.endswith('.mp4')), None)
         reader = imageio.get_reader(video_path)
@@ -116,4 +113,22 @@ def youtube_transcribe(url: str) -> str:
 
 @tool
 def transcribe_audio(audio_file_path: str) -> str:
-    return whisper.load_model("small").transcribe(audio_file_path)['text']
+    return whisper.load_model("small").transcribe(audio_file_path)['text']
+
+# 🔧 Stub Tools for Mistral Compatibility
+
+@tool
+def web_search(query: str) -> str:
+    return f"(Stub) Web search results for: {query}"
+
+@tool
+def wikipedia_search(topic: str) -> str:
+    return f"(Stub) Wikipedia summary for: {topic}"
+
+@tool
+def visit_webpage(url: str) -> str:
+    return f"(Stub) Visited webpage at: {url}"
+
+@tool
+def final_answer(response: str) -> str:
+    return f"(Stub) Final answer: {response}"