Update agent.py

agent.py

@@ -220,7 +220,8 @@ tools=[
     TavilySearchTool()
 ]
 
-model = LiteLLMModel(model="gemini-2.0-flash", temperature=0)
+model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-exp",
+                     api_key=os.getenv("GEMINI_API_KEY"))
 
 def delay_execution_10(pagent, **kwargs) -> bool:
     """