feat: update openai/qianwen/gemini models (#306)

pull/308/head
sigoden 4 months ago committed by GitHub
parent 25b27c148a
commit 3bf0c371e4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -14,10 +14,9 @@ use serde_json::{json, Value};
const API_BASE: &str = "https://generativelanguage.googleapis.com/v1beta/models/";
const MODELS: [(&str, usize, &str); 3] = [
const MODELS: [(&str, usize, &str); 2] = [
("gemini-pro", 32768, "text"),
("gemini-pro-vision", 16384, "vision"),
("gemini-ultra", 32768, "text"),
];
const TOKENS_COUNT_FACTORS: TokensCountFactors = (5, 2);

@ -13,12 +13,12 @@ use serde_json::{json, Value};
const API_BASE: &str = "https://api.openai.com/v1";
const MODELS: [(&str, usize, &str); 7] = [
("gpt-3.5-turbo", 4096, "text"),
("gpt-3.5-turbo-16k", 16385, "text"),
("gpt-3.5-turbo", 16385, "text"),
("gpt-3.5-turbo-1106", 16385, "text"),
("gpt-4", 8192, "text"),
("gpt-4-32k", 32768, "text"),
("gpt-4-1106-preview", 128000, "text"),
("gpt-4-turbo-preview", 128000, "text"),
("gpt-4-vision-preview", 128000, "text,vision"),
];

@ -24,12 +24,13 @@ const API_URL: &str =
const API_URL_VL: &str =
"https://dashscope.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation";
const MODELS: [(&str, usize, &str); 5] = [
const MODELS: [(&str, usize, &str); 6] = [
("qwen-turbo", 8192, "text"),
("qwen-plus", 32768, "text"),
("qwen-max", 8192, "text"),
("qwen-max-longcontext", 30720, "text"),
("qwen-vl-plus", 0, "text,vision"),
("qwen-vl-max", 0, "text,vision"),
];
#[derive(Debug, Clone, Deserialize, Default)]

Loading…
Cancel
Save