fix: hit testing throws errors cause internal server error (#1865)
This commit is contained in:
@@ -1,6 +1,8 @@
|
|||||||
import logging
|
import logging
|
||||||
|
|
||||||
from flask_login import current_user
|
from flask_login import current_user
|
||||||
|
|
||||||
|
from core.model_runtime.errors.invoke import InvokeError
|
||||||
from libs.login import login_required
|
from libs.login import login_required
|
||||||
from flask_restful import Resource, reqparse, marshal
|
from flask_restful import Resource, reqparse, marshal
|
||||||
from werkzeug.exceptions import InternalServerError, NotFound, Forbidden
|
from werkzeug.exceptions import InternalServerError, NotFound, Forbidden
|
||||||
@@ -8,7 +10,7 @@ from werkzeug.exceptions import InternalServerError, NotFound, Forbidden
|
|||||||
import services
|
import services
|
||||||
from controllers.console import api
|
from controllers.console import api
|
||||||
from controllers.console.app.error import ProviderNotInitializeError, ProviderQuotaExceededError, \
|
from controllers.console.app.error import ProviderNotInitializeError, ProviderQuotaExceededError, \
|
||||||
ProviderModelCurrentlyNotSupportError
|
ProviderModelCurrentlyNotSupportError, CompletionRequestError
|
||||||
from controllers.console.datasets.error import HighQualityDatasetOnlyError, DatasetNotInitializedError
|
from controllers.console.datasets.error import HighQualityDatasetOnlyError, DatasetNotInitializedError
|
||||||
from controllers.console.setup import setup_required
|
from controllers.console.setup import setup_required
|
||||||
from controllers.console.wraps import account_initialization_required
|
from controllers.console.wraps import account_initialization_required
|
||||||
@@ -69,6 +71,8 @@ class HitTestingApi(Resource):
|
|||||||
raise ProviderNotInitializeError(
|
raise ProviderNotInitializeError(
|
||||||
f"No Embedding Model or Reranking Model available. Please configure a valid provider "
|
f"No Embedding Model or Reranking Model available. Please configure a valid provider "
|
||||||
f"in the Settings -> Model Provider.")
|
f"in the Settings -> Model Provider.")
|
||||||
|
except InvokeError as e:
|
||||||
|
raise CompletionRequestError(e.description)
|
||||||
except ValueError as e:
|
except ValueError as e:
|
||||||
raise ValueError(str(e))
|
raise ValueError(str(e))
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
Reference in New Issue
Block a user