Update agent.py

agent.py

@@ -184,7 +184,7 @@ class BasicAgent:
         model = self.select_model(provider)
         client = InferenceClientModel()
         tools = [
-            GoogleSearchTool(),
+            # GoogleSearchTool(),
             DuckDuckGoSearchTool(),
             GeminiVideoQA(GEMINI_MODEL_NAME),
             WikiTitleFinder(),
@@ -254,16 +254,17 @@ class BasicAgent:
         )
 
     def select_model(self, provider: str):
-        if provider == "openai":
-            return LiteLLMModel(model_id=OPENAI_MODEL_NAME, api_key=os.getenv("OPENAI_API_KEY"))
-        elif provider == "groq":
-            return LiteLLMModel(model_id=GROQ_MODEL_NAME, api_key=os.getenv("GROQ_API_KEY"))
-        elif provider == "deepseek":
-            return LiteLLMModel(model_id=DEEPSEEK_MODEL_NAME, api_key=os.getenv("DEEPSEEK_API_KEY"))
-        elif provider == "hf":
-            return InferenceClientModel()
-        else:
-            return LiteLLMModel(model_id=GEMINI_MODEL_NAME, api_key=os.getenv("GOOGLE_API_KEY"))
+        # if provider == "openai":
+        #     return LiteLLMModel(model_id=OPENAI_MODEL_NAME, api_key=os.getenv("OPENAI_API_KEY"))
+        # elif provider == "groq":
+        #     return LiteLLMModel(model_id=GROQ_MODEL_NAME, api_key=os.getenv("GROQ_API_KEY"))
+        # elif provider == "deepseek":
+        #     return LiteLLMModel(model_id=DEEPSEEK_MODEL_NAME, api_key=os.getenv("DEEPSEEK_API_KEY"))
+        # elif provider == "hf":
+        #     return InferenceClientModel()
+        # else:
+        #     return LiteLLMModel(model_id=GEMINI_MODEL_NAME, api_key=os.getenv("GOOGLE_API_KEY"))
+        return LiteLLMModel(model_id=GEMINI_MODEL_NAME, api_key=os.getenv("GOOGLE_API_KEY"))
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")