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