Skip to content

Commit 3379944

Browse files
authored
fix: 修复升级langchain后千帆模型无法报错 (#638)
1 parent 5c697da commit 3379944

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

apps/setting/models_provider/impl/wenxin_model_provider/wenxin_model_provider.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
2828
model_type_list = WenxinModelProvider().get_model_type_list()
2929
if not any(list(filter(lambda mt: mt.get('value') == model_type, model_type_list))):
3030
raise AppApiException(ValidCode.valid_error.value, f'{model_type} 模型类型不支持')
31-
model_info = [model.lower() for model in ChatCompletion.models()]
31+
model = WenxinModelProvider().get_model(model_type, model_name, model_credential)
32+
model_info = [model.lower() for model in model.client.models()]
3233
if not model_info.__contains__(model_name.lower()):
3334
raise AppApiException(ValidCode.valid_error.value, f'{model_name} 模型不支持')
3435
for key in ['api_key', 'secret_key']:
@@ -38,7 +39,7 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
3839
else:
3940
return False
4041
try:
41-
WenxinModelProvider().get_model(model_type, model_name, model_credential).invoke(
42+
model.invoke(
4243
[HumanMessage(content='你好')])
4344
except Exception as e:
4445
raise e

0 commit comments

Comments
 (0)