Update agent.py

agent.py

@@ -389,18 +389,18 @@ class MagAgent:
         self.rate_limiter = rate_limiter
 
         print("Initializing MagAgent with search tools...")
-#        model = LiteLLMModel(
-#            model_id="gemini/gemini-2.0-flash-preview-image-generation",
-#            api_key= os.environ.get("GEMINI_KEY"),
-#            max_tokens=8192
-#        )
-
         model = LiteLLMModel(
-            model_id="gemini/gemini-1.5-flash",  # Use standard multimodal model
-            api_key=os.environ.get("GEMINI_KEY"),
-            max_tokens=8192,
-            api_base="https://generativelanguage.googleapis.com/v1beta"  # Correct endpoint
+            model_id="gemini/gemini-2.0-flash",
+            api_key= os.environ.get("GEMINI_KEY"),
+            max_tokens=8192
         )
+
+#        model = LiteLLMModel(
+#            model_id="gemini/gemini-1.5-flash",  # Use standard multimodal model
+#            api_key=os.environ.get("GEMINI_KEY"),
+#            max_tokens=8192,
+#            api_base="https://generativelanguage.googleapis.com/v1beta"  # Correct endpoint
+#        )
         
         # Load prompt templates
         with open("prompts.yaml", 'r') as stream: