Update agent.py

agent.py

@@ -15,8 +15,12 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
         # now for the agent:
-        my_gemini_model = os.getenv(key="GEMINI_MODEL")
-        model = LiteLLMModel(model_id=f"gemini/{my_gemini_model}", api_key=os.getenv(key="MY_GEMINI_API_KEY"))
+        
+        model = LiteLLMModel(model_id="gemini/gemini-2.5-pro-preview-06-05", api_key=os.getenv(key="MY_GEMINI_API_KEY"))
+        
+        # my_gemini_model = os.getenv(key="GEMINI_MODEL")
+        # model = LiteLLMModel(model_id=f"gemini/{my_gemini_model}", api_key=os.getenv(key="MY_GEMINI_API_KEY"))
+        
         agent = CodeAgent(
             model=model,
             tools=[DuckDuckGoSearchTool()]