diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index b5f14a4c..d08f44da 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -262,6 +262,7 @@ async def run_task( model=run_request.model, max_screenshot_scrolling_times=run_request.max_screenshot_scrolls, extra_http_headers=run_request.extra_http_headers, + browser_session_id=run_request.browser_session_id, browser_address=run_request.browser_address, ) except MissingBrowserAddressError as e: diff --git a/skyvern/forge/sdk/schemas/tasks.py b/skyvern/forge/sdk/schemas/tasks.py index 0c42dff1..b7b98895 100644 --- a/skyvern/forge/sdk/schemas/tasks.py +++ b/skyvern/forge/sdk/schemas/tasks.py @@ -364,6 +364,7 @@ class Task(TaskBase): workflow_run_id=self.workflow_run_id, max_screenshot_scrolls=self.max_screenshot_scrolls, step_count=step_count, + browser_session_id=self.browser_session_id, ) @@ -390,6 +391,7 @@ class TaskResponse(BaseModel): finished_at: datetime | None = None max_screenshot_scrolls: int | None = None step_count: int | None = None + browser_session_id: str | None = None class TaskOutput(BaseModel):