diff --git a/src/google/adk/models/lite_llm.py b/src/google/adk/models/lite_llm.py index b5dd232bb9..af3b2e7ffb 100644 --- a/src/google/adk/models/lite_llm.py +++ b/src/google/adk/models/lite_llm.py @@ -2419,6 +2419,7 @@ async def _get_completion_inputs( "max_output_tokens", "top_p", "top_k", + "seed", "stop_sequences", "presence_penalty", "frequency_penalty", diff --git a/tests/unittests/models/test_litellm.py b/tests/unittests/models/test_litellm.py index 899c453e86..23e142a931 100644 --- a/tests/unittests/models/test_litellm.py +++ b/tests/unittests/models/test_litellm.py @@ -4758,6 +4758,7 @@ async def test_get_completion_inputs_generation_params(): max_output_tokens=123, top_p=0.88, top_k=7, + seed=42, stop_sequences=["foo", "bar"], presence_penalty=0.1, frequency_penalty=0.2, @@ -4771,6 +4772,7 @@ async def test_get_completion_inputs_generation_params(): assert generation_params["max_completion_tokens"] == 123 assert generation_params["top_p"] == 0.88 assert generation_params["top_k"] == 7 + assert generation_params["seed"] == 42 assert generation_params["stop"] == ["foo", "bar"] assert generation_params["presence_penalty"] == 0.1 assert generation_params["frequency_penalty"] == 0.2