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

BE: allow workflow run UI to show browser stream if workflow run has one (#3834)

Browse Source
This commit is contained in:
Jonathan Dobson
2025-10-28 10:18:12 -04:00
committed by GitHub
parent ba0b25cb4b
commit f9b2cb0fb3
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1878,6 +1878,7 @@ class AgentDB:
sequential_key: str | None = None, sequential_key: str | None = None,
ai_fallback: bool | None = None, ai_fallback: bool | None = None,
depends_on_workflow_run_id: str | None = None, depends_on_workflow_run_id: str | None = None,
browser_session_id: str | None = None,
) -> WorkflowRun: ) -> WorkflowRun:
async with self.Session() as session: async with self.Session() as session:
workflow_run = ( workflow_run = (
@@ -1908,6 +1909,8 @@ class AgentDB:
workflow_run.ai_fallback = ai_fallback workflow_run.ai_fallback = ai_fallback
if depends_on_workflow_run_id: if depends_on_workflow_run_id:
workflow_run.depends_on_workflow_run_id = depends_on_workflow_run_id workflow_run.depends_on_workflow_run_id = depends_on_workflow_run_id
if browser_session_id:
workflow_run.browser_session_id = browser_session_id
await session.commit() await session.commit()
await session.refresh(workflow_run) await session.refresh(workflow_run)
await save_workflow_run_logs(workflow_run_id) await save_workflow_run_logs(workflow_run_id)

4
skyvern/forge/sdk/workflow/service.py
View File

@@ -465,6 +465,10 @@ class WorkflowService:
if browser_session: if browser_session:
browser_session_id = browser_session.persistent_browser_session_id browser_session_id = browser_session.persistent_browser_session_id
close_browser_on_completion = True close_browser_on_completion = True
await app.DATABASE.update_workflow_run(
workflow_run_id=workflow_run.workflow_run_id,
browser_session_id=browser_session_id,
)
# Check if there's a related workflow script that should be used instead # Check if there's a related workflow script that should be used instead
workflow_script, _ = await workflow_script_service.get_workflow_script(workflow, workflow_run, block_labels) workflow_script, _ = await workflow_script_service.get_workflow_script(workflow, workflow_run, block_labels)