From 7baef2a1bd259992502fc08a610de76e55e2f0b9 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Thu, 15 May 2025 11:35:33 -0700 Subject: [PATCH] persistent browser session - screenshot streaming (#2352) --- skyvern/forge/sdk/db/client.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index a916c94d..4f18ee98 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -2679,6 +2679,34 @@ class AgentDB: LOG.error("UnexpectedError", exc_info=True) raise + async def get_persistent_browser_session_by_runnable_id( + self, runnable_id: str, organization_id: str | None = None + ) -> PersistentBrowserSession | None: + """Get a specific persistent browser session.""" + try: + async with self.Session() as session: + query = ( + select(PersistentBrowserSessionModel) + .filter_by(runnable_id=runnable_id) + .filter_by(deleted_at=None) + .filter_by(completed_at=None) + ) + if organization_id: + query = query.filter_by(organization_id=organization_id) + persistent_browser_session = (await session.scalars(query)).first() + if persistent_browser_session: + return PersistentBrowserSession.model_validate(persistent_browser_session) + raise NotFoundError(f"PersistentBrowserSession {runnable_id} not found") + except NotFoundError: + LOG.error("NotFoundError", exc_info=True) + raise + except SQLAlchemyError: + LOG.error("SQLAlchemyError", exc_info=True) + raise + except Exception: + LOG.error("UnexpectedError", exc_info=True) + raise + async def get_persistent_browser_session( self, session_id: str,