Add browser_session_id to context

Skyvern-AI · Dec 16, 2024 · 1078402 · 1078402
1 parent 9b6f0f3
commit 1078402
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/skyvern/forge/sdk/core/skyvern_context.py b/skyvern/forge/sdk/core/skyvern_context.py
@@ -12,6 +12,7 @@ class SkyvernContext:
     workflow_run_id: str | None = None
     max_steps_override: int | None = None
     tz_info: ZoneInfo | None = None
+    browser_session_id: str | None = None
     totp_codes: dict[str, str | None] = field(default_factory=dict)
 
     def __repr__(self) -> str:

diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py
@@ -122,7 +122,7 @@ async def setup_workflow_run(
             )
             if reusable_browser_session is None:
                 raise BrowserSessionNotFound(browser_session_id=workflow_request.browser_session_id)
-            reusable_browser_session.reset_timeout()
+            # reusable_browser_session.reset_timeout()
 
         if workflow_request.proxy_location is None and workflow.proxy_location is not None:
             workflow_request.proxy_location = workflow.proxy_location
@@ -150,7 +150,7 @@ async def setup_workflow_run(
                 workflow_id=workflow_id,
                 workflow_run_id=workflow_run.workflow_run_id,
                 max_steps_override=max_steps_override,
-                reuse_browser_session=workflow_request.browser_session_id,
+                browser_session_id=workflow_request.browser_session_id,
             )
         )