diff options
| author | sigoden <sigoden@gmail.com> | 2024-04-06 09:10:42 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-06 09:10:42 +0800 |
| commit | e28b53a405256f040db329fbb23676b0e9f10774 (patch) | |
| tree | 0117999e705599a582c7e0baa6e3112365f96cf9 /src/client/ernie.rs | |
| parent | 2654542601df9cc472ccc17487d8852dcb2c0d1d (diff) | |
| download | aichat-e28b53a405256f040db329fbb23676b0e9f10774.tar.gz | |
refactor: update models (#388)
Diffstat (limited to 'src/client/ernie.rs')
| -rw-r--r-- | src/client/ernie.rs | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/client/ernie.rs b/src/client/ernie.rs index 5c8b563..060f89b 100644 --- a/src/client/ernie.rs +++ b/src/client/ernie.rs @@ -15,7 +15,7 @@ use std::{env, sync::Mutex}; 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, usize, &str); 6] = [ +const MODELS: [(&str, usize, &str); 7] = [ // https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t ("ernie-4.0-8k", 5120, "/wenxinworkshop/chat/completions_pro"), ( @@ -35,6 +35,7 @@ const MODELS: [(&str, usize, &str); 6] = [ "/wenxinworkshop/chat/ernie-speed-128k", ), ("ernie-lite-8k", 7168, "/wenxinworkshop/chat/ernie-lite-8k"), + ("ernie-tiny-8k", 7168, "/wenxinworkshop/chat/ernie-tiny-8k"), ]; lazy_static! { |