feat: support hyperparameters in distillation tuning

PiperOrigin-RevId: 882708166

Author: speedstorm1

Google.GenAI/Tunings.cs

@@ -355,6 +355,11 @@ internal JsonNode CreateTuningJobConfigToVertex(JsonNode fromObject, JsonObject
           Common.SetValueByPath(parentObject, new string[] { "supervisedTuningSpec", "tuningMode" },
                                 Common.GetValueByPath(fromObject, new string[] { "tuningMode" }));
         }
+      } else if (discriminatorValueTuningMode == "DISTILLATION") {
+        if (Common.GetValueByPath(fromObject, new string[] { "tuningMode" }) != null) {
+          Common.SetValueByPath(parentObject, new string[] { "distillationSpec", "tuningMode" },
+                                Common.GetValueByPath(fromObject, new string[] { "tuningMode" }));
+        }
       }
       if (Common.GetValueByPath(fromObject, new string[] { "customBaseModel" }) != null) {
         Common.SetValueByPath(
@@ -373,6 +378,12 @@ internal JsonNode CreateTuningJobConfigToVertex(JsonNode fromObject, JsonObject
               parentObject, new string[] { "supervisedTuningSpec", "hyperParameters", "batchSize" },
               Common.GetValueByPath(fromObject, new string[] { "batchSize" }));
         }
+      } else if (discriminatorValueBatchSize == "DISTILLATION") {
+        if (Common.GetValueByPath(fromObject, new string[] { "batchSize" }) != null) {
+          Common.SetValueByPath(parentObject,
+                                new string[] { "distillationSpec", "hyperParameters", "batchSize" },
+                                Common.GetValueByPath(fromObject, new string[] { "batchSize" }));
+        }
       }
 
       JsonNode discriminatorLearningRate =
@@ -387,6 +398,12 @@ internal JsonNode CreateTuningJobConfigToVertex(JsonNode fromObject, JsonObject
               new string[] { "supervisedTuningSpec", "hyperParameters", "learningRate" },
               Common.GetValueByPath(fromObject, new string[] { "learningRate" }));
         }
+      } else if (discriminatorValueLearningRate == "DISTILLATION") {
+        if (Common.GetValueByPath(fromObject, new string[] { "learningRate" }) != null) {
+          Common.SetValueByPath(
+              parentObject, new string[] { "distillationSpec", "hyperParameters", "learningRate" },
+              Common.GetValueByPath(fromObject, new string[] { "learningRate" }));
+        }
       }
 
       if (Common.GetValueByPath(fromObject, new string[] { "labels" }) != null) {

Google.GenAI/types/CreateTuningJobConfig.cs

@@ -129,7 +129,7 @@ public AdapterSize
           }
 
     /// <summary>
-    /// Tuning mode for SFT tuning.
+    /// Tuning mode for tuning.
     /// </summary>
     [JsonPropertyName("tuningMode")]
     [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]

Google.GenAI/types/DistillationHyperParameters.cs

@@ -23,7 +23,7 @@
 
 namespace Google.GenAI.Types {
   /// <summary>
-  /// Hyperparameters for Distillation. This data type is not supported in Gemini API.
+  /// Hyperparameters for distillation.
   /// </summary>
 
   public record DistillationHyperParameters {
@@ -56,6 +56,26 @@ public double
             get; set;
           }
 
+    /// <summary>
+    /// The batch size hyperparameter for tuning. This is only supported for OSS models in Vertex.
+    /// </summary>
+    [JsonPropertyName("batchSize")]
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public int
+        ? BatchSize {
+            get; set;
+          }
+
+    /// <summary>
+    /// The learning rate for tuning. OSS models only.
+    /// </summary>
+    [JsonPropertyName("learningRate")]
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public double
+        ? LearningRate {
+            get; set;
+          }
+
     /// <summary>
     /// Deserializes a JSON string to a DistillationHyperParameters object.
     /// </summary>

Google.GenAI/types/DistillationSpec.cs

@@ -111,6 +111,16 @@ public string
             get; set;
           }
 
+    /// <summary>
+    /// Tuning mode for tuning.
+    /// </summary>
+    [JsonPropertyName("tuningMode")]
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public TuningMode
+        ? TuningMode {
+            get; set;
+          }
+
     /// <summary>
     /// Deserializes a JSON string to a DistillationSpec object.
     /// </summary>