vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Make get_browser_state async (#1775)

Browse Source
This commit is contained in:
Maksim Ivanov
2025-02-16 13:10:00 +01:00
committed by GitHub
parent 4ba32fa7ee
commit 63a37cf799
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/webeye/browser_manager.py
View File

@@ -81,7 +81,7 @@ class BrowserManager:
"Getting browser state for task from persistent sessions manager", "Getting browser state for task from persistent sessions manager",
browser_session_id=browser_session_id, 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: if browser_state is None:
LOG.warning( LOG.warning(
"Browser state not found in persistent sessions manager", "Browser state not found in persistent sessions manager",
@@ -139,7 +139,7 @@ class BrowserManager:
"Getting browser state for workflow run from persistent sessions manager", "Getting browser state for workflow run from persistent sessions manager",
browser_session_id=browser_session_id, 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: if browser_state is None:
LOG.warning( LOG.warning(
"Browser state not found in persistent sessions manager", browser_session_id=browser_session_id "Browser state not found in persistent sessions manager", browser_session_id=browser_session_id

2
skyvern/webeye/persistent_sessions_manager.py
View File

@@ -39,7 +39,7 @@ class PersistentSessionsManager:
"""Get all active sessions for an organization.""" """Get all active sessions for an organization."""
return await self.database.get_active_persistent_browser_sessions(organization_id) 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.""" """Get a specific browser session's state by session ID."""
browser_session = self._browser_sessions.get(session_id) browser_session = self._browser_sessions.get(session_id)
return browser_session.browser_state if browser_session else None return browser_session.browser_state if browser_session else None