From fe9341a29caad102222c98e549207a0042620d2d Mon Sep 17 00:00:00 2001 From: Wang Wei Date: Wed, 29 Nov 2023 11:22:23 +0800 Subject: [PATCH] feat: Add ERNIE-Bot-8K model support for ErnieBotChat. (#13716) - **Description:** According to the document https://cloud.baidu.com/doc/WENXINWORKSHOP/s/6lp69is2a, add ERNIE-Bot-8K model support for ErnieBotChat. - **Dependencies:** Before using the ERNIE-Bot-8K, 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 e23002cd362..5b36548619d 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-8K": "ernie_bot_8k", "ERNIE-Bot-4": "completions_pro", "BLOOMZ-7B": "bloomz_7b1", "Llama-2-7b-chat": "llama_2_7b",