Update agent.py

agent.py

@@ -1,7 +1,6 @@
 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):
@@ -221,7 +220,7 @@ tools=[
     TavilySearchTool()
 ]
 
-model = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+model = LiteLLMModel(model="gemini-2.0-flash", temperature=0)
 
 def delay_execution_10(pagent, **kwargs) -> bool:
     """