Unverified Commit 4562e83b authored by takatost's avatar takatost Committed by GitHub

fix: hit testing throws errors cause internal server error (#1865)

parent 7be77c19
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 ...@@ -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): ...@@ -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:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment