Adding text generation model supports for ge

BatsResearch · May 19, 2024 · cedff96 · cedff96
1 parent 4868157
commit cedff96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alfred/fm/google.py b/alfred/fm/google.py
@@ -25,7 +25,7 @@
 
 GOOGLE_GENAI_MODELS = ("gemini-pro", "gemini-1.5-flash", "gemini-1.5-pro-latest")
 
-GOOGLE_GENAI_VISION_MODELS = ("gemini-pro-vision") 
+GOOGLE_GENAI_VISION_MODELS = ("gemini-pro-vision",) 
 
 GOOGLE_GENAI_EMBEDDING_MODELS = ("embedding-001",)