zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Increase log output

Browse Source
This commit is contained in:
wxg0103 2025-04-02 14:17:48 +08:00
parent 27d4603b02
commit 6d7b5eb219
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/setting/models_provider/impl/openai_model_provider/credential/llm.py
View File

@ -11,6 +11,7 @@ from typing import Dict
from django.utils.translation import gettext_lazy as _, gettext from django.utils.translation import gettext_lazy as _, gettext
from langchain_core.messages import HumanMessage from langchain_core.messages import HumanMessage
from openai import BadRequestError
from common import forms from common import forms
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
@ -58,7 +59,7 @@ class OpenAILLMModelCredential(BaseForm, BaseModelCredential):
model.invoke([HumanMessage(content=gettext('Hello'))]) model.invoke([HumanMessage(content=gettext('Hello'))])
except Exception as e: except Exception as e:
traceback.print_exc() traceback.print_exc()
if isinstance(e, AppApiException): if isinstance(e, AppApiException) or isinstance(e, BadRequestError):
raise e raise e
if raise_exception: if raise_exception:
raise AppApiException(ValidCode.valid_error.value, raise AppApiException(ValidCode.valid_error.value,