Accomondating Gemini 1.5 Models

BatsResearch · May 19, 2024 · f2ba05d · f2ba05d
1 parent 5408b6c
commit f2ba05d
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",)
+GOOGLE_GENAI_MODELS = ("gemini-pro", "gemini-1.5-flash")
 
-GOOGLE_GENAI_VISION_MODELS = ("gemini-pro-vision",)
+GOOGLE_GENAI_VISION_MODELS = ("gemini-pro-vision", "gemini-1.5-pro-latest", "gemini-1.5-flash")
 
 GOOGLE_GENAI_EMBEDDING_MODELS = ("embedding-001",)