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