File tree Expand file tree Collapse file tree 2 files changed +3
-0
lines changed
pydantic_ai_slim/pydantic_ai Expand file tree Collapse file tree 2 files changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -254,6 +254,7 @@ async def count_tokens(
254
254
stop_sequences = generation_config .get ('stop_sequences' ),
255
255
presence_penalty = generation_config .get ('presence_penalty' ),
256
256
frequency_penalty = generation_config .get ('frequency_penalty' ),
257
+ seed = generation_config .get ('seed' ),
257
258
thinking_config = generation_config .get ('thinking_config' ),
258
259
media_resolution = generation_config .get ('media_resolution' ),
259
260
response_mime_type = generation_config .get ('response_mime_type' ),
@@ -397,6 +398,7 @@ async def _build_content_and_config(
397
398
stop_sequences = model_settings .get ('stop_sequences' ),
398
399
presence_penalty = model_settings .get ('presence_penalty' ),
399
400
frequency_penalty = model_settings .get ('frequency_penalty' ),
401
+ seed = model_settings .get ('seed' ),
400
402
safety_settings = model_settings .get ('google_safety_settings' ),
401
403
thinking_config = model_settings .get ('google_thinking_config' ),
402
404
labels = model_settings .get ('google_labels' ),
Original file line number Diff line number Diff line change @@ -94,6 +94,7 @@ class ModelSettings(TypedDict, total=False):
94
94
* Groq
95
95
* Cohere
96
96
* Mistral
97
+ * Gemini
97
98
"""
98
99
99
100
presence_penalty : float
You can’t perform that action at this time.
0 commit comments