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

SkyvernContext shouldn't be a blocker for running agent (#295)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-05-10 11:11:13 -04:00
committed by GitHub
parent cf01e81ba2
commit 25a7a3ad8e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/forge/agent.py
View File

@@ -1112,8 +1112,8 @@ class ForgeAgent:
await self.update_task(task, status=TaskStatus.terminated, failure_reason=failure_reason) await self.update_task(task, status=TaskStatus.terminated, failure_reason=failure_reason)
return False, last_step, None return False, last_step, None
# If the max steps are exceeded, mark the current step as the last step and conclude the task # If the max steps are exceeded, mark the current step as the last step and conclude the task
context = skyvern_context.ensure_context() context = skyvern_context.current()
override_max_steps_per_run = context.max_steps_override override_max_steps_per_run = context.max_steps_override if context else None
max_steps_per_run = ( max_steps_per_run = (
override_max_steps_per_run override_max_steps_per_run
or organization.max_steps_per_run or organization.max_steps_per_run