deepkoluguri
/
agents-runtime
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update model to gemini-2.5-flash
Build Agents Runtime / build (push) Has been cancelled Details

This commit is contained in:
Deep Koluguri 2026-06-29 10:38:41 -04:00
parent 7c3ad4447b
commit 194a9d1228
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/agent.py
View File

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