summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-03-19 16:06:53 +0800
committerGitHub <noreply@github.com>2024-03-19 16:06:53 +0800
commit4d52479545082254315171d6716ee9c3588f76f5 (patch)
tree138b44e5548cece3d5b217cbe5a21253822bec53
parent7279e431416673716b217ab16ed238ad082812f1 (diff)
downloadaichat-4d52479545082254315171d6716ee9c3588f76f5.tar.gz
fix: vertexai model names (#365)
-rw-r--r--src/client/vertexai.rs4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs
index 47d9432..ffe9151 100644
--- a/src/client/vertexai.rs
+++ b/src/client/vertexai.rs
@@ -17,9 +17,9 @@ use std::path::PathBuf;
// https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models
const MODELS: [(&str, usize, &str); 5] = [
("gemini-1.0-pro", 24568, "text"),
- ("gemini.1.0-pro-vision", 14336, "text,vision"),
+ ("gemini-1.0-pro-vision", 14336, "text,vision"),
("gemini-1.0-ultra", 8192, "text"),
- ("gemini.1.0-ultra-vision", 8192, "text,vision"),
+ ("gemini-1.0-ultra-vision", 8192, "text,vision"),
("gemini-1.5-pro", 1000000, "text"),
];