diff --git a/app/constant.ts b/app/constant.ts index a146200d63b..e0b3d227eab 100644 --- a/app/constant.ts +++ b/app/constant.ts @@ -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 = [