add gemini-1.5-flash to GeminiPro provider

This commit is contained in:
mcareddu 2024-06-13 14:32:52 +02:00
parent d6ac24e875
commit e678417a7c

View File

@ -18,7 +18,7 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin):
needs_auth = True
default_model = "gemini-1.5-pro-latest"
default_vision_model = default_model
models = [default_model, "gemini-pro", "gemini-pro-vision"]
models = [default_model, "gemini-pro", "gemini-pro-vision", "gemini-1.5-flash"]
@classmethod
async def create_async_generator(