diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index 436eed43..8c337192 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -1533,7 +1533,7 @@ async def get_workflow_run_with_workflow_id( browser_session_id = browser_session.persistent_browser_session_id if browser_session else None - return_dict["browser_session_id"] = browser_session_id + return_dict["browser_session_id"] = browser_session_id or return_dict.get("browser_session_id") task_v2 = await app.DATABASE.get_task_v2_by_workflow_run_id( workflow_run_id=workflow_run_id, diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py index 1cdbfa67..5090c73f 100644 --- a/skyvern/forge/sdk/workflow/service.py +++ b/skyvern/forge/sdk/workflow/service.py @@ -1243,6 +1243,7 @@ class WorkflowService: total_steps=total_steps, total_cost=total_cost, workflow_title=workflow.title, + browser_session_id=workflow_run.browser_session_id, max_screenshot_scrolls=workflow_run.max_screenshot_scrolls, )