15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: 修复升级langchain后千帆模型无法报错 (#638)

This commit is contained in:
shaohuzhang1 2024-06-13 16:15:16 +08:00 committed by GitHub
parent 5c697da8dd
commit 337994439c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/setting/models_provider/impl/wenxin_model_provider/wenxin_model_provider.py
View File

@ -28,7 +28,8 @@ class WenxinLLMModelCredential(BaseForm, BaseModelCredential):
model_type_list = WenxinModelProvider().get_model_type_list()
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
raise AppApiException(ValidCode.valid_error.value, f'{model_type} 模型类型不支持')
model_info = [model.lower() for model in ChatCompletion.models()]
model = WenxinModelProvider().get_model(model_type, model_name, model_credential)
model_info = [model.lower() for model in model.client.models()]
if not model_info.__contains__(model_name.lower()):
raise AppApiException(ValidCode.valid_error.value, f'{model_name} 模型不支持')
for key in ['api_key', 'secret_key']:
@ -38,7 +39,7 @@ class WenxinLLMModelCredential(BaseForm, BaseModelCredential):
else:
return False
try:
WenxinModelProvider().get_model(model_type, model_name, model_credential).invoke(
model.invoke(
[HumanMessage(content='你好')])
except Exception as e:
raise e