diff --git a/apps/setting/models_provider/impl/openai_model_provider/credential/llm.py b/apps/setting/models_provider/impl/openai_model_provider/credential/llm.py index 13b70980d..eb8622648 100644 --- a/apps/setting/models_provider/impl/openai_model_provider/credential/llm.py +++ b/apps/setting/models_provider/impl/openai_model_provider/credential/llm.py @@ -11,6 +11,7 @@ from typing import Dict from django.utils.translation import gettext_lazy as _, gettext from langchain_core.messages import HumanMessage +from openai import BadRequestError from common import forms from common.exception.app_exception import AppApiException @@ -58,7 +59,7 @@ class OpenAILLMModelCredential(BaseForm, BaseModelCredential): model.invoke([HumanMessage(content=gettext('Hello'))]) except Exception as e: traceback.print_exc() - if isinstance(e, AppApiException): + if isinstance(e, AppApiException) or isinstance(e, BadRequestError): raise e if raise_exception: raise AppApiException(ValidCode.valid_error.value,