feat: Add labels to create tuning job config

sararob · copybara-github · commit 695e17a7b1ad · 2025-09-09T14:39:59.000-07:00
PiperOrigin-RevId: 805069788
diff --git a/src/main/java/com/google/genai/Tunings.java b/src/main/java/com/google/genai/Tunings.java
@@ -248,6 +248,10 @@ ObjectNode createTuningJobConfigToMldev(JsonNode fromObject, ObjectNode parentOb
           Common.getValueByPath(fromObject, new String[] {"learningRate"}));
     }
 
+    if (!Common.isZero(Common.getValueByPath(fromObject, new String[] {"labels"}))) {
+      throw new IllegalArgumentException("labels parameter is not supported in Gemini API.");
+    }
+
     return toObject;
   }
 
@@ -490,6 +494,13 @@ ObjectNode createTuningJobConfigToVertex(JsonNode fromObject, ObjectNode parentO
       throw new IllegalArgumentException("learningRate parameter is not supported in Vertex AI.");
     }
 
+    if (Common.getValueByPath(fromObject, new String[] {"labels"}) != null) {
+      Common.setValueByPath(
+          parentObject,
+          new String[] {"labels"},
+          Common.getValueByPath(fromObject, new String[] {"labels"}));
+    }
+
     return toObject;
   }
 
diff --git a/src/main/java/com/google/genai/types/CreateTuningJobConfig.java b/src/main/java/com/google/genai/types/CreateTuningJobConfig.java
@@ -24,6 +24,7 @@
 import com.google.auto.value.AutoValue;
 import com.google.errorprone.annotations.CanIgnoreReturnValue;
 import com.google.genai.JsonSerializable;
+import java.util.Map;
 import java.util.Optional;
 
 /** Supervised fine-tuning job creation request - optional fields. */
@@ -89,6 +90,16 @@ public abstract class CreateTuningJobConfig extends JsonSerializable {
   @JsonProperty("learningRate")
   public abstract Optional<Float> learningRate();
 
+  /**
+   * Optional. The labels with user-defined metadata to organize TuningJob and generated resources
+   * such as Model and Endpoint. Label keys and values can be no longer than 64 characters (Unicode
+   * codepoints), can only contain lowercase letters, numeric characters, underscores and dashes.
+   * International characters are allowed. See https://goo.gl/xmQnxf for more information and
+   * examples of labels.
+   */
+  @JsonProperty("labels")
+  public abstract Optional<Map<String, String>> labels();
+
   /** Instantiates a builder for CreateTuningJobConfig. */
   @ExcludeFromGeneratedCoverageReport
   public static Builder builder() {
@@ -242,6 +253,18 @@ public Builder adapterSize(String adapterSize) {
     @JsonProperty("learningRate")
     public abstract Builder learningRate(Float learningRate);
 
+    /**
+     * Setter for labels.
+     *
+     * <p>labels: Optional. The labels with user-defined metadata to organize TuningJob and
+     * generated resources such as Model and Endpoint. Label keys and values can be no longer than
+     * 64 characters (Unicode codepoints), can only contain lowercase letters, numeric characters,
+     * underscores and dashes. International characters are allowed. See https://goo.gl/xmQnxf for
+     * more information and examples of labels.
+     */
+    @JsonProperty("labels")
+    public abstract Builder labels(Map<String, String> labels);
+
     public abstract CreateTuningJobConfig build();
   }
 

Original file line number	Diff line number	Diff line change
`@@ -248,6 +248,10 @@ ObjectNode createTuningJobConfigToMldev(JsonNode fromObject, ObjectNode parentOb`
`248`	`248`	`Common.getValueByPath(fromObject, new String[] {"learningRate"}));`
`249`	`249`	`}`
`250`	`250`
	`251`	`+ if (!Common.isZero(Common.getValueByPath(fromObject, new String[] {"labels"}))) {`
	`252`	`+ throw new IllegalArgumentException("labels parameter is not supported in Gemini API.");`
	`253`	`+ }`
	`254`	`+`
`251`	`255`	`return toObject;`
`252`	`256`	`}`
`253`	`257`
`@@ -490,6 +494,13 @@ ObjectNode createTuningJobConfigToVertex(JsonNode fromObject, ObjectNode parentO`
`490`	`494`	`throw new IllegalArgumentException("learningRate parameter is not supported in Vertex AI.");`
`491`	`495`	`}`
`492`	`496`
	`497`	`+ if (Common.getValueByPath(fromObject, new String[] {"labels"}) != null) {`
	`498`	`+ Common.setValueByPath(`
	`499`	`+ parentObject,`
	`500`	`+ new String[] {"labels"},`
	`501`	`+ Common.getValueByPath(fromObject, new String[] {"labels"}));`
	`502`	`+ }`
	`503`	`+`
`493`	`504`	`return toObject;`
`494`	`505`	`}`
`495`	`506`