From 12588b9180e458afd60dbe8437c269f441952dc4 Mon Sep 17 00:00:00 2001
From: Google Team Member <genai-sdk-bot@google.com>
Date: Tue, 17 Feb 2026 16:15:04 -0800
Subject: [PATCH] feat: Add inference_generation_config to EvaluationConfig for
 Tuning

PiperOrigin-RevId: 871544397
---
 src/test/java/com/google/genai/TuningsTest.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/test/java/com/google/genai/TuningsTest.java b/src/test/java/com/google/genai/TuningsTest.java
index 3f26067de60..f9ad4dbd2d4 100644
--- a/src/test/java/com/google/genai/TuningsTest.java
+++ b/src/test/java/com/google/genai/TuningsTest.java
@@ -254,6 +254,8 @@ public void testTuneWithEvaluationConfig(boolean vertexAI) {
             .autoraterConfig(
                 AutoraterConfig.builder().autoraterModel("test-model").samplingCount(1).build())
             .metrics(metrics)
+            .inferenceGenerationConfig(
+                GenerationConfig.builder().temperature(0.5f).maxOutputTokens(1024).build())
             .build();
     CreateTuningJobConfig tuningConfig =
         CreateTuningJobConfig.builder()
@@ -277,6 +279,7 @@ public void testTuneWithEvaluationConfig(boolean vertexAI) {
     // Assert
     assertNotNull(currentJob);
     assertTrue(currentJob.evaluationConfig().isPresent());
+    assertTrue(currentJob.evaluationConfig().get().inferenceGenerationConfig().isPresent());
     assertTrue(currentJob.state().get().knownEnum() == JobState.Known.JOB_STATE_PENDING);
   }
 }