From 5a0d0c0b75a9eb4d9a6853457195dd954bf528c3 Mon Sep 17 00:00:00 2001
From: consistent-k <67736912+consistent-k@users.noreply.github.com>
Date: Wed, 17 Jul 2024 08:54:53 +0000
Subject: [PATCH 1/3] update constant for baidu add ernie-speed model

---
 app/constant.ts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/app/constant.ts b/app/constant.ts
index a146200d63b..a6c9eadc033 100644
--- a/app/constant.ts
+++ b/app/constant.ts
@@ -134,6 +134,9 @@ export const Baidu = {
     if (modelName === "ernie-3.5-8k") {
       endpoint = "completions";
     }
+    if (modelName === "ernie-speed") {
+      endpoint = "ernie_speed";
+    }
     return `rpc/2.0/ai_custom/v1/wenxinworkshop/chat/${endpoint}`;
   },
 };
@@ -225,6 +228,7 @@ const baiduModels = [
   "ernie-4.0-8k-latest",
   "ernie-3.5-8k",
   "ernie-3.5-8k-0205",
+  "ernie-speed"
 ];
 
 const bytedanceModels = [

From 5dc731bc7713f6506f685a8a6eaa4ffa4d52f6ab Mon Sep 17 00:00:00 2001
From: consistent-k <67736912+consistent-k@users.noreply.github.com>
Date: Wed, 17 Jul 2024 09:29:43 +0000
Subject: [PATCH 2/3] update constant for baidu add ernie-speed model

---
 app/constant.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/app/constant.ts b/app/constant.ts
index a6c9eadc033..2488a57afdb 100644
--- a/app/constant.ts
+++ b/app/constant.ts
@@ -134,7 +134,10 @@ export const Baidu = {
     if (modelName === "ernie-3.5-8k") {
       endpoint = "completions";
     }
-    if (modelName === "ernie-speed") {
+    if (modelName === "ernie-speed-128k") {
+      endpoint = "ernie-speed-128k";
+    }
+    if (modelName === "ernie-speed-8k") {
       endpoint = "ernie_speed";
     }
     return `rpc/2.0/ai_custom/v1/wenxinworkshop/chat/${endpoint}`;
@@ -228,7 +231,8 @@ const baiduModels = [
   "ernie-4.0-8k-latest",
   "ernie-3.5-8k",
   "ernie-3.5-8k-0205",
-  "ernie-speed"
+  "ernie-speed-128k",
+  "ernie-speed-8k"
 ];
 
 const bytedanceModels = [

From 9aabc4ad6add38e662d6338b3ae724f18690322e Mon Sep 17 00:00:00 2001
From: consistent-k <67736912+consistent-k@users.noreply.github.com>
Date: Wed, 17 Jul 2024 10:41:31 +0000
Subject: [PATCH 3/3] update constant for baidu add ernie-tiny model

---
 app/constant.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/app/constant.ts b/app/constant.ts
index 2488a57afdb..e0b3d227eab 100644
--- a/app/constant.ts
+++ b/app/constant.ts
@@ -232,7 +232,9 @@ const baiduModels = [
   "ernie-3.5-8k",
   "ernie-3.5-8k-0205",
   "ernie-speed-128k",
-  "ernie-speed-8k"
+  "ernie-speed-8k",
+  "ernie-lite-8k",
+  "ernie-tiny-8k"
 ];
 
 const bytedanceModels = [