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

fix get active browser sessions (#3055)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-08-08 15:46:49 -07:00
committed by GitHub
parent bea8905a8c
commit 4bc5cb3b38
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
skyvern/forge/sdk/db/client.py
View File

@@ -2982,7 +2982,11 @@ class AgentDB:
for workflow_run_block in workflow_run_blocks for workflow_run_block in workflow_run_blocks
] ]
async def get_active_persistent_browser_sessions(self, organization_id: str) -> list[PersistentBrowserSession]: async def get_active_persistent_browser_sessions(
self,
organization_id: str,
active_hours: int = 24,
) -> list[PersistentBrowserSession]:
"""Get all active persistent browser sessions for an organization.""" """Get all active persistent browser sessions for an organization."""
try: try:
async with self.Session() as session: async with self.Session() as session:
@@ -2990,6 +2994,10 @@ class AgentDB:
select(PersistentBrowserSessionModel) select(PersistentBrowserSessionModel)
.filter_by(organization_id=organization_id) .filter_by(organization_id=organization_id)
.filter_by(deleted_at=None) .filter_by(deleted_at=None)
.filter_by(completed_at=None)
.filter(
PersistentBrowserSessionModel.created_at > datetime.utcnow() - timedelta(hours=active_hours)
)
) )
sessions = result.scalars().all() sessions = result.scalars().all()
return [PersistentBrowserSession.model_validate(session) for session in sessions] return [PersistentBrowserSession.model_validate(session) for session in sessions]