Просмотр исходного кода

Merge pull request #5040 from consistent-k/main

update constant for baidu  add ENIRE Speed  ENIRE Lite model
Dogtiti 1 год назад
Родитель
Сommit
2e9f701bb7
1 измененных файлов с 10 добавлено и 0 удалено
  1. 10 0
      app/constant.ts

+ 10 - 0
app/constant.ts

@@ -134,6 +134,12 @@ export const Baidu = {
     if (modelName === "ernie-3.5-8k") {
     if (modelName === "ernie-3.5-8k") {
       endpoint = "completions";
       endpoint = "completions";
     }
     }
+    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}`;
     return `rpc/2.0/ai_custom/v1/wenxinworkshop/chat/${endpoint}`;
   },
   },
 };
 };
@@ -225,6 +231,10 @@ const baiduModels = [
   "ernie-4.0-8k-latest",
   "ernie-4.0-8k-latest",
   "ernie-3.5-8k",
   "ernie-3.5-8k",
   "ernie-3.5-8k-0205",
   "ernie-3.5-8k-0205",
+  "ernie-speed-128k",
+  "ernie-speed-8k",
+  "ernie-lite-8k",
+  "ernie-tiny-8k"
 ];
 ];
 
 
 const bytedanceModels = [
 const bytedanceModels = [