update constant for baidu add ernie-speed model

ChatGPTNextWeb · Jul 17, 2024 · 5a0d0c0 · 5a0d0c0
1 parent ec95292
commit 5a0d0c0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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 = [