return pbs id on workflow run endpoint (#2699)

skyvern/forge/sdk/routes/agent_protocol.py

@@ -1357,12 +1357,31 @@ async def get_workflow_run_with_workflow_id(
         include_cost=True,
     )
     return_dict = workflow_run_status_response.model_dump()
+
+    browser_session = await app.DATABASE.get_persistent_browser_session_by_runnable_id(
+        runnable_id=workflow_run_id,
+        organization_id=current_org.organization_id,
+    )
+
+    browser_session_id = browser_session.persistent_browser_session_id if browser_session else None
+
+    LOG.info(
+        "workflow-run.assign-browser-session-id",
+        runnable_id=workflow_run_id,
+        browser_session_id=browser_session_id,
+        organization_id=current_org.organization_id,
+    )
+
+    return_dict["browser_session_id"] = browser_session_id
+
     task_v2 = await app.DATABASE.get_task_v2_by_workflow_run_id(
         workflow_run_id=workflow_run_id,
         organization_id=current_org.organization_id,
     )
+
     if task_v2:
         return_dict["task_v2"] = task_v2.model_dump(by_alias=True)
+
     return return_dict
 
 

skyvern/forge/sdk/workflow/models/workflow.py

@@ -161,3 +161,4 @@ class WorkflowRunResponseBase(BaseModel):
     total_cost: float | None = None
     task_v2: TaskV2 | None = None
     workflow_title: str | None = None
+    browser_session_id: str | None = None