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

This commit is contained in:
Shuchang Zheng
2024-05-10 11:11:13 -04:00
committed by GitHub
parent cf01e81ba2
commit 25a7a3ad8e

View File

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