From e26559f512472497df9ecf1aea007a1ef39b6380 Mon Sep 17 00:00:00 2001 From: Wang Wei Date: Thu, 19 Oct 2023 05:55:29 +0800 Subject: [PATCH] Add ERNIE-Bot-4 model support for ErnieBotChat. (#11969) - **Description:** According to the document https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t, add ERNIE-Bot-4 model support for ErnieBotChat. - **Dependencies:** Before using the ERNIE-Bot-4, you should have the model's access authority. --- libs/langchain/langchain/chat_models/ernie.py | 1 + 1 file changed, 1 insertion(+) diff --git a/libs/langchain/langchain/chat_models/ernie.py b/libs/langchain/langchain/chat_models/ernie.py index dd7c37ed966..bebfb374947 100644 --- a/libs/langchain/langchain/chat_models/ernie.py +++ b/libs/langchain/langchain/chat_models/ernie.py @@ -107,6 +107,7 @@ class ErnieBotChat(BaseChatModel): model_paths = { "ERNIE-Bot-turbo": "eb-instant", "ERNIE-Bot": "completions", + "ERNIE-Bot-4": "completions_pro", "BLOOMZ-7B": "bloomz_7b1", "Llama-2-7b-chat": "llama_2_7b", "Llama-2-13b-chat": "llama_2_13b",