add max_steps_per_run to task (#297)

Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
2024-05-11 14:13:21 -07:00
parent 6feddbde6a
commit 270642c60c
8 changed files with 39 additions and 0 deletions
--- a/skyvern/forge/sdk/db/client.py
+++ b/skyvern/forge/sdk/db/client.py
@@ -83,6 +83,7 @@ class AgentDB:
        workflow_run_id: str | None = None,
        order: int | None = None,
        retry: int | None = None,
+        max_steps_per_run: int | None = None,
        error_code_mapping: dict[str, str] | None = None,
    ) -> Task:
        try:
@@ -101,6 +102,7 @@ class AgentDB:
                    workflow_run_id=workflow_run_id,
                    order=order,
                    retry=retry,
+                    max_steps_per_run=max_steps_per_run,
                    error_code_mapping=error_code_mapping,
                )
                session.add(new_task)