diff options
| author | sigoden <sigoden@gmail.com> | 2024-04-10 07:44:36 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-10 07:44:36 +0800 |
| commit | 07605d892a4bcfb75f3c8daf873e180f424c335b (patch) | |
| tree | a919e18e2c582674f2541504ab078c06a1e81c13 /src/client/gemini.rs | |
| parent | 26ad75c00583d5f36c44ce79083a7ba69c1201bf (diff) | |
| download | aichat-07605d892a4bcfb75f3c8daf873e180f424c335b.tar.gz | |
refactor: update gemini/vertexai models list (#395)
Diffstat (limited to 'src/client/gemini.rs')
| -rw-r--r-- | src/client/gemini.rs | 8 |
1 files changed, 4 insertions, 4 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); |