diff options
| -rw-r--r-- | src/client/gemini.rs | 8 | ||||
| -rw-r--r-- | src/client/vertexai.rs | 6 |
2 files changed, 6 insertions, 8 deletions
diff --git a/src/client/gemini.rs b/src/client/gemini.rs index 39371fd..19e656a 100644 --- a/src/client/gemini.rs +++ b/src/client/gemini.rs @@ -10,11 +10,11 @@ use serde::Deserialize; const API_BASE: &str = "https://generativelanguage.googleapis.com/v1beta/models/"; -const MODELS: [(&str, usize, &str); 2] = [ +const MODELS: [(&str, usize, &str); 3] = [ // https://ai.google.dev/models/gemini - ("gemini-pro", 30720, "text"), - ("gemini-pro-vision", 12288, "vision"), - // ("gemini-1.5-pro", 1048576, "text,vision"), + ("gemini-1.0-pro-latest", 30720, "text"), + ("gemini-1.0-pro-vision-latest", 12288, "text,vision"), + ("gemini-1.5-pro-latest", 1048576, "text,vision"), ]; const TOKENS_COUNT_FACTORS: TokensCountFactors = (5, 2); diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs index 3c663f4..1c1fd46 100644 --- a/src/client/vertexai.rs +++ b/src/client/vertexai.rs @@ -14,13 +14,11 @@ use serde::Deserialize; use serde_json::{json, Value}; use std::path::PathBuf; -const MODELS: [(&str, usize, &str); 5] = [ +const MODELS: [(&str, usize, &str); 3] = [ // https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models ("gemini-1.0-pro", 24568, "text"), ("gemini-1.0-pro-vision", 14336, "text,vision"), - ("gemini-1.0-ultra", 8192, "text"), - ("gemini-1.0-ultra-vision", 8192, "text,vision"), - ("gemini-1.5-pro", 1000000, "text"), + ("gemini-1.5-pro-preview-0409", 1000000, "text,vision"), ]; const TOKENS_COUNT_FACTORS: TokensCountFactors = (5, 2); |