Update basic_agent.py

basic_agent.py

@@ -9,43 +9,17 @@ class BasicAgent:
         except Exception as e:
             return f"Error: {e}"
 
-    def select_tool(self, question: str):
-        q = question.lower()
+ 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
+    if "youtube.com" in q or "youtu.be" in q or "in the video" in q:
+        return analyze_youtube_video  # direct reference, not index
+
+    elif "how many" in q or "number of" in q:
+        return extract_number_from_text
+
+    elif "wikipedia" in q or "encyclopedia" in q or "who is" in q:
+        return search_wikipedia
+
+    return extract_number_from_text  # fallback tool