vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add ai_fallback to workflow_runs (#3581)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-10-01 14:13:56 -07:00
committed by GitHub
parent db024d42bb
commit f97b53975f
7 changed files with 62 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
skyvern/forge/sdk/db/client.py
View File

@@ -1699,6 +1699,7 @@ class AgentDB:
sequential_key: str | None = None,
run_with: str | None = None,
debug_session_id: str | None = None,
ai_fallback: bool | None = None,
) -> WorkflowRun:
try:
async with self.Session() as session:
@@ -1719,6 +1720,7 @@ class AgentDB:
sequential_key=sequential_key,
run_with=run_with,
debug_session_id=debug_session_id,
ai_fallback=ai_fallback,
)
session.add(workflow_run)
await session.commit()
@@ -1738,6 +1740,7 @@ class AgentDB:
job_id: str | None = None,
run_with: str | None = None,
sequential_key: str | None = None,
ai_fallback: bool | None = None,
) -> WorkflowRun:
async with self.Session() as session:
workflow_run = (
@@ -1764,6 +1767,8 @@ class AgentDB:
workflow_run.run_with = run_with
if sequential_key:
workflow_run.sequential_key = sequential_key
if ai_fallback is not None:
workflow_run.ai_fallback = ai_fallback
await session.commit()
await session.refresh(workflow_run)
await save_workflow_run_logs(workflow_run_id)