From 194a9d1228cfa4b50270188dba9e116984ee6f4a Mon Sep 17 00:00:00 2001
From: Deep Koluguri <deepkoluguri@gmail.com>
Date: Mon, 29 Jun 2026 10:38:41 -0400
Subject: [PATCH] Update model to gemini-2.5-flash

---
 src/agent.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/agent.py b/src/agent.py
index fae066c..0e229fc 100644
--- a/src/agent.py
+++ b/src/agent.py
@@ -109,8 +109,8 @@ def run_agent_activity(payload: dict) -> str:
     target_jid = payload.get("targetJid", "default")
     sender = payload.get("sender", "Unknown")
 
-    primary_llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash")
-    fallback_llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro")
+    primary_llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash")
+    fallback_llm = ChatGoogleGenerativeAI(model="gemini-2.5-pro")
     llm = primary_llm.with_fallbacks([fallback_llm])
     
     system_prompt = f"""You are a highly capable AI assistant operating as an orchestrator.