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(
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' ),
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments