From 05960aa8947eef222d2163f496af63cc2e5fd316 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sun, 2 Mar 2025 23:39:18 -0500 Subject: [PATCH] Add browser session id to global context (#1867) --- skyvern/forge/sdk/core/skyvern_context.py | 1 + skyvern/forge/sdk/forge_log.py | 2 ++ skyvern/forge/sdk/services/task_v2_service.py | 1 + 3 files changed, 4 insertions(+) diff --git a/skyvern/forge/sdk/core/skyvern_context.py b/skyvern/forge/sdk/core/skyvern_context.py index 4f6a9210..d86a91d2 100644 --- a/skyvern/forge/sdk/core/skyvern_context.py +++ b/skyvern/forge/sdk/core/skyvern_context.py @@ -14,6 +14,7 @@ class SkyvernContext: workflow_run_id: str | None = None task_v2_id: str | None = None max_steps_override: int | None = None + browser_session_id: str | None = None tz_info: ZoneInfo | None = None totp_codes: dict[str, str | None] = field(default_factory=dict) log: list[dict] = field(default_factory=list) diff --git a/skyvern/forge/sdk/forge_log.py b/skyvern/forge/sdk/forge_log.py index 95ace7a8..375ef739 100644 --- a/skyvern/forge/sdk/forge_log.py +++ b/skyvern/forge/sdk/forge_log.py @@ -34,6 +34,8 @@ def add_kv_pairs_to_msg(logger: logging.Logger, method_name: str, event_dict: Ev event_dict["workflow_run_id"] = context.workflow_run_id if context.task_v2_id: event_dict["task_v2_id"] = context.task_v2_id + if context.browser_session_id: + event_dict["browser_session_id"] = context.browser_session_id # Add env to the log event_dict["env"] = settings.ENV diff --git a/skyvern/forge/sdk/services/task_v2_service.py b/skyvern/forge/sdk/services/task_v2_service.py index da606bff..dc0faf2e 100644 --- a/skyvern/forge/sdk/services/task_v2_service.py +++ b/skyvern/forge/sdk/services/task_v2_service.py @@ -359,6 +359,7 @@ async def run_task_v2_helper( workflow_run_id=workflow_run_id, request_id=request_id, task_v2_id=task_v2_id, + browser_session_id=browser_session_id, ) )