feat: Add inference_generation_config to EvaluationConfig for Tuning

PiperOrigin-RevId: 879306544

Author: google-genai-bot

google/genai/tests/tunings/test_tune.py

@@ -383,6 +383,10 @@ def test_eval_config_with_unified_metrics(client):
             sampling_count=1,
             autorater_model="test-model",
         ),
+        inference_generation_config=genai_types.GenerationConfig(
+            temperature=0.5,
+            max_output_tokens=1024,
+        ),
     )
     tuning_job = client.tunings.tune(
       base_model="gemini-2.5-flash",

google/genai/tunings.py

@@ -585,6 +585,17 @@ def _EvaluationConfig_from_vertex(
         ),
     )
 
+  if getv(from_object, ['inferenceGenerationConfig']) is not None:
+    setv(
+        to_object,
+        ['inference_generation_config'],
+        _GenerationConfig_from_vertex(
+            getv(from_object, ['inferenceGenerationConfig']),
+            to_object,
+            root_object,
+        ),
+    )
+
   return to_object
 
 
@@ -609,6 +620,17 @@ def _EvaluationConfig_to_vertex(
         ),
     )
 
+  if getv(from_object, ['inference_generation_config']) is not None:
+    setv(
+        to_object,
+        ['inferenceGenerationConfig'],
+        _GenerationConfig_to_vertex(
+            getv(from_object, ['inference_generation_config']),
+            to_object,
+            root_object,
+        ),
+    )
+
   return to_object
 
 

google/genai/types.py

@@ -11454,6 +11454,9 @@ class EvaluationConfig(_common.BaseModel):
   autorater_config: Optional[AutoraterConfig] = Field(
       default=None, description="""Autorater config for evaluation."""
   )
+  inference_generation_config: Optional[GenerationConfig] = Field(
+      default=None, description="""Generation config for inference."""
+  )
 
 
 class EvaluationConfigDict(TypedDict, total=False):
@@ -11468,6 +11471,9 @@ class EvaluationConfigDict(TypedDict, total=False):
   autorater_config: Optional[AutoraterConfigDict]
   """Autorater config for evaluation."""
 
+  inference_generation_config: Optional[GenerationConfigDict]
+  """Generation config for inference."""
+
 
 EvaluationConfigOrDict = Union[EvaluationConfig, EvaluationConfigDict]