Update agent.py

agent.py

@@ -46,8 +46,14 @@ sys_msg = SystemMessage(content=system_prompt)
 
 # Loading the assistant
 #chat = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
-#chat = ChatGroq(model_name="mistral-saba-24b", temperature=0)
-chat = ChatGroq(model_name="llama-3.1-8b-instant", temperature=0)
+chat = ChatGoogleGenerativeAI(
+    model="gemini-2.5-flash-preview-04-17",
+    temperature=0,
+    max_retries=1,
+    rate_limiter=InMemoryRateLimiter(
+        requests_per_second=1/60,
+    ),
+)
 
 tools = [add, 
          substract,