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

Cleanup executor logic (#84)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-03-14 23:07:04 -07:00
committed by GitHub
parent ff4be0de9e
commit eda6e07d36
2 changed files with 26 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
skyvern/forge/sdk/routes/agent_protocol.py
View File

@@ -96,8 +96,8 @@ async def create_agent_task(
LOG.info("Overriding max steps per run", max_steps_override=x_max_steps_override)
await AsyncExecutorFactory.get_executor().execute_task(
background_tasks=background_tasks,
task=created_task,
organization=current_org,
task_id=created_task.task_id,
organization_id=current_org.organization_id,
max_steps_override=x_max_steps_override,
api_key=x_api_key,
)
@@ -422,7 +422,7 @@ async def execute_workflow(
LOG.info("Overriding max steps per run", max_steps_override=x_max_steps_override)
await AsyncExecutorFactory.get_executor().execute_workflow(
background_tasks=background_tasks,
organization=current_org,
organization_id=current_org.organization_id,
workflow_id=workflow_id,
workflow_run_id=workflow_run.workflow_run_id,
max_steps_override=x_max_steps_override,