add disable_cache to Validation and TaskV2 (#3721)
This commit is contained in:
@@ -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,
|
||||
)
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user