diff options
| author | sigoden <sigoden@gmail.com> | 2023-11-27 16:06:35 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-27 16:06:35 +0800 |
| commit | 18f16c6511e20d11ea1ae4b4d81b43d4d35486b3 (patch) | |
| tree | b2e3022c39c7db4927f1a35eecc1193f46c953e8 /src/client/ernie.rs | |
| parent | 2508d56598a37844e369ab623ceb5bc7b2c78d38 (diff) | |
| download | aichat-18f16c6511e20d11ea1ae4b4d81b43d4d35486b3.tar.gz | |
feat: add ernie:ernie-bot-8k qianwen:qwen-max (#252)
Diffstat (limited to 'src/client/ernie.rs')
| -rw-r--r-- | src/client/ernie.rs | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/client/ernie.rs b/src/client/ernie.rs index 4bb3435..75a362a 100644 --- a/src/client/ernie.rs +++ b/src/client/ernie.rs @@ -18,9 +18,10 @@ use std::env; const API_BASE: &str = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1"; const ACCESS_TOKEN_URL: &str = "https://aip.baidubce.com/oauth/2.0/token"; -const MODELS: [(&str, &str); 3] = [ - ("eb-instant", "/wenxinworkshop/chat/eb-instant"), +const MODELS: [(&str, &str); 4] = [ + ("ernie-bot-turbo", "/wenxinworkshop/chat/eb-instant"), ("ernie-bot", "/wenxinworkshop/chat/completions"), + ("ernie-bot-8k", "/wenxinworkshop/chat/ernie_bot_8k"), ("ernie-bot-4", "/wenxinworkshop/chat/completions_pro"), ]; |