Adding text generation model supports for gemini 1.5

BatsResearch · May 19, 2024 · 4868157 · 4868157
1 parent f2ba05d
commit 4868157
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alfred/fm/google.py b/alfred/fm/google.py
@@ -23,9 +23,9 @@
         "Google GenAI module not found. Please install google-generativeai to use the Google model."
     )
 
-GOOGLE_GENAI_MODELS = ("gemini-pro", "gemini-1.5-flash")
+GOOGLE_GENAI_MODELS = ("gemini-pro", "gemini-1.5-flash", "gemini-1.5-pro-latest")
 
-GOOGLE_GENAI_VISION_MODELS = ("gemini-pro-vision", "gemini-1.5-pro-latest", "gemini-1.5-flash")
+GOOGLE_GENAI_VISION_MODELS = ("gemini-pro-vision") 
 
 GOOGLE_GENAI_EMBEDDING_MODELS = ("embedding-001",)