Update agent.py

agent.py

@@ -27,9 +27,11 @@ YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma sepa
 Your answer should only start with "FINAL ANSWER: ", then follows with the answer.
 """
 
-api_key = os.getenv("OPENAI_API_KEY")
+#api_key = os.getenv("OPENAI_API_KEY")
+api_key = os.getenv("GEMINI_API_KEY")
 
-model = ChatOpenAI(model="gpt-4o-mini", api_key=api_key, temperature=0)
+#model = ChatOpenAI(model="gpt-4o-mini", api_key=api_key, temperature=0)
+model = ChatGoogleGenerativeAI(model="gemini-2.5-flash", temperature=0, api_key=api_key)
 
 @tool
 def search_wiki(query: str, max_results: int = 2) -> str: