From 23479af94252ad8eecfcc99dc160f9a7fff9128b Mon Sep 17 00:00:00 2001 From: LawyZheng Date: Thu, 21 Nov 2024 17:48:56 +0800 Subject: [PATCH] fix extraction block bug (#1232) --- 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 165e5907..6b646ab1 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -44,7 +44,7 @@ from skyvern.forge.sdk.models import Organization, Step, StepStatus from skyvern.forge.sdk.schemas.tasks import Task, TaskRequest, TaskStatus from skyvern.forge.sdk.settings_manager import SettingsManager from skyvern.forge.sdk.workflow.context_manager import WorkflowRunContext -from skyvern.forge.sdk.workflow.models.block import ActionBlock, BaseTaskBlock +from skyvern.forge.sdk.workflow.models.block import ActionBlock, BaseTaskBlock, ValidationBlock from skyvern.forge.sdk.workflow.models.workflow import Workflow, WorkflowRun, WorkflowRunStatus from skyvern.webeye.actions.actions import ( Action, @@ -632,7 +632,7 @@ class ForgeAgent: actions: list[Action] using_cached_action_plan = False - if not task.workflow_run_id and not task.navigation_goal: + if not task.navigation_goal and not isinstance(task_block, ValidationBlock): actions = [ CompleteAction( reasoning="Task has no navigation goal.",