speech_to_text wiki_search

app.py

@@ -7,7 +7,7 @@ import pandas as pd
 # Import our custom tools from their modules
 from huggingface_hub import login
 from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel
-from tools import web_search, visit_webpage, final_answer
+from tools import web_search, visit_webpage, final_answer, speech_to_text, wiki_search
 # from tools import web_search, visit_webpage, final_answer, go_back, close_popups, search_item_ctrl_f
 from retriever import load_guest_dataset
 from functools import lru_cache
@@ -34,6 +34,10 @@ class BasicAgent:
         self.visit_webpage = visit_webpage
 
         self.final_answer = final_answer
+        
+        self.speech_to_text = speech_to_text
+        
+        self.wiki_search = wiki_search
 
         # self.go_back = go_back 
         
@@ -70,7 +74,7 @@ class BasicAgent:
         
         # Create the agent with tools
         self.agent = CodeAgent(
-            tools=[web_search, self.visit_webpage, self.final_answer],
+            tools=[web_search, self.visit_webpage, self.final_answer, wiki_search, speech_to_text],
             # tools=[self.web_search, self.visit_webpage, self.final_answer, go_back, close_popups, search_item_ctrl_f],
             model=self.model,
             additional_authorized_imports=["pandas", "openpyxl", "yt_dlp", "requests", "io", "json", "whisper", "bs4"],