From 7c2c949f01ba1919ff2d2a8fb4c8d54a39fffcb7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Little=20=E7=BE=8A?=
 <61880568+Littlesheepxy@users.noreply.github.com>
Date: Fri, 12 Jul 2024 19:54:53 +0800
Subject: [PATCH] Update ernie_bot.py (#6236)

---
 api/core/model_runtime/model_providers/wenxin/llm/ernie_bot.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/api/core/model_runtime/model_providers/wenxin/llm/ernie_bot.py b/api/core/model_runtime/model_providers/wenxin/llm/ernie_bot.py
index 305769c1c1..9aeab04cd2 100644
--- a/api/core/model_runtime/model_providers/wenxin/llm/ernie_bot.py
+++ b/api/core/model_runtime/model_providers/wenxin/llm/ernie_bot.py
@@ -138,6 +138,7 @@ class ErnieBotModel:
         'ernie-lite-8k-0922': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/eb-instant',
         'ernie-lite-8k-0308': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-lite-8k',
         'ernie-character-8k-0321': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-char-8k',
+        'ernie-4.0-tutbo-8k-preview': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-4.0-turbo-8k-preview',
     }
 
     function_calling_supports = [
@@ -149,6 +150,7 @@ class ErnieBotModel:
         'ernie-3.5-4k-0205',
         'ernie-3.5-128k',
         'ernie-4.0-8k'
+        'ernie-4.0-turbo-8k-preview'
     ]
 
     api_key: str = ''