diff --git a/skyvern/webeye/browser_manager.py b/skyvern/webeye/browser_manager.py index 3dd0f924..4d273b9f 100644 --- a/skyvern/webeye/browser_manager.py +++ b/skyvern/webeye/browser_manager.py @@ -81,7 +81,7 @@ class BrowserManager: "Getting browser state for task from persistent sessions manager", browser_session_id=browser_session_id, ) - browser_state = app.PERSISTENT_SESSIONS_MANAGER.get_browser_state(browser_session_id) + browser_state = await app.PERSISTENT_SESSIONS_MANAGER.get_browser_state(browser_session_id) if browser_state is None: LOG.warning( "Browser state not found in persistent sessions manager", @@ -139,7 +139,7 @@ class BrowserManager: "Getting browser state for workflow run from persistent sessions manager", browser_session_id=browser_session_id, ) - browser_state = app.PERSISTENT_SESSIONS_MANAGER.get_browser_state(browser_session_id) + browser_state = await app.PERSISTENT_SESSIONS_MANAGER.get_browser_state(browser_session_id) if browser_state is None: LOG.warning( "Browser state not found in persistent sessions manager", browser_session_id=browser_session_id diff --git a/skyvern/webeye/persistent_sessions_manager.py b/skyvern/webeye/persistent_sessions_manager.py index f5228fc6..d55724d1 100644 --- a/skyvern/webeye/persistent_sessions_manager.py +++ b/skyvern/webeye/persistent_sessions_manager.py @@ -39,7 +39,7 @@ class PersistentSessionsManager: """Get all active sessions for an organization.""" return await self.database.get_active_persistent_browser_sessions(organization_id) - def get_browser_state(self, session_id: str) -> BrowserState | None: + async def get_browser_state(self, session_id: str) -> BrowserState | None: """Get a specific browser session's state by session ID.""" browser_session = self._browser_sessions.get(session_id) return browser_session.browser_state if browser_session else None