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, ) )