Skip to content

Commit 668e024

Browse files
chelsealongcopybara-github
authored andcommitted
fix: forward GenerateContentConfig.seed to LiteLLM
Merge #6516 Closes #6513 PiperOrigin-RevId: 978131122
1 parent b89037d commit 668e024

2 files changed

Lines changed: 3 additions & 0 deletions

File tree

‎src/google/adk/models/lite_llm.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2800,6 +2800,7 @@ async def _get_completion_inputs(
28002800
"max_output_tokens",
28012801
"top_p",
28022802
"top_k",
2803+
"seed",
28032804
"stop_sequences",
28042805
"presence_penalty",
28052806
"frequency_penalty",

‎tests/unittests/models/test_litellm.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5683,6 +5683,7 @@ async def test_get_completion_inputs_generation_params():
56835683
max_output_tokens=123,
56845684
top_p=0.88,
56855685
top_k=7,
5686+
seed=42,
56865687
stop_sequences=["foo", "bar"],
56875688
presence_penalty=0.1,
56885689
frequency_penalty=0.2,
@@ -5696,6 +5697,7 @@ async def test_get_completion_inputs_generation_params():
56965697
assert generation_params["max_completion_tokens"] == 123
56975698
assert generation_params["top_p"] == 0.88
56985699
assert generation_params["top_k"] == 7
5700+
assert generation_params["seed"] == 42
56995701
assert generation_params["stop"] == ["foo", "bar"]
57005702
assert generation_params["presence_penalty"] == 0.1
57015703
assert generation_params["frequency_penalty"] == 0.2

0 commit comments

Comments
 (0)