Add support for new Gemini model 'gemini-2.0-flash-exp' (#223)

Etelis · EtelisIBM · web-flow · commit 560ecb21812d · 2024-12-12T11:44:45.000-05:00
Co-authored-by: Itay Etelis &lt;itay.etelis@ibm.com&gt;
diff --git a/pydantic_ai_slim/pydantic_ai/models/__init__.py b/pydantic_ai_slim/pydantic_ai/models/__init__.py
@@ -48,6 +48,7 @@
     'groq:gemma-7b-it',
     'gemini-1.5-flash',
     'gemini-1.5-pro',
+    'gemini-2.0-flash-exp',
     'vertexai:gemini-1.5-flash',
     'vertexai:gemini-1.5-pro',
     'ollama:codellama',
diff --git a/pydantic_ai_slim/pydantic_ai/models/gemini.py b/pydantic_ai_slim/pydantic_ai/models/gemini.py
@@ -37,7 +37,9 @@
     get_user_agent,
 )
 
-GeminiModelName = Literal['gemini-1.5-flash', 'gemini-1.5-flash-8b', 'gemini-1.5-pro', 'gemini-1.0-pro']
+GeminiModelName = Literal[
+    'gemini-1.5-flash', 'gemini-1.5-flash-8b', 'gemini-1.5-pro', 'gemini-1.0-pro', 'gemini-2.0-flash-exp'
+]
 """Named Gemini models.
 
 See [the Gemini API docs](https://ai.google.dev/gemini-api/docs/models/gemini#model-variations) for a full list.

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,9 @@`
`37`	`37`	`get_user_agent,`
`38`	`38`	`)`
`39`	`39`
`40`		`-GeminiModelName = Literal['gemini-1.5-flash', 'gemini-1.5-flash-8b', 'gemini-1.5-pro', 'gemini-1.0-pro']`
	`40`	`+GeminiModelName = Literal[`
	`41`	`+ 'gemini-1.5-flash', 'gemini-1.5-flash-8b', 'gemini-1.5-pro', 'gemini-1.0-pro', 'gemini-2.0-flash-exp'`
	`42`	`+]`
`41`	`43`	`"""Named Gemini models.`
`42`	`44`
`43`	`45`	`See [the Gemini API docs](https://ai.google.dev/gemini-api/docs/models/gemini#model-variations) for a full list.`