Skip to content

Commit 085a4c3

Browse files
DouweMclaude[bot]
andauthored
Support ModelSettings.seed in GoogleModel (#2842)
Co-authored-by: claude[bot] <209825114+claude[bot]@users.noreply.github.com> Co-authored-by: Douwe Maan <[email protected]>
1 parent 0047a68 commit 085a4c3

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

pydantic_ai_slim/pydantic_ai/models/google.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,7 @@ async def count_tokens(
254254
stop_sequences=generation_config.get('stop_sequences'),
255255
presence_penalty=generation_config.get('presence_penalty'),
256256
frequency_penalty=generation_config.get('frequency_penalty'),
257+
seed=generation_config.get('seed'),
257258
thinking_config=generation_config.get('thinking_config'),
258259
media_resolution=generation_config.get('media_resolution'),
259260
response_mime_type=generation_config.get('response_mime_type'),
@@ -397,6 +398,7 @@ async def _build_content_and_config(
397398
stop_sequences=model_settings.get('stop_sequences'),
398399
presence_penalty=model_settings.get('presence_penalty'),
399400
frequency_penalty=model_settings.get('frequency_penalty'),
401+
seed=model_settings.get('seed'),
400402
safety_settings=model_settings.get('google_safety_settings'),
401403
thinking_config=model_settings.get('google_thinking_config'),
402404
labels=model_settings.get('google_labels'),

pydantic_ai_slim/pydantic_ai/settings.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,7 @@ class ModelSettings(TypedDict, total=False):
9494
* Groq
9595
* Cohere
9696
* Mistral
97+
* Gemini
9798
"""
9899

99100
presence_penalty: float

0 commit comments

Comments
 (0)