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

validate browser session id when running tasks or workflows (#2874)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-07-04 01:49:51 -07:00
committed by GitHub
parent 133cee4358
commit f832206f38
3 changed files with 28 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
skyvern/forge/agent.py
View File

@@ -27,6 +27,7 @@ from skyvern.constants import (
ScrapeType,
)
from skyvern.exceptions import (
BrowserSessionNotFound,
BrowserStateMissingPage,
DownloadFileMaxWaitingTime,
EmptyScrapePage,
@@ -223,6 +224,15 @@ class ForgeAgent:
async def create_task(self, task_request: TaskRequest, organization_id: str) -> Task:
webhook_callback_url = str(task_request.webhook_callback_url) if task_request.webhook_callback_url else None
totp_verification_url = str(task_request.totp_verification_url) if task_request.totp_verification_url else None
# validate browser session id
if task_request.browser_session_id:
browser_session = await app.DATABASE.get_persistent_browser_session(
session_id=task_request.browser_session_id,
organization_id=organization_id,
)
if not browser_session:
raise BrowserSessionNotFound(browser_session_id=task_request.browser_session_id)
task = await app.DATABASE.create_task(
url=str(task_request.url),
title=task_request.title,