Adding new gemini experimental models (#811)

sydney-runkle · web-flow · commit 1cf76e6d1b56 · 2025-01-30T07:18:36.000-07:00
diff --git a/pydantic_ai_slim/pydantic_ai/models/__init__.py b/pydantic_ai_slim/pydantic_ai/models/__init__.py
@@ -52,11 +52,15 @@
     'google-gla:gemini-1.5-flash-8b',
     'google-gla:gemini-1.5-pro',
     'google-gla:gemini-2.0-flash-exp',
+    'google-gla:gemini-2.0-flash-thinking-exp-01-21',
+    'google-gla:gemini-exp-1206',
     'google-vertex:gemini-1.0-pro',
     'google-vertex:gemini-1.5-flash',
     'google-vertex:gemini-1.5-flash-8b',
     'google-vertex:gemini-1.5-pro',
     'google-vertex:gemini-2.0-flash-exp',
+    'google-vertex:gemini-2.0-flash-thinking-exp-01-21',
+    'google-vertex:gemini-exp-1206',
     'gpt-3.5-turbo',
     'gpt-3.5-turbo-0125',
     'gpt-3.5-turbo-0301',
diff --git a/pydantic_ai_slim/pydantic_ai/models/gemini.py b/pydantic_ai_slim/pydantic_ai/models/gemini.py
@@ -40,7 +40,13 @@
 )
 
 GeminiModelName = Literal[
-    'gemini-1.5-flash', 'gemini-1.5-flash-8b', 'gemini-1.5-pro', 'gemini-1.0-pro', 'gemini-2.0-flash-exp'
+    'gemini-1.5-flash',
+    'gemini-1.5-flash-8b',
+    'gemini-1.5-pro',
+    'gemini-1.0-pro',
+    'gemini-2.0-flash-exp',
+    'gemini-2.0-flash-thinking-exp-01-21',
+    'gemini-exp-1206',
 ]
 """Named Gemini models.
 

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,13 @@`
`40`	`40`	`)`
`41`	`41`
`42`	`42`	`GeminiModelName = Literal[`
`43`		`- 'gemini-1.5-flash', 'gemini-1.5-flash-8b', 'gemini-1.5-pro', 'gemini-1.0-pro', 'gemini-2.0-flash-exp'`
	`43`	`+ 'gemini-1.5-flash',`
	`44`	`+ 'gemini-1.5-flash-8b',`
	`45`	`+ 'gemini-1.5-pro',`
	`46`	`+ 'gemini-1.0-pro',`
	`47`	`+ 'gemini-2.0-flash-exp',`
	`48`	`+ 'gemini-2.0-flash-thinking-exp-01-21',`
	`49`	`+ 'gemini-exp-1206',`
`44`	`50`	`]`
`45`	`51`	`"""Named Gemini models.`
`46`	`52`