Add browser session id to global context (#1867)

This commit is contained in:
Shuchang Zheng
2025-03-02 23:39:18 -05:00
committed by GitHub
parent d271a84eff
commit 05960aa894
3 changed files with 4 additions and 0 deletions

View File

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

View File

@@ -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

View File

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