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,
|
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.
|
# 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,
|
max_steps_per_run=2,
|
||||||
|
disable_cache=block_yaml.disable_cache,
|
||||||
model=block_yaml.model,
|
model=block_yaml.model,
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -2834,6 +2835,7 @@ class WorkflowService:
|
|||||||
totp_identifier=block_yaml.totp_identifier,
|
totp_identifier=block_yaml.totp_identifier,
|
||||||
max_iterations=block_yaml.max_iterations,
|
max_iterations=block_yaml.max_iterations,
|
||||||
max_steps=block_yaml.max_steps,
|
max_steps=block_yaml.max_steps,
|
||||||
|
disable_cache=block_yaml.disable_cache,
|
||||||
model=block_yaml.model,
|
model=block_yaml.model,
|
||||||
output_parameter=output_parameter,
|
output_parameter=output_parameter,
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -341,6 +341,7 @@ class ValidationBlockYAML(BlockYAML):
|
|||||||
terminate_criterion: str | None = None
|
terminate_criterion: str | None = None
|
||||||
error_code_mapping: dict[str, str] | None = None
|
error_code_mapping: dict[str, str] | None = None
|
||||||
parameter_keys: list[str] | None = None
|
parameter_keys: list[str] | None = None
|
||||||
|
disable_cache: bool = False
|
||||||
|
|
||||||
|
|
||||||
class ActionBlockYAML(BlockYAML):
|
class ActionBlockYAML(BlockYAML):
|
||||||
@@ -462,6 +463,7 @@ class TaskV2BlockYAML(BlockYAML):
|
|||||||
totp_identifier: str | None = None
|
totp_identifier: str | None = None
|
||||||
max_iterations: int = settings.MAX_ITERATIONS_PER_TASK_V2
|
max_iterations: int = settings.MAX_ITERATIONS_PER_TASK_V2
|
||||||
max_steps: int = settings.MAX_STEPS_PER_TASK_V2
|
max_steps: int = settings.MAX_STEPS_PER_TASK_V2
|
||||||
|
disable_cache: bool = False
|
||||||
|
|
||||||
|
|
||||||
class HttpRequestBlockYAML(BlockYAML):
|
class HttpRequestBlockYAML(BlockYAML):
|
||||||
|
|||||||
Reference in New Issue
Block a user