Skip to content

Commit 86aeb75

Browse files
MarkDaoustcopybara-github
authored andcommitted
fix: Fix service_tier enums.
FUTURE_COPYBARA_INTEGRATE_REVIEW=#904 from googleapis:release-please--branches--main 4d7573e PiperOrigin-RevId: 888287753
1 parent f3dcf87 commit 86aeb75

3 files changed

Lines changed: 15 additions & 10 deletions

File tree

src/main/java/com/google/genai/Models.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1871,8 +1871,11 @@ ObjectNode generateContentConfigToVertex(
18711871
Common.getValueByPath(fromObject, new String[] {"modelArmorConfig"}));
18721872
}
18731873

1874-
if (!Common.isZero(Common.getValueByPath(fromObject, new String[] {"serviceTier"}))) {
1875-
throw new IllegalArgumentException("serviceTier parameter is not supported in Vertex AI.");
1874+
if (Common.getValueByPath(fromObject, new String[] {"serviceTier"}) != null) {
1875+
Common.setValueByPath(
1876+
parentObject,
1877+
new String[] {"serviceTier"},
1878+
Common.getValueByPath(fromObject, new String[] {"serviceTier"}));
18761879
}
18771880

18781881
return toObject;

src/main/java/com/google/genai/types/GenerateContentConfig.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,7 @@ public abstract class GenerateContentConfig extends JsonSerializable {
239239
@JsonProperty("modelArmorConfig")
240240
public abstract Optional<ModelArmorConfig> modelArmorConfig();
241241

242-
/** The service tier to use for the request. For example, SERVICE_TIER_FLEX. */
242+
/** The service tier to use for the request. For example, ServiceTier.FLEX. */
243243
@JsonProperty("serviceTier")
244244
public abstract Optional<ServiceTier> serviceTier();
245245

@@ -1110,7 +1110,7 @@ public Builder clearModelArmorConfig() {
11101110
/**
11111111
* Setter for serviceTier.
11121112
*
1113-
* <p>serviceTier: The service tier to use for the request. For example, SERVICE_TIER_FLEX.
1113+
* <p>serviceTier: The service tier to use for the request. For example, ServiceTier.FLEX.
11141114
*/
11151115
@JsonProperty("serviceTier")
11161116
public abstract Builder serviceTier(ServiceTier serviceTier);
@@ -1128,7 +1128,7 @@ public Builder clearServiceTier() {
11281128
/**
11291129
* Setter for serviceTier given a known enum.
11301130
*
1131-
* <p>serviceTier: The service tier to use for the request. For example, SERVICE_TIER_FLEX.
1131+
* <p>serviceTier: The service tier to use for the request. For example, ServiceTier.FLEX.
11321132
*/
11331133
@CanIgnoreReturnValue
11341134
public Builder serviceTier(ServiceTier.Known knownType) {
@@ -1138,7 +1138,7 @@ public Builder serviceTier(ServiceTier.Known knownType) {
11381138
/**
11391139
* Setter for serviceTier given a string.
11401140
*
1141-
* <p>serviceTier: The service tier to use for the request. For example, SERVICE_TIER_FLEX.
1141+
* <p>serviceTier: The service tier to use for the request. For example, ServiceTier.FLEX.
11421142
*/
11431143
@CanIgnoreReturnValue
11441144
public Builder serviceTier(String serviceTier) {

src/main/java/com/google/genai/types/ServiceTier.java

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,16 +29,18 @@ public class ServiceTier {
2929
/** Enum representing the known values for ServiceTier. */
3030
public enum Known {
3131
/** Default service tier, which is standard. */
32-
SERVICE_TIER_UNSPECIFIED,
32+
UNSPECIFIED,
3333

3434
/** Flex service tier. */
35-
SERVICE_TIER_FLEX,
35+
FLEX,
3636

3737
/** Standard service tier. */
38-
SERVICE_TIER_STANDARD,
38+
STANDARD,
3939

4040
/** Priority service tier. */
41-
SERVICE_TIER_PRIORITY
41+
PRIORITY,
42+
43+
SERVICE_TIER_UNSPECIFIED
4244
}
4345

4446
private Known serviceTierEnum;

0 commit comments

Comments
 (0)