From 337994439c2d32438bc833f26bd2add2cd377406 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Thu, 13 Jun 2024 16:15:16 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E5=8D=87=E7=BA=A7lang?= =?UTF-8?q?chain=E5=90=8E=E5=8D=83=E5=B8=86=E6=A8=A1=E5=9E=8B=E6=97=A0?= =?UTF-8?q?=E6=B3=95=E6=8A=A5=E9=94=99=20(#638)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../impl/wenxin_model_provider/wenxin_model_provider.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/apps/setting/models_provider/impl/wenxin_model_provider/wenxin_model_provider.py b/apps/setting/models_provider/impl/wenxin_model_provider/wenxin_model_provider.py index 3d7c9a7d9..b61e2f6b4 100644 --- a/apps/setting/models_provider/impl/wenxin_model_provider/wenxin_model_provider.py +++ b/apps/setting/models_provider/impl/wenxin_model_provider/wenxin_model_provider.py @@ -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