Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions src/main/java/com/google/genai/Models.java
Original file line number Diff line number Diff line change
Expand Up @@ -1871,8 +1871,11 @@ ObjectNode generateContentConfigToVertex(
Common.getValueByPath(fromObject, new String[] {"modelArmorConfig"}));
}

if (!Common.isZero(Common.getValueByPath(fromObject, new String[] {"serviceTier"}))) {
throw new IllegalArgumentException("serviceTier parameter is not supported in Vertex AI.");
if (Common.getValueByPath(fromObject, new String[] {"serviceTier"}) != null) {
Common.setValueByPath(
parentObject,
new String[] {"serviceTier"},
Common.getValueByPath(fromObject, new String[] {"serviceTier"}));
}

return toObject;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -239,7 +239,7 @@ public abstract class GenerateContentConfig extends JsonSerializable {
@JsonProperty("modelArmorConfig")
public abstract Optional<ModelArmorConfig> modelArmorConfig();

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

Expand Down Expand Up @@ -1110,7 +1110,7 @@ public Builder clearModelArmorConfig() {
/**
* Setter for serviceTier.
*
* <p>serviceTier: The service tier to use for the request. For example, SERVICE_TIER_FLEX.
* <p>serviceTier: The service tier to use for the request. For example, ServiceTier.FLEX.
*/
@JsonProperty("serviceTier")
public abstract Builder serviceTier(ServiceTier serviceTier);
Expand All @@ -1128,7 +1128,7 @@ public Builder clearServiceTier() {
/**
* Setter for serviceTier given a known enum.
*
* <p>serviceTier: The service tier to use for the request. For example, SERVICE_TIER_FLEX.
* <p>serviceTier: The service tier to use for the request. For example, ServiceTier.FLEX.
*/
@CanIgnoreReturnValue
public Builder serviceTier(ServiceTier.Known knownType) {
Expand All @@ -1138,7 +1138,7 @@ public Builder serviceTier(ServiceTier.Known knownType) {
/**
* Setter for serviceTier given a string.
*
* <p>serviceTier: The service tier to use for the request. For example, SERVICE_TIER_FLEX.
* <p>serviceTier: The service tier to use for the request. For example, ServiceTier.FLEX.
*/
@CanIgnoreReturnValue
public Builder serviceTier(String serviceTier) {
Expand Down
36 changes: 27 additions & 9 deletions src/main/java/com/google/genai/types/ImagePromptLanguage.java
Original file line number Diff line number Diff line change
Expand Up @@ -29,30 +29,48 @@ public class ImagePromptLanguage {
/** Enum representing the known values for ImagePromptLanguage. */
public enum Known {
/** Auto-detect the language. */
AUTO,
AUTO("auto"),

/** English */
EN,
EN("en"),

/** Japanese */
JA,
JA("ja"),

/** Korean */
KO,
KO("ko"),

/** Hindi */
HI,
HI("hi"),

/** Chinese */
ZH,
ZH("zh"),

/** Portuguese */
PT,
PT("pt"),

/** Spanish */
ES,
ES("es"),

IMAGE_PROMPT_LANGUAGE_UNSPECIFIED
IMAGE_PROMPT_LANGUAGE_UNSPECIFIED;
private final String value;

Known() {
this.value = this.name();
}

Known(String value) {
this.value = value;
}

public String value() {
return value;
}

@Override
public String toString() {
return value;
}
}

private Known imagePromptLanguageEnum;
Expand Down
28 changes: 24 additions & 4 deletions src/main/java/com/google/genai/types/ServiceTier.java
Original file line number Diff line number Diff line change
Expand Up @@ -29,16 +29,36 @@ public class ServiceTier {
/** Enum representing the known values for ServiceTier. */
public enum Known {
/** Default service tier, which is standard. */
SERVICE_TIER_UNSPECIFIED,
UNSPECIFIED("unspecified"),

/** Flex service tier. */
SERVICE_TIER_FLEX,
FLEX("flex"),

/** Standard service tier. */
SERVICE_TIER_STANDARD,
STANDARD("standard"),

/** Priority service tier. */
SERVICE_TIER_PRIORITY
PRIORITY("priority"),

SERVICE_TIER_UNSPECIFIED;
private final String value;

Known() {
this.value = this.name();
}

Known(String value) {
this.value = value;
}

public String value() {
return value;
}

@Override
public String toString() {
return value;
}
}

private Known serviceTierEnum;
Expand Down
Loading