dynamic tool picking

basic_agent.py

@@ -1,10 +1,51 @@
-from tools import search_duckduckgo
+from tools import all_tools
 
 class BasicAgent:
     def __call__(self, question: str) -> str:
         try:
-            result = search_duckduckgo.invoke({"query": question})
+            tool = self.select_tool(question)
+            result = tool.invoke({"query": question})
             return result
         except Exception as e:
-            return f"Error searching: {e}"
+            return f"Error: {e}"
+
+    def select_tool(self, question: str):
+        q = question.lower()
+
+        if "wikipedia" in q or "encyclopedia" in q:
+            return all_tools[1]  # search_wikipedia
+        elif "youtube" in q or "video" in q:
+            return all_tools[2]  # search_youtube_transcript
+        elif "opposite" in q or "reverse" in q:
+            return all_tools[3]  # reverse_sentence
+        elif "chess" in q or "position" in q:
+            return all_tools[4]  # solve_chess_position
+        elif "table" in q or "commutative" in q:
+            return all_tools[5]  # math_solver
+        elif "libretext" in q or "document" in q:
+            return all_tools[6]  # document_search
+        elif "vegetable" in q or "botany" in q:
+            return all_tools[7]  # classify_vegetable
+        elif "audio" in q or "transcribe" in q or "mp3" in q:
+            return all_tools[8]  # transcribe_audio
+        elif "imdb" in q or "movie" in q or "tv show" in q:
+            return all_tools[9]  # search_imdb
+        elif "python" in q or "code" in q:
+            return all_tools[10]  # run_python_code
+        elif "yankee" in q or "baseball" in q and "walks" in q:
+            return all_tools[11]  # search_baseball_stats
+        elif "nasa" in q or "universe today" in q:
+            return all_tools[12]  # scrape_document
+        elif "specimen" in q or "zoological" in q:
+            return all_tools[13]  # search_zoological_database
+        elif "olympic" in q or "athletes" in q:
+            return all_tools[14]  # search_olympic_database
+        elif "roster" in q or "pitcher" in q:
+            return all_tools[15]  # search_baseball_roster
+        elif "excel" in q or "sales" in q or "menu" in q:
+            return all_tools[16]  # process_excel
+        elif "malko" in q or "music competition" in q:
+            return all_tools[17]  # search_classical_music_database
+        else:
+            return all_tools[0]  # fallback to search_duckduckgo