expose max_steps_per_run in the task endpoint

Skyvern-AI · Jul 18, 2024 · 5b5454c · 5b5454c
1 parent 941c9b8
commit 5b5454c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/skyvern/forge/sdk/schemas/tasks.py b/skyvern/forge/sdk/schemas/tasks.py
@@ -221,6 +221,7 @@ def to_task_response(
             screenshot_url=screenshot_url,
             recording_url=recording_url,
             errors=self.errors,
+            max_steps_per_run=self.max_steps_per_run,
         )
 
 
@@ -236,6 +237,7 @@ class TaskResponse(BaseModel):
     recording_url: str | None = None
     failure_reason: str | None = None
     errors: list[dict[str, Any]] = []
+    max_steps_per_run: int | None = None
 
 
 class TaskOutput(BaseModel):