mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2025-09-02 10:41:04 +00:00
90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
import abc
|
|
|
|
import structlog
|
|
from fastapi import BackgroundTasks
|
|
|
|
from skyvern.forge import app
|
|
from skyvern.forge.sdk.core import skyvern_context
|
|
from skyvern.forge.sdk.core.skyvern_context import SkyvernContext
|
|
from skyvern.forge.sdk.models import Organization
|
|
from skyvern.forge.sdk.schemas.tasks import Task, TaskStatus
|
|
|
|
LOG = structlog.get_logger()
|
|
|
|
|
|
class AsyncExecutor(abc.ABC):
|
|
@abc.abstractmethod
|
|
async def execute_task(
|
|
self,
|
|
background_tasks: BackgroundTasks,
|
|
task: Task,
|
|
organization: Organization,
|
|
max_steps_override: int | None,
|
|
api_key: str | None,
|
|
) -> None:
|
|
pass
|
|
|
|
@abc.abstractmethod
|
|
async def execute_workflow(
|
|
self,
|
|
background_tasks: BackgroundTasks,
|
|
organization: Organization,
|
|
workflow_id: str,
|
|
workflow_run_id: str,
|
|
max_steps_override: int | None,
|
|
api_key: str | None,
|
|
) -> None:
|
|
pass
|
|
|
|
|
|
class BackgroundTaskExecutor(AsyncExecutor):
|
|
async def execute_task(
|
|
self,
|
|
background_tasks: BackgroundTasks,
|
|
task: Task,
|
|
organization: Organization,
|
|
max_steps_override: int | None,
|
|
api_key: str | None,
|
|
) -> None:
|
|
LOG.info("Executing task using background task executor", task_id=task.task_id)
|
|
step = await app.DATABASE.create_step(
|
|
task.task_id,
|
|
order=0,
|
|
retry_index=0,
|
|
organization_id=organization.organization_id,
|
|
)
|
|
|
|
task = await app.DATABASE.update_task(
|
|
task.task_id,
|
|
status=TaskStatus.running,
|
|
organization_id=organization.organization_id,
|
|
)
|
|
|
|
context: SkyvernContext = skyvern_context.ensure_context()
|
|
context.task_id = task.task_id
|
|
context.organization_id = organization.organization_id
|
|
context.max_steps_override = max_steps_override
|
|
|
|
background_tasks.add_task(
|
|
app.agent.execute_step,
|
|
organization,
|
|
task,
|
|
step,
|
|
api_key,
|
|
)
|
|
|
|
async def execute_workflow(
|
|
self,
|
|
background_tasks: BackgroundTasks,
|
|
organization: Organization,
|
|
workflow_id: str,
|
|
workflow_run_id: str,
|
|
max_steps_override: int | None,
|
|
api_key: str | None,
|
|
) -> None:
|
|
LOG.info("Executing workflow using background task executor", workflow_run_id=workflow_run_id)
|
|
background_tasks.add_task(
|
|
app.WORKFLOW_SERVICE.execute_workflow,
|
|
workflow_run_id=workflow_run_id,
|
|
api_key=api_key,
|
|
)
|