summaryrefslogtreecommitdiffstats
path: root/src/client/vertexai.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-03-25 09:21:46 +0800
committerGitHub <noreply@github.com>2024-03-25 09:21:46 +0800
commitbbd0c287261f8876d8c66a4e8015f5a0d8ce6548 (patch)
tree0bc74b4d8e228d0bc1a64188a75a250b366c46c3 /src/client/vertexai.rs
parent774d991144915368f9eee9740091448b92c17e62 (diff)
downloadaichat-bbd0c287261f8876d8c66a4e8015f5a0d8ce6548.tar.gz
refactor: reorder models (#372)
The capable ones come first.
Diffstat (limited to 'src/client/vertexai.rs')
-rw-r--r--src/client/vertexai.rs10
1 files changed, 5 insertions, 5 deletions
diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs
index ffe9151..ba43796 100644
--- a/src/client/vertexai.rs
+++ b/src/client/vertexai.rs
@@ -14,13 +14,13 @@ use serde::Deserialize;
use serde_json::{json, Value};
use std::path::PathBuf;
-// https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models
-const MODELS: [(&str, usize, &str); 5] = [
+const MODELS: [(&str, usize, &str); 2] = [
+ // 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.0-ultra", 8192, "text"),
+ // ("gemini-1.0-ultra-vision", 8192, "text,vision"),
+ // ("gemini-1.5-pro", 1000000, "text"),
];
const TOKENS_COUNT_FACTORS: TokensCountFactors = (5, 2);