Update basic_agent.py

basic_agent.py

@@ -11,6 +11,7 @@ class BasicAgent:
             "analyze_youtube_video": (analyze_youtube_video, "url"),
             "extract_number_from_text": (extract_number_from_text, "text"),
             "search_wikipedia": (search_wikipedia, "query"),
+            "check_commutativity_table": (check_commutativity_table, "table_text"),
         }
 
     def __call__(self, question: str) -> str:
@@ -32,6 +33,9 @@ class BasicAgent:
         elif "wikipedia" in q or "encyclopedia" in q or "who is" in q:
             return self.tool_registry["search_wikipedia"]
 
+        elif "commutative" in q or "counter-examples" in q:
+            return self.tool_registry["check_commutativity_table"]
+
         return self.tool_registry["extract_number_from_text"]  # fallback