Skip to content

Commit 9f1e073

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: GenAI - Added seed parameter to the GenerationConfig class
PiperOrigin-RevId: 665144542
1 parent 24393ec commit 9f1e073

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

tests/system/vertexai/test_generative_models.py

+1
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,7 @@ def test_generate_content_with_gemini_15_parameters(self, api_endpoint):
227227
top_p=0.95,
228228
top_k=20,
229229
candidate_count=1,
230+
seed=5,
230231
max_output_tokens=100,
231232
stop_sequences=["STOP!"],
232233
presence_penalty=0.0,

vertexai/generative_models/_generative_models.py

+4
Original file line numberDiff line numberDiff line change
@@ -1406,6 +1406,7 @@ def __init__(
14061406
frequency_penalty: Optional[float] = None,
14071407
response_mime_type: Optional[str] = None,
14081408
response_schema: Optional[Dict[str, Any]] = None,
1409+
seed: Optional[int] = None,
14091410
):
14101411
r"""Constructs a GenerationConfig object.
14111412
@@ -1414,6 +1415,7 @@ def __init__(
14141415
top_p: If specified, nucleus sampling will be used. Range: (0.0, 1.0]
14151416
top_k: If specified, top-k sampling will be used.
14161417
candidate_count: Number of candidates to generate.
1418+
seed: Random seed for the generation.
14171419
max_output_tokens: The maximum number of output tokens to generate per message.
14181420
stop_sequences: A list of stop sequences.
14191421
presence_penalty: Positive values penalize tokens that have appeared in the generated text,
@@ -1442,6 +1444,7 @@ def __init__(
14421444
candidate_count=1,
14431445
max_output_tokens=100,
14441446
stop_sequences=["\n\n\n"],
1447+
seed=5,
14451448
)
14461449
)
14471450
```
@@ -1462,6 +1465,7 @@ def __init__(
14621465
frequency_penalty=frequency_penalty,
14631466
response_mime_type=response_mime_type,
14641467
response_schema=raw_schema,
1468+
seed=seed,
14651469
)
14661470

14671471
@classmethod

0 commit comments

Comments
 (0)