From dde2d650ff0b40de08243a5dd76a40553fadf0ac Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Thu, 9 Jan 2025 11:26:30 -0800 Subject: [PATCH] WorkflowRunNotFound should be a SkyvernHTTPException error (#1528) --- skyvern/exceptions.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/skyvern/exceptions.py b/skyvern/exceptions.py index da7b1eae..5136bee6 100644 --- a/skyvern/exceptions.py +++ b/skyvern/exceptions.py @@ -140,9 +140,9 @@ class WorkflowNotFound(SkyvernHTTPException): ) -class WorkflowRunNotFound(SkyvernException): +class WorkflowRunNotFound(SkyvernHTTPException): def __init__(self, workflow_run_id: str) -> None: - super().__init__(f"WorkflowRun {workflow_run_id} not found") + super().__init__(f"WorkflowRun {workflow_run_id} not found", status_code=status.HTTP_404_NOT_FOUND) class MissingValueForParameter(SkyvernHTTPException):