From 4562e83b242130021ee6c5058b4afaa2672e59bc Mon Sep 17 00:00:00 2001 From: takatost Date: Wed, 3 Jan 2024 08:57:39 +0800 Subject: [PATCH] fix: hit testing throws errors cause internal server error (#1865) --- api/controllers/console/datasets/hit_testing.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/api/controllers/console/datasets/hit_testing.py b/api/controllers/console/datasets/hit_testing.py index 79c7878717..06f75e2706 100644 --- a/api/controllers/console/datasets/hit_testing.py +++ b/api/controllers/console/datasets/hit_testing.py @@ -1,6 +1,8 @@ import logging from flask_login import current_user + +from core.model_runtime.errors.invoke import InvokeError from libs.login import login_required from flask_restful import Resource, reqparse, marshal from werkzeug.exceptions import InternalServerError, NotFound, Forbidden @@ -8,7 +10,7 @@ from werkzeug.exceptions import InternalServerError, NotFound, Forbidden import services from controllers.console import api from controllers.console.app.error import ProviderNotInitializeError, ProviderQuotaExceededError, \ - ProviderModelCurrentlyNotSupportError + ProviderModelCurrentlyNotSupportError, CompletionRequestError from controllers.console.datasets.error import HighQualityDatasetOnlyError, DatasetNotInitializedError from controllers.console.setup import setup_required from controllers.console.wraps import account_initialization_required @@ -69,6 +71,8 @@ class HitTestingApi(Resource): raise ProviderNotInitializeError( f"No Embedding Model or Reranking Model available. Please configure a valid provider " f"in the Settings -> Model Provider.") + except InvokeError as e: + raise CompletionRequestError(e.description) except ValueError as e: raise ValueError(str(e)) except Exception as e: