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

Use persistent browser session in runnables (#1510)

Browse Source 
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
Co-authored-by: Shuchang Zheng <shu@skyvern.com>
This commit is contained in:
Maksim Ivanov
2025-01-09 22:04:53 +01:00
committed by GitHub
parent 5ed7e5ad8e
commit a4744ed9f5
12 changed files with 506 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
skyvern/forge/sdk/executor/async_executor.py
View File

@@ -25,6 +25,7 @@ class AsyncExecutor(abc.ABC):
organization_id: str,
max_steps_override: int | None,
api_key: str | None,
browser_session_id: str | None,
**kwargs: dict,
) -> None:
pass
@@ -39,6 +40,7 @@ class AsyncExecutor(abc.ABC):
workflow_run_id: str,
max_steps_override: int | None,
api_key: str | None,
browser_session_id: str | None,
**kwargs: dict,
) -> None:
pass
@@ -51,6 +53,7 @@ class AsyncExecutor(abc.ABC):
organization_id: str,
observer_cruise_id: str,
max_iterations_override: int | None,
browser_session_id: str | None,
**kwargs: dict,
) -> None:
pass
@@ -65,10 +68,13 @@ class BackgroundTaskExecutor(AsyncExecutor):
organization_id: str,
max_steps_override: int | None,
api_key: str | None,
browser_session_id: str | None,
**kwargs: dict,
) -> None:
LOG.info("Executing task using background task executor", task_id=task_id)
close_browser_on_completion = browser_session_id is None
organization = await app.DATABASE.get_organization(organization_id)
if organization is None:
raise OrganizationNotFound(organization_id)
@@ -98,6 +104,8 @@ class BackgroundTaskExecutor(AsyncExecutor):
task,
step,
api_key,
close_browser_on_completion=close_browser_on_completion,
browser_session_id=browser_session_id,
)
async def execute_workflow(
@@ -109,6 +117,7 @@ class BackgroundTaskExecutor(AsyncExecutor):
workflow_run_id: str,
max_steps_override: int | None,
api_key: str | None,
browser_session_id: str | None,
**kwargs: dict,
) -> None:
LOG.info(
@@ -126,6 +135,7 @@ class BackgroundTaskExecutor(AsyncExecutor):
workflow_run_id=workflow_run_id,
api_key=api_key,
organization=organization,
browser_session_id=browser_session_id,
)
async def execute_cruise(
@@ -135,6 +145,7 @@ class BackgroundTaskExecutor(AsyncExecutor):
organization_id: str,
observer_cruise_id: str,
max_iterations_override: int | None,
browser_session_id: str | None,
**kwargs: dict,
) -> None:
LOG.info(
@@ -169,4 +180,5 @@ class BackgroundTaskExecutor(AsyncExecutor):
organization=organization,
observer_cruise_id=observer_cruise_id,
max_iterations_override=max_iterations_override,
browser_session_id=browser_session_id,
)