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

3 files changed, 5 insertions, 5 deletions

diff --git a/src/client/gemini.rs b/src/client/gemini.rs
index 6ff3f95..6f3141c 100644
--- a/src/client/gemini.rs
+++ b/src/client/gemini.rs
@@ -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);
diff --git a/src/client/openai.rs b/src/client/openai.rs
index cf6cc89..9c28473 100644
--- a/src/client/openai.rs
+++ b/src/client/openai.rs
@@ -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"),
 ];
 
diff --git a/src/client/qianwen.rs b/src/client/qianwen.rs
index b9254ff..70c251a 100644
--- a/src/client/qianwen.rs
+++ b/src/client/qianwen.rs
@@ -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)]