From 25a7a3ad8e28cbf126adcdf21e8a1625069c3af0 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Fri, 10 May 2024 11:11:13 -0400 Subject: [PATCH] SkyvernContext shouldn't be a blocker for running agent (#295) --- skyvern/forge/agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py index 6821f288..f16425bb 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -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