Update agent.py

agent.py

@@ -23,6 +23,7 @@ import whisper
 #    """
 #    cse_id = os.environ.get("GOOGLE_CSE_ID")
 #    if not api_key or not cse_id:
+
 #        raise ValueError("GOOGLE_API_KEY and GOOGLE_CSE_ID must be set in environment variables.")
 #    url = "https://www.googleapis.com/customsearch/v1"
 #    params = {
@@ -173,7 +174,7 @@ class MagAgent:
         self.rate_limiter = rate_limiter
         print("Initializing MagAgent with search tools...")
         model = LiteLLMModel(
-            model_id="gemini/gemini-2.0-flash-lite",
+            model_id="gemini/gemini-2.0-flash",
             api_key= os.environ.get("GEMINI_KEY"),
             max_tokens=8192
         )