add max_steps_per_run to task (#297)

Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
This commit is contained in:
Kerem Yilmaz 2024-05-11 14:13:21 -07:00 committed by GitHub
parent 6feddbde6a
commit 270642c60c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 39 additions and 0 deletions

View file

@ -111,6 +111,7 @@ class TaskBlock(Block):
# error code to error description for the LLM
error_code_mapping: dict[str, str] | None = None
max_retries: int = 0
max_steps_per_run: int | None = None
parameters: list[PARAMETER_TYPE] = []
def get_all_parameters(