Jelajahi Sumber

update constant for baidu add ernie-speed model

consistent-k 1 tahun lalu
induk
melakukan
5dc731bc77
1 mengubah file dengan 6 tambahan dan 2 penghapusan
  1. 6 2
      app/constant.ts

+ 6 - 2
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 = [