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

add Request context to async_executor (#709)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-08-16 08:25:10 +03:00
committed by GitHub
parent 4a8b1473ec
commit fd5fdb9d32
4 changed files with 47 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,7 +1,7 @@
import abc
import structlog
from fastapi import BackgroundTasks
from fastapi import BackgroundTasks, Request
from skyvern.exceptions import OrganizationNotFound
from skyvern.forge import app
@@ -16,6 +16,7 @@ class AsyncExecutor(abc.ABC):
@abc.abstractmethod
async def execute_task(
self,
request: Request | None,
background_tasks: BackgroundTasks,
task_id: str,
organization_id: str,
@@ -28,6 +29,7 @@ class AsyncExecutor(abc.ABC):
@abc.abstractmethod
async def execute_workflow(
self,
request: Request | None,
background_tasks: BackgroundTasks,
organization_id: str,
workflow_id: str,
@@ -42,6 +44,7 @@ class AsyncExecutor(abc.ABC):
class BackgroundTaskExecutor(AsyncExecutor):
async def execute_task(
self,
request: Request | None,
background_tasks: BackgroundTasks,
task_id: str,
organization_id: str,
@@ -83,6 +86,7 @@ class BackgroundTaskExecutor(AsyncExecutor):
async def execute_workflow(
self,
request: Request | None,
background_tasks: BackgroundTasks,
organization_id: str,
workflow_id: str,