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

Use gemini-2.0-flash
Build Agents Runtime / build (push) Has been cancelled Details

This commit is contained in:
Deep Koluguri 2026-06-28 19:43:05 -04:00
parent b491b76644
commit e0ace519c8
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/agent.py
View File

@ -109,7 +109,7 @@ 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")
llm = ChatGoogleGenerativeAI(model="gemini-flash-latest") llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash")
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.

2
src/api.py
View File

@ -21,7 +21,9 @@ async def trigger_workflow(payload: dict):
result = await asyncio.to_thread(run_agent_activity, payload) result = await asyncio.to_thread(run_agent_activity, payload)
print(f"Agent finished. Result: {result}") print(f"Agent finished. Result: {result}")
except Exception as e: except Exception as e:
import traceback
print(f"Failed to trigger workflow: {e}") print(f"Failed to trigger workflow: {e}")
print(traceback.format_exc())
@app.post("/webhook") @app.post("/webhook")
async def webhook(payload: WebhookPayload, background_tasks: BackgroundTasks): async def webhook(payload: WebhookPayload, background_tasks: BackgroundTasks):