diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py index 59dfd47c..10f03900 100644 --- a/skyvern/forge/sdk/workflow/service.py +++ b/skyvern/forge/sdk/workflow/service.py @@ -2666,6 +2666,7 @@ class WorkflowService: continue_on_failure=block_yaml.continue_on_failure, # Should only need one step for validation block, but we allow 2 in case the LLM has an unexpected failure and we need to retry. max_steps_per_run=2, + disable_cache=block_yaml.disable_cache, model=block_yaml.model, ) @@ -2834,6 +2835,7 @@ class WorkflowService: totp_identifier=block_yaml.totp_identifier, max_iterations=block_yaml.max_iterations, max_steps=block_yaml.max_steps, + disable_cache=block_yaml.disable_cache, model=block_yaml.model, output_parameter=output_parameter, ) diff --git a/skyvern/schemas/workflows.py b/skyvern/schemas/workflows.py index be067350..b92a4930 100644 --- a/skyvern/schemas/workflows.py +++ b/skyvern/schemas/workflows.py @@ -341,6 +341,7 @@ class ValidationBlockYAML(BlockYAML): terminate_criterion: str | None = None error_code_mapping: dict[str, str] | None = None parameter_keys: list[str] | None = None + disable_cache: bool = False class ActionBlockYAML(BlockYAML): @@ -462,6 +463,7 @@ class TaskV2BlockYAML(BlockYAML): totp_identifier: str | None = None max_iterations: int = settings.MAX_ITERATIONS_PER_TASK_V2 max_steps: int = settings.MAX_STEPS_PER_TASK_V2 + disable_cache: bool = False class HttpRequestBlockYAML(BlockYAML):