| @@ -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: | |||