Merge pull request #5040 from consistent-k/main

update constant for baidu  add ENIRE Speed  ENIRE Lite model
This commit is contained in:
Dogtiti
2024-07-17 19:05:31 +08:00
committed by GitHub

View File

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