From 858d60016aecf7d8f65912e9bbf2eba2e7b7508b Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Thu, 18 Jul 2024 22:46:58 -0700 Subject: [PATCH] fix update_task bug (#626) --- skyvern/forge/sdk/db/client.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index 62460c05..53298113 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -379,7 +379,13 @@ class AgentDB: max_steps_per_run: int | None = None, organization_id: str | None = None, ) -> Task: - if status is None and extracted_information is None and failure_reason is None and errors is None: + if ( + status is None + and extracted_information is None + and failure_reason is None + and errors is None + and max_steps_per_run is None + ): raise ValueError( "At least one of status, extracted_information, or failure_reason must be provided to update the task" )