feat(gemini): change model

blackfoot-makers · Dec 16, 2023 · 87d744a · 87d744a
1 parent 963dec7
commit 87d744a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/features/gemini.rs b/src/features/gemini.rs
@@ -103,7 +103,7 @@ struct GeminiResult {
 async fn query_gemini(question: &str) -> Result<Vec<GeminiResult>, &str> {
   const API_ENDPOINT: &str = "us-east4-aiplatform.googleapis.com";
   const PROJECT_ID: &str = "blackfoot-dev";
-  const MODEL_ID: &str = "gemini-pro-vision";
+  const MODEL_ID: &str = "gemini-pro";
   const LOCATION_ID: &str = "us-east4";
 
   let url = format!("https://{API_ENDPOINT}/v1beta1/projects/{PROJECT_ID}/locations/{LOCATION_ID}/publishers/google/models/{MODEL_ID}:streamGenerateContent");