From b20f3e13c02018ce7af11c6b0f7dec5292fd3077 Mon Sep 17 00:00:00 2001
From: Kerem Yilmaz <kerem@skyvern.com>
Date: Thu, 25 Jul 2024 13:29:23 -0700
Subject: [PATCH] Fix dynamic browsers for workflows (#650)

---
 skyvern/webeye/browser_manager.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/skyvern/webeye/browser_manager.py b/skyvern/webeye/browser_manager.py
index 1d527081..446c8f54 100644
--- a/skyvern/webeye/browser_manager.py
+++ b/skyvern/webeye/browser_manager.py
@@ -25,6 +25,7 @@ class BrowserManager:
         proxy_location: ProxyLocation | None = None,
         url: str | None = None,
         task_id: str | None = None,
+        workflow_run_id: str | None = None,
     ) -> BrowserState:
         pw = await async_playwright().start()
         (
@@ -36,6 +37,7 @@ class BrowserManager:
             proxy_location=proxy_location,
             url=url,
             task_id=task_id,
+            workflow_run_id=workflow_run_id,
         )
         return BrowserState(
             pw=pw,
@@ -76,7 +78,9 @@ class BrowserManager:
             "Creating browser state for workflow run",
             workflow_run_id=workflow_run.workflow_run_id,
         )
-        browser_state = await self._create_browser_state(workflow_run.proxy_location, url=url)
+        browser_state = await self._create_browser_state(
+            workflow_run.proxy_location, url=url, workflow_run_id=workflow_run.workflow_run_id
+        )
 
         # The URL here is only used when creating a new page, and not when using an existing page.
         # This will make sure browser_state.page is not None.