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

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

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