Unify predict errors

nuclia · Nov 24, 2023 · c5b96a3 · c5b96a3
1 parent 9875755
commit c5b96a3
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/nucliadb/nucliadb/search/api/v1/chat.py b/nucliadb/nucliadb/search/api/v1/chat.py
@@ -106,7 +106,10 @@ async def chat_knowledgebox_endpoint(
     except LimitsExceededError as exc:
         return HTTPClientError(status_code=exc.status_code, detail=exc.detail)
     except predict.ProxiedPredictAPIError as err:
-        return HTTPClientError(status_code=503, detail=f"{err.status}: {err.detail}")
+        return HTTPClientError(
+            status_code=503,
+            detail=f"Chat service unavailable. {err.status}: {err.detail}",
+        )
     except IncompleteFindResultsError:
         return HTTPClientError(
             status_code=529,

diff --git a/nucliadb/nucliadb/search/api/v1/resource/chat.py b/nucliadb/nucliadb/search/api/v1/resource/chat.py
@@ -71,7 +71,10 @@ async def resource_chat_endpoint(
     except LimitsExceededError as exc:
         return HTTPClientError(status_code=exc.status_code, detail=exc.detail)
     except predict.ProxiedPredictAPIError as err:
-        return HTTPClientError(status_code=503, detail=f"{err.status}: {err.detail}")
+        return HTTPClientError(
+            status_code=503,
+            detail=f"Chat service unavailable. {err.status}: {err.detail}",
+        )
     except IncompleteFindResultsError:
         return HTTPClientError(
             status_code=529,

diff --git a/nucliadb/nucliadb/search/api/v1/summarize.py b/nucliadb/nucliadb/search/api/v1/summarize.py
@@ -59,4 +59,7 @@ async def summarize_endpoint(
     except LimitsExceededError as exc:
         return HTTPClientError(status_code=exc.status_code, detail=exc.detail)
     except predict.ProxiedPredictAPIError as err:
-        return HTTPClientError(status_code=503, detail=f"{err.status}: {err.detail}")
+        return HTTPClientError(
+            status_code=503,
+            detail=f"Summarize service unavailable. {err.status}: {err.detail}",
+        )