update constant for baidu add ernie-speed model

This commit is contained in:
consistent-k 2024-07-17 09:29:43 +00:00
parent 5a0d0c0b75
commit 5dc731bc77
1 changed files with 6 additions and 2 deletions

View File

@ -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 = [