gemini model

agent.py

@@ -2,6 +2,9 @@ import os
 from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool
 from tools import VisitWebpageTool, DownloadTaskAttachmentTool
 
+MODEL_ID = "gemini/gemini-2.5-pro-preview-06-05"
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+
 class TheAgent:
     """Agent that processes questions using LLM and various tools."""
 
@@ -9,8 +12,8 @@ class TheAgent:
         # Initialize the agent with model and tools
         self.agent = CodeAgent(
             model=LiteLLMModel(
-                model_id="openrouter/meta-llama/llama-4-maverick:free",
-                api_key=os.getenv("OPENROUTER_KEY")
+                model_id=MODEL_ID,
+                api_key=GOOGLE_API_KEY
             ),
             tools=[
                 DuckDuckGoSearchTool(),