fix get_or_create_page by ensuring the page is returned

Skyvern-AI · Feb 24, 2025 · 0a0aa53 · 0a0aa53
1 parent 2a2c6f1
commit 0a0aa53
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/skyvern/webeye/browser_factory.py b/skyvern/webeye/browser_factory.py
@@ -536,7 +536,7 @@ async def get_or_create_page(
                 workflow_run_id=workflow_run_id,
                 organization_id=organization_id,
             )
-        await self.__assert_page()
+        page = await self.__assert_page()
 
         if not await BrowserContextFactory.validate_browser_context(await self.get_working_page()):
             await self.close_current_open_page()
@@ -547,8 +547,7 @@ async def get_or_create_page(
                 workflow_run_id=workflow_run_id,
                 organization_id=organization_id,
             )
-            await self.__assert_page()
-
+            page = await self.__assert_page()
         return page
 
     async def close_current_open_page(self) -> None: