Update agent.py

agent.py

@@ -1,6 +1,7 @@
 from smolagents import HfApiModel, CodeAgent, DuckDuckGoSearchTool, WikipediaSearchTool, Tool, LiteLLMModel
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
+from langchain_google_genai import ChatGoogleGenerativeAI
 
 
 class add(Tool):
@@ -220,11 +221,7 @@ tools=[
     TavilySearchTool()
 ]
 
-model = LiteLLMModel(
-    model_id='ollama_chat/gemma3:27b',
-    api_base="http://127.0.0.1:11434",
-    num_ctx=8192
-)
+model = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
 
 def delay_execution_10(pagent, **kwargs) -> bool:
     """