diff --git a/alfred/fm/google.py b/alfred/fm/google.py index 76a708d..ee31250 100644 --- 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",)