+ * LINT: LEGACY_NAMES * Represents a hardware accelerator type. ** @@ -141,6 +142,26 @@ public enum AcceleratorType implements com.google.protobuf.ProtocolMessageEnum { *
NVIDIA_H100_MEGA_80GB = 14;
*/
NVIDIA_H100_MEGA_80GB(14),
+ /**
+ *
+ *
+ * + * Nvidia H200 141Gb GPU. + *+ * + *
NVIDIA_H200_141GB = 15;
+ */
+ NVIDIA_H200_141GB(15),
+ /**
+ *
+ *
+ * + * Nvidia B200 GPU. + *+ * + *
NVIDIA_B200 = 16;
+ */
+ NVIDIA_B200(16),
/**
*
*
@@ -306,6 +327,28 @@ public enum AcceleratorType implements com.google.protobuf.ProtocolMessageEnum {
*/
public static final int NVIDIA_H100_MEGA_80GB_VALUE = 14;
+ /**
+ *
+ *
+ * + * Nvidia H200 141Gb GPU. + *+ * + *
NVIDIA_H200_141GB = 15;
+ */
+ public static final int NVIDIA_H200_141GB_VALUE = 15;
+
+ /**
+ *
+ *
+ * + * Nvidia B200 GPU. + *+ * + *
NVIDIA_B200 = 16;
+ */
+ public static final int NVIDIA_B200_VALUE = 16;
+
/**
*
*
@@ -396,6 +439,10 @@ public static AcceleratorType forNumber(int value) {
return NVIDIA_H100_80GB;
case 14:
return NVIDIA_H100_MEGA_80GB;
+ case 15:
+ return NVIDIA_H200_141GB;
+ case 16:
+ return NVIDIA_B200;
case 6:
return TPU_V2;
case 7:
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/AcceleratorTypeProto.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/AcceleratorTypeProto.java
index fee0664eb589..e9ac40d2c077 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/AcceleratorTypeProto.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/AcceleratorTypeProto.java
@@ -38,13 +38,14 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
java.lang.String[] descriptorData = {
"\n1google/cloud/aiplatform/v1/accelerator"
+ "_type.proto\022\032google.cloud.aiplatform.v1*"
- + "\316\002\n\017AcceleratorType\022 \n\034ACCELERATOR_TYPE_"
+ + "\366\002\n\017AcceleratorType\022 \n\034ACCELERATOR_TYPE_"
+ "UNSPECIFIED\020\000\022\030\n\020NVIDIA_TESLA_K80\020\001\032\002\010\001\022"
+ "\025\n\021NVIDIA_TESLA_P100\020\002\022\025\n\021NVIDIA_TESLA_V"
+ "100\020\003\022\023\n\017NVIDIA_TESLA_P4\020\004\022\023\n\017NVIDIA_TES"
+ "LA_T4\020\005\022\025\n\021NVIDIA_TESLA_A100\020\010\022\024\n\020NVIDIA"
+ "_A100_80GB\020\t\022\r\n\tNVIDIA_L4\020\013\022\024\n\020NVIDIA_H1"
- + "00_80GB\020\r\022\031\n\025NVIDIA_H100_MEGA_80GB\020\016\022\n\n\006"
+ + "00_80GB\020\r\022\031\n\025NVIDIA_H100_MEGA_80GB\020\016\022\025\n\021"
+ + "NVIDIA_H200_141GB\020\017\022\017\n\013NVIDIA_B200\020\020\022\n\n\006"
+ "TPU_V2\020\006\022\n\n\006TPU_V3\020\007\022\016\n\nTPU_V4_POD\020\n\022\022\n\016"
+ "TPU_V5_LITEPOD\020\014B\322\001\n\036com.google.cloud.ai"
+ "platform.v1B\024AcceleratorTypeProtoP\001Z>clo"
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/ImportRagFilesConfig.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/ImportRagFilesConfig.java
index 08e6e2426973..83fd55ade164 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/ImportRagFilesConfig.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/ImportRagFilesConfig.java
@@ -521,7 +521,7 @@ public com.google.cloud.aiplatform.v1.SharePointSourcesOrBuilder getSharePointSo
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=413
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=453
* @return Whether the partialFailureGcsSink field is set.
*/
@java.lang.Override
@@ -543,7 +543,7 @@ public boolean hasPartialFailureGcsSink() {
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=413
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=453
* @return The partialFailureGcsSink.
*/
@java.lang.Override
@@ -597,7 +597,7 @@ public com.google.cloud.aiplatform.v1.GcsDestination getPartialFailureGcsSink()
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_bigquery_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=422
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=462
* @return Whether the partialFailureBigquerySink field is set.
*/
@java.lang.Override
@@ -624,7 +624,7 @@ public boolean hasPartialFailureBigquerySink() {
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_bigquery_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=422
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=462
* @return The partialFailureBigquerySink.
*/
@java.lang.Override
@@ -935,6 +935,31 @@ public int getMaxEmbeddingRequestsPerMin() {
return maxEmbeddingRequestsPerMin_;
}
+ public static final int REBUILD_ANN_INDEX_FIELD_NUMBER = 19;
+ private boolean rebuildAnnIndex_ = false;
+
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @return The rebuildAnnIndex.
+ */
+ @java.lang.Override
+ public boolean getRebuildAnnIndex() {
+ return rebuildAnnIndex_;
+ }
+
private byte memoizedIsInitialized = -1;
@java.lang.Override
@@ -987,6 +1012,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
if (((bitField0_ & 0x00000001) != 0)) {
output.writeMessage(16, getRagFileTransformationConfig());
}
+ if (rebuildAnnIndex_ != false) {
+ output.writeBool(19, rebuildAnnIndex_);
+ }
getUnknownFields().writeTo(output);
}
@@ -1054,6 +1082,9 @@ public int getSerializedSize() {
com.google.protobuf.CodedOutputStream.computeMessageSize(
16, getRagFileTransformationConfig());
}
+ if (rebuildAnnIndex_ != false) {
+ size += com.google.protobuf.CodedOutputStream.computeBoolSize(19, rebuildAnnIndex_);
+ }
size += getUnknownFields().getSerializedSize();
memoizedSize = size;
return size;
@@ -1080,6 +1111,7 @@ public boolean equals(final java.lang.Object obj) {
if (!getRagFileParsingConfig().equals(other.getRagFileParsingConfig())) return false;
}
if (getMaxEmbeddingRequestsPerMin() != other.getMaxEmbeddingRequestsPerMin()) return false;
+ if (getRebuildAnnIndex() != other.getRebuildAnnIndex()) return false;
if (!getImportSourceCase().equals(other.getImportSourceCase())) return false;
switch (importSourceCase_) {
case 2:
@@ -1145,6 +1177,8 @@ public int hashCode() {
}
hash = (37 * hash) + MAX_EMBEDDING_REQUESTS_PER_MIN_FIELD_NUMBER;
hash = (53 * hash) + getMaxEmbeddingRequestsPerMin();
+ hash = (37 * hash) + REBUILD_ANN_INDEX_FIELD_NUMBER;
+ hash = (53 * hash) + com.google.protobuf.Internal.hashBoolean(getRebuildAnnIndex());
switch (importSourceCase_) {
case 2:
hash = (37 * hash) + GCS_SOURCE_FIELD_NUMBER;
@@ -1381,6 +1415,7 @@ public Builder clear() {
ragFileParsingConfigBuilder_ = null;
}
maxEmbeddingRequestsPerMin_ = 0;
+ rebuildAnnIndex_ = false;
importSourceCase_ = 0;
importSource_ = null;
partialFailureSinkCase_ = 0;
@@ -1442,6 +1477,9 @@ private void buildPartial0(com.google.cloud.aiplatform.v1.ImportRagFilesConfig r
if (((from_bitField0_ & 0x00000800) != 0)) {
result.maxEmbeddingRequestsPerMin_ = maxEmbeddingRequestsPerMin_;
}
+ if (((from_bitField0_ & 0x00001000) != 0)) {
+ result.rebuildAnnIndex_ = rebuildAnnIndex_;
+ }
result.bitField0_ |= to_bitField0_;
}
@@ -1536,6 +1574,9 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1.ImportRagFilesConfig oth
if (other.getMaxEmbeddingRequestsPerMin() != 0) {
setMaxEmbeddingRequestsPerMin(other.getMaxEmbeddingRequestsPerMin());
}
+ if (other.getRebuildAnnIndex() != false) {
+ setRebuildAnnIndex(other.getRebuildAnnIndex());
+ }
switch (other.getImportSourceCase()) {
case GCS_SOURCE:
{
@@ -1705,6 +1746,12 @@ public Builder mergeFrom(
bitField0_ |= 0x00000200;
break;
} // case 130
+ case 152:
+ {
+ rebuildAnnIndex_ = input.readBool();
+ bitField0_ |= 0x00001000;
+ break;
+ } // case 152
default:
{
if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -2914,7 +2961,7 @@ public com.google.cloud.aiplatform.v1.SharePointSources.Builder getSharePointSou
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=413
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=453
* @return Whether the partialFailureGcsSink field is set.
*/
@java.lang.Override
@@ -2936,7 +2983,7 @@ public boolean hasPartialFailureGcsSink() {
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=413
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=453
* @return The partialFailureGcsSink.
*/
@java.lang.Override
@@ -3178,7 +3225,7 @@ public com.google.cloud.aiplatform.v1.GcsDestination.Builder getPartialFailureGc
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_bigquery_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=422
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=462
* @return Whether the partialFailureBigquerySink field is set.
*/
@java.lang.Override
@@ -3205,7 +3252,7 @@ public boolean hasPartialFailureBigquerySink() {
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_bigquery_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=422
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=462
* @return The partialFailureBigquerySink.
*/
@java.lang.Override
@@ -4478,6 +4525,80 @@ public Builder clearMaxEmbeddingRequestsPerMin() {
return this;
}
+ private boolean rebuildAnnIndex_;
+
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @return The rebuildAnnIndex.
+ */
+ @java.lang.Override
+ public boolean getRebuildAnnIndex() {
+ return rebuildAnnIndex_;
+ }
+
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @param value The rebuildAnnIndex to set.
+ * @return This builder for chaining.
+ */
+ public Builder setRebuildAnnIndex(boolean value) {
+
+ rebuildAnnIndex_ = value;
+ bitField0_ |= 0x00001000;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearRebuildAnnIndex() {
+ bitField0_ = (bitField0_ & ~0x00001000);
+ rebuildAnnIndex_ = false;
+ onChanged();
+ return this;
+ }
+
@java.lang.Override
public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.setUnknownFields(unknownFields);
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/ImportRagFilesConfigOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/ImportRagFilesConfigOrBuilder.java
index 558a4f3248cc..da99b4bed121 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/ImportRagFilesConfigOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/ImportRagFilesConfigOrBuilder.java
@@ -234,7 +234,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=413
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=453
* @return Whether the partialFailureGcsSink field is set.
*/
@java.lang.Deprecated
@@ -253,7 +253,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=413
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=453
* @return The partialFailureGcsSink.
*/
@java.lang.Deprecated
@@ -292,7 +292,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_bigquery_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=422
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=462
* @return Whether the partialFailureBigquerySink field is set.
*/
@java.lang.Deprecated
@@ -316,7 +316,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1.ImportRagFilesConfig.partial_failure_bigquery_sink is
- * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=422
+ * deprecated. See google/cloud/aiplatform/v1/vertex_rag_data.proto;l=462
* @return The partialFailureBigquerySink.
*/
@java.lang.Deprecated
@@ -541,6 +541,25 @@ public interface ImportRagFilesConfigOrBuilder
*/
int getMaxEmbeddingRequestsPerMin();
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @return The rebuildAnnIndex.
+ */
+ boolean getRebuildAnnIndex();
+
com.google.cloud.aiplatform.v1.ImportRagFilesConfig.ImportSourceCase getImportSourceCase();
com.google.cloud.aiplatform.v1.ImportRagFilesConfig.PartialFailureSinkCase
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/OpenApiProto.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/OpenApiProto.java
index 1286f3a5a912..0b83cd431164 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/OpenApiProto.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/OpenApiProto.java
@@ -52,7 +52,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
"\n"
+ "(google/cloud/aiplatform/v1/openapi.pro"
+ "to\022\032google.cloud.aiplatform.v1\032\037google/a"
- + "pi/field_behavior.proto\032\034google/protobuf/struct.proto\"\324\007\n"
+ + "pi/field_behavior.proto\032\034google/protobuf/struct.proto\"\220\010\n"
+ "\006Schema\0223\n"
+ "\004type\030\001 \001(\0162 .google.cloud.aiplatform.v1.TypeB\003\340A\001\022\023\n"
+ "\006format\030\007 \001(\tB\003\340A\001\022\022\n"
@@ -77,11 +77,12 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "max_length\030\023 \001(\003B\003\340A\001\022\024\n"
+ "\007pattern\030\024 \001(\tB\003\340A\001\022,\n"
+ "\007example\030\004 \001(\0132\026.google.protobuf.ValueB\003\340A\001\0227\n"
- + "\006any_of\030\013"
- + " \003(\0132\".google.cloud.aiplatform.v1.SchemaB\003\340A\001\022\020\n"
+ + "\006any_of\030\013 \003(\0132\".google.cloud.aiplatform.v1.SchemaB\003\340A\001\022:\n"
+ + "\025additional_properties\030\032"
+ + " \001(\0132\026.google.protobuf.ValueB\003\340A\001\022\020\n"
+ "\003ref\030\033 \001(\tB\003\340A\001\022?\n"
- + "\004defs\030\034"
- + " \003(\0132,.google.cloud.aiplatform.v1.Schema.DefsEntryB\003\340A\001\032U\n"
+ + "\004defs\030\034 \003(\0132,."
+ + "google.cloud.aiplatform.v1.Schema.DefsEntryB\003\340A\001\032U\n"
+ "\017PropertiesEntry\022\013\n"
+ "\003key\030\001 \001(\t\0221\n"
+ "\005value\030\002 \001(\0132\".google.cloud.aiplatform.v1.Schema:\0028\001\032O\n"
@@ -90,18 +91,16 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\005value\030\002 \001(\0132\".google.cloud.aiplatform.v1.Schema:\0028\001*e\n"
+ "\004Type\022\024\n"
+ "\020TYPE_UNSPECIFIED\020\000\022\n\n"
- + "\006STRING\020\001\022\n"
- + "\n"
+ + "\006STRING\020\001\022\n\n"
+ "\006NUMBER\020\002\022\013\n"
+ "\007INTEGER\020\003\022\013\n"
+ "\007BOOLEAN\020\004\022\t\n"
+ "\005ARRAY\020\005\022\n\n"
+ "\006OBJECT\020\006B\312\001\n"
- + "\036com.google.cloud.aiplatform.v1B\014OpenApiProtoP\001Z>cloud.go"
- + "ogle.com/go/aiplatform/apiv1/aiplatformp"
- + "b;aiplatformpb\252\002\032Google.Cloud.AIPlatform"
- + ".V1\312\002\032Google\\Cloud\\AIPlatform\\V1\352\002\035Googl"
- + "e::Cloud::AIPlatform::V1b\006proto3"
+ + "\036com.google.cloud.aiplatform.v1B\014OpenApiProtoP\001Z>cloud.google.com/go/aiplatfo"
+ + "rm/apiv1/aiplatformpb;aiplatformpb\252\002\032Goo"
+ + "gle.Cloud.AIPlatform.V1\312\002\032Google\\Cloud\\A"
+ + "IPlatform\\V1\352\002\035Google::Cloud::AIPlatform::V1b\006proto3"
};
descriptor =
com.google.protobuf.Descriptors.FileDescriptor.internalBuildGeneratedFileFrom(
@@ -138,6 +137,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
"Pattern",
"Example",
"AnyOf",
+ "AdditionalProperties",
"Ref",
"Defs",
});
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/RagVectorDbConfig.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/RagVectorDbConfig.java
index bd65c8c2f873..e977e2e1a90f 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/RagVectorDbConfig.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/RagVectorDbConfig.java
@@ -65,7 +65,91 @@ public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() {
public interface RagManagedDbOrBuilder
extends
// @@protoc_insertion_point(interface_extends:google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb)
- com.google.protobuf.MessageOrBuilder {}
+ com.google.protobuf.MessageOrBuilder {
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return Whether the knn field is set.
+ */
+ boolean hasKnn();
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return The knn.
+ */
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN getKnn();
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder getKnnOrBuilder();
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return Whether the ann field is set.
+ */
+ boolean hasAnn();
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return The ann.
+ */
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN getAnn();
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder getAnnOrBuilder();
+
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.RetrievalStrategyCase
+ getRetrievalStrategyCase();
+ }
/**
*
@@ -110,230 +194,1653 @@ public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() {
com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.Builder.class);
}
- private byte memoizedIsInitialized = -1;
-
- @java.lang.Override
- public final boolean isInitialized() {
- byte isInitialized = memoizedIsInitialized;
- if (isInitialized == 1) return true;
- if (isInitialized == 0) return false;
-
- memoizedIsInitialized = 1;
- return true;
- }
-
- @java.lang.Override
- public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException {
- getUnknownFields().writeTo(output);
- }
-
- @java.lang.Override
- public int getSerializedSize() {
- int size = memoizedSize;
- if (size != -1) return size;
-
- size = 0;
- size += getUnknownFields().getSerializedSize();
- memoizedSize = size;
- return size;
- }
-
- @java.lang.Override
- public boolean equals(final java.lang.Object obj) {
- if (obj == this) {
- return true;
- }
- if (!(obj instanceof com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb)) {
- return super.equals(obj);
- }
- com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb other =
- (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb) obj;
-
- if (!getUnknownFields().equals(other.getUnknownFields())) return false;
- return true;
- }
-
- @java.lang.Override
- public int hashCode() {
- if (memoizedHashCode != 0) {
- return memoizedHashCode;
- }
- int hash = 41;
- hash = (19 * hash) + getDescriptor().hashCode();
- hash = (29 * hash) + getUnknownFields().hashCode();
- memoizedHashCode = hash;
- return hash;
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- java.nio.ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- com.google.protobuf.ByteString data)
- throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- com.google.protobuf.ByteString data,
- com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- byte[] data) throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- java.io.InputStream input) throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
- PARSER, input, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseDelimitedFrom(
- java.io.InputStream input) throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(PARSER, input);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseDelimitedFrom(
- java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(
- PARSER, input, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- com.google.protobuf.CodedInputStream input) throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
- }
-
- public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
- com.google.protobuf.CodedInputStream input,
- com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
- PARSER, input, extensionRegistry);
- }
-
- @java.lang.Override
- public Builder newBuilderForType() {
- return newBuilder();
- }
-
- public static Builder newBuilder() {
- return DEFAULT_INSTANCE.toBuilder();
- }
-
- public static Builder newBuilder(
- com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb prototype) {
- return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
- }
-
- @java.lang.Override
- public Builder toBuilder() {
- return this == DEFAULT_INSTANCE ? new Builder() : new Builder().mergeFrom(this);
- }
-
- @java.lang.Override
- protected Builder newBuilderForType(
- com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
- Builder builder = new Builder(parent);
- return builder;
- }
+ public interface KNNOrBuilder
+ extends
+ // @@protoc_insertion_point(interface_extends:google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN)
+ com.google.protobuf.MessageOrBuilder {}
/**
*
*
*
- * The config for the default RAG-managed Vector DB.
+ * Config for KNN search.
*
*
- * Protobuf type {@code google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb}
+ * Protobuf type {@code google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN}
*/
- public static final class Builder
- extends com.google.protobuf.GeneratedMessageV3.Builder+ * Config for KNN search. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN} + */ + public static final class Builder + extends com.google.protobuf.GeneratedMessageV3.Builder
+ * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @return The treeDepth.
+ */
+ int getTreeDepth();
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @return The leafCount.
+ */
+ int getLeafCount();
+ }
+
+ /**
+ *
+ *
+ * + * Config for ANN search. + * + * RagManagedDb uses a tree-based structure to partition data and + * facilitate faster searches. As a tradeoff, it requires longer indexing + * time and manual triggering of index rebuild via the ImportRagFiles and + * UpdateRagCorpus API. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN} + */ + public static final class ANN extends com.google.protobuf.GeneratedMessageV3 + implements + // @@protoc_insertion_point(message_implements:google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN) + ANNOrBuilder { + private static final long serialVersionUID = 0L; + + // Use ANN.newBuilder() to construct. + private ANN(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + + private ANN() {} + + @java.lang.Override + @SuppressWarnings({"unused"}) + protected java.lang.Object newInstance(UnusedPrivateParameter unused) { + return new ANN(); + } + + public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() { + return com.google.cloud.aiplatform.v1.VertexRagDataProto + .internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_ANN_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return com.google.cloud.aiplatform.v1.VertexRagDataProto + .internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_ANN_fieldAccessorTable + .ensureFieldAccessorsInitialized( + com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN.class, + com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN.Builder.class); + } + + public static final int TREE_DEPTH_FIELD_NUMBER = 1; + private int treeDepth_ = 0; + + /** + * + * + *
+ * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @return The treeDepth.
+ */
+ @java.lang.Override
+ public int getTreeDepth() {
+ return treeDepth_;
+ }
+
+ public static final int LEAF_COUNT_FIELD_NUMBER = 2;
+ private int leafCount_ = 0;
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @return The leafCount.
+ */
+ @java.lang.Override
+ public int getLeafCount() {
+ return leafCount_;
+ }
+
+ private byte memoizedIsInitialized = -1;
+
+ @java.lang.Override
+ public final boolean isInitialized() {
+ byte isInitialized = memoizedIsInitialized;
+ if (isInitialized == 1) return true;
+ if (isInitialized == 0) return false;
+
+ memoizedIsInitialized = 1;
+ return true;
+ }
+
+ @java.lang.Override
+ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException {
+ if (treeDepth_ != 0) {
+ output.writeInt32(1, treeDepth_);
+ }
+ if (leafCount_ != 0) {
+ output.writeInt32(2, leafCount_);
+ }
+ getUnknownFields().writeTo(output);
+ }
+
+ @java.lang.Override
+ public int getSerializedSize() {
+ int size = memoizedSize;
+ if (size != -1) return size;
+
+ size = 0;
+ if (treeDepth_ != 0) {
+ size += com.google.protobuf.CodedOutputStream.computeInt32Size(1, treeDepth_);
+ }
+ if (leafCount_ != 0) {
+ size += com.google.protobuf.CodedOutputStream.computeInt32Size(2, leafCount_);
+ }
+ size += getUnknownFields().getSerializedSize();
+ memoizedSize = size;
+ return size;
+ }
+
+ @java.lang.Override
+ public boolean equals(final java.lang.Object obj) {
+ if (obj == this) {
+ return true;
+ }
+ if (!(obj instanceof com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)) {
+ return super.equals(obj);
+ }
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN other =
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN) obj;
+
+ if (getTreeDepth() != other.getTreeDepth()) return false;
+ if (getLeafCount() != other.getLeafCount()) return false;
+ if (!getUnknownFields().equals(other.getUnknownFields())) return false;
+ return true;
+ }
+
+ @java.lang.Override
+ public int hashCode() {
+ if (memoizedHashCode != 0) {
+ return memoizedHashCode;
+ }
+ int hash = 41;
+ hash = (19 * hash) + getDescriptor().hashCode();
+ hash = (37 * hash) + TREE_DEPTH_FIELD_NUMBER;
+ hash = (53 * hash) + getTreeDepth();
+ hash = (37 * hash) + LEAF_COUNT_FIELD_NUMBER;
+ hash = (53 * hash) + getLeafCount();
+ hash = (29 * hash) + getUnknownFields().hashCode();
+ memoizedHashCode = hash;
+ return hash;
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ java.nio.ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ com.google.protobuf.ByteString data)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ com.google.protobuf.ByteString data,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ byte[] data) throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ parseDelimitedFrom(java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ parseDelimitedFrom(
+ java.io.InputStream input,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ com.google.protobuf.CodedInputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN parseFrom(
+ com.google.protobuf.CodedInputStream input,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ @java.lang.Override
+ public Builder newBuilderForType() {
+ return newBuilder();
+ }
+
+ public static Builder newBuilder() {
+ return DEFAULT_INSTANCE.toBuilder();
+ }
+
+ public static Builder newBuilder(
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN prototype) {
+ return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
+ }
+
+ @java.lang.Override
+ public Builder toBuilder() {
+ return this == DEFAULT_INSTANCE ? new Builder() : new Builder().mergeFrom(this);
+ }
+
+ @java.lang.Override
+ protected Builder newBuilderForType(
+ com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
+ Builder builder = new Builder(parent);
+ return builder;
+ }
+
+ /**
+ *
+ *
+ * + * Config for ANN search. + * + * RagManagedDb uses a tree-based structure to partition data and + * facilitate faster searches. As a tradeoff, it requires longer indexing + * time and manual triggering of index rebuild via the ImportRagFiles and + * UpdateRagCorpus API. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN} + */ + public static final class Builder + extends com.google.protobuf.GeneratedMessageV3.Builder
+ * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @return The treeDepth.
+ */
+ @java.lang.Override
+ public int getTreeDepth() {
+ return treeDepth_;
+ }
+
+ /**
+ *
+ *
+ * + * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @param value The treeDepth to set.
+ * @return This builder for chaining.
+ */
+ public Builder setTreeDepth(int value) {
+
+ treeDepth_ = value;
+ bitField0_ |= 0x00000001;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearTreeDepth() {
+ bitField0_ = (bitField0_ & ~0x00000001);
+ treeDepth_ = 0;
+ onChanged();
+ return this;
+ }
+
+ private int leafCount_;
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @return The leafCount.
+ */
+ @java.lang.Override
+ public int getLeafCount() {
+ return leafCount_;
+ }
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @param value The leafCount to set.
+ * @return This builder for chaining.
+ */
+ public Builder setLeafCount(int value) {
+
+ leafCount_ = value;
+ bitField0_ |= 0x00000002;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearLeafCount() {
+ bitField0_ = (bitField0_ & ~0x00000002);
+ leafCount_ = 0;
+ onChanged();
+ return this;
+ }
+
+ @java.lang.Override
+ public final Builder setUnknownFields(
+ final com.google.protobuf.UnknownFieldSet unknownFields) {
+ return super.setUnknownFields(unknownFields);
+ }
+
+ @java.lang.Override
+ public final Builder mergeUnknownFields(
+ final com.google.protobuf.UnknownFieldSet unknownFields) {
+ return super.mergeUnknownFields(unknownFields);
+ }
+
+ // @@protoc_insertion_point(builder_scope:google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)
+ }
+
+ // @@protoc_insertion_point(class_scope:google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)
+ private static final com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ DEFAULT_INSTANCE;
+
+ static {
+ DEFAULT_INSTANCE = new com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN();
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ getDefaultInstance() {
+ return DEFAULT_INSTANCE;
+ }
+
+ private static final com.google.protobuf.Parser+ * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return Whether the knn field is set.
+ */
+ @java.lang.Override
+ public boolean hasKnn() {
+ return retrievalStrategyCase_ == 1;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return The knn.
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN getKnn() {
+ if (retrievalStrategyCase_ == 1) {
+ return (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN.getDefaultInstance();
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder
+ getKnnOrBuilder() {
+ if (retrievalStrategyCase_ == 1) {
+ return (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN.getDefaultInstance();
+ }
+
+ public static final int ANN_FIELD_NUMBER = 2;
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return Whether the ann field is set.
+ */
+ @java.lang.Override
+ public boolean hasAnn() {
+ return retrievalStrategyCase_ == 2;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return The ann.
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN getAnn() {
+ if (retrievalStrategyCase_ == 2) {
+ return (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN.getDefaultInstance();
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder
+ getAnnOrBuilder() {
+ if (retrievalStrategyCase_ == 2) {
+ return (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN.getDefaultInstance();
+ }
+
+ private byte memoizedIsInitialized = -1;
+
+ @java.lang.Override
+ public final boolean isInitialized() {
+ byte isInitialized = memoizedIsInitialized;
+ if (isInitialized == 1) return true;
+ if (isInitialized == 0) return false;
+
+ memoizedIsInitialized = 1;
+ return true;
+ }
+
+ @java.lang.Override
+ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException {
+ if (retrievalStrategyCase_ == 1) {
+ output.writeMessage(
+ 1,
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN) retrievalStrategy_);
+ }
+ if (retrievalStrategyCase_ == 2) {
+ output.writeMessage(
+ 2,
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN) retrievalStrategy_);
+ }
+ getUnknownFields().writeTo(output);
+ }
+
+ @java.lang.Override
+ public int getSerializedSize() {
+ int size = memoizedSize;
+ if (size != -1) return size;
+
+ size = 0;
+ if (retrievalStrategyCase_ == 1) {
+ size +=
+ com.google.protobuf.CodedOutputStream.computeMessageSize(
+ 1,
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_);
+ }
+ if (retrievalStrategyCase_ == 2) {
+ size +=
+ com.google.protobuf.CodedOutputStream.computeMessageSize(
+ 2,
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_);
+ }
+ size += getUnknownFields().getSerializedSize();
+ memoizedSize = size;
+ return size;
+ }
+
+ @java.lang.Override
+ public boolean equals(final java.lang.Object obj) {
+ if (obj == this) {
+ return true;
+ }
+ if (!(obj instanceof com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb)) {
+ return super.equals(obj);
+ }
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb other =
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb) obj;
+
+ if (!getRetrievalStrategyCase().equals(other.getRetrievalStrategyCase())) return false;
+ switch (retrievalStrategyCase_) {
+ case 1:
+ if (!getKnn().equals(other.getKnn())) return false;
+ break;
+ case 2:
+ if (!getAnn().equals(other.getAnn())) return false;
+ break;
+ case 0:
+ default:
+ }
+ if (!getUnknownFields().equals(other.getUnknownFields())) return false;
+ return true;
+ }
+
+ @java.lang.Override
+ public int hashCode() {
+ if (memoizedHashCode != 0) {
+ return memoizedHashCode;
+ }
+ int hash = 41;
+ hash = (19 * hash) + getDescriptor().hashCode();
+ switch (retrievalStrategyCase_) {
+ case 1:
+ hash = (37 * hash) + KNN_FIELD_NUMBER;
+ hash = (53 * hash) + getKnn().hashCode();
+ break;
+ case 2:
+ hash = (37 * hash) + ANN_FIELD_NUMBER;
+ hash = (53 * hash) + getAnn().hashCode();
+ break;
+ case 0:
+ default:
+ }
+ hash = (29 * hash) + getUnknownFields().hashCode();
+ memoizedHashCode = hash;
+ return hash;
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ java.nio.ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ com.google.protobuf.ByteString data)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ com.google.protobuf.ByteString data,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ byte[] data) throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseDelimitedFrom(
+ java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseDelimitedFrom(
+ java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ com.google.protobuf.CodedInputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb parseFrom(
+ com.google.protobuf.CodedInputStream input,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ @java.lang.Override
+ public Builder newBuilderForType() {
+ return newBuilder();
+ }
+
+ public static Builder newBuilder() {
+ return DEFAULT_INSTANCE.toBuilder();
+ }
+
+ public static Builder newBuilder(
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb prototype) {
+ return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
+ }
+
+ @java.lang.Override
+ public Builder toBuilder() {
+ return this == DEFAULT_INSTANCE ? new Builder() : new Builder().mergeFrom(this);
+ }
+
+ @java.lang.Override
+ protected Builder newBuilderForType(
+ com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
+ Builder builder = new Builder(parent);
+ return builder;
+ }
+
+ /**
+ *
+ *
+ * + * The config for the default RAG-managed Vector DB. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb} + */ + public static final class Builder + extends com.google.protobuf.GeneratedMessageV3.Builder
+ * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return Whether the knn field is set.
+ */
+ @java.lang.Override
+ public boolean hasKnn() {
+ return retrievalStrategyCase_ == 1;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return The knn.
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN getKnn() {
+ if (knnBuilder_ == null) {
+ if (retrievalStrategyCase_ == 1) {
+ return (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ } else {
+ if (retrievalStrategyCase_ == 1) {
+ return knnBuilder_.getMessage();
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public Builder setKnn(
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN value) {
+ if (knnBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ retrievalStrategy_ = value;
+ onChanged();
+ } else {
+ knnBuilder_.setMessage(value);
+ }
+ retrievalStrategyCase_ = 1;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public Builder setKnn(
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN.Builder
+ builderForValue) {
+ if (knnBuilder_ == null) {
+ retrievalStrategy_ = builderForValue.build();
+ onChanged();
+ } else {
+ knnBuilder_.setMessage(builderForValue.build());
+ }
+ retrievalStrategyCase_ = 1;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public Builder mergeKnn(
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN value) {
+ if (knnBuilder_ == null) {
+ if (retrievalStrategyCase_ == 1
+ && retrievalStrategy_
+ != com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance()) {
+ retrievalStrategy_ =
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN
+ .newBuilder(
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_)
+ .mergeFrom(value)
+ .buildPartial();
+ } else {
+ retrievalStrategy_ = value;
+ }
+ onChanged();
+ } else {
+ if (retrievalStrategyCase_ == 1) {
+ knnBuilder_.mergeFrom(value);
+ } else {
+ knnBuilder_.setMessage(value);
+ }
+ }
+ retrievalStrategyCase_ = 1;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public Builder clearKnn() {
+ if (knnBuilder_ == null) {
+ if (retrievalStrategyCase_ == 1) {
+ retrievalStrategyCase_ = 0;
+ retrievalStrategy_ = null;
+ onChanged();
+ }
+ } else {
+ if (retrievalStrategyCase_ == 1) {
+ retrievalStrategyCase_ = 0;
+ retrievalStrategy_ = null;
+ }
+ knnBuilder_.clear();
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN.Builder
+ getKnnBuilder() {
+ return getKnnFieldBuilder().getBuilder();
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder
+ getKnnOrBuilder() {
+ if ((retrievalStrategyCase_ == 1) && (knnBuilder_ != null)) {
+ return knnBuilder_.getMessageOrBuilder();
+ } else {
+ if (retrievalStrategyCase_ == 1) {
+ return (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN.Builder,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder>
+ getKnnFieldBuilder() {
+ if (knnBuilder_ == null) {
+ if (!(retrievalStrategyCase_ == 1)) {
+ retrievalStrategy_ =
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ }
+ knnBuilder_ =
+ new com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN.Builder,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder>(
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_,
+ getParentForChildren(),
+ isClean());
+ retrievalStrategy_ = null;
+ }
+ retrievalStrategyCase_ = 1;
+ onChanged();
+ return knnBuilder_;
+ }
+
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN.Builder,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder>
+ annBuilder_;
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return Whether the ann field is set.
+ */
+ @java.lang.Override
+ public boolean hasAnn() {
+ return retrievalStrategyCase_ == 2;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return The ann.
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN getAnn() {
+ if (annBuilder_ == null) {
+ if (retrievalStrategyCase_ == 2) {
+ return (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ } else {
+ if (retrievalStrategyCase_ == 2) {
+ return annBuilder_.getMessage();
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public Builder setAnn(
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN value) {
+ if (annBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ retrievalStrategy_ = value;
+ onChanged();
+ } else {
+ annBuilder_.setMessage(value);
+ }
+ retrievalStrategyCase_ = 2;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public Builder setAnn(
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN.Builder
+ builderForValue) {
+ if (annBuilder_ == null) {
+ retrievalStrategy_ = builderForValue.build();
+ onChanged();
+ } else {
+ annBuilder_.setMessage(builderForValue.build());
+ }
+ retrievalStrategyCase_ = 2;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public Builder mergeAnn(
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN value) {
+ if (annBuilder_ == null) {
+ if (retrievalStrategyCase_ == 2
+ && retrievalStrategy_
+ != com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance()) {
+ retrievalStrategy_ =
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ .newBuilder(
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_)
+ .mergeFrom(value)
+ .buildPartial();
+ } else {
+ retrievalStrategy_ = value;
+ }
+ onChanged();
+ } else {
+ if (retrievalStrategyCase_ == 2) {
+ annBuilder_.mergeFrom(value);
+ } else {
+ annBuilder_.setMessage(value);
+ }
+ }
+ retrievalStrategyCase_ = 2;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public Builder clearAnn() {
+ if (annBuilder_ == null) {
+ if (retrievalStrategyCase_ == 2) {
+ retrievalStrategyCase_ = 0;
+ retrievalStrategy_ = null;
+ onChanged();
+ }
+ } else {
+ if (retrievalStrategyCase_ == 2) {
+ retrievalStrategyCase_ = 0;
+ retrievalStrategy_ = null;
+ }
+ annBuilder_.clear();
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN.Builder
+ getAnnBuilder() {
+ return getAnnFieldBuilder().getBuilder();
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder
+ getAnnOrBuilder() {
+ if ((retrievalStrategyCase_ == 2) && (annBuilder_ != null)) {
+ return annBuilder_.getMessageOrBuilder();
+ } else {
+ if (retrievalStrategyCase_ == 2) {
+ return (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN.Builder,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder>
+ getAnnFieldBuilder() {
+ if (annBuilder_ == null) {
+ if (!(retrievalStrategyCase_ == 2)) {
+ retrievalStrategy_ =
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ }
+ annBuilder_ =
+ new com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN.Builder,
+ com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder>(
+ (com.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_,
+ getParentForChildren(),
+ isClean());
+ retrievalStrategy_ = null;
+ }
+ retrievalStrategyCase_ = 2;
+ onChanged();
+ return annBuilder_;
+ }
+
@java.lang.Override
public final Builder setUnknownFields(
final com.google.protobuf.UnknownFieldSet unknownFields) {
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/Schema.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/Schema.java
index bd81dfdbee92..0f04c21e3e14 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/Schema.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/Schema.java
@@ -1130,6 +1130,68 @@ public com.google.cloud.aiplatform.v1.SchemaOrBuilder getAnyOfOrBuilder(int inde
return anyOf_.get(index);
}
+ public static final int ADDITIONAL_PROPERTIES_FIELD_NUMBER = 26;
+ private com.google.protobuf.Value additionalProperties_;
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return Whether the additionalProperties field is set.
+ */
+ @java.lang.Override
+ public boolean hasAdditionalProperties() {
+ return ((bitField0_ & 0x00000008) != 0);
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return The additionalProperties.
+ */
+ @java.lang.Override
+ public com.google.protobuf.Value getAdditionalProperties() {
+ return additionalProperties_ == null
+ ? com.google.protobuf.Value.getDefaultInstance()
+ : additionalProperties_;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ @java.lang.Override
+ public com.google.protobuf.ValueOrBuilder getAdditionalPropertiesOrBuilder() {
+ return additionalProperties_ == null
+ ? com.google.protobuf.Value.getDefaultInstance()
+ : additionalProperties_;
+ }
+
public static final int REF_FIELD_NUMBER = 27;
@SuppressWarnings("serial")
@@ -1428,6 +1490,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
for (int i = 0; i < propertyOrdering_.size(); i++) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 25, propertyOrdering_.getRaw(i));
}
+ if (((bitField0_ & 0x00000008) != 0)) {
+ output.writeMessage(26, getAdditionalProperties());
+ }
if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(ref_)) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 27, ref_);
}
@@ -1531,6 +1596,10 @@ public int getSerializedSize() {
size += dataSize;
size += 2 * getPropertyOrderingList().size();
}
+ if (((bitField0_ & 0x00000008) != 0)) {
+ size +=
+ com.google.protobuf.CodedOutputStream.computeMessageSize(26, getAdditionalProperties());
+ }
if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(ref_)) {
size += com.google.protobuf.GeneratedMessageV3.computeStringSize(27, ref_);
}
@@ -1592,6 +1661,10 @@ public boolean equals(final java.lang.Object obj) {
if (!getExample().equals(other.getExample())) return false;
}
if (!getAnyOfList().equals(other.getAnyOfList())) return false;
+ if (hasAdditionalProperties() != other.hasAdditionalProperties()) return false;
+ if (hasAdditionalProperties()) {
+ if (!getAdditionalProperties().equals(other.getAdditionalProperties())) return false;
+ }
if (!getRef().equals(other.getRef())) return false;
if (!internalGetDefs().equals(other.internalGetDefs())) return false;
if (!getUnknownFields().equals(other.getUnknownFields())) return false;
@@ -1671,6 +1744,10 @@ public int hashCode() {
hash = (37 * hash) + ANY_OF_FIELD_NUMBER;
hash = (53 * hash) + getAnyOfList().hashCode();
}
+ if (hasAdditionalProperties()) {
+ hash = (37 * hash) + ADDITIONAL_PROPERTIES_FIELD_NUMBER;
+ hash = (53 * hash) + getAdditionalProperties().hashCode();
+ }
hash = (37 * hash) + REF_FIELD_NUMBER;
hash = (53 * hash) + getRef().hashCode();
if (!internalGetDefs().getMap().isEmpty()) {
@@ -1850,6 +1927,7 @@ private void maybeForceBuilderInitialization() {
getItemsFieldBuilder();
getExampleFieldBuilder();
getAnyOfFieldBuilder();
+ getAdditionalPropertiesFieldBuilder();
}
}
@@ -1897,6 +1975,11 @@ public Builder clear() {
anyOfBuilder_.clear();
}
bitField0_ = (bitField0_ & ~0x00200000);
+ additionalProperties_ = null;
+ if (additionalPropertiesBuilder_ != null) {
+ additionalPropertiesBuilder_.dispose();
+ additionalPropertiesBuilder_ = null;
+ }
ref_ = "";
internalGetMutableDefs().clear();
return this;
@@ -2020,9 +2103,16 @@ private void buildPartial0(com.google.cloud.aiplatform.v1.Schema result) {
to_bitField0_ |= 0x00000004;
}
if (((from_bitField0_ & 0x00400000) != 0)) {
- result.ref_ = ref_;
+ result.additionalProperties_ =
+ additionalPropertiesBuilder_ == null
+ ? additionalProperties_
+ : additionalPropertiesBuilder_.build();
+ to_bitField0_ |= 0x00000008;
}
if (((from_bitField0_ & 0x00800000) != 0)) {
+ result.ref_ = ref_;
+ }
+ if (((from_bitField0_ & 0x01000000) != 0)) {
result.defs_ = internalGetDefs().build(DefsDefaultEntryHolder.defaultEntry);
}
result.bitField0_ |= to_bitField0_;
@@ -2191,13 +2281,16 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1.Schema other) {
}
}
}
+ if (other.hasAdditionalProperties()) {
+ mergeAdditionalProperties(other.getAdditionalProperties());
+ }
if (!other.getRef().isEmpty()) {
ref_ = other.ref_;
- bitField0_ |= 0x00400000;
+ bitField0_ |= 0x00800000;
onChanged();
}
internalGetMutableDefs().mergeFrom(other.internalGetDefs());
- bitField0_ |= 0x00800000;
+ bitField0_ |= 0x01000000;
this.mergeUnknownFields(other.getUnknownFields());
onChanged();
return this;
@@ -2374,10 +2467,17 @@ public Builder mergeFrom(
propertyOrdering_.add(s);
break;
} // case 202
+ case 210:
+ {
+ input.readMessage(
+ getAdditionalPropertiesFieldBuilder().getBuilder(), extensionRegistry);
+ bitField0_ |= 0x00400000;
+ break;
+ } // case 210
case 218:
{
ref_ = input.readStringRequireUtf8();
- bitField0_ |= 0x00400000;
+ bitField0_ |= 0x00800000;
break;
} // case 218
case 226:
@@ -2389,7 +2489,7 @@ public Builder mergeFrom(
DefsDefaultEntryHolder.defaultEntry.getParserForType(),
extensionRegistry);
internalGetMutableDefs().ensureBuilderMap().put(defs__.getKey(), defs__.getValue());
- bitField0_ |= 0x00800000;
+ bitField0_ |= 0x01000000;
break;
} // case 226
default:
@@ -5395,6 +5495,227 @@ public java.util.List+ * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return Whether the additionalProperties field is set.
+ */
+ public boolean hasAdditionalProperties() {
+ return ((bitField0_ & 0x00400000) != 0);
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return The additionalProperties.
+ */
+ public com.google.protobuf.Value getAdditionalProperties() {
+ if (additionalPropertiesBuilder_ == null) {
+ return additionalProperties_ == null
+ ? com.google.protobuf.Value.getDefaultInstance()
+ : additionalProperties_;
+ } else {
+ return additionalPropertiesBuilder_.getMessage();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder setAdditionalProperties(com.google.protobuf.Value value) {
+ if (additionalPropertiesBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ additionalProperties_ = value;
+ } else {
+ additionalPropertiesBuilder_.setMessage(value);
+ }
+ bitField0_ |= 0x00400000;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder setAdditionalProperties(com.google.protobuf.Value.Builder builderForValue) {
+ if (additionalPropertiesBuilder_ == null) {
+ additionalProperties_ = builderForValue.build();
+ } else {
+ additionalPropertiesBuilder_.setMessage(builderForValue.build());
+ }
+ bitField0_ |= 0x00400000;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder mergeAdditionalProperties(com.google.protobuf.Value value) {
+ if (additionalPropertiesBuilder_ == null) {
+ if (((bitField0_ & 0x00400000) != 0)
+ && additionalProperties_ != null
+ && additionalProperties_ != com.google.protobuf.Value.getDefaultInstance()) {
+ getAdditionalPropertiesBuilder().mergeFrom(value);
+ } else {
+ additionalProperties_ = value;
+ }
+ } else {
+ additionalPropertiesBuilder_.mergeFrom(value);
+ }
+ if (additionalProperties_ != null) {
+ bitField0_ |= 0x00400000;
+ onChanged();
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder clearAdditionalProperties() {
+ bitField0_ = (bitField0_ & ~0x00400000);
+ additionalProperties_ = null;
+ if (additionalPropertiesBuilder_ != null) {
+ additionalPropertiesBuilder_.dispose();
+ additionalPropertiesBuilder_ = null;
+ }
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public com.google.protobuf.Value.Builder getAdditionalPropertiesBuilder() {
+ bitField0_ |= 0x00400000;
+ onChanged();
+ return getAdditionalPropertiesFieldBuilder().getBuilder();
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public com.google.protobuf.ValueOrBuilder getAdditionalPropertiesOrBuilder() {
+ if (additionalPropertiesBuilder_ != null) {
+ return additionalPropertiesBuilder_.getMessageOrBuilder();
+ } else {
+ return additionalProperties_ == null
+ ? com.google.protobuf.Value.getDefaultInstance()
+ : additionalProperties_;
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.protobuf.Value,
+ com.google.protobuf.Value.Builder,
+ com.google.protobuf.ValueOrBuilder>
+ getAdditionalPropertiesFieldBuilder() {
+ if (additionalPropertiesBuilder_ == null) {
+ additionalPropertiesBuilder_ =
+ new com.google.protobuf.SingleFieldBuilderV3<
+ com.google.protobuf.Value,
+ com.google.protobuf.Value.Builder,
+ com.google.protobuf.ValueOrBuilder>(
+ getAdditionalProperties(), getParentForChildren(), isClean());
+ additionalProperties_ = null;
+ }
+ return additionalPropertiesBuilder_;
+ }
+
private java.lang.Object ref_ = "";
/**
@@ -5520,7 +5841,7 @@ public Builder setRef(java.lang.String value) {
throw new NullPointerException();
}
ref_ = value;
- bitField0_ |= 0x00400000;
+ bitField0_ |= 0x00800000;
onChanged();
return this;
}
@@ -5558,7 +5879,7 @@ public Builder setRef(java.lang.String value) {
*/
public Builder clearRef() {
ref_ = getDefaultInstance().getRef();
- bitField0_ = (bitField0_ & ~0x00400000);
+ bitField0_ = (bitField0_ & ~0x00800000);
onChanged();
return this;
}
@@ -5601,7 +5922,7 @@ public Builder setRefBytes(com.google.protobuf.ByteString value) {
}
checkByteStringIsUtf8(value);
ref_ = value;
- bitField0_ |= 0x00400000;
+ bitField0_ |= 0x00800000;
onChanged();
return this;
}
@@ -5658,7 +5979,7 @@ public com.google.cloud.aiplatform.v1.Schema build(
if (defs_ == null) {
defs_ = new com.google.protobuf.MapFieldBuilder<>(defsConverter);
}
- bitField0_ |= 0x00800000;
+ bitField0_ |= 0x01000000;
onChanged();
return defs_;
}
@@ -5762,7 +6083,7 @@ public com.google.cloud.aiplatform.v1.Schema getDefsOrThrow(java.lang.String key
}
public Builder clearDefs() {
- bitField0_ = (bitField0_ & ~0x00800000);
+ bitField0_ = (bitField0_ & ~0x01000000);
internalGetMutableDefs().clear();
return this;
}
@@ -5790,7 +6111,7 @@ public Builder removeDefs(java.lang.String key) {
/** Use alternate mutation accessors instead. */
@java.lang.Deprecated
public java.util.Map+ * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return Whether the additionalProperties field is set.
+ */
+ boolean hasAdditionalProperties();
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return The additionalProperties.
+ */
+ com.google.protobuf.Value getAdditionalProperties();
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ com.google.protobuf.ValueOrBuilder getAdditionalPropertiesOrBuilder();
+
/**
*
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedHyperParameters.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedHyperParameters.java
index 27fbbd36f01d..053b0944cb80 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedHyperParameters.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedHyperParameters.java
@@ -94,6 +94,16 @@ public enum AdapterSize implements com.google.protobuf.ProtocolMessageEnum {
* ADAPTER_SIZE_ONE = 1;
*/
ADAPTER_SIZE_ONE(1),
+ /**
+ *
+ *
+ * + * Adapter size 2. + *+ * + *
ADAPTER_SIZE_TWO = 6;
+ */
+ ADAPTER_SIZE_TWO(6),
/**
*
*
@@ -124,6 +134,16 @@ public enum AdapterSize implements com.google.protobuf.ProtocolMessageEnum {
* ADAPTER_SIZE_SIXTEEN = 4;
*/
ADAPTER_SIZE_SIXTEEN(4),
+ /**
+ *
+ *
+ * + * Adapter size 32. + *+ * + *
ADAPTER_SIZE_THIRTY_TWO = 5;
+ */
+ ADAPTER_SIZE_THIRTY_TWO(5),
UNRECOGNIZED(-1),
;
@@ -149,6 +169,17 @@ public enum AdapterSize implements com.google.protobuf.ProtocolMessageEnum {
*/
public static final int ADAPTER_SIZE_ONE_VALUE = 1;
+ /**
+ *
+ *
+ * + * Adapter size 2. + *+ * + *
ADAPTER_SIZE_TWO = 6;
+ */
+ public static final int ADAPTER_SIZE_TWO_VALUE = 6;
+
/**
*
*
@@ -182,6 +213,17 @@ public enum AdapterSize implements com.google.protobuf.ProtocolMessageEnum {
*/
public static final int ADAPTER_SIZE_SIXTEEN_VALUE = 4;
+ /**
+ *
+ *
+ * + * Adapter size 32. + *+ * + *
ADAPTER_SIZE_THIRTY_TWO = 5;
+ */
+ public static final int ADAPTER_SIZE_THIRTY_TWO_VALUE = 5;
+
public final int getNumber() {
if (this == UNRECOGNIZED) {
throw new java.lang.IllegalArgumentException(
@@ -210,12 +252,16 @@ public static AdapterSize forNumber(int value) {
return ADAPTER_SIZE_UNSPECIFIED;
case 1:
return ADAPTER_SIZE_ONE;
+ case 6:
+ return ADAPTER_SIZE_TWO;
case 2:
return ADAPTER_SIZE_FOUR;
case 3:
return ADAPTER_SIZE_EIGHT;
case 4:
return ADAPTER_SIZE_SIXTEEN;
+ case 5:
+ return ADAPTER_SIZE_THIRTY_TWO;
default:
return null;
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningDataStats.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningDataStats.java
index eeafb4934377..37772f2a8b6c 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningDataStats.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningDataStats.java
@@ -42,6 +42,7 @@ private SupervisedTuningDataStats(com.google.protobuf.GeneratedMessageV3.Builder
private SupervisedTuningDataStats() {
userDatasetExamples_ = java.util.Collections.emptyList();
truncatedExampleIndices_ = emptyLongList();
+ droppedExampleReasons_ = com.google.protobuf.LazyStringArrayList.emptyList();
}
@java.lang.Override
@@ -121,7 +122,7 @@ public long getTotalTuningCharacterCount() {
*
*
* @deprecated google.cloud.aiplatform.v1.SupervisedTuningDataStats.total_billable_character_count
- * is deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=221
+ * is deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=228
* @return The totalBillableCharacterCount.
*/
@java.lang.Override
@@ -448,11 +449,13 @@ public com.google.cloud.aiplatform.v1.ContentOrBuilder getUserDatasetExamplesOrB
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The totalTruncatedExampleCount.
*/
@@ -470,11 +473,13 @@ public long getTotalTruncatedExampleCount() {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return A list containing the truncatedExampleIndices.
*/
@@ -487,11 +492,13 @@ public java.util.List
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The count of truncatedExampleIndices.
*/
@@ -503,11 +510,13 @@ public int getTruncatedExampleIndicesCount() {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param index The index of the element to return.
* @return The truncatedExampleIndices at the given index.
@@ -518,6 +527,86 @@ public long getTruncatedExampleIndices(int index) {
private int truncatedExampleIndicesMemoizedSerializedSize = -1;
+ public static final int DROPPED_EXAMPLE_REASONS_FIELD_NUMBER = 12;
+
+ @SuppressWarnings("serial")
+ private com.google.protobuf.LazyStringArrayList droppedExampleReasons_ =
+ com.google.protobuf.LazyStringArrayList.emptyList();
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return A list containing the droppedExampleReasons.
+ */
+ public com.google.protobuf.ProtocolStringList getDroppedExampleReasonsList() {
+ return droppedExampleReasons_;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return The count of droppedExampleReasons.
+ */
+ public int getDroppedExampleReasonsCount() {
+ return droppedExampleReasons_.size();
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the element to return.
+ * @return The droppedExampleReasons at the given index.
+ */
+ public java.lang.String getDroppedExampleReasons(int index) {
+ return droppedExampleReasons_.get(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the value to return.
+ * @return The bytes of the droppedExampleReasons at the given index.
+ */
+ public com.google.protobuf.ByteString getDroppedExampleReasonsBytes(int index) {
+ return droppedExampleReasons_.getByteString(index);
+ }
+
private byte memoizedIsInitialized = -1;
@java.lang.Override
@@ -570,6 +659,10 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
for (int i = 0; i < truncatedExampleIndices_.size(); i++) {
output.writeInt64NoTag(truncatedExampleIndices_.getLong(i));
}
+ for (int i = 0; i < droppedExampleReasons_.size(); i++) {
+ com.google.protobuf.GeneratedMessageV3.writeString(
+ output, 12, droppedExampleReasons_.getRaw(i));
+ }
getUnknownFields().writeTo(output);
}
@@ -632,6 +725,14 @@ public int getSerializedSize() {
}
truncatedExampleIndicesMemoizedSerializedSize = dataSize;
}
+ {
+ int dataSize = 0;
+ for (int i = 0; i < droppedExampleReasons_.size(); i++) {
+ dataSize += computeStringSizeNoTag(droppedExampleReasons_.getRaw(i));
+ }
+ size += dataSize;
+ size += 1 * getDroppedExampleReasonsList().size();
+ }
size += getUnknownFields().getSerializedSize();
memoizedSize = size;
return size;
@@ -673,6 +774,7 @@ public boolean equals(final java.lang.Object obj) {
if (getTotalTruncatedExampleCount() != other.getTotalTruncatedExampleCount()) return false;
if (!getTruncatedExampleIndicesList().equals(other.getTruncatedExampleIndicesList()))
return false;
+ if (!getDroppedExampleReasonsList().equals(other.getDroppedExampleReasonsList())) return false;
if (!getUnknownFields().equals(other.getUnknownFields())) return false;
return true;
}
@@ -716,6 +818,10 @@ public int hashCode() {
hash = (37 * hash) + TRUNCATED_EXAMPLE_INDICES_FIELD_NUMBER;
hash = (53 * hash) + getTruncatedExampleIndicesList().hashCode();
}
+ if (getDroppedExampleReasonsCount() > 0) {
+ hash = (37 * hash) + DROPPED_EXAMPLE_REASONS_FIELD_NUMBER;
+ hash = (53 * hash) + getDroppedExampleReasonsList().hashCode();
+ }
hash = (29 * hash) + getUnknownFields().hashCode();
memoizedHashCode = hash;
return hash;
@@ -898,6 +1004,7 @@ public Builder clear() {
bitField0_ = (bitField0_ & ~0x00000100);
totalTruncatedExampleCount_ = 0L;
truncatedExampleIndices_ = emptyLongList();
+ droppedExampleReasons_ = com.google.protobuf.LazyStringArrayList.emptyList();
return this;
}
@@ -992,6 +1099,10 @@ private void buildPartial0(com.google.cloud.aiplatform.v1.SupervisedTuningDataSt
truncatedExampleIndices_.makeImmutable();
result.truncatedExampleIndices_ = truncatedExampleIndices_;
}
+ if (((from_bitField0_ & 0x00000800) != 0)) {
+ droppedExampleReasons_.makeImmutable();
+ result.droppedExampleReasons_ = droppedExampleReasons_;
+ }
result.bitField0_ |= to_bitField0_;
}
@@ -1106,6 +1217,16 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1.SupervisedTuningDataStat
}
onChanged();
}
+ if (!other.droppedExampleReasons_.isEmpty()) {
+ if (droppedExampleReasons_.isEmpty()) {
+ droppedExampleReasons_ = other.droppedExampleReasons_;
+ bitField0_ |= 0x00000800;
+ } else {
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.addAll(other.droppedExampleReasons_);
+ }
+ onChanged();
+ }
this.mergeUnknownFields(other.getUnknownFields());
onChanged();
return this;
@@ -1221,6 +1342,13 @@ public Builder mergeFrom(
input.popLimit(limit);
break;
} // case 90
+ case 98:
+ {
+ java.lang.String s = input.readStringRequireUtf8();
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.add(s);
+ break;
+ } // case 98
default:
{
if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -1373,7 +1501,7 @@ public Builder clearTotalTuningCharacterCount() {
*
* @deprecated
* google.cloud.aiplatform.v1.SupervisedTuningDataStats.total_billable_character_count is
- * deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=221
+ * deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=228
* @return The totalBillableCharacterCount.
*/
@java.lang.Override
@@ -1395,7 +1523,7 @@ public long getTotalBillableCharacterCount() {
*
* @deprecated
* google.cloud.aiplatform.v1.SupervisedTuningDataStats.total_billable_character_count is
- * deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=221
+ * deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=228
* @param value The totalBillableCharacterCount to set.
* @return This builder for chaining.
*/
@@ -1421,7 +1549,7 @@ public Builder setTotalBillableCharacterCount(long value) {
*
* @deprecated
* google.cloud.aiplatform.v1.SupervisedTuningDataStats.total_billable_character_count is
- * deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=221
+ * deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=228
* @return This builder for chaining.
*/
@java.lang.Deprecated
@@ -2633,11 +2761,13 @@ public com.google.cloud.aiplatform.v1.Content.Builder addUserDatasetExamplesBuil
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The totalTruncatedExampleCount.
*/
@@ -2650,11 +2780,13 @@ public long getTotalTruncatedExampleCount() {
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param value The totalTruncatedExampleCount to set.
* @return This builder for chaining.
@@ -2671,11 +2803,13 @@ public Builder setTotalTruncatedExampleCount(long value) {
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return This builder for chaining.
*/
@@ -2699,11 +2833,13 @@ private void ensureTruncatedExampleIndicesIsMutable() {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return A list containing the truncatedExampleIndices.
*/
@@ -2716,11 +2852,13 @@ public java.util.List
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The count of truncatedExampleIndices.
*/
@@ -2732,11 +2870,13 @@ public int getTruncatedExampleIndicesCount() {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param index The index of the element to return.
* @return The truncatedExampleIndices at the given index.
@@ -2749,11 +2889,13 @@ public long getTruncatedExampleIndices(int index) {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param index The index to set the value at.
* @param value The truncatedExampleIndices to set.
@@ -2772,11 +2914,13 @@ public Builder setTruncatedExampleIndices(int index, long value) {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param value The truncatedExampleIndices to add.
* @return This builder for chaining.
@@ -2794,11 +2938,13 @@ public Builder addTruncatedExampleIndices(long value) {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param values The truncatedExampleIndices to add.
* @return This builder for chaining.
@@ -2816,11 +2962,13 @@ public Builder addAllTruncatedExampleIndices(
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return This builder for chaining.
*/
@@ -2831,6 +2979,217 @@ public Builder clearTruncatedExampleIndices() {
return this;
}
+ private com.google.protobuf.LazyStringArrayList droppedExampleReasons_ =
+ com.google.protobuf.LazyStringArrayList.emptyList();
+
+ private void ensureDroppedExampleReasonsIsMutable() {
+ if (!droppedExampleReasons_.isModifiable()) {
+ droppedExampleReasons_ =
+ new com.google.protobuf.LazyStringArrayList(droppedExampleReasons_);
+ }
+ bitField0_ |= 0x00000800;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return A list containing the droppedExampleReasons.
+ */
+ public com.google.protobuf.ProtocolStringList getDroppedExampleReasonsList() {
+ droppedExampleReasons_.makeImmutable();
+ return droppedExampleReasons_;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return The count of droppedExampleReasons.
+ */
+ public int getDroppedExampleReasonsCount() {
+ return droppedExampleReasons_.size();
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the element to return.
+ * @return The droppedExampleReasons at the given index.
+ */
+ public java.lang.String getDroppedExampleReasons(int index) {
+ return droppedExampleReasons_.get(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the value to return.
+ * @return The bytes of the droppedExampleReasons at the given index.
+ */
+ public com.google.protobuf.ByteString getDroppedExampleReasonsBytes(int index) {
+ return droppedExampleReasons_.getByteString(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index to set the value at.
+ * @param value The droppedExampleReasons to set.
+ * @return This builder for chaining.
+ */
+ public Builder setDroppedExampleReasons(int index, java.lang.String value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.set(index, value);
+ bitField0_ |= 0x00000800;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param value The droppedExampleReasons to add.
+ * @return This builder for chaining.
+ */
+ public Builder addDroppedExampleReasons(java.lang.String value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.add(value);
+ bitField0_ |= 0x00000800;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param values The droppedExampleReasons to add.
+ * @return This builder for chaining.
+ */
+ public Builder addAllDroppedExampleReasons(java.lang.Iterable+ * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearDroppedExampleReasons() {
+ droppedExampleReasons_ = com.google.protobuf.LazyStringArrayList.emptyList();
+ bitField0_ = (bitField0_ & ~0x00000800);
+ ;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param value The bytes of the droppedExampleReasons to add.
+ * @return This builder for chaining.
+ */
+ public Builder addDroppedExampleReasonsBytes(com.google.protobuf.ByteString value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ checkByteStringIsUtf8(value);
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.add(value);
+ bitField0_ |= 0x00000800;
+ onChanged();
+ return this;
+ }
+
@java.lang.Override
public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.setUnknownFields(unknownFields);
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningDataStatsOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningDataStatsOrBuilder.java
index 26b6cc2adf74..472841287997 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningDataStatsOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningDataStatsOrBuilder.java
@@ -64,7 +64,7 @@ public interface SupervisedTuningDataStatsOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1.SupervisedTuningDataStats.total_billable_character_count
- * is deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=221
+ * is deprecated. See google/cloud/aiplatform/v1/tuning_job.proto;l=228
* @return The totalBillableCharacterCount.
*/
@java.lang.Deprecated
@@ -301,11 +301,13 @@ public interface SupervisedTuningDataStatsOrBuilder
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The totalTruncatedExampleCount.
*/
@@ -315,11 +317,13 @@ public interface SupervisedTuningDataStatsOrBuilder
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return A list containing the truncatedExampleIndices.
*/
@@ -329,11 +333,13 @@ public interface SupervisedTuningDataStatsOrBuilder
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The count of truncatedExampleIndices.
*/
@@ -343,14 +349,82 @@ public interface SupervisedTuningDataStatsOrBuilder
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param index The index of the element to return.
* @return The truncatedExampleIndices at the given index.
*/
long getTruncatedExampleIndices(int index);
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return A list containing the droppedExampleReasons.
+ */
+ java.util.List+ * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return The count of droppedExampleReasons.
+ */
+ int getDroppedExampleReasonsCount();
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the element to return.
+ * @return The droppedExampleReasons at the given index.
+ */
+ java.lang.String getDroppedExampleReasons(int index);
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the value to return.
+ * @return The bytes of the droppedExampleReasons at the given index.
+ */
+ com.google.protobuf.ByteString getDroppedExampleReasonsBytes(int index);
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningSpec.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningSpec.java
index 0c838ed056e6..b8eb6a2f7bac 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningSpec.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningSpec.java
@@ -236,6 +236,27 @@ public com.google.cloud.aiplatform.v1.SupervisedHyperParameters getHyperParamete
: hyperParameters_;
}
+ public static final int EXPORT_LAST_CHECKPOINT_ONLY_FIELD_NUMBER = 6;
+ private boolean exportLastCheckpointOnly_ = false;
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @return The exportLastCheckpointOnly.
+ */
+ @java.lang.Override
+ public boolean getExportLastCheckpointOnly() {
+ return exportLastCheckpointOnly_;
+ }
+
private byte memoizedIsInitialized = -1;
@java.lang.Override
@@ -259,6 +280,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
if (((bitField0_ & 0x00000001) != 0)) {
output.writeMessage(3, getHyperParameters());
}
+ if (exportLastCheckpointOnly_ != false) {
+ output.writeBool(6, exportLastCheckpointOnly_);
+ }
getUnknownFields().writeTo(output);
}
@@ -277,6 +301,9 @@ public int getSerializedSize() {
if (((bitField0_ & 0x00000001) != 0)) {
size += com.google.protobuf.CodedOutputStream.computeMessageSize(3, getHyperParameters());
}
+ if (exportLastCheckpointOnly_ != false) {
+ size += com.google.protobuf.CodedOutputStream.computeBoolSize(6, exportLastCheckpointOnly_);
+ }
size += getUnknownFields().getSerializedSize();
memoizedSize = size;
return size;
@@ -299,6 +326,7 @@ public boolean equals(final java.lang.Object obj) {
if (hasHyperParameters()) {
if (!getHyperParameters().equals(other.getHyperParameters())) return false;
}
+ if (getExportLastCheckpointOnly() != other.getExportLastCheckpointOnly()) return false;
if (!getUnknownFields().equals(other.getUnknownFields())) return false;
return true;
}
@@ -318,6 +346,8 @@ public int hashCode() {
hash = (37 * hash) + HYPER_PARAMETERS_FIELD_NUMBER;
hash = (53 * hash) + getHyperParameters().hashCode();
}
+ hash = (37 * hash) + EXPORT_LAST_CHECKPOINT_ONLY_FIELD_NUMBER;
+ hash = (53 * hash) + com.google.protobuf.Internal.hashBoolean(getExportLastCheckpointOnly());
hash = (29 * hash) + getUnknownFields().hashCode();
memoizedHashCode = hash;
return hash;
@@ -474,6 +504,7 @@ public Builder clear() {
hyperParametersBuilder_.dispose();
hyperParametersBuilder_ = null;
}
+ exportLastCheckpointOnly_ = false;
return this;
}
@@ -522,6 +553,9 @@ private void buildPartial0(com.google.cloud.aiplatform.v1.SupervisedTuningSpec r
hyperParametersBuilder_ == null ? hyperParameters_ : hyperParametersBuilder_.build();
to_bitField0_ |= 0x00000001;
}
+ if (((from_bitField0_ & 0x00000008) != 0)) {
+ result.exportLastCheckpointOnly_ = exportLastCheckpointOnly_;
+ }
result.bitField0_ |= to_bitField0_;
}
@@ -584,6 +618,9 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1.SupervisedTuningSpec oth
if (other.hasHyperParameters()) {
mergeHyperParameters(other.getHyperParameters());
}
+ if (other.getExportLastCheckpointOnly() != false) {
+ setExportLastCheckpointOnly(other.getExportLastCheckpointOnly());
+ }
this.mergeUnknownFields(other.getUnknownFields());
onChanged();
return this;
@@ -628,6 +665,12 @@ public Builder mergeFrom(
bitField0_ |= 0x00000004;
break;
} // case 26
+ case 48:
+ {
+ exportLastCheckpointOnly_ = input.readBool();
+ bitField0_ |= 0x00000008;
+ break;
+ } // case 48
default:
{
if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -1097,6 +1140,68 @@ public Builder clearHyperParameters() {
return hyperParametersBuilder_;
}
+ private boolean exportLastCheckpointOnly_;
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @return The exportLastCheckpointOnly.
+ */
+ @java.lang.Override
+ public boolean getExportLastCheckpointOnly() {
+ return exportLastCheckpointOnly_;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @param value The exportLastCheckpointOnly to set.
+ * @return This builder for chaining.
+ */
+ public Builder setExportLastCheckpointOnly(boolean value) {
+
+ exportLastCheckpointOnly_ = value;
+ bitField0_ |= 0x00000008;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearExportLastCheckpointOnly() {
+ bitField0_ = (bitField0_ & ~0x00000008);
+ exportLastCheckpointOnly_ = false;
+ onChanged();
+ return this;
+ }
+
@java.lang.Override
public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.setUnknownFields(unknownFields);
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningSpecOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningSpecOrBuilder.java
index 35d78dd610c1..92e0de88187e 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningSpecOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/SupervisedTuningSpecOrBuilder.java
@@ -122,4 +122,19 @@ public interface SupervisedTuningSpecOrBuilder
*
*/
com.google.cloud.aiplatform.v1.SupervisedHyperParametersOrBuilder getHyperParametersOrBuilder();
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @return The exportLastCheckpointOnly.
+ */
+ boolean getExportLastCheckpointOnly();
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TunedModel.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TunedModel.java
index b298884c6268..bf73f097bd3d 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TunedModel.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TunedModel.java
@@ -23,7 +23,7 @@
*
*
* - * The Model Registry Model and Online Prediction Endpoint assiociated with + * The Model Registry Model and Online Prediction Endpoint associated with * this [TuningJob][google.cloud.aiplatform.v1.TuningJob]. ** @@ -43,6 +43,7 @@ private TunedModel(com.google.protobuf.GeneratedMessageV3.Builder builder) { private TunedModel() { model_ = ""; endpoint_ = ""; + checkpoints_ = java.util.Collections.emptyList(); } @java.lang.Override @@ -184,6 +185,103 @@ public com.google.protobuf.ByteString getEndpointBytes() { } } + public static final int CHECKPOINTS_FIELD_NUMBER = 3; + + @SuppressWarnings("serial") + private java.util.List
+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public java.util.List+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public java.util.List
+ getCheckpointsOrBuilderList() {
+ return checkpoints_;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public int getCheckpointsCount() {
+ return checkpoints_.size();
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.TunedModelCheckpoint getCheckpoints(int index) {
+ return checkpoints_.get(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1.TunedModelCheckpointOrBuilder getCheckpointsOrBuilder(
+ int index) {
+ return checkpoints_.get(index);
+ }
+
private byte memoizedIsInitialized = -1;
@java.lang.Override
@@ -204,6 +302,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(endpoint_)) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 2, endpoint_);
}
+ for (int i = 0; i < checkpoints_.size(); i++) {
+ output.writeMessage(3, checkpoints_.get(i));
+ }
getUnknownFields().writeTo(output);
}
@@ -219,6 +320,9 @@ public int getSerializedSize() {
if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(endpoint_)) {
size += com.google.protobuf.GeneratedMessageV3.computeStringSize(2, endpoint_);
}
+ for (int i = 0; i < checkpoints_.size(); i++) {
+ size += com.google.protobuf.CodedOutputStream.computeMessageSize(3, checkpoints_.get(i));
+ }
size += getUnknownFields().getSerializedSize();
memoizedSize = size;
return size;
@@ -237,6 +341,7 @@ public boolean equals(final java.lang.Object obj) {
if (!getModel().equals(other.getModel())) return false;
if (!getEndpoint().equals(other.getEndpoint())) return false;
+ if (!getCheckpointsList().equals(other.getCheckpointsList())) return false;
if (!getUnknownFields().equals(other.getUnknownFields())) return false;
return true;
}
@@ -252,6 +357,10 @@ public int hashCode() {
hash = (53 * hash) + getModel().hashCode();
hash = (37 * hash) + ENDPOINT_FIELD_NUMBER;
hash = (53 * hash) + getEndpoint().hashCode();
+ if (getCheckpointsCount() > 0) {
+ hash = (37 * hash) + CHECKPOINTS_FIELD_NUMBER;
+ hash = (53 * hash) + getCheckpointsList().hashCode();
+ }
hash = (29 * hash) + getUnknownFields().hashCode();
memoizedHashCode = hash;
return hash;
@@ -357,7 +466,7 @@ protected Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.Build
*
*
*
- * The Model Registry Model and Online Prediction Endpoint assiociated with
+ * The Model Registry Model and Online Prediction Endpoint associated with
* this [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -395,6 +504,13 @@ public Builder clear() {
bitField0_ = 0;
model_ = "";
endpoint_ = "";
+ if (checkpointsBuilder_ == null) {
+ checkpoints_ = java.util.Collections.emptyList();
+ } else {
+ checkpoints_ = null;
+ checkpointsBuilder_.clear();
+ }
+ bitField0_ = (bitField0_ & ~0x00000004);
return this;
}
@@ -422,6 +538,7 @@ public com.google.cloud.aiplatform.v1.TunedModel build() {
public com.google.cloud.aiplatform.v1.TunedModel buildPartial() {
com.google.cloud.aiplatform.v1.TunedModel result =
new com.google.cloud.aiplatform.v1.TunedModel(this);
+ buildPartialRepeatedFields(result);
if (bitField0_ != 0) {
buildPartial0(result);
}
@@ -429,6 +546,18 @@ public com.google.cloud.aiplatform.v1.TunedModel buildPartial() {
return result;
}
+ private void buildPartialRepeatedFields(com.google.cloud.aiplatform.v1.TunedModel result) {
+ if (checkpointsBuilder_ == null) {
+ if (((bitField0_ & 0x00000004) != 0)) {
+ checkpoints_ = java.util.Collections.unmodifiableList(checkpoints_);
+ bitField0_ = (bitField0_ & ~0x00000004);
+ }
+ result.checkpoints_ = checkpoints_;
+ } else {
+ result.checkpoints_ = checkpointsBuilder_.build();
+ }
+ }
+
private void buildPartial0(com.google.cloud.aiplatform.v1.TunedModel result) {
int from_bitField0_ = bitField0_;
if (((from_bitField0_ & 0x00000001) != 0)) {
@@ -494,6 +623,33 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1.TunedModel other) {
bitField0_ |= 0x00000002;
onChanged();
}
+ if (checkpointsBuilder_ == null) {
+ if (!other.checkpoints_.isEmpty()) {
+ if (checkpoints_.isEmpty()) {
+ checkpoints_ = other.checkpoints_;
+ bitField0_ = (bitField0_ & ~0x00000004);
+ } else {
+ ensureCheckpointsIsMutable();
+ checkpoints_.addAll(other.checkpoints_);
+ }
+ onChanged();
+ }
+ } else {
+ if (!other.checkpoints_.isEmpty()) {
+ if (checkpointsBuilder_.isEmpty()) {
+ checkpointsBuilder_.dispose();
+ checkpointsBuilder_ = null;
+ checkpoints_ = other.checkpoints_;
+ bitField0_ = (bitField0_ & ~0x00000004);
+ checkpointsBuilder_ =
+ com.google.protobuf.GeneratedMessageV3.alwaysUseFieldBuilders
+ ? getCheckpointsFieldBuilder()
+ : null;
+ } else {
+ checkpointsBuilder_.addAllMessages(other.checkpoints_);
+ }
+ }
+ }
this.mergeUnknownFields(other.getUnknownFields());
onChanged();
return this;
@@ -532,6 +688,20 @@ public Builder mergeFrom(
bitField0_ |= 0x00000002;
break;
} // case 18
+ case 26:
+ {
+ com.google.cloud.aiplatform.v1.TunedModelCheckpoint m =
+ input.readMessage(
+ com.google.cloud.aiplatform.v1.TunedModelCheckpoint.parser(),
+ extensionRegistry);
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(m);
+ } else {
+ checkpointsBuilder_.addMessage(m);
+ }
+ break;
+ } // case 26
default:
{
if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -803,6 +973,454 @@ public Builder setEndpointBytes(com.google.protobuf.ByteString value) {
return this;
}
+ private java.util.List+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public java.util.List+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public int getCheckpointsCount() {
+ if (checkpointsBuilder_ == null) {
+ return checkpoints_.size();
+ } else {
+ return checkpointsBuilder_.getCount();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1.TunedModelCheckpoint getCheckpoints(int index) {
+ if (checkpointsBuilder_ == null) {
+ return checkpoints_.get(index);
+ } else {
+ return checkpointsBuilder_.getMessage(index);
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder setCheckpoints(
+ int index, com.google.cloud.aiplatform.v1.TunedModelCheckpoint value) {
+ if (checkpointsBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureCheckpointsIsMutable();
+ checkpoints_.set(index, value);
+ onChanged();
+ } else {
+ checkpointsBuilder_.setMessage(index, value);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder setCheckpoints(
+ int index, com.google.cloud.aiplatform.v1.TunedModelCheckpoint.Builder builderForValue) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.set(index, builderForValue.build());
+ onChanged();
+ } else {
+ checkpointsBuilder_.setMessage(index, builderForValue.build());
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addCheckpoints(com.google.cloud.aiplatform.v1.TunedModelCheckpoint value) {
+ if (checkpointsBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(value);
+ onChanged();
+ } else {
+ checkpointsBuilder_.addMessage(value);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addCheckpoints(
+ int index, com.google.cloud.aiplatform.v1.TunedModelCheckpoint value) {
+ if (checkpointsBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(index, value);
+ onChanged();
+ } else {
+ checkpointsBuilder_.addMessage(index, value);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addCheckpoints(
+ com.google.cloud.aiplatform.v1.TunedModelCheckpoint.Builder builderForValue) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(builderForValue.build());
+ onChanged();
+ } else {
+ checkpointsBuilder_.addMessage(builderForValue.build());
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addCheckpoints(
+ int index, com.google.cloud.aiplatform.v1.TunedModelCheckpoint.Builder builderForValue) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(index, builderForValue.build());
+ onChanged();
+ } else {
+ checkpointsBuilder_.addMessage(index, builderForValue.build());
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addAllCheckpoints(
+ java.lang.Iterable values) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ com.google.protobuf.AbstractMessageLite.Builder.addAll(values, checkpoints_);
+ onChanged();
+ } else {
+ checkpointsBuilder_.addAllMessages(values);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder clearCheckpoints() {
+ if (checkpointsBuilder_ == null) {
+ checkpoints_ = java.util.Collections.emptyList();
+ bitField0_ = (bitField0_ & ~0x00000004);
+ onChanged();
+ } else {
+ checkpointsBuilder_.clear();
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder removeCheckpoints(int index) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.remove(index);
+ onChanged();
+ } else {
+ checkpointsBuilder_.remove(index);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1.TunedModelCheckpoint.Builder getCheckpointsBuilder(
+ int index) {
+ return getCheckpointsFieldBuilder().getBuilder(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1.TunedModelCheckpointOrBuilder getCheckpointsOrBuilder(
+ int index) {
+ if (checkpointsBuilder_ == null) {
+ return checkpoints_.get(index);
+ } else {
+ return checkpointsBuilder_.getMessageOrBuilder(index);
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public java.util.List
+ getCheckpointsOrBuilderList() {
+ if (checkpointsBuilder_ != null) {
+ return checkpointsBuilder_.getMessageOrBuilderList();
+ } else {
+ return java.util.Collections.unmodifiableList(checkpoints_);
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1.TunedModelCheckpoint.Builder addCheckpointsBuilder() {
+ return getCheckpointsFieldBuilder()
+ .addBuilder(com.google.cloud.aiplatform.v1.TunedModelCheckpoint.getDefaultInstance());
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1.TunedModelCheckpoint.Builder addCheckpointsBuilder(
+ int index) {
+ return getCheckpointsFieldBuilder()
+ .addBuilder(
+ index, com.google.cloud.aiplatform.v1.TunedModelCheckpoint.getDefaultInstance());
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public java.util.List+ * TunedModelCheckpoint for the Tuned Model of a Tuning Job. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1.TunedModelCheckpoint} + */ +public final class TunedModelCheckpoint extends com.google.protobuf.GeneratedMessageV3 + implements + // @@protoc_insertion_point(message_implements:google.cloud.aiplatform.v1.TunedModelCheckpoint) + TunedModelCheckpointOrBuilder { + private static final long serialVersionUID = 0L; + + // Use TunedModelCheckpoint.newBuilder() to construct. + private TunedModelCheckpoint(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + + private TunedModelCheckpoint() { + checkpointId_ = ""; + endpoint_ = ""; + } + + @java.lang.Override + @SuppressWarnings({"unused"}) + protected java.lang.Object newInstance(UnusedPrivateParameter unused) { + return new TunedModelCheckpoint(); + } + + public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() { + return com.google.cloud.aiplatform.v1.TuningJobProto + .internal_static_google_cloud_aiplatform_v1_TunedModelCheckpoint_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return com.google.cloud.aiplatform.v1.TuningJobProto + .internal_static_google_cloud_aiplatform_v1_TunedModelCheckpoint_fieldAccessorTable + .ensureFieldAccessorsInitialized( + com.google.cloud.aiplatform.v1.TunedModelCheckpoint.class, + com.google.cloud.aiplatform.v1.TunedModelCheckpoint.Builder.class); + } + + public static final int CHECKPOINT_ID_FIELD_NUMBER = 1; + + @SuppressWarnings("serial") + private volatile java.lang.Object checkpointId_ = ""; + + /** + * + * + *
+ * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The checkpointId.
+ */
+ @java.lang.Override
+ public java.lang.String getCheckpointId() {
+ java.lang.Object ref = checkpointId_;
+ if (ref instanceof java.lang.String) {
+ return (java.lang.String) ref;
+ } else {
+ com.google.protobuf.ByteString bs = (com.google.protobuf.ByteString) ref;
+ java.lang.String s = bs.toStringUtf8();
+ checkpointId_ = s;
+ return s;
+ }
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The bytes for checkpointId.
+ */
+ @java.lang.Override
+ public com.google.protobuf.ByteString getCheckpointIdBytes() {
+ java.lang.Object ref = checkpointId_;
+ if (ref instanceof java.lang.String) {
+ com.google.protobuf.ByteString b =
+ com.google.protobuf.ByteString.copyFromUtf8((java.lang.String) ref);
+ checkpointId_ = b;
+ return b;
+ } else {
+ return (com.google.protobuf.ByteString) ref;
+ }
+ }
+
+ public static final int EPOCH_FIELD_NUMBER = 2;
+ private long epoch_ = 0L;
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @return The epoch.
+ */
+ @java.lang.Override
+ public long getEpoch() {
+ return epoch_;
+ }
+
+ public static final int STEP_FIELD_NUMBER = 3;
+ private long step_ = 0L;
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @return The step.
+ */
+ @java.lang.Override
+ public long getStep() {
+ return step_;
+ }
+
+ public static final int ENDPOINT_FIELD_NUMBER = 4;
+
+ @SuppressWarnings("serial")
+ private volatile java.lang.Object endpoint_ = "";
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The endpoint.
+ */
+ @java.lang.Override
+ public java.lang.String getEndpoint() {
+ java.lang.Object ref = endpoint_;
+ if (ref instanceof java.lang.String) {
+ return (java.lang.String) ref;
+ } else {
+ com.google.protobuf.ByteString bs = (com.google.protobuf.ByteString) ref;
+ java.lang.String s = bs.toStringUtf8();
+ endpoint_ = s;
+ return s;
+ }
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The bytes for endpoint.
+ */
+ @java.lang.Override
+ public com.google.protobuf.ByteString getEndpointBytes() {
+ java.lang.Object ref = endpoint_;
+ if (ref instanceof java.lang.String) {
+ com.google.protobuf.ByteString b =
+ com.google.protobuf.ByteString.copyFromUtf8((java.lang.String) ref);
+ endpoint_ = b;
+ return b;
+ } else {
+ return (com.google.protobuf.ByteString) ref;
+ }
+ }
+
+ private byte memoizedIsInitialized = -1;
+
+ @java.lang.Override
+ public final boolean isInitialized() {
+ byte isInitialized = memoizedIsInitialized;
+ if (isInitialized == 1) return true;
+ if (isInitialized == 0) return false;
+
+ memoizedIsInitialized = 1;
+ return true;
+ }
+
+ @java.lang.Override
+ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException {
+ if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(checkpointId_)) {
+ com.google.protobuf.GeneratedMessageV3.writeString(output, 1, checkpointId_);
+ }
+ if (epoch_ != 0L) {
+ output.writeInt64(2, epoch_);
+ }
+ if (step_ != 0L) {
+ output.writeInt64(3, step_);
+ }
+ if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(endpoint_)) {
+ com.google.protobuf.GeneratedMessageV3.writeString(output, 4, endpoint_);
+ }
+ getUnknownFields().writeTo(output);
+ }
+
+ @java.lang.Override
+ public int getSerializedSize() {
+ int size = memoizedSize;
+ if (size != -1) return size;
+
+ size = 0;
+ if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(checkpointId_)) {
+ size += com.google.protobuf.GeneratedMessageV3.computeStringSize(1, checkpointId_);
+ }
+ if (epoch_ != 0L) {
+ size += com.google.protobuf.CodedOutputStream.computeInt64Size(2, epoch_);
+ }
+ if (step_ != 0L) {
+ size += com.google.protobuf.CodedOutputStream.computeInt64Size(3, step_);
+ }
+ if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(endpoint_)) {
+ size += com.google.protobuf.GeneratedMessageV3.computeStringSize(4, endpoint_);
+ }
+ size += getUnknownFields().getSerializedSize();
+ memoizedSize = size;
+ return size;
+ }
+
+ @java.lang.Override
+ public boolean equals(final java.lang.Object obj) {
+ if (obj == this) {
+ return true;
+ }
+ if (!(obj instanceof com.google.cloud.aiplatform.v1.TunedModelCheckpoint)) {
+ return super.equals(obj);
+ }
+ com.google.cloud.aiplatform.v1.TunedModelCheckpoint other =
+ (com.google.cloud.aiplatform.v1.TunedModelCheckpoint) obj;
+
+ if (!getCheckpointId().equals(other.getCheckpointId())) return false;
+ if (getEpoch() != other.getEpoch()) return false;
+ if (getStep() != other.getStep()) return false;
+ if (!getEndpoint().equals(other.getEndpoint())) return false;
+ if (!getUnknownFields().equals(other.getUnknownFields())) return false;
+ return true;
+ }
+
+ @java.lang.Override
+ public int hashCode() {
+ if (memoizedHashCode != 0) {
+ return memoizedHashCode;
+ }
+ int hash = 41;
+ hash = (19 * hash) + getDescriptor().hashCode();
+ hash = (37 * hash) + CHECKPOINT_ID_FIELD_NUMBER;
+ hash = (53 * hash) + getCheckpointId().hashCode();
+ hash = (37 * hash) + EPOCH_FIELD_NUMBER;
+ hash = (53 * hash) + com.google.protobuf.Internal.hashLong(getEpoch());
+ hash = (37 * hash) + STEP_FIELD_NUMBER;
+ hash = (53 * hash) + com.google.protobuf.Internal.hashLong(getStep());
+ hash = (37 * hash) + ENDPOINT_FIELD_NUMBER;
+ hash = (53 * hash) + getEndpoint().hashCode();
+ hash = (29 * hash) + getUnknownFields().hashCode();
+ memoizedHashCode = hash;
+ return hash;
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(
+ java.nio.ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(
+ java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(
+ com.google.protobuf.ByteString data)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(
+ com.google.protobuf.ByteString data,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(byte[] data)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(
+ byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(
+ java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(
+ java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseDelimitedFrom(
+ java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseDelimitedFrom(
+ java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(
+ com.google.protobuf.CodedInputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint parseFrom(
+ com.google.protobuf.CodedInputStream input,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ @java.lang.Override
+ public Builder newBuilderForType() {
+ return newBuilder();
+ }
+
+ public static Builder newBuilder() {
+ return DEFAULT_INSTANCE.toBuilder();
+ }
+
+ public static Builder newBuilder(com.google.cloud.aiplatform.v1.TunedModelCheckpoint prototype) {
+ return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
+ }
+
+ @java.lang.Override
+ public Builder toBuilder() {
+ return this == DEFAULT_INSTANCE ? new Builder() : new Builder().mergeFrom(this);
+ }
+
+ @java.lang.Override
+ protected Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
+ Builder builder = new Builder(parent);
+ return builder;
+ }
+
+ /**
+ *
+ *
+ * + * TunedModelCheckpoint for the Tuned Model of a Tuning Job. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1.TunedModelCheckpoint} + */ + public static final class Builder extends com.google.protobuf.GeneratedMessageV3.Builder
+ * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The checkpointId.
+ */
+ public java.lang.String getCheckpointId() {
+ java.lang.Object ref = checkpointId_;
+ if (!(ref instanceof java.lang.String)) {
+ com.google.protobuf.ByteString bs = (com.google.protobuf.ByteString) ref;
+ java.lang.String s = bs.toStringUtf8();
+ checkpointId_ = s;
+ return s;
+ } else {
+ return (java.lang.String) ref;
+ }
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The bytes for checkpointId.
+ */
+ public com.google.protobuf.ByteString getCheckpointIdBytes() {
+ java.lang.Object ref = checkpointId_;
+ if (ref instanceof String) {
+ com.google.protobuf.ByteString b =
+ com.google.protobuf.ByteString.copyFromUtf8((java.lang.String) ref);
+ checkpointId_ = b;
+ return b;
+ } else {
+ return (com.google.protobuf.ByteString) ref;
+ }
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @param value The checkpointId to set.
+ * @return This builder for chaining.
+ */
+ public Builder setCheckpointId(java.lang.String value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ checkpointId_ = value;
+ bitField0_ |= 0x00000001;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearCheckpointId() {
+ checkpointId_ = getDefaultInstance().getCheckpointId();
+ bitField0_ = (bitField0_ & ~0x00000001);
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @param value The bytes for checkpointId to set.
+ * @return This builder for chaining.
+ */
+ public Builder setCheckpointIdBytes(com.google.protobuf.ByteString value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ checkByteStringIsUtf8(value);
+ checkpointId_ = value;
+ bitField0_ |= 0x00000001;
+ onChanged();
+ return this;
+ }
+
+ private long epoch_;
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @return The epoch.
+ */
+ @java.lang.Override
+ public long getEpoch() {
+ return epoch_;
+ }
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @param value The epoch to set.
+ * @return This builder for chaining.
+ */
+ public Builder setEpoch(long value) {
+
+ epoch_ = value;
+ bitField0_ |= 0x00000002;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearEpoch() {
+ bitField0_ = (bitField0_ & ~0x00000002);
+ epoch_ = 0L;
+ onChanged();
+ return this;
+ }
+
+ private long step_;
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @return The step.
+ */
+ @java.lang.Override
+ public long getStep() {
+ return step_;
+ }
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @param value The step to set.
+ * @return This builder for chaining.
+ */
+ public Builder setStep(long value) {
+
+ step_ = value;
+ bitField0_ |= 0x00000004;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearStep() {
+ bitField0_ = (bitField0_ & ~0x00000004);
+ step_ = 0L;
+ onChanged();
+ return this;
+ }
+
+ private java.lang.Object endpoint_ = "";
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The endpoint.
+ */
+ public java.lang.String getEndpoint() {
+ java.lang.Object ref = endpoint_;
+ if (!(ref instanceof java.lang.String)) {
+ com.google.protobuf.ByteString bs = (com.google.protobuf.ByteString) ref;
+ java.lang.String s = bs.toStringUtf8();
+ endpoint_ = s;
+ return s;
+ } else {
+ return (java.lang.String) ref;
+ }
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The bytes for endpoint.
+ */
+ public com.google.protobuf.ByteString getEndpointBytes() {
+ java.lang.Object ref = endpoint_;
+ if (ref instanceof String) {
+ com.google.protobuf.ByteString b =
+ com.google.protobuf.ByteString.copyFromUtf8((java.lang.String) ref);
+ endpoint_ = b;
+ return b;
+ } else {
+ return (com.google.protobuf.ByteString) ref;
+ }
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @param value The endpoint to set.
+ * @return This builder for chaining.
+ */
+ public Builder setEndpoint(java.lang.String value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ endpoint_ = value;
+ bitField0_ |= 0x00000008;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearEndpoint() {
+ endpoint_ = getDefaultInstance().getEndpoint();
+ bitField0_ = (bitField0_ & ~0x00000008);
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @param value The bytes for endpoint to set.
+ * @return This builder for chaining.
+ */
+ public Builder setEndpointBytes(com.google.protobuf.ByteString value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ checkByteStringIsUtf8(value);
+ endpoint_ = value;
+ bitField0_ |= 0x00000008;
+ onChanged();
+ return this;
+ }
+
+ @java.lang.Override
+ public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
+ return super.setUnknownFields(unknownFields);
+ }
+
+ @java.lang.Override
+ public final Builder mergeUnknownFields(
+ final com.google.protobuf.UnknownFieldSet unknownFields) {
+ return super.mergeUnknownFields(unknownFields);
+ }
+
+ // @@protoc_insertion_point(builder_scope:google.cloud.aiplatform.v1.TunedModelCheckpoint)
+ }
+
+ // @@protoc_insertion_point(class_scope:google.cloud.aiplatform.v1.TunedModelCheckpoint)
+ private static final com.google.cloud.aiplatform.v1.TunedModelCheckpoint DEFAULT_INSTANCE;
+
+ static {
+ DEFAULT_INSTANCE = new com.google.cloud.aiplatform.v1.TunedModelCheckpoint();
+ }
+
+ public static com.google.cloud.aiplatform.v1.TunedModelCheckpoint getDefaultInstance() {
+ return DEFAULT_INSTANCE;
+ }
+
+ private static final com.google.protobuf.Parser+ * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The checkpointId.
+ */
+ java.lang.String getCheckpointId();
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The bytes for checkpointId.
+ */
+ com.google.protobuf.ByteString getCheckpointIdBytes();
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @return The epoch.
+ */
+ long getEpoch();
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @return The step.
+ */
+ long getStep();
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The endpoint.
+ */
+ java.lang.String getEndpoint();
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The bytes for endpoint.
+ */
+ com.google.protobuf.ByteString getEndpointBytes();
+}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TunedModelOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TunedModelOrBuilder.java
index 9aa35078510d..17fc95ac3e2e 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TunedModelOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TunedModelOrBuilder.java
@@ -87,4 +87,80 @@ public interface TunedModelOrBuilder
* @return The bytes for endpoint.
*/
com.google.protobuf.ByteString getEndpointBytes();
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ java.util.List+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ com.google.cloud.aiplatform.v1.TunedModelCheckpoint getCheckpoints(int index);
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ int getCheckpointsCount();
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ java.util.List
+ getCheckpointsOrBuilderList();
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ com.google.cloud.aiplatform.v1.TunedModelCheckpointOrBuilder getCheckpointsOrBuilder(int index);
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJob.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJob.java
index 61ac3d8b8169..a5abef18b170 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJob.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJob.java
@@ -182,7 +182,8 @@ public TuningSpecCase getTuningSpecCase() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -197,7 +198,8 @@ public boolean hasBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -225,7 +227,8 @@ public java.lang.String getBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -1006,7 +1009,7 @@ public com.google.protobuf.ByteString getExperimentBytes() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -1025,7 +1028,7 @@ public boolean hasTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -1046,7 +1049,7 @@ public com.google.cloud.aiplatform.v1.TunedModel getTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -2174,7 +2177,8 @@ public Builder clearTuningSpec() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2190,7 +2194,8 @@ public boolean hasBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2219,7 +2224,8 @@ public java.lang.String getBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2248,7 +2254,8 @@ public com.google.protobuf.ByteString getBaseModelBytes() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2270,7 +2277,8 @@ public Builder setBaseModel(java.lang.String value) {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2290,7 +2298,8 @@ public Builder clearBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -4451,7 +4460,7 @@ public Builder setExperimentBytes(com.google.protobuf.ByteString value) {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -4469,7 +4478,7 @@ public boolean hasTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -4493,7 +4502,7 @@ public com.google.cloud.aiplatform.v1.TunedModel getTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -4519,7 +4528,7 @@ public Builder setTunedModel(com.google.cloud.aiplatform.v1.TunedModel value) {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -4543,7 +4552,7 @@ public Builder setTunedModel(
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -4574,7 +4583,7 @@ public Builder mergeTunedModel(com.google.cloud.aiplatform.v1.TunedModel value)
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -4597,7 +4606,7 @@ public Builder clearTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -4615,7 +4624,7 @@ public com.google.cloud.aiplatform.v1.TunedModel.Builder getTunedModelBuilder()
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -4637,7 +4646,7 @@ public com.google.cloud.aiplatform.v1.TunedModelOrBuilder getTunedModelOrBuilder
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJobOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJobOrBuilder.java
index 7c693683b6a2..268a355f5d98 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJobOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJobOrBuilder.java
@@ -28,7 +28,8 @@ public interface TuningJobOrBuilder
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -41,7 +42,8 @@ public interface TuningJobOrBuilder
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -54,7 +56,8 @@ public interface TuningJobOrBuilder
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -585,7 +588,7 @@ java.lang.String getLabelsOrDefault(
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -601,7 +604,7 @@ java.lang.String getLabelsOrDefault(
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -617,7 +620,7 @@ java.lang.String getLabelsOrDefault(
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJobProto.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJobProto.java
index 82d01c002cb5..f3859a7fd8b3 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJobProto.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/TuningJobProto.java
@@ -68,6 +68,10 @@ public static void registerAllExtensions(com.google.protobuf.ExtensionRegistry r
internal_static_google_cloud_aiplatform_v1_TunedModelRef_descriptor;
static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
internal_static_google_cloud_aiplatform_v1_TunedModelRef_fieldAccessorTable;
+ static final com.google.protobuf.Descriptors.Descriptor
+ internal_static_google_cloud_aiplatform_v1_TunedModelCheckpoint_descriptor;
+ static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
+ internal_static_google_cloud_aiplatform_v1_TunedModelCheckpoint_fieldAccessorTable;
public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
return descriptor;
@@ -119,12 +123,14 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "oject}/locations/{location}/tuningJobs/{tuning_job}*\n"
+ "tuningJobs2\ttuningJobB\016\n"
+ "\014source_modelB\r\n"
- + "\013tuning_spec\"\202\001\n\n"
+ + "\013tuning_spec\"\316\001\n\n"
+ "TunedModel\0226\n"
+ "\005model\030\001 \001(\tB\'\340A\003\372A!\n"
+ "\037aiplatform.googleapis.com/Model\022<\n"
+ "\010endpoint\030\002 \001(\tB*\340A\003\372A$\n"
- + "\"aiplatform.googleapis.com/Endpoint\"\362\002\n"
+ + "\"aiplatform.googleapis.com/Endpoint\022J\n"
+ + "\013checkpoints\030\003"
+ + " \003(\01320.google.cloud.aiplatform.v1.TunedModelCheckpointB\003\340A\003\"\362\002\n"
+ "#SupervisedTuningDatasetDistribution\022\020\n"
+ "\003sum\030\001 \001(\003B\003\340A\003\022\031\n"
+ "\014billable_sum\030\t \001(\003B\003\340A\003\022\020\n"
@@ -134,49 +140,54 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\006median\030\005 \001(\001B\003\340A\003\022\017\n"
+ "\002p5\030\006 \001(\001B\003\340A\003\022\020\n"
+ "\003p95\030\007 \001(\001B\003\340A\003\022c\n"
- + "\007buckets\030\010 \003(\0132M.google.cloud.aiplatform.v1"
- + ".SupervisedTuningDatasetDistribution.DatasetBucketB\003\340A\003\032J\n\r"
+ + "\007buckets\030\010 \003(\0132M.google.cloud.aiplatform.v1.Sup"
+ + "ervisedTuningDatasetDistribution.DatasetBucketB\003\340A\003\032J\n\r"
+ "DatasetBucket\022\022\n"
+ "\005count\030\001 \001(\001B\003\340A\003\022\021\n"
+ "\004left\030\002 \001(\001B\003\340A\003\022\022\n"
- + "\005right\030\003 \001(\001B\003\340A\003\"\314\005\n"
+ + "\005right\030\003 \001(\001B\003\340A\003\"\374\005\n"
+ "\031SupervisedTuningDataStats\022)\n"
+ "\034tuning_dataset_example_count\030\001 \001(\003B\003\340A\003\022)\n"
+ "\034total_tuning_character_count\030\002 \001(\003B\003\340A\003\022-\n"
+ "\036total_billable_character_count\030\003 \001(\003B\005\030\001\340A\003\022\'\n"
+ "\032total_billable_token_count\030\t \001(\003B\003\340A\003\022\036\n"
+ "\021tuning_step_count\030\004 \001(\003B\003\340A\003\022k\n"
- + "\035user_input_token_distribution\030\005"
- + " \001(\0132?.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\003\340A\003\022l\n"
- + "\036user_output_token_distribution\030\006 \001(\0132?.g"
- + "oogle.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\003\340A\003\022s\n"
- + "%user_message_per_example_distribution\030\007 \001(\0132?.goog"
- + "le.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\003\340A\003\022G\n"
+ + "\035user_input_token_distribution\030\005 \001"
+ + "(\0132?.google.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\003\340A\003\022l\n"
+ + "\036user_output_token_distribution\030\006 \001(\0132?.googl"
+ + "e.cloud.aiplatform.v1.SupervisedTuningDatasetDistributionB\003\340A\003\022s\n"
+ + "%user_message_per_example_distribution\030\007 \001(\0132?.google.c"
+ + "loud.aiplatform.v1.SupervisedTuningDatasetDistributionB\003\340A\003\022G\n"
+ "\025user_dataset_examples\030\010"
- + " \003(\0132#.google.cloud.aiplatform.v1.ContentB\003\340A\003\022%\n"
+ + " \003(\0132#.google.cloud.aiplatform.v1.ContentB\003\340A\003\022*\n"
+ "\035total_truncated_example_count\030\n"
- + " \001(\003\022!\n"
- + "\031truncated_example_indices\030\013 \003(\003\"\205\001\n"
+ + " \001(\003B\003\340A\003\022&\n"
+ + "\031truncated_example_indices\030\013 \003(\003B\003\340A\003\022$\n"
+ + "\027dropped_example_reasons\030\014 \003(\tB\003\340A\003\"\205\001\n"
+ "\017TuningDataStats\022]\n"
- + "\034supervised_tuning_data_stats\030\001 \001(\01325.google.cl"
- + "oud.aiplatform.v1.SupervisedTuningDataStatsH\000B\023\n"
- + "\021tuning_data_stats\"\307\002\n"
+ + "\034supervised_tuning_data_stats\030\001 \001(\01325.googl"
+ + "e.cloud.aiplatform.v1.SupervisedTuningDataStatsH\000B\023\n"
+ + "\021tuning_data_stats\"\372\002\n"
+ "\031SupervisedHyperParameters\022\030\n"
+ "\013epoch_count\030\001 \001(\003B\003\340A\001\022%\n"
+ "\030learning_rate_multiplier\030\002 \001(\001B\003\340A\001\022\\\n"
- + "\014adapter_size\030\003 \001(\0162A.google.cloud.a"
- + "iplatform.v1.SupervisedHyperParameters.AdapterSizeB\003\340A\001\"\212\001\n"
+ + "\014adapter_size\030\003 \001(\0162A.google.clo"
+ + "ud.aiplatform.v1.SupervisedHyperParameters.AdapterSizeB\003\340A\001\"\275\001\n"
+ "\013AdapterSize\022\034\n"
+ "\030ADAPTER_SIZE_UNSPECIFIED\020\000\022\024\n"
- + "\020ADAPTER_SIZE_ONE\020\001\022\025\n"
+ + "\020ADAPTER_SIZE_ONE\020\001\022\024\n"
+ + "\020ADAPTER_SIZE_TWO\020\006\022\025\n"
+ "\021ADAPTER_SIZE_FOUR\020\002\022\026\n"
+ "\022ADAPTER_SIZE_EIGHT\020\003\022\030\n"
- + "\024ADAPTER_SIZE_SIXTEEN\020\004\"\264\001\n"
+ + "\024ADAPTER_SIZE_SIXTEEN\020\004\022\033\n"
+ + "\027ADAPTER_SIZE_THIRTY_TWO\020\005\"\336\001\n"
+ "\024SupervisedTuningSpec\022!\n"
+ "\024training_dataset_uri\030\001 \001(\tB\003\340A\002\022#\n"
+ "\026validation_dataset_uri\030\002 \001(\tB\003\340A\001\022T\n"
- + "\020hyper_parameters\030\003 \001(\0132"
- + "5.google.cloud.aiplatform.v1.SupervisedHyperParametersB\003\340A\001\"\343\001\n\r"
+ + "\020hyper_parameters\030\003"
+ + " \001(\01325.google.cloud.aiplatform.v1.SupervisedHyperParametersB\003\340A\001\022(\n"
+ + "\033export_last_checkpoint_only\030\006 \001(\010B\003\340A\001\"\343\001\n"
+ + "\r"
+ "TunedModelRef\022;\n"
+ "\013tuned_model\030\001 \001(\tB$\372A!\n"
+ "\037aiplatform.googleapis.com/ModelH\000\022>\n\n"
@@ -184,11 +195,16 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "#aiplatform.googleapis.com/TuningJobH\000\022B\n"
+ "\014pipeline_job\030\003 \001(\tB*\372A\'\n"
+ "%aiplatform.googleapis.com/PipelineJobH\000B\021\n"
- + "\017tuned_model_refB\314\001\n"
- + "\036com.google.cloud.aiplatform.v1B\016TuningJobProtoP\001Z>cloud.google.com"
- + "/go/aiplatform/apiv1/aiplatformpb;aiplat"
- + "formpb\252\002\032Google.Cloud.AIPlatform.V1\312\002\032Go"
- + "ogle\\Cloud\\AIPlatform\\V1\352\002\035Google::Cloud::AIPlatform::V1b\006proto3"
+ + "\017tuned_model_ref\"\\\n"
+ + "\024TunedModelCheckpoint\022\025\n\r"
+ + "checkpoint_id\030\001 \001(\t\022\r\n"
+ + "\005epoch\030\002 \001(\003\022\014\n"
+ + "\004step\030\003 \001(\003\022\020\n"
+ + "\010endpoint\030\004 \001(\tB\314\001\n"
+ + "\036com.google.cloud.aiplatform.v1B\016TuningJobProtoP\001Z>cloud.google.com/go/aipla"
+ + "tform/apiv1/aiplatformpb;aiplatformpb\252\002\032"
+ + "Google.Cloud.AIPlatform.V1\312\002\032Google\\Clou"
+ + "d\\AIPlatform\\V1\352\002\035Google::Cloud::AIPlatform::V1b\006proto3"
};
descriptor =
com.google.protobuf.Descriptors.FileDescriptor.internalBuildGeneratedFileFrom(
@@ -242,7 +258,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
internal_static_google_cloud_aiplatform_v1_TunedModel_descriptor,
new java.lang.String[] {
- "Model", "Endpoint",
+ "Model", "Endpoint", "Checkpoints",
});
internal_static_google_cloud_aiplatform_v1_SupervisedTuningDatasetDistribution_descriptor =
getDescriptor().getMessageTypes().get(2);
@@ -279,6 +295,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
"UserDatasetExamples",
"TotalTruncatedExampleCount",
"TruncatedExampleIndices",
+ "DroppedExampleReasons",
});
internal_static_google_cloud_aiplatform_v1_TuningDataStats_descriptor =
getDescriptor().getMessageTypes().get(4);
@@ -302,7 +319,10 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
internal_static_google_cloud_aiplatform_v1_SupervisedTuningSpec_descriptor,
new java.lang.String[] {
- "TrainingDatasetUri", "ValidationDatasetUri", "HyperParameters",
+ "TrainingDatasetUri",
+ "ValidationDatasetUri",
+ "HyperParameters",
+ "ExportLastCheckpointOnly",
});
internal_static_google_cloud_aiplatform_v1_TunedModelRef_descriptor =
getDescriptor().getMessageTypes().get(7);
@@ -312,6 +332,14 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
new java.lang.String[] {
"TunedModel", "TuningJob", "PipelineJob", "TunedModelRef",
});
+ internal_static_google_cloud_aiplatform_v1_TunedModelCheckpoint_descriptor =
+ getDescriptor().getMessageTypes().get(8);
+ internal_static_google_cloud_aiplatform_v1_TunedModelCheckpoint_fieldAccessorTable =
+ new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
+ internal_static_google_cloud_aiplatform_v1_TunedModelCheckpoint_descriptor,
+ new java.lang.String[] {
+ "CheckpointId", "Epoch", "Step", "Endpoint",
+ });
com.google.protobuf.ExtensionRegistry registry =
com.google.protobuf.ExtensionRegistry.newInstance();
registry.add(com.google.api.FieldBehaviorProto.fieldBehavior);
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/VertexRagDataProto.java b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/VertexRagDataProto.java
index 3dbe7808d0e8..7add795017e1 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/VertexRagDataProto.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/VertexRagDataProto.java
@@ -44,6 +44,14 @@ public static void registerAllExtensions(com.google.protobuf.ExtensionRegistry r
internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_descriptor;
static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_fieldAccessorTable;
+ static final com.google.protobuf.Descriptors.Descriptor
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_KNN_descriptor;
+ static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_KNN_fieldAccessorTable;
+ static final com.google.protobuf.Descriptors.Descriptor
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_ANN_descriptor;
+ static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_ANN_fieldAccessorTable;
static final com.google.protobuf.Descriptors.Descriptor
internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_Pinecone_descriptor;
static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
@@ -136,7 +144,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\005model\030\002 \001(\tB\'\340A\003\372A!\n"
+ "\037aiplatform.googleapis.com/Model\022\035\n"
+ "\020model_version_id\030\003 \001(\tB\003\340A\003B\016\n"
- + "\014model_config\"\251\004\n"
+ + "\014model_config\"\224\006\n"
+ "\021RagVectorDbConfig\022T\n"
+ "\016rag_managed_db\030\001 \001(\0132:.go"
+ "ogle.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDbH\000\022J\n"
@@ -146,8 +154,17 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\001(\0132@.google.cloud.aiplatform.v1.RagVectorDbConfig.VertexVectorSearchH\000\0225\n"
+ "\010api_auth\030\005 \001(\0132#.google.cloud.aiplatform.v1.ApiAuth\022_\n"
+ "\032rag_embedding_model_config\030\007 \001"
- + "(\01323.google.cloud.aiplatform.v1.RagEmbeddingModelConfigB\006\340A\001\340A\005\032\016\n"
- + "\014RagManagedDb\032\036\n"
+ + "(\01323.google.cloud.aiplatform.v1.RagEmbeddingModelConfigB\006\340A\001\340A\005\032\370\001\n"
+ + "\014RagManagedDb\022M\n"
+ + "\003knn\030\001 \001(\0132>.google.cloud.aiplatform."
+ + "v1.RagVectorDbConfig.RagManagedDb.KNNH\000\022M\n"
+ + "\003ann\030\002 \001(\0132>.google.cloud.aiplatform.v"
+ + "1.RagVectorDbConfig.RagManagedDb.ANNH\000\032\005\n"
+ + "\003KNN\032-\n"
+ + "\003ANN\022\022\n\n"
+ + "tree_depth\030\001 \001(\005\022\022\n\n"
+ + "leaf_count\030\002 \001(\005B\024\n"
+ + "\022retrieval_strategy\032\036\n"
+ "\010Pinecone\022\022\n\n"
+ "index_name\030\001 \001(\t\032;\n"
+ "\022VertexVectorSearch\022\026\n"
@@ -165,8 +182,8 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\024VertexAiSearchConfig\022\026\n"
+ "\016serving_config\030\001 \001(\t\"\253\001\n"
+ "\014CorpusStatus\022B\n"
- + "\005state\030\001 \001(\0162..google.cl"
- + "oud.aiplatform.v1.CorpusStatus.StateB\003\340A\003\022\031\n"
+ + "\005state\030\001"
+ + " \001(\0162..google.cloud.aiplatform.v1.CorpusStatus.StateB\003\340A\003\022\031\n"
+ "\014error_status\030\002 \001(\tB\003\340A\003\"<\n"
+ "\005State\022\013\n"
+ "\007UNKNOWN\020\000\022\017\n"
@@ -177,32 +194,31 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\020vector_db_config\030\t"
+ " \001(\0132-.google.cloud.aiplatform.v1.RagVectorDbConfigB\006\340A\001\340A\005H\000\022[\n"
+ "\027vertex_ai_search_config\030\n"
- + " \001(\01320.google.cloud.aipl"
- + "atform.v1.VertexAiSearchConfigB\006\340A\001\340A\005H\000\022\021\n"
+ + " \001(\01320.google.cloud.aiplatform.v1.VertexAiSearchConfigB\006\340A\001\340A\005H\000\022\021\n"
+ "\004name\030\001 \001(\tB\003\340A\003\022\031\n"
+ "\014display_name\030\002 \001(\tB\003\340A\002\022\030\n"
+ "\013description\030\003 \001(\tB\003\340A\001\0224\n"
+ "\013create_time\030\004 \001(\0132\032.google.protobuf.TimestampB\003\340A\003\0224\n"
+ "\013update_time\030\005"
+ " \001(\0132\032.google.protobuf.TimestampB\003\340A\003\022D\n\r"
- + "corpus_status\030\010 "
- + "\001(\0132(.google.cloud.aiplatform.v1.CorpusStatusB\003\340A\003:\200\001\352A}\n"
- + "#aiplatform.googleapis."
- + "com/RagCorpus\022?projects/{project}/locations/{location}/ragCorpora/{rag_corpus}*\n"
+ + "corpus_status\030\010 \001(\0132("
+ + ".google.cloud.aiplatform.v1.CorpusStatusB\003\340A\003:\200\001\352A}\n"
+ + "#aiplatform.googleapis.com/R"
+ + "agCorpus\022?projects/{project}/locations/{location}/ragCorpora/{rag_corpus}*\n"
+ "ragCorpora2\tragCorpusB\020\n"
+ "\016backend_config\"\334\006\n"
+ "\007RagFile\022@\n\n"
+ "gcs_source\030\010"
+ " \001(\0132%.google.cloud.aiplatform.v1.GcsSourceB\003\340A\003H\000\022Q\n"
- + "\023google_drive_source\030\t \001(\0132-.google.clou"
- + "d.aiplatform.v1.GoogleDriveSourceB\003\340A\003H\000\022S\n"
+ + "\023google_drive_source\030\t"
+ + " \001(\0132-.google.cloud.aiplatform.v1.GoogleDriveSourceB\003\340A\003H\000\022S\n"
+ "\024direct_upload_source\030\n"
+ " \001(\0132..google.cloud.aiplatform.v1.DirectUploadSourceB\003\340A\003H\000\022?\n"
+ "\014slack_source\030\013"
+ " \001(\0132\'.google.cloud.aiplatform.v1.SlackSourceH\000\022=\n"
+ "\013jira_source\030\014 \001(\0132&.google.cloud.aiplatform.v1.JiraSourceH\000\022L\n"
- + "\023share_point_sources\030\016 \001"
- + "(\0132-.google.cloud.aiplatform.v1.SharePointSourcesH\000\022\021\n"
+ + "\023share_point_sources\030\016 \001(\0132-."
+ + "google.cloud.aiplatform.v1.SharePointSourcesH\000\022\021\n"
+ "\004name\030\001 \001(\tB\003\340A\003\022\031\n"
+ "\014display_name\030\002 \001(\tB\003\340A\002\022\030\n"
+ "\013description\030\003 \001(\tB\003\340A\001\0224\n"
@@ -210,21 +226,21 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\013update_time\030\007 \001(\0132\032.google.protobuf.TimestampB\003\340A\003\022@\n"
+ "\013file_status\030\r"
+ " \001(\0132&.google.cloud.aiplatform.v1.FileStatusB\003\340A\003:\217\001\352A\213\001\n"
- + "!aiplatform.googleapis.com/RagFile\022Sprojects/{project}/"
- + "locations/{location}/ragCorpora/{rag_cor"
- + "pus}/ragFiles/{rag_file}*\010ragFiles2\007ragFileB\021\n"
+ + "!aiplatform.googleapis.com/RagFile\022Sprojects/{project}/locat"
+ + "ions/{location}/ragCorpora/{rag_corpus}/"
+ + "ragFiles/{rag_file}*\010ragFiles2\007ragFileB\021\n"
+ "\017rag_file_source\"\240\001\n"
+ "\010RagChunk\022\014\n"
+ "\004text\030\001 \001(\t\022E\n"
- + "\tpage_span\030\002 \001(\0132-.google.cl"
- + "oud.aiplatform.v1.RagChunk.PageSpanH\000\210\001\001\0321\n"
+ + "\tpage_span\030\002"
+ + " \001(\0132-.google.cloud.aiplatform.v1.RagChunk.PageSpanH\000\210\001\001\0321\n"
+ "\010PageSpan\022\022\n\n"
+ "first_page\030\001 \001(\005\022\021\n"
+ "\tlast_page\030\002 \001(\005B\014\n\n"
+ "_page_span\"\324\001\n"
+ "\025RagFileChunkingConfig\022f\n"
- + "\025fixed_length_chunking\030\003 \001"
- + "(\0132E.google.cloud.aiplatform.v1.RagFileChunkingConfig.FixedLengthChunkingH\000\032@\n"
+ + "\025fixed_length_chunking\030\003 \001(\0132E."
+ + "google.cloud.aiplatform.v1.RagFileChunkingConfig.FixedLengthChunkingH\000\032@\n"
+ "\023FixedLengthChunking\022\022\n\n"
+ "chunk_size\030\001 \001(\005\022\025\n\r"
+ "chunk_overlap\030\002 \001(\005B\021\n"
@@ -233,51 +249,53 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\030rag_file_chunking_config\030\001"
+ " \001(\01321.google.cloud.aiplatform.v1.RagFileChunkingConfig\"\376\002\n"
+ "\024RagFileParsingConfig\022V\n\r"
- + "layout_parser\030\004"
- + " \001(\0132=.google.cloud.aiplatform.v1.RagFileParsingConfig.LayoutParserH\000\022P\n\n"
+ + "layout_parser\030\004 \001(\0132"
+ + "=.google.cloud.aiplatform.v1.RagFileParsingConfig.LayoutParserH\000\022P\n\n"
+ "llm_parser\030\005"
+ " \001(\0132:.google.cloud.aiplatform.v1.RagFileParsingConfig.LlmParserH\000\032L\n"
+ "\014LayoutParser\022\026\n"
+ "\016processor_name\030\001 \001(\t\022$\n"
+ "\034max_parsing_requests_per_min\030\002 \001(\005\032d\n"
- + "\tLlmParser\022\022\n\n"
+ + "\tLlmParser\022\022\n"
+ + "\n"
+ "model_name\030\001 \001(\t\022$\n"
+ "\034max_parsing_requests_per_min\030\002 \001(\005\022\035\n"
+ "\025custom_parsing_prompt\030\003 \001(\tB\010\n"
+ "\006parser\"v\n"
+ "\023UploadRagFileConfig\022_\n"
- + "\036rag_file_transformation_config\030\003"
- + " \001(\01327.google.cloud.aiplatform.v1.RagFileTransformationConfig\"\354\007\n"
+ + "\036rag_file_transformation_config\030\003 \001(\0132"
+ + "7.google.cloud.aiplatform.v1.RagFileTransformationConfig\"\207\010\n"
+ "\024ImportRagFilesConfig\022;\n\n"
+ "gcs_source\030\002 \001(\0132%.google.cloud.aiplatform.v1.GcsSourceH\000\022L\n"
+ "\023google_drive_source\030\003"
+ " \001(\0132-.google.cloud.aiplatform.v1.GoogleDriveSourceH\000\022?\n"
- + "\014slack_source\030\006"
- + " \001(\0132\'.google.cloud.aiplatform.v1.SlackSourceH\000\022=\n"
+ + "\014slack_source\030\006 \001"
+ + "(\0132\'.google.cloud.aiplatform.v1.SlackSourceH\000\022=\n"
+ "\013jira_source\030\007 \001(\0132&.google.cloud.aiplatform.v1.JiraSourceH\000\022L\n"
+ "\023share_point_sources\030\r"
+ " \001(\0132-.google.cloud.aiplatform.v1.SharePointSourcesH\000\022R\n"
+ "\030partial_failure_gcs_sink\030\013"
+ " \001(\0132*.google.cloud.aiplatform.v1.GcsDestinationB\002\030\001H\001\022\\\n"
- + "\035partial_failure_bigquery_sink\030\014 \001(\0132/.goo"
- + "gle.cloud.aiplatform.v1.BigQueryDestinationB\002\030\001H\001\022L\n"
- + "\026import_result_gcs_sink\030\016 \001("
- + "\0132*.google.cloud.aiplatform.v1.GcsDestinationH\002\022V\n"
- + "\033import_result_bigquery_sink\030\017"
- + " \001(\0132/.google.cloud.aiplatform.v1.BigQueryDestinationH\002\022_\n"
+ + "\035partial_failure_bigquery_sink\030\014 \001(\0132/.google.c"
+ + "loud.aiplatform.v1.BigQueryDestinationB\002\030\001H\001\022L\n"
+ + "\026import_result_gcs_sink\030\016 \001(\0132*.g"
+ + "oogle.cloud.aiplatform.v1.GcsDestinationH\002\022V\n"
+ + "\033import_result_bigquery_sink\030\017 \001(\0132"
+ + "/.google.cloud.aiplatform.v1.BigQueryDestinationH\002\022_\n"
+ "\036rag_file_transformation_config\030\020"
+ " \001(\01327.google.cloud.aiplatform.v1.RagFileTransformationConfig\022V\n"
- + "\027rag_file_parsing_config\030\010 \001(\01320.google.cloud"
- + ".aiplatform.v1.RagFileParsingConfigB\003\340A\001\022+\n"
- + "\036max_embedding_requests_per_min\030\005 \001(\005B\003\340A\001B\017\n\r"
+ + "\027rag_file_parsing_config\030\010"
+ + " \001(\01320.google.cloud.aiplatform.v1.RagFileParsingConfigB\003\340A\001\022+\n"
+ + "\036max_embedding_requests_per_min\030\005 \001(\005B\003\340A\001\022\031\n"
+ + "\021rebuild_ann_index\030\023 \001(\010B\017\n\r"
+ "import_sourceB\026\n"
+ "\024partial_failure_sinkB\024\n"
+ "\022import_result_sinkB\320\001\n"
- + "\036com.google.cloud.aiplatform.v1B\022VertexRagDataPr"
- + "otoP\001Z>cloud.google.com/go/aiplatform/ap"
- + "iv1/aiplatformpb;aiplatformpb\252\002\032Google.C"
- + "loud.AIPlatform.V1\312\002\032Google\\Cloud\\AIPlat"
- + "form\\V1\352\002\035Google::Cloud::AIPlatform::V1b\006proto3"
+ + "\036com.google.cloud.aiplatform.v1B\022VertexRagDataProtoP\001Z>cloud.googl"
+ + "e.com/go/aiplatform/apiv1/aiplatformpb;a"
+ + "iplatformpb\252\002\032Google.Cloud.AIPlatform.V1"
+ + "\312\002\032Google\\Cloud\\AIPlatform\\V1\352\002\035Google::"
+ + "Cloud::AIPlatform::V1b\006proto3"
};
descriptor =
com.google.protobuf.Descriptors.FileDescriptor.internalBuildGeneratedFileFrom(
@@ -327,7 +345,27 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_fieldAccessorTable =
new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_descriptor,
+ new java.lang.String[] {
+ "Knn", "Ann", "RetrievalStrategy",
+ });
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_KNN_descriptor =
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_descriptor
+ .getNestedTypes()
+ .get(0);
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_KNN_fieldAccessorTable =
+ new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_KNN_descriptor,
new java.lang.String[] {});
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_ANN_descriptor =
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_descriptor
+ .getNestedTypes()
+ .get(1);
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_ANN_fieldAccessorTable =
+ new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
+ internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_RagManagedDb_ANN_descriptor,
+ new java.lang.String[] {
+ "TreeDepth", "LeafCount",
+ });
internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_Pinecone_descriptor =
internal_static_google_cloud_aiplatform_v1_RagVectorDbConfig_descriptor
.getNestedTypes()
@@ -504,6 +542,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
"RagFileTransformationConfig",
"RagFileParsingConfig",
"MaxEmbeddingRequestsPerMin",
+ "RebuildAnnIndex",
"ImportSource",
"PartialFailureSink",
"ImportResultSink",
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/accelerator_type.proto b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/accelerator_type.proto
index 30a5aa734328..14cf7fa90736 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/accelerator_type.proto
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/accelerator_type.proto
@@ -24,6 +24,7 @@ option java_package = "com.google.cloud.aiplatform.v1";
option php_namespace = "Google\\Cloud\\AIPlatform\\V1";
option ruby_package = "Google::Cloud::AIPlatform::V1";
+// LINT: LEGACY_NAMES
// Represents a hardware accelerator type.
enum AcceleratorType {
// Unspecified accelerator type, which means no accelerator.
@@ -60,6 +61,12 @@ enum AcceleratorType {
// Nvidia H100 Mega 80Gb GPU.
NVIDIA_H100_MEGA_80GB = 14;
+ // Nvidia H200 141Gb GPU.
+ NVIDIA_H200_141GB = 15;
+
+ // Nvidia B200 GPU.
+ NVIDIA_B200 = 16;
+
// TPU v2.
TPU_V2 = 6;
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/openapi.proto b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/openapi.proto
index 6fc973c7ad19..f952b3348df7 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/openapi.proto
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/openapi.proto
@@ -142,6 +142,11 @@ message Schema {
// subschemas in the list.
repeated Schema any_of = 11 [(google.api.field_behavior) = OPTIONAL];
+ // Optional. Can either be a boolean or an object; controls the presence of
+ // additional properties.
+ google.protobuf.Value additional_properties = 26
+ [(google.api.field_behavior) = OPTIONAL];
+
// Optional. Allows indirect references between schema nodes. The value should
// be a valid reference to a child of the root `defs`.
//
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/tuning_job.proto b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/tuning_job.proto
index 677debb54f22..dc773999069a 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/tuning_job.proto
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/tuning_job.proto
@@ -42,7 +42,8 @@ message TuningJob {
};
oneof source_model {
- // The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ // The base model that is being tuned. See [Supported
+ // models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
string base_model = 4;
}
@@ -118,7 +119,7 @@ message TuningJob {
}
];
- // Output only. The tuned model resources assiociated with this
+ // Output only. The tuned model resources associated with this
// [TuningJob][google.cloud.aiplatform.v1.TuningJob].
TunedModel tuned_model = 14 [(google.api.field_behavior) = OUTPUT_ONLY];
@@ -142,7 +143,7 @@ message TuningJob {
string service_account = 22;
}
-// The Model Registry Model and Online Prediction Endpoint assiociated with
+// The Model Registry Model and Online Prediction Endpoint associated with
// this [TuningJob][google.cloud.aiplatform.v1.TuningJob].
message TunedModel {
// Output only. The resource name of the TunedModel. Format:
@@ -162,6 +163,12 @@ message TunedModel {
type: "aiplatform.googleapis.com/Endpoint"
}
];
+
+ // Output only. The checkpoints associated with this TunedModel.
+ // This field is only populated for tuning jobs that enable intermediate
+ // checkpoints.
+ repeated TunedModelCheckpoint checkpoints = 3
+ [(google.api.field_behavior) = OUTPUT_ONLY];
}
// Dataset distribution for Supervised Tuning.
@@ -245,13 +252,21 @@ message SupervisedTuningDataStats {
repeated Content user_dataset_examples = 8
[(google.api.field_behavior) = OUTPUT_ONLY];
- // The number of examples in the dataset that have been truncated by any
- // amount.
- int64 total_truncated_example_count = 10;
+ // Output only. The number of examples in the dataset that have been dropped.
+ // An example can be dropped for reasons including: too many tokens, contains
+ // an invalid image, contains too many images, etc.
+ int64 total_truncated_example_count = 10
+ [(google.api.field_behavior) = OUTPUT_ONLY];
+
+ // Output only. A partial sample of the indices (starting from 1) of the
+ // dropped examples.
+ repeated int64 truncated_example_indices = 11
+ [(google.api.field_behavior) = OUTPUT_ONLY];
- // A partial sample of the indices (starting from 1) of the truncated
- // examples.
- repeated int64 truncated_example_indices = 11;
+ // Output only. For each index in `truncated_example_indices`, the user-facing
+ // reason why the example was dropped.
+ repeated string dropped_example_reasons = 12
+ [(google.api.field_behavior) = OUTPUT_ONLY];
}
// The tuning data statistic values for
@@ -273,6 +288,9 @@ message SupervisedHyperParameters {
// Adapter size 1.
ADAPTER_SIZE_ONE = 1;
+ // Adapter size 2.
+ ADAPTER_SIZE_TWO = 6;
+
// Adapter size 4.
ADAPTER_SIZE_FOUR = 2;
@@ -281,6 +299,9 @@ message SupervisedHyperParameters {
// Adapter size 16.
ADAPTER_SIZE_SIXTEEN = 4;
+
+ // Adapter size 32.
+ ADAPTER_SIZE_THIRTY_TWO = 5;
}
// Optional. Number of complete passes the model makes over the entire
@@ -307,6 +328,11 @@ message SupervisedTuningSpec {
// Optional. Hyperparameters for SFT.
SupervisedHyperParameters hyper_parameters = 3
[(google.api.field_behavior) = OPTIONAL];
+
+ // Optional. If set to true, disable intermediate checkpoints for SFT and only
+ // the last checkpoint will be exported. Otherwise, enable intermediate
+ // checkpoints for SFT. Default is false.
+ bool export_last_checkpoint_only = 6 [(google.api.field_behavior) = OPTIONAL];
}
// TunedModel Reference for legacy model migration.
@@ -331,3 +357,19 @@ message TunedModelRef {
}];
}
}
+
+// TunedModelCheckpoint for the Tuned Model of a Tuning Job.
+message TunedModelCheckpoint {
+ // The ID of the checkpoint.
+ string checkpoint_id = 1;
+
+ // The epoch of the checkpoint.
+ int64 epoch = 2;
+
+ // The step of the checkpoint.
+ int64 step = 3;
+
+ // The Endpoint resource name that the checkpoint is deployed to. Format:
+ // `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ string endpoint = 4;
+}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/vertex_rag_data.proto b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/vertex_rag_data.proto
index e4720667ae59..30410294c21c 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/vertex_rag_data.proto
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/vertex_rag_data.proto
@@ -76,7 +76,47 @@ message RagEmbeddingModelConfig {
// Config for the Vector DB to use for RAG.
message RagVectorDbConfig {
// The config for the default RAG-managed Vector DB.
- message RagManagedDb {}
+ message RagManagedDb {
+ // Config for KNN search.
+ message KNN {}
+
+ // Config for ANN search.
+ //
+ // RagManagedDb uses a tree-based structure to partition data and
+ // facilitate faster searches. As a tradeoff, it requires longer indexing
+ // time and manual triggering of index rebuild via the ImportRagFiles and
+ // UpdateRagCorpus API.
+ message ANN {
+ // The depth of the tree-based structure. Only depth values of 2 and 3 are
+ // supported.
+ //
+ // Recommended value is 2 if you have if you have O(10K) files in the
+ // RagCorpus and set this to 3 if more than that.
+ //
+ // Default value is 2.
+ int32 tree_depth = 1;
+
+ // Number of leaf nodes in the tree-based structure. Each leaf node
+ // contains groups of closely related vectors along with their
+ // corresponding centroid.
+ //
+ // Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus).
+ //
+ // Default value is 500.
+ int32 leaf_count = 2;
+ }
+
+ // Choice of retrieval strategy.
+ oneof retrieval_strategy {
+ // Performs a KNN search on RagCorpus.
+ // Default choice if not specified.
+ KNN knn = 1;
+
+ // Performs an ANN search on RagCorpus. Use this if you have a lot of
+ // files (> 10K) in your RagCorpus and want to reduce the search latency.
+ ANN ann = 2;
+ }
+ }
// The config for the Pinecone.
message Pinecone {
@@ -453,4 +493,13 @@ message ImportRagFilesConfig {
// If unspecified, a default value of 1,000 QPM would be used.
int32 max_embedding_requests_per_min = 5
[(google.api.field_behavior) = OPTIONAL];
+
+ // Rebuilds the ANN index to optimize for recall on the imported data.
+ // Only applicable for RagCorpora running on RagManagedDb with
+ // `retrieval_strategy` set to `ANN`. The rebuild will be performed using the
+ // existing ANN config set on the RagCorpus. To change the ANN config, please
+ // use the UpdateRagCorpus API.
+ //
+ // Default is false, i.e., index is not rebuilt.
+ bool rebuild_ann_index = 19;
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorType.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorType.java
index e8e347fbdfa9..44b303006835 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorType.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorType.java
@@ -23,6 +23,7 @@
*
*
* + * LINT: LEGACY_NAMES * Represents a hardware accelerator type. ** @@ -141,6 +142,26 @@ public enum AcceleratorType implements com.google.protobuf.ProtocolMessageEnum { *
NVIDIA_H100_MEGA_80GB = 14;
*/
NVIDIA_H100_MEGA_80GB(14),
+ /**
+ *
+ *
+ * + * Nvidia H200 141Gb GPU. + *+ * + *
NVIDIA_H200_141GB = 15;
+ */
+ NVIDIA_H200_141GB(15),
+ /**
+ *
+ *
+ * + * Nvidia B200 GPU. + *+ * + *
NVIDIA_B200 = 16;
+ */
+ NVIDIA_B200(16),
/**
*
*
@@ -306,6 +327,28 @@ public enum AcceleratorType implements com.google.protobuf.ProtocolMessageEnum {
*/
public static final int NVIDIA_H100_MEGA_80GB_VALUE = 14;
+ /**
+ *
+ *
+ * + * Nvidia H200 141Gb GPU. + *+ * + *
NVIDIA_H200_141GB = 15;
+ */
+ public static final int NVIDIA_H200_141GB_VALUE = 15;
+
+ /**
+ *
+ *
+ * + * Nvidia B200 GPU. + *+ * + *
NVIDIA_B200 = 16;
+ */
+ public static final int NVIDIA_B200_VALUE = 16;
+
/**
*
*
@@ -396,6 +439,10 @@ public static AcceleratorType forNumber(int value) {
return NVIDIA_H100_80GB;
case 14:
return NVIDIA_H100_MEGA_80GB;
+ case 15:
+ return NVIDIA_H200_141GB;
+ case 16:
+ return NVIDIA_B200;
case 6:
return TPU_V2;
case 7:
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorTypeProto.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorTypeProto.java
index 94dfbd96cb50..6a168e042f87 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorTypeProto.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/AcceleratorTypeProto.java
@@ -38,14 +38,15 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
java.lang.String[] descriptorData = {
"\n6google/cloud/aiplatform/v1beta1/accele"
+ "rator_type.proto\022\037google.cloud.aiplatfor"
- + "m.v1beta1*\316\002\n\017AcceleratorType\022 \n\034ACCELER"
+ + "m.v1beta1*\366\002\n\017AcceleratorType\022 \n\034ACCELER"
+ "ATOR_TYPE_UNSPECIFIED\020\000\022\030\n\020NVIDIA_TESLA_"
+ "K80\020\001\032\002\010\001\022\025\n\021NVIDIA_TESLA_P100\020\002\022\025\n\021NVID"
+ "IA_TESLA_V100\020\003\022\023\n\017NVIDIA_TESLA_P4\020\004\022\023\n\017"
+ "NVIDIA_TESLA_T4\020\005\022\025\n\021NVIDIA_TESLA_A100\020\010"
+ "\022\024\n\020NVIDIA_A100_80GB\020\t\022\r\n\tNVIDIA_L4\020\013\022\024\n"
+ "\020NVIDIA_H100_80GB\020\r\022\031\n\025NVIDIA_H100_MEGA_"
- + "80GB\020\016\022\n\n\006TPU_V2\020\006\022\n\n\006TPU_V3\020\007\022\016\n\nTPU_V4"
+ + "80GB\020\016\022\025\n\021NVIDIA_H200_141GB\020\017\022\017\n\013NVIDIA_"
+ + "B200\020\020\022\n\n\006TPU_V2\020\006\022\n\n\006TPU_V3\020\007\022\016\n\nTPU_V4"
+ "_POD\020\n\022\022\n\016TPU_V5_LITEPOD\020\014B\353\001\n#com.googl"
+ "e.cloud.aiplatform.v1beta1B\024AcceleratorT"
+ "ypeProtoP\001ZCcloud.google.com/go/aiplatfo"
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/DistillationSpec.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/DistillationSpec.java
index cbd8fc75d286..8191d729b388 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/DistillationSpec.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/DistillationSpec.java
@@ -124,8 +124,8 @@ public TeacherModelCase getTeacherModelCase() {
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -140,8 +140,8 @@ public boolean hasBaseTeacherModel() {
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -169,8 +169,8 @@ public java.lang.String getBaseTeacherModel() {
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -1130,8 +1130,8 @@ public Builder clearTeacherModel() {
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -1147,8 +1147,8 @@ public boolean hasBaseTeacherModel() {
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -1177,8 +1177,8 @@ public java.lang.String getBaseTeacherModel() {
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -1207,8 +1207,8 @@ public com.google.protobuf.ByteString getBaseTeacherModelBytes() {
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -1230,8 +1230,8 @@ public Builder setBaseTeacherModel(java.lang.String value) {
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -1251,8 +1251,8 @@ public Builder clearBaseTeacherModel() {
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/DistillationSpecOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/DistillationSpecOrBuilder.java
index 076c06ff2ab4..1f66aeeaece8 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/DistillationSpecOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/DistillationSpecOrBuilder.java
@@ -28,8 +28,8 @@ public interface DistillationSpecOrBuilder
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -42,8 +42,8 @@ public interface DistillationSpecOrBuilder
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
@@ -56,8 +56,8 @@ public interface DistillationSpecOrBuilder
*
*
*
- * The base teacher model that is being distilled, e.g.,
- * "gemini-1.0-pro-002".
+ * The base teacher model that is being distilled. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_teacher_model = 5;
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/ImportRagFilesConfig.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/ImportRagFilesConfig.java
index c03bda5c41cf..c8200dbf4234 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/ImportRagFilesConfig.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/ImportRagFilesConfig.java
@@ -523,7 +523,7 @@ public com.google.cloud.aiplatform.v1beta1.SharePointSources getSharePointSource
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=555
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=605
* @return Whether the partialFailureGcsSink field is set.
*/
@java.lang.Override
@@ -545,7 +545,7 @@ public boolean hasPartialFailureGcsSink() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=555
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=605
* @return The partialFailureGcsSink.
*/
@java.lang.Override
@@ -599,7 +599,7 @@ public com.google.cloud.aiplatform.v1beta1.GcsDestination getPartialFailureGcsSi
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_bigquery_sink
- * is deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
+ * is deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=614
* @return Whether the partialFailureBigquerySink field is set.
*/
@java.lang.Override
@@ -626,7 +626,7 @@ public boolean hasPartialFailureBigquerySink() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_bigquery_sink
- * is deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
+ * is deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=614
* @return The partialFailureBigquerySink.
*/
@java.lang.Override
@@ -808,7 +808,7 @@ public com.google.cloud.aiplatform.v1beta1.BigQueryDestination getImportResultBi
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=583
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=633
* @return Whether the ragFileChunkingConfig field is set.
*/
@java.lang.Override
@@ -829,7 +829,7 @@ public boolean hasRagFileChunkingConfig() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=583
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=633
* @return The ragFileChunkingConfig.
*/
@java.lang.Override
@@ -1035,6 +1035,31 @@ public int getGlobalMaxEmbeddingRequestsPerMin() {
return globalMaxEmbeddingRequestsPerMin_;
}
+ public static final int REBUILD_ANN_INDEX_FIELD_NUMBER = 19;
+ private boolean rebuildAnnIndex_ = false;
+
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @return The rebuildAnnIndex.
+ */
+ @java.lang.Override
+ public boolean getRebuildAnnIndex() {
+ return rebuildAnnIndex_;
+ }
+
private byte memoizedIsInitialized = -1;
@java.lang.Override
@@ -1096,6 +1121,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
if (globalMaxEmbeddingRequestsPerMin_ != 0) {
output.writeInt32(18, globalMaxEmbeddingRequestsPerMin_);
}
+ if (rebuildAnnIndex_ != false) {
+ output.writeBool(19, rebuildAnnIndex_);
+ }
getUnknownFields().writeTo(output);
}
@@ -1172,6 +1200,9 @@ public int getSerializedSize() {
com.google.protobuf.CodedOutputStream.computeInt32Size(
18, globalMaxEmbeddingRequestsPerMin_);
}
+ if (rebuildAnnIndex_ != false) {
+ size += com.google.protobuf.CodedOutputStream.computeBoolSize(19, rebuildAnnIndex_);
+ }
size += getUnknownFields().getSerializedSize();
memoizedSize = size;
return size;
@@ -1204,6 +1235,7 @@ public boolean equals(final java.lang.Object obj) {
if (getMaxEmbeddingRequestsPerMin() != other.getMaxEmbeddingRequestsPerMin()) return false;
if (getGlobalMaxEmbeddingRequestsPerMin() != other.getGlobalMaxEmbeddingRequestsPerMin())
return false;
+ if (getRebuildAnnIndex() != other.getRebuildAnnIndex()) return false;
if (!getImportSourceCase().equals(other.getImportSourceCase())) return false;
switch (importSourceCase_) {
case 2:
@@ -1275,6 +1307,8 @@ public int hashCode() {
hash = (53 * hash) + getMaxEmbeddingRequestsPerMin();
hash = (37 * hash) + GLOBAL_MAX_EMBEDDING_REQUESTS_PER_MIN_FIELD_NUMBER;
hash = (53 * hash) + getGlobalMaxEmbeddingRequestsPerMin();
+ hash = (37 * hash) + REBUILD_ANN_INDEX_FIELD_NUMBER;
+ hash = (53 * hash) + com.google.protobuf.Internal.hashBoolean(getRebuildAnnIndex());
switch (importSourceCase_) {
case 2:
hash = (37 * hash) + GCS_SOURCE_FIELD_NUMBER;
@@ -1519,6 +1553,7 @@ public Builder clear() {
}
maxEmbeddingRequestsPerMin_ = 0;
globalMaxEmbeddingRequestsPerMin_ = 0;
+ rebuildAnnIndex_ = false;
importSourceCase_ = 0;
importSource_ = null;
partialFailureSinkCase_ = 0;
@@ -1590,6 +1625,9 @@ private void buildPartial0(com.google.cloud.aiplatform.v1beta1.ImportRagFilesCon
if (((from_bitField0_ & 0x00002000) != 0)) {
result.globalMaxEmbeddingRequestsPerMin_ = globalMaxEmbeddingRequestsPerMin_;
}
+ if (((from_bitField0_ & 0x00004000) != 0)) {
+ result.rebuildAnnIndex_ = rebuildAnnIndex_;
+ }
result.bitField0_ |= to_bitField0_;
}
@@ -1691,6 +1729,9 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1beta1.ImportRagFilesConfi
if (other.getGlobalMaxEmbeddingRequestsPerMin() != 0) {
setGlobalMaxEmbeddingRequestsPerMin(other.getGlobalMaxEmbeddingRequestsPerMin());
}
+ if (other.getRebuildAnnIndex() != false) {
+ setRebuildAnnIndex(other.getRebuildAnnIndex());
+ }
switch (other.getImportSourceCase()) {
case GCS_SOURCE:
{
@@ -1873,6 +1914,12 @@ public Builder mergeFrom(
bitField0_ |= 0x00002000;
break;
} // case 144
+ case 152:
+ {
+ rebuildAnnIndex_ = input.readBool();
+ bitField0_ |= 0x00004000;
+ break;
+ } // case 152
default:
{
if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -3094,7 +3141,7 @@ public Builder clearSharePointSources() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=555
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=605
* @return Whether the partialFailureGcsSink field is set.
*/
@java.lang.Override
@@ -3116,7 +3163,7 @@ public boolean hasPartialFailureGcsSink() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=555
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=605
* @return The partialFailureGcsSink.
*/
@java.lang.Override
@@ -3363,7 +3410,7 @@ public Builder clearPartialFailureGcsSink() {
*
* @deprecated
* google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_bigquery_sink is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=614
* @return Whether the partialFailureBigquerySink field is set.
*/
@java.lang.Override
@@ -3391,7 +3438,7 @@ public boolean hasPartialFailureBigquerySink() {
*
* @deprecated
* google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_bigquery_sink is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=614
* @return The partialFailureBigquerySink.
*/
@java.lang.Override
@@ -4169,7 +4216,7 @@ public Builder clearImportResultBigquerySink() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=583
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=633
* @return Whether the ragFileChunkingConfig field is set.
*/
@java.lang.Deprecated
@@ -4189,7 +4236,7 @@ public boolean hasRagFileChunkingConfig() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=583
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=633
* @return The ragFileChunkingConfig.
*/
@java.lang.Deprecated
@@ -4976,6 +5023,80 @@ public Builder clearGlobalMaxEmbeddingRequestsPerMin() {
return this;
}
+ private boolean rebuildAnnIndex_;
+
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @return The rebuildAnnIndex.
+ */
+ @java.lang.Override
+ public boolean getRebuildAnnIndex() {
+ return rebuildAnnIndex_;
+ }
+
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @param value The rebuildAnnIndex to set.
+ * @return This builder for chaining.
+ */
+ public Builder setRebuildAnnIndex(boolean value) {
+
+ rebuildAnnIndex_ = value;
+ bitField0_ |= 0x00004000;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearRebuildAnnIndex() {
+ bitField0_ = (bitField0_ & ~0x00004000);
+ rebuildAnnIndex_ = false;
+ onChanged();
+ return this;
+ }
+
@java.lang.Override
public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.setUnknownFields(unknownFields);
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/ImportRagFilesConfigOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/ImportRagFilesConfigOrBuilder.java
index cf34b2aab8ea..16e53c21e3f0 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/ImportRagFilesConfigOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/ImportRagFilesConfigOrBuilder.java
@@ -234,7 +234,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=555
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=605
* @return Whether the partialFailureGcsSink field is set.
*/
@java.lang.Deprecated
@@ -253,7 +253,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_gcs_sink is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=555
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=605
* @return The partialFailureGcsSink.
*/
@java.lang.Deprecated
@@ -292,7 +292,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_bigquery_sink
- * is deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
+ * is deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=614
* @return Whether the partialFailureBigquerySink field is set.
*/
@java.lang.Deprecated
@@ -316,7 +316,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.partial_failure_bigquery_sink
- * is deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
+ * is deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=614
* @return The partialFailureBigquerySink.
*/
@java.lang.Deprecated
@@ -448,7 +448,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=583
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=633
* @return Whether the ragFileChunkingConfig field is set.
*/
@java.lang.Deprecated
@@ -466,7 +466,7 @@ public interface ImportRagFilesConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=583
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=633
* @return The ragFileChunkingConfig.
*/
@java.lang.Deprecated
@@ -616,6 +616,25 @@ public interface ImportRagFilesConfigOrBuilder
*/
int getGlobalMaxEmbeddingRequestsPerMin();
+ /**
+ *
+ *
+ * + * Rebuilds the ANN index to optimize for recall on the imported data. + * Only applicable for RagCorpora running on RagManagedDb with + * `retrieval_strategy` set to `ANN`. The rebuild will be performed using the + * existing ANN config set on the RagCorpus. To change the ANN config, please + * use the UpdateRagCorpus API. + * + * Default is false, i.e., index is not rebuilt. + *+ * + *
bool rebuild_ann_index = 19;
+ *
+ * @return The rebuildAnnIndex.
+ */
+ boolean getRebuildAnnIndex();
+
com.google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.ImportSourceCase getImportSourceCase();
com.google.cloud.aiplatform.v1beta1.ImportRagFilesConfig.PartialFailureSinkCase
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/NotebookSoftwareConfig.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/NotebookSoftwareConfig.java
index 109b71b75ef7..6a711945e366 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/NotebookSoftwareConfig.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/NotebookSoftwareConfig.java
@@ -19,15 +19,7 @@
// Protobuf Java Version: 3.25.5
package com.google.cloud.aiplatform.v1beta1;
-/**
- *
- *
- * - * Notebook Software Config. - *- * - * Protobuf type {@code google.cloud.aiplatform.v1beta1.NotebookSoftwareConfig} - */ +/** Protobuf type {@code google.cloud.aiplatform.v1beta1.NotebookSoftwareConfig} */ public final class NotebookSoftwareConfig extends com.google.protobuf.GeneratedMessageV3 implements // @@protoc_insertion_point(message_implements:google.cloud.aiplatform.v1beta1.NotebookSoftwareConfig) @@ -160,12 +152,6 @@ public com.google.cloud.aiplatform.v1beta1.EnvVarOrBuilder getEnvOrBuilder(int i private com.google.cloud.aiplatform.v1beta1.PostStartupScriptConfig postStartupScriptConfig_; /** - * - * - *
- * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -178,12 +164,6 @@ public boolean hasPostStartupScriptConfig() {
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -198,12 +178,6 @@ public com.google.cloud.aiplatform.v1beta1.PostStartupScriptConfig getPostStartu
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -394,15 +368,7 @@ protected Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.Build
return builder;
}
- /**
- *
- *
- * - * Notebook Software Config. - *- * - * Protobuf type {@code google.cloud.aiplatform.v1beta1.NotebookSoftwareConfig} - */ + /** Protobuf type {@code google.cloud.aiplatform.v1beta1.NotebookSoftwareConfig} */ public static final class Builder extends com.google.protobuf.GeneratedMessageV3.Builder
- * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1101,12 +1061,6 @@ public boolean hasPostStartupScriptConfig() {
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1125,12 +1079,6 @@ public boolean hasPostStartupScriptConfig() {
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1151,12 +1099,6 @@ public Builder setPostStartupScriptConfig(
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1174,12 +1116,6 @@ public Builder setPostStartupScriptConfig(
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1207,12 +1143,6 @@ public Builder mergePostStartupScriptConfig(
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1229,12 +1159,6 @@ public Builder clearPostStartupScriptConfig() {
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1247,12 +1171,6 @@ public Builder clearPostStartupScriptConfig() {
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1269,12 +1187,6 @@ public Builder clearPostStartupScriptConfig() {
}
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/NotebookSoftwareConfigOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/NotebookSoftwareConfigOrBuilder.java
index c9a720ed4cc8..c679fbc3e449 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/NotebookSoftwareConfigOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/NotebookSoftwareConfigOrBuilder.java
@@ -96,12 +96,6 @@ public interface NotebookSoftwareConfigOrBuilder
com.google.cloud.aiplatform.v1beta1.EnvVarOrBuilder getEnvOrBuilder(int index);
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -111,12 +105,6 @@ public interface NotebookSoftwareConfigOrBuilder
boolean hasPostStartupScriptConfig();
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -126,12 +114,6 @@ public interface NotebookSoftwareConfigOrBuilder
com.google.cloud.aiplatform.v1beta1.PostStartupScriptConfig getPostStartupScriptConfig();
/**
- *
- *
- * - * Optional. Post startup script config. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig post_startup_script_config = 2 [(.google.api.field_behavior) = OPTIONAL];
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/OpenApiProto.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/OpenApiProto.java
index 404281580ad6..5e29f6e4ec1f 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/OpenApiProto.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/OpenApiProto.java
@@ -51,7 +51,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
java.lang.String[] descriptorData = {
"\n"
+ "-google/cloud/aiplatform/v1beta1/openapi.proto\022\037google.cloud.aiplatform.v1beta1"
- + "\032\037google/api/field_behavior.proto\032\034google/protobuf/struct.proto\"\367\007\n"
+ + "\032\037google/api/field_behavior.proto\032\034google/protobuf/struct.proto\"\263\010\n"
+ "\006Schema\0228\n"
+ "\004type\030\001"
+ " \001(\0162%.google.cloud.aiplatform.v1beta1.TypeB\003\340A\001\022\023\n"
@@ -78,10 +78,12 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\007pattern\030\024 \001(\tB\003\340A\001\022,\n"
+ "\007example\030\004 \001(\0132\026.google.protobuf.ValueB\003\340A\001\022<\n"
+ "\006any_of\030\013"
- + " \003(\0132\'.google.cloud.aiplatform.v1beta1.SchemaB\003\340A\001\022\020\n"
+ + " \003(\0132\'.google.cloud.aiplatform.v1beta1.SchemaB\003\340A\001\022:\n"
+ + "\025additional_properties\030\032"
+ + " \001(\0132\026.google.protobuf.ValueB\003\340A\001\022\020\n"
+ "\003ref\030\033 \001(\tB\003\340A\001\022D\n"
- + "\004defs\030\034"
- + " \003(\01321.google.cloud.aiplatform.v1beta1.Schema.DefsEntryB\003\340A\001\032Z\n"
+ + "\004defs\030\034 \003(\01321.google.clo"
+ + "ud.aiplatform.v1beta1.Schema.DefsEntryB\003\340A\001\032Z\n"
+ "\017PropertiesEntry\022\013\n"
+ "\003key\030\001 \001(\t\0226\n"
+ "\005value\030\002 \001(\0132\'.google.cloud.aiplatform.v1beta1.Schema:\0028\001\032T\n"
@@ -96,11 +98,11 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\007BOOLEAN\020\004\022\t\n"
+ "\005ARRAY\020\005\022\n\n"
+ "\006OBJECT\020\006B\343\001\n"
- + "#com.google.cloud.aiplatform.v1beta1B\014OpenApiProtoP\001"
- + "ZCcloud.google.com/go/aiplatform/apiv1be"
- + "ta1/aiplatformpb;aiplatformpb\252\002\037Google.C"
- + "loud.AIPlatform.V1Beta1\312\002\037Google\\Cloud\\A"
- + "IPlatform\\V1beta1\352\002\"Google::Cloud::AIPlatform::V1beta1b\006proto3"
+ + "#com.google.cloud.aiplatform.v1beta1B\014OpenApiProtoP\001ZCcloud.google.com/g"
+ + "o/aiplatform/apiv1beta1/aiplatformpb;aip"
+ + "latformpb\252\002\037Google.Cloud.AIPlatform.V1Be"
+ + "ta1\312\002\037Google\\Cloud\\AIPlatform\\V1beta1\352\002\""
+ + "Google::Cloud::AIPlatform::V1beta1b\006proto3"
};
descriptor =
com.google.protobuf.Descriptors.FileDescriptor.internalBuildGeneratedFileFrom(
@@ -137,6 +139,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
"Pattern",
"Example",
"AnyOf",
+ "AdditionalProperties",
"Ref",
"Defs",
});
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/PostStartupScriptConfig.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/PostStartupScriptConfig.java
index 9351271ee903..03ae979d24a2 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/PostStartupScriptConfig.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/PostStartupScriptConfig.java
@@ -19,15 +19,7 @@
// Protobuf Java Version: 3.25.5
package com.google.cloud.aiplatform.v1beta1;
-/**
- *
- *
- * - * Post startup script config. - *- * - * Protobuf type {@code google.cloud.aiplatform.v1beta1.PostStartupScriptConfig} - */ +/** Protobuf type {@code google.cloud.aiplatform.v1beta1.PostStartupScriptConfig} */ public final class PostStartupScriptConfig extends com.google.protobuf.GeneratedMessageV3 implements // @@protoc_insertion_point(message_implements:google.cloud.aiplatform.v1beta1.PostStartupScriptConfig) @@ -67,101 +59,31 @@ public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() { } /** - * - * - *
- * Represents a notebook runtime post startup script behavior. - *- * * Protobuf enum {@code * google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior} */ public enum PostStartupScriptBehavior implements com.google.protobuf.ProtocolMessageEnum { - /** - * - * - *
- * Unspecified post startup script behavior. - *- * - *
POST_STARTUP_SCRIPT_BEHAVIOR_UNSPECIFIED = 0;
- */
+ /** POST_STARTUP_SCRIPT_BEHAVIOR_UNSPECIFIED = 0; */
POST_STARTUP_SCRIPT_BEHAVIOR_UNSPECIFIED(0),
- /**
- *
- *
- * - * Run post startup script after runtime is started. - *- * - *
RUN_ONCE = 1;
- */
+ /** RUN_ONCE = 1; */
RUN_ONCE(1),
- /**
- *
- *
- * - * Run post startup script after runtime is stopped. - *- * - *
RUN_EVERY_START = 2;
- */
+ /** RUN_EVERY_START = 2; */
RUN_EVERY_START(2),
- /**
- *
- *
- * - * Download and run post startup script every time runtime is started. - *- * - *
DOWNLOAD_AND_RUN_EVERY_START = 3;
- */
+ /** DOWNLOAD_AND_RUN_EVERY_START = 3; */
DOWNLOAD_AND_RUN_EVERY_START(3),
UNRECOGNIZED(-1),
;
- /**
- *
- *
- * - * Unspecified post startup script behavior. - *- * - *
POST_STARTUP_SCRIPT_BEHAVIOR_UNSPECIFIED = 0;
- */
+ /** POST_STARTUP_SCRIPT_BEHAVIOR_UNSPECIFIED = 0; */
public static final int POST_STARTUP_SCRIPT_BEHAVIOR_UNSPECIFIED_VALUE = 0;
- /**
- *
- *
- * - * Run post startup script after runtime is started. - *- * - *
RUN_ONCE = 1;
- */
+ /** RUN_ONCE = 1; */
public static final int RUN_ONCE_VALUE = 1;
- /**
- *
- *
- * - * Run post startup script after runtime is stopped. - *- * - *
RUN_EVERY_START = 2;
- */
+ /** RUN_EVERY_START = 2; */
public static final int RUN_EVERY_START_VALUE = 2;
- /**
- *
- *
- * - * Download and run post startup script every time runtime is started. - *- * - *
DOWNLOAD_AND_RUN_EVERY_START = 3;
- */
+ /** DOWNLOAD_AND_RUN_EVERY_START = 3; */
public static final int DOWNLOAD_AND_RUN_EVERY_START_VALUE = 3;
public final int getNumber() {
@@ -260,12 +182,6 @@ private PostStartupScriptBehavior(int value) {
private volatile java.lang.Object postStartupScript_ = "";
/**
- *
- *
- * - * Optional. Post startup script to run after runtime is started. - *- * *
string post_startup_script = 1 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The postStartupScript.
@@ -284,12 +200,6 @@ public java.lang.String getPostStartupScript() {
}
/**
- *
- *
- * - * Optional. Post startup script to run after runtime is started. - *- * *
string post_startup_script = 1 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The bytes for postStartupScript.
@@ -313,13 +223,6 @@ public com.google.protobuf.ByteString getPostStartupScriptBytes() {
private volatile java.lang.Object postStartupScriptUrl_ = "";
/**
- *
- *
- * - * Optional. Post startup script url to download. Example: - * https://bucket/script.sh - *- * *
string post_startup_script_url = 2 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The postStartupScriptUrl.
@@ -338,13 +241,6 @@ public java.lang.String getPostStartupScriptUrl() {
}
/**
- *
- *
- * - * Optional. Post startup script url to download. Example: - * https://bucket/script.sh - *- * *
string post_startup_script_url = 2 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The bytes for postStartupScriptUrl.
@@ -366,13 +262,6 @@ public com.google.protobuf.ByteString getPostStartupScriptUrlBytes() {
private int postStartupScriptBehavior_ = 0;
/**
- *
- *
- * - * Optional. Post startup script behavior that defines download and execution - * behavior. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior post_startup_script_behavior = 3 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -385,13 +274,6 @@ public int getPostStartupScriptBehaviorValue() {
}
/**
- *
- *
- * - * Optional. Post startup script behavior that defines download and execution - * behavior. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior post_startup_script_behavior = 3 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -595,15 +477,7 @@ protected Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.Build
return builder;
}
- /**
- *
- *
- * - * Post startup script config. - *- * - * Protobuf type {@code google.cloud.aiplatform.v1beta1.PostStartupScriptConfig} - */ + /** Protobuf type {@code google.cloud.aiplatform.v1beta1.PostStartupScriptConfig} */ public static final class Builder extends com.google.protobuf.GeneratedMessageV3.Builder
- * Optional. Post startup script to run after runtime is started. - *- * *
string post_startup_script = 1 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The postStartupScript.
@@ -832,12 +700,6 @@ public java.lang.String getPostStartupScript() {
}
/**
- *
- *
- * - * Optional. Post startup script to run after runtime is started. - *- * *
string post_startup_script = 1 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The bytes for postStartupScript.
@@ -855,12 +717,6 @@ public com.google.protobuf.ByteString getPostStartupScriptBytes() {
}
/**
- *
- *
- * - * Optional. Post startup script to run after runtime is started. - *- * *
string post_startup_script = 1 [(.google.api.field_behavior) = OPTIONAL];
*
* @param value The postStartupScript to set.
@@ -877,12 +733,6 @@ public Builder setPostStartupScript(java.lang.String value) {
}
/**
- *
- *
- * - * Optional. Post startup script to run after runtime is started. - *- * *
string post_startup_script = 1 [(.google.api.field_behavior) = OPTIONAL];
*
* @return This builder for chaining.
@@ -895,12 +745,6 @@ public Builder clearPostStartupScript() {
}
/**
- *
- *
- * - * Optional. Post startup script to run after runtime is started. - *- * *
string post_startup_script = 1 [(.google.api.field_behavior) = OPTIONAL];
*
* @param value The bytes for postStartupScript to set.
@@ -920,13 +764,6 @@ public Builder setPostStartupScriptBytes(com.google.protobuf.ByteString value) {
private java.lang.Object postStartupScriptUrl_ = "";
/**
- *
- *
- * - * Optional. Post startup script url to download. Example: - * https://bucket/script.sh - *- * *
string post_startup_script_url = 2 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The postStartupScriptUrl.
@@ -944,13 +781,6 @@ public java.lang.String getPostStartupScriptUrl() {
}
/**
- *
- *
- * - * Optional. Post startup script url to download. Example: - * https://bucket/script.sh - *- * *
string post_startup_script_url = 2 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The bytes for postStartupScriptUrl.
@@ -968,13 +798,6 @@ public com.google.protobuf.ByteString getPostStartupScriptUrlBytes() {
}
/**
- *
- *
- * - * Optional. Post startup script url to download. Example: - * https://bucket/script.sh - *- * *
string post_startup_script_url = 2 [(.google.api.field_behavior) = OPTIONAL];
*
* @param value The postStartupScriptUrl to set.
@@ -991,13 +814,6 @@ public Builder setPostStartupScriptUrl(java.lang.String value) {
}
/**
- *
- *
- * - * Optional. Post startup script url to download. Example: - * https://bucket/script.sh - *- * *
string post_startup_script_url = 2 [(.google.api.field_behavior) = OPTIONAL];
*
* @return This builder for chaining.
@@ -1010,13 +826,6 @@ public Builder clearPostStartupScriptUrl() {
}
/**
- *
- *
- * - * Optional. Post startup script url to download. Example: - * https://bucket/script.sh - *- * *
string post_startup_script_url = 2 [(.google.api.field_behavior) = OPTIONAL];
*
* @param value The bytes for postStartupScriptUrl to set.
@@ -1036,13 +845,6 @@ public Builder setPostStartupScriptUrlBytes(com.google.protobuf.ByteString value
private int postStartupScriptBehavior_ = 0;
/**
- *
- *
- * - * Optional. Post startup script behavior that defines download and execution - * behavior. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior post_startup_script_behavior = 3 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1055,13 +857,6 @@ public int getPostStartupScriptBehaviorValue() {
}
/**
- *
- *
- * - * Optional. Post startup script behavior that defines download and execution - * behavior. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior post_startup_script_behavior = 3 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1077,13 +872,6 @@ public Builder setPostStartupScriptBehaviorValue(int value) {
}
/**
- *
- *
- * - * Optional. Post startup script behavior that defines download and execution - * behavior. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior post_startup_script_behavior = 3 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1103,13 +891,6 @@ public Builder setPostStartupScriptBehaviorValue(int value) {
}
/**
- *
- *
- * - * Optional. Post startup script behavior that defines download and execution - * behavior. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior post_startup_script_behavior = 3 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -1130,13 +911,6 @@ public Builder setPostStartupScriptBehavior(
}
/**
- *
- *
- * - * Optional. Post startup script behavior that defines download and execution - * behavior. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior post_startup_script_behavior = 3 [(.google.api.field_behavior) = OPTIONAL];
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/PostStartupScriptConfigOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/PostStartupScriptConfigOrBuilder.java
index 19835aa53618..4428c8fad69d 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/PostStartupScriptConfigOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/PostStartupScriptConfigOrBuilder.java
@@ -25,12 +25,6 @@ public interface PostStartupScriptConfigOrBuilder
com.google.protobuf.MessageOrBuilder {
/**
- *
- *
- * - * Optional. Post startup script to run after runtime is started. - *- * *
string post_startup_script = 1 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The postStartupScript.
@@ -38,12 +32,6 @@ public interface PostStartupScriptConfigOrBuilder
java.lang.String getPostStartupScript();
/**
- *
- *
- * - * Optional. Post startup script to run after runtime is started. - *- * *
string post_startup_script = 1 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The bytes for postStartupScript.
@@ -51,13 +39,6 @@ public interface PostStartupScriptConfigOrBuilder
com.google.protobuf.ByteString getPostStartupScriptBytes();
/**
- *
- *
- * - * Optional. Post startup script url to download. Example: - * https://bucket/script.sh - *- * *
string post_startup_script_url = 2 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The postStartupScriptUrl.
@@ -65,13 +46,6 @@ public interface PostStartupScriptConfigOrBuilder
java.lang.String getPostStartupScriptUrl();
/**
- *
- *
- * - * Optional. Post startup script url to download. Example: - * https://bucket/script.sh - *- * *
string post_startup_script_url = 2 [(.google.api.field_behavior) = OPTIONAL];
*
* @return The bytes for postStartupScriptUrl.
@@ -79,13 +53,6 @@ public interface PostStartupScriptConfigOrBuilder
com.google.protobuf.ByteString getPostStartupScriptUrlBytes();
/**
- *
- *
- * - * Optional. Post startup script behavior that defines download and execution - * behavior. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior post_startup_script_behavior = 3 [(.google.api.field_behavior) = OPTIONAL];
*
@@ -95,13 +62,6 @@ public interface PostStartupScriptConfigOrBuilder
int getPostStartupScriptBehaviorValue();
/**
- *
- *
- * - * Optional. Post startup script behavior that defines download and execution - * behavior. - *- * *
* .google.cloud.aiplatform.v1beta1.PostStartupScriptConfig.PostStartupScriptBehavior post_startup_script_behavior = 3 [(.google.api.field_behavior) = OPTIONAL];
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagCorpus.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagCorpus.java
index c95e204b0c9e..4150aeda25b8 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagCorpus.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagCorpus.java
@@ -418,7 +418,7 @@ public com.google.protobuf.ByteString getDescriptionBytes() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_embedding_model_config is deprecated.
- * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=284
+ * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=325
* @return Whether the ragEmbeddingModelConfig field is set.
*/
@java.lang.Override
@@ -439,7 +439,7 @@ public boolean hasRagEmbeddingModelConfig() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_embedding_model_config is deprecated.
- * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=284
+ * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=325
* @return The ragEmbeddingModelConfig.
*/
@java.lang.Override
@@ -485,7 +485,7 @@ public com.google.cloud.aiplatform.v1beta1.RagEmbeddingModelConfig getRagEmbeddi
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_vector_db_config is deprecated. See
- * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=291
+ * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=332
* @return Whether the ragVectorDbConfig field is set.
*/
@java.lang.Override
@@ -506,7 +506,7 @@ public boolean hasRagVectorDbConfig() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_vector_db_config is deprecated. See
- * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=291
+ * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=332
* @return The ragVectorDbConfig.
*/
@java.lang.Override
@@ -719,6 +719,74 @@ public int getRagFilesCount() {
return ragFilesCount_;
}
+ public static final int ENCRYPTION_SPEC_FIELD_NUMBER = 12;
+ private com.google.cloud.aiplatform.v1beta1.EncryptionSpec encryptionSpec_;
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return Whether the encryptionSpec field is set.
+ */
+ @java.lang.Override
+ public boolean hasEncryptionSpec() {
+ return ((bitField0_ & 0x00000020) != 0);
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return The encryptionSpec.
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.EncryptionSpec getEncryptionSpec() {
+ return encryptionSpec_ == null
+ ? com.google.cloud.aiplatform.v1beta1.EncryptionSpec.getDefaultInstance()
+ : encryptionSpec_;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.EncryptionSpecOrBuilder getEncryptionSpecOrBuilder() {
+ return encryptionSpec_ == null
+ ? com.google.cloud.aiplatform.v1beta1.EncryptionSpec.getDefaultInstance()
+ : encryptionSpec_;
+ }
+
private byte memoizedIsInitialized = -1;
@java.lang.Override
@@ -768,6 +836,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
if (ragFilesCount_ != 0) {
output.writeInt32(11, ragFilesCount_);
}
+ if (((bitField0_ & 0x00000020) != 0)) {
+ output.writeMessage(12, getEncryptionSpec());
+ }
getUnknownFields().writeTo(output);
}
@@ -815,6 +886,9 @@ public int getSerializedSize() {
if (ragFilesCount_ != 0) {
size += com.google.protobuf.CodedOutputStream.computeInt32Size(11, ragFilesCount_);
}
+ if (((bitField0_ & 0x00000020) != 0)) {
+ size += com.google.protobuf.CodedOutputStream.computeMessageSize(12, getEncryptionSpec());
+ }
size += getUnknownFields().getSerializedSize();
memoizedSize = size;
return size;
@@ -855,6 +929,10 @@ public boolean equals(final java.lang.Object obj) {
if (!getCorpusStatus().equals(other.getCorpusStatus())) return false;
}
if (getRagFilesCount() != other.getRagFilesCount()) return false;
+ if (hasEncryptionSpec() != other.hasEncryptionSpec()) return false;
+ if (hasEncryptionSpec()) {
+ if (!getEncryptionSpec().equals(other.getEncryptionSpec())) return false;
+ }
if (!getBackendConfigCase().equals(other.getBackendConfigCase())) return false;
switch (backendConfigCase_) {
case 9:
@@ -905,6 +983,10 @@ public int hashCode() {
}
hash = (37 * hash) + RAG_FILES_COUNT_FIELD_NUMBER;
hash = (53 * hash) + getRagFilesCount();
+ if (hasEncryptionSpec()) {
+ hash = (37 * hash) + ENCRYPTION_SPEC_FIELD_NUMBER;
+ hash = (53 * hash) + getEncryptionSpec().hashCode();
+ }
switch (backendConfigCase_) {
case 9:
hash = (37 * hash) + VECTOR_DB_CONFIG_FIELD_NUMBER;
@@ -1064,6 +1146,7 @@ private void maybeForceBuilderInitialization() {
getCreateTimeFieldBuilder();
getUpdateTimeFieldBuilder();
getCorpusStatusFieldBuilder();
+ getEncryptionSpecFieldBuilder();
}
}
@@ -1106,6 +1189,11 @@ public Builder clear() {
corpusStatusBuilder_ = null;
}
ragFilesCount_ = 0;
+ encryptionSpec_ = null;
+ if (encryptionSpecBuilder_ != null) {
+ encryptionSpecBuilder_.dispose();
+ encryptionSpecBuilder_ = null;
+ }
backendConfigCase_ = 0;
backendConfig_ = null;
return this;
@@ -1185,6 +1273,11 @@ private void buildPartial0(com.google.cloud.aiplatform.v1beta1.RagCorpus result)
if (((from_bitField0_ & 0x00000400) != 0)) {
result.ragFilesCount_ = ragFilesCount_;
}
+ if (((from_bitField0_ & 0x00000800) != 0)) {
+ result.encryptionSpec_ =
+ encryptionSpecBuilder_ == null ? encryptionSpec_ : encryptionSpecBuilder_.build();
+ to_bitField0_ |= 0x00000020;
+ }
result.bitField0_ |= to_bitField0_;
}
@@ -1277,6 +1370,9 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1beta1.RagCorpus other) {
if (other.getRagFilesCount() != 0) {
setRagFilesCount(other.getRagFilesCount());
}
+ if (other.hasEncryptionSpec()) {
+ mergeEncryptionSpec(other.getEncryptionSpec());
+ }
switch (other.getBackendConfigCase()) {
case VECTOR_DB_CONFIG:
{
@@ -1388,6 +1484,12 @@ public Builder mergeFrom(
bitField0_ |= 0x00000400;
break;
} // case 88
+ case 98:
+ {
+ input.readMessage(getEncryptionSpecFieldBuilder().getBuilder(), extensionRegistry);
+ bitField0_ |= 0x00000800;
+ break;
+ } // case 98
default:
{
if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -2266,7 +2368,7 @@ public Builder setDescriptionBytes(com.google.protobuf.ByteString value) {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_embedding_model_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=284
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=325
* @return Whether the ragEmbeddingModelConfig field is set.
*/
@java.lang.Deprecated
@@ -2286,7 +2388,7 @@ public boolean hasRagEmbeddingModelConfig() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_embedding_model_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=284
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=325
* @return The ragEmbeddingModelConfig.
*/
@java.lang.Deprecated
@@ -2498,7 +2600,7 @@ public Builder clearRagEmbeddingModelConfig() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_vector_db_config is deprecated. See
- * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=291
+ * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=332
* @return Whether the ragVectorDbConfig field is set.
*/
@java.lang.Deprecated
@@ -2518,7 +2620,7 @@ public boolean hasRagVectorDbConfig() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_vector_db_config is deprecated. See
- * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=291
+ * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=332
* @return The ragVectorDbConfig.
*/
@java.lang.Deprecated
@@ -3403,6 +3505,248 @@ public Builder clearRagFilesCount() {
return this;
}
+ private com.google.cloud.aiplatform.v1beta1.EncryptionSpec encryptionSpec_;
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpec,
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpec.Builder,
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpecOrBuilder>
+ encryptionSpecBuilder_;
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return Whether the encryptionSpec field is set.
+ */
+ public boolean hasEncryptionSpec() {
+ return ((bitField0_ & 0x00000800) != 0);
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return The encryptionSpec.
+ */
+ public com.google.cloud.aiplatform.v1beta1.EncryptionSpec getEncryptionSpec() {
+ if (encryptionSpecBuilder_ == null) {
+ return encryptionSpec_ == null
+ ? com.google.cloud.aiplatform.v1beta1.EncryptionSpec.getDefaultInstance()
+ : encryptionSpec_;
+ } else {
+ return encryptionSpecBuilder_.getMessage();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder setEncryptionSpec(com.google.cloud.aiplatform.v1beta1.EncryptionSpec value) {
+ if (encryptionSpecBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ encryptionSpec_ = value;
+ } else {
+ encryptionSpecBuilder_.setMessage(value);
+ }
+ bitField0_ |= 0x00000800;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder setEncryptionSpec(
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpec.Builder builderForValue) {
+ if (encryptionSpecBuilder_ == null) {
+ encryptionSpec_ = builderForValue.build();
+ } else {
+ encryptionSpecBuilder_.setMessage(builderForValue.build());
+ }
+ bitField0_ |= 0x00000800;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder mergeEncryptionSpec(com.google.cloud.aiplatform.v1beta1.EncryptionSpec value) {
+ if (encryptionSpecBuilder_ == null) {
+ if (((bitField0_ & 0x00000800) != 0)
+ && encryptionSpec_ != null
+ && encryptionSpec_
+ != com.google.cloud.aiplatform.v1beta1.EncryptionSpec.getDefaultInstance()) {
+ getEncryptionSpecBuilder().mergeFrom(value);
+ } else {
+ encryptionSpec_ = value;
+ }
+ } else {
+ encryptionSpecBuilder_.mergeFrom(value);
+ }
+ if (encryptionSpec_ != null) {
+ bitField0_ |= 0x00000800;
+ onChanged();
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder clearEncryptionSpec() {
+ bitField0_ = (bitField0_ & ~0x00000800);
+ encryptionSpec_ = null;
+ if (encryptionSpecBuilder_ != null) {
+ encryptionSpecBuilder_.dispose();
+ encryptionSpecBuilder_ = null;
+ }
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public com.google.cloud.aiplatform.v1beta1.EncryptionSpec.Builder getEncryptionSpecBuilder() {
+ bitField0_ |= 0x00000800;
+ onChanged();
+ return getEncryptionSpecFieldBuilder().getBuilder();
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public com.google.cloud.aiplatform.v1beta1.EncryptionSpecOrBuilder
+ getEncryptionSpecOrBuilder() {
+ if (encryptionSpecBuilder_ != null) {
+ return encryptionSpecBuilder_.getMessageOrBuilder();
+ } else {
+ return encryptionSpec_ == null
+ ? com.google.cloud.aiplatform.v1beta1.EncryptionSpec.getDefaultInstance()
+ : encryptionSpec_;
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpec,
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpec.Builder,
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpecOrBuilder>
+ getEncryptionSpecFieldBuilder() {
+ if (encryptionSpecBuilder_ == null) {
+ encryptionSpecBuilder_ =
+ new com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpec,
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpec.Builder,
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpecOrBuilder>(
+ getEncryptionSpec(), getParentForChildren(), isClean());
+ encryptionSpec_ = null;
+ }
+ return encryptionSpecBuilder_;
+ }
+
@java.lang.Override
public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.setUnknownFields(unknownFields);
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagCorpusOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagCorpusOrBuilder.java
index df7c058e42de..6f8d0a5bcc58 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagCorpusOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagCorpusOrBuilder.java
@@ -205,7 +205,7 @@ public interface RagCorpusOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_embedding_model_config is deprecated.
- * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=284
+ * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=325
* @return Whether the ragEmbeddingModelConfig field is set.
*/
@java.lang.Deprecated
@@ -223,7 +223,7 @@ public interface RagCorpusOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_embedding_model_config is deprecated.
- * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=284
+ * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=325
* @return The ragEmbeddingModelConfig.
*/
@java.lang.Deprecated
@@ -256,7 +256,7 @@ public interface RagCorpusOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_vector_db_config is deprecated. See
- * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=291
+ * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=332
* @return Whether the ragVectorDbConfig field is set.
*/
@java.lang.Deprecated
@@ -274,7 +274,7 @@ public interface RagCorpusOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.RagCorpus.rag_vector_db_config is deprecated. See
- * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=291
+ * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=332
* @return The ragVectorDbConfig.
*/
@java.lang.Deprecated
@@ -430,5 +430,57 @@ public interface RagCorpusOrBuilder
*/
int getRagFilesCount();
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return Whether the encryptionSpec field is set.
+ */
+ boolean hasEncryptionSpec();
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return The encryptionSpec.
+ */
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpec getEncryptionSpec();
+
+ /**
+ *
+ *
+ * + * Optional. Immutable. The CMEK key name used to encrypt at-rest data related + * to this Corpus. Only applicable to RagManagedDb option for Vector DB. This + * field can only be set at corpus creation time, and cannot be updated or + * deleted. + *+ * + *
+ * .google.cloud.aiplatform.v1beta1.EncryptionSpec encryption_spec = 12 [(.google.api.field_behavior) = IMMUTABLE, (.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpecOrBuilder getEncryptionSpecOrBuilder();
+
com.google.cloud.aiplatform.v1beta1.RagCorpus.BackendConfigCase getBackendConfigCase();
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileChunkingConfig.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileChunkingConfig.java
index 4714079fdc43..6204658facfc 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileChunkingConfig.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileChunkingConfig.java
@@ -873,7 +873,7 @@ public boolean hasFixedLengthChunking() {
* int32 chunk_size = 1 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_size is deprecated. See
- * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=426
+ * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=476
* @return The chunkSize.
*/
@java.lang.Override
@@ -895,7 +895,7 @@ public int getChunkSize() {
* int32 chunk_overlap = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_overlap is deprecated.
- * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=429
+ * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=479
* @return The chunkOverlap.
*/
@java.lang.Override
@@ -1613,7 +1613,7 @@ public Builder clearFixedLengthChunking() {
* int32 chunk_size = 1 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_size is deprecated.
- * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=426
+ * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=476
* @return The chunkSize.
*/
@java.lang.Override
@@ -1632,7 +1632,7 @@ public int getChunkSize() {
* int32 chunk_size = 1 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_size is deprecated.
- * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=426
+ * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=476
* @param value The chunkSize to set.
* @return This builder for chaining.
*/
@@ -1655,7 +1655,7 @@ public Builder setChunkSize(int value) {
* int32 chunk_size = 1 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_size is deprecated.
- * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=426
+ * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=476
* @return This builder for chaining.
*/
@java.lang.Deprecated
@@ -1678,7 +1678,7 @@ public Builder clearChunkSize() {
* int32 chunk_overlap = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_overlap is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=429
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=479
* @return The chunkOverlap.
*/
@java.lang.Override
@@ -1697,7 +1697,7 @@ public int getChunkOverlap() {
* int32 chunk_overlap = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_overlap is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=429
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=479
* @param value The chunkOverlap to set.
* @return This builder for chaining.
*/
@@ -1720,7 +1720,7 @@ public Builder setChunkOverlap(int value) {
* int32 chunk_overlap = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_overlap is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=429
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=479
* @return This builder for chaining.
*/
@java.lang.Deprecated
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileChunkingConfigOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileChunkingConfigOrBuilder.java
index a8b60d1b3e9b..76c5d1d50208 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileChunkingConfigOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileChunkingConfigOrBuilder.java
@@ -79,7 +79,7 @@ public interface RagFileChunkingConfigOrBuilder
* int32 chunk_size = 1 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_size is deprecated. See
- * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=426
+ * google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=476
* @return The chunkSize.
*/
@java.lang.Deprecated
@@ -95,7 +95,7 @@ public interface RagFileChunkingConfigOrBuilder
* int32 chunk_overlap = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileChunkingConfig.chunk_overlap is deprecated.
- * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=429
+ * See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=479
* @return The chunkOverlap.
*/
@java.lang.Deprecated
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileParsingConfig.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileParsingConfig.java
index 71e055f89c84..4498a10d3183 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileParsingConfig.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileParsingConfig.java
@@ -3036,7 +3036,7 @@ public com.google.cloud.aiplatform.v1beta1.RagFileParsingConfig.LlmParser getLlm
* bool use_advanced_pdf_parsing = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileParsingConfig.use_advanced_pdf_parsing is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=514
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
* @return The useAdvancedPdfParsing.
*/
@java.lang.Override
@@ -4256,7 +4256,7 @@ public Builder clearLlmParser() {
* bool use_advanced_pdf_parsing = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileParsingConfig.use_advanced_pdf_parsing is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=514
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
* @return The useAdvancedPdfParsing.
*/
@java.lang.Override
@@ -4275,7 +4275,7 @@ public boolean getUseAdvancedPdfParsing() {
* bool use_advanced_pdf_parsing = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileParsingConfig.use_advanced_pdf_parsing is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=514
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
* @param value The useAdvancedPdfParsing to set.
* @return This builder for chaining.
*/
@@ -4298,7 +4298,7 @@ public Builder setUseAdvancedPdfParsing(boolean value) {
* bool use_advanced_pdf_parsing = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileParsingConfig.use_advanced_pdf_parsing is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=514
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
* @return This builder for chaining.
*/
@java.lang.Deprecated
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileParsingConfigOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileParsingConfigOrBuilder.java
index c9cc292f1cac..f29fe61c8bea 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileParsingConfigOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagFileParsingConfigOrBuilder.java
@@ -154,7 +154,7 @@ public interface RagFileParsingConfigOrBuilder
* bool use_advanced_pdf_parsing = 2 [deprecated = true];
*
* @deprecated google.cloud.aiplatform.v1beta1.RagFileParsingConfig.use_advanced_pdf_parsing is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=514
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=564
* @return The useAdvancedPdfParsing.
*/
@java.lang.Deprecated
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagVectorDbConfig.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagVectorDbConfig.java
index ada4e2ca6f8e..cd47c32e400b 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagVectorDbConfig.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/RagVectorDbConfig.java
@@ -65,7 +65,93 @@ public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() {
public interface RagManagedDbOrBuilder
extends
// @@protoc_insertion_point(interface_extends:google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb)
- com.google.protobuf.MessageOrBuilder {}
+ com.google.protobuf.MessageOrBuilder {
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return Whether the knn field is set.
+ */
+ boolean hasKnn();
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return The knn.
+ */
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN getKnn();
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder
+ getKnnOrBuilder();
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return Whether the ann field is set.
+ */
+ boolean hasAnn();
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return The ann.
+ */
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN getAnn();
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder
+ getAnnOrBuilder();
+
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.RetrievalStrategyCase
+ getRetrievalStrategyCase();
+ }
/**
*
@@ -110,233 +196,1692 @@ public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() {
com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.Builder.class);
}
- private byte memoizedIsInitialized = -1;
-
- @java.lang.Override
- public final boolean isInitialized() {
- byte isInitialized = memoizedIsInitialized;
- if (isInitialized == 1) return true;
- if (isInitialized == 0) return false;
-
- memoizedIsInitialized = 1;
- return true;
- }
-
- @java.lang.Override
- public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException {
- getUnknownFields().writeTo(output);
- }
-
- @java.lang.Override
- public int getSerializedSize() {
- int size = memoizedSize;
- if (size != -1) return size;
-
- size = 0;
- size += getUnknownFields().getSerializedSize();
- memoizedSize = size;
- return size;
- }
-
- @java.lang.Override
- public boolean equals(final java.lang.Object obj) {
- if (obj == this) {
- return true;
- }
- if (!(obj instanceof com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb)) {
- return super.equals(obj);
- }
- com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb other =
- (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb) obj;
-
- if (!getUnknownFields().equals(other.getUnknownFields())) return false;
- return true;
- }
-
- @java.lang.Override
- public int hashCode() {
- if (memoizedHashCode != 0) {
- return memoizedHashCode;
- }
- int hash = 41;
- hash = (19 * hash) + getDescriptor().hashCode();
- hash = (29 * hash) + getUnknownFields().hashCode();
- memoizedHashCode = hash;
- return hash;
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- java.nio.ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- com.google.protobuf.ByteString data)
- throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- com.google.protobuf.ByteString data,
- com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- byte[] data) throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws com.google.protobuf.InvalidProtocolBufferException {
- return PARSER.parseFrom(data, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- java.io.InputStream input) throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
- PARSER, input, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb
- parseDelimitedFrom(java.io.InputStream input) throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(PARSER, input);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb
- parseDelimitedFrom(
- java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(
- PARSER, input, extensionRegistry);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- com.google.protobuf.CodedInputStream input) throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
- }
-
- public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
- com.google.protobuf.CodedInputStream input,
- com.google.protobuf.ExtensionRegistryLite extensionRegistry)
- throws java.io.IOException {
- return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
- PARSER, input, extensionRegistry);
- }
-
- @java.lang.Override
- public Builder newBuilderForType() {
- return newBuilder();
- }
-
- public static Builder newBuilder() {
- return DEFAULT_INSTANCE.toBuilder();
- }
-
- public static Builder newBuilder(
- com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb prototype) {
- return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
- }
-
- @java.lang.Override
- public Builder toBuilder() {
- return this == DEFAULT_INSTANCE ? new Builder() : new Builder().mergeFrom(this);
- }
-
- @java.lang.Override
- protected Builder newBuilderForType(
- com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
- Builder builder = new Builder(parent);
- return builder;
- }
+ public interface KNNOrBuilder
+ extends
+ // @@protoc_insertion_point(interface_extends:google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN)
+ com.google.protobuf.MessageOrBuilder {}
/**
*
*
*
- * The config for the default RAG-managed Vector DB.
+ * Config for KNN search.
*
*
- * Protobuf type {@code google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb}
+ * Protobuf type {@code google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN}
*/
- public static final class Builder
- extends com.google.protobuf.GeneratedMessageV3.Builder+ * Config for KNN search. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN} + */ + public static final class Builder + extends com.google.protobuf.GeneratedMessageV3.Builder
+ * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @return The treeDepth.
+ */
+ int getTreeDepth();
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @return The leafCount.
+ */
+ int getLeafCount();
+ }
+
+ /**
+ *
+ *
+ * + * Config for ANN search. + * + * RagManagedDb uses a tree-based structure to partition data and + * facilitate faster searches. As a tradeoff, it requires longer indexing + * time and manual triggering of index rebuild via the ImportRagFiles and + * UpdateRagCorpus API. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN} + */ + public static final class ANN extends com.google.protobuf.GeneratedMessageV3 + implements + // @@protoc_insertion_point(message_implements:google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN) + ANNOrBuilder { + private static final long serialVersionUID = 0L; + + // Use ANN.newBuilder() to construct. + private ANN(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + + private ANN() {} + + @java.lang.Override + @SuppressWarnings({"unused"}) + protected java.lang.Object newInstance(UnusedPrivateParameter unused) { + return new ANN(); + } + + public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() { + return com.google.cloud.aiplatform.v1beta1.VertexRagDataProto + .internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_ANN_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return com.google.cloud.aiplatform.v1beta1.VertexRagDataProto + .internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_ANN_fieldAccessorTable + .ensureFieldAccessorsInitialized( + com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN.class, + com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN + .Builder + .class); + } + + public static final int TREE_DEPTH_FIELD_NUMBER = 1; + private int treeDepth_ = 0; + + /** + * + * + *
+ * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @return The treeDepth.
+ */
+ @java.lang.Override
+ public int getTreeDepth() {
+ return treeDepth_;
+ }
+
+ public static final int LEAF_COUNT_FIELD_NUMBER = 2;
+ private int leafCount_ = 0;
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @return The leafCount.
+ */
+ @java.lang.Override
+ public int getLeafCount() {
+ return leafCount_;
+ }
+
+ private byte memoizedIsInitialized = -1;
+
+ @java.lang.Override
+ public final boolean isInitialized() {
+ byte isInitialized = memoizedIsInitialized;
+ if (isInitialized == 1) return true;
+ if (isInitialized == 0) return false;
+
+ memoizedIsInitialized = 1;
+ return true;
+ }
+
+ @java.lang.Override
+ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException {
+ if (treeDepth_ != 0) {
+ output.writeInt32(1, treeDepth_);
+ }
+ if (leafCount_ != 0) {
+ output.writeInt32(2, leafCount_);
+ }
+ getUnknownFields().writeTo(output);
+ }
+
+ @java.lang.Override
+ public int getSerializedSize() {
+ int size = memoizedSize;
+ if (size != -1) return size;
+
+ size = 0;
+ if (treeDepth_ != 0) {
+ size += com.google.protobuf.CodedOutputStream.computeInt32Size(1, treeDepth_);
+ }
+ if (leafCount_ != 0) {
+ size += com.google.protobuf.CodedOutputStream.computeInt32Size(2, leafCount_);
+ }
+ size += getUnknownFields().getSerializedSize();
+ memoizedSize = size;
+ return size;
+ }
+
+ @java.lang.Override
+ public boolean equals(final java.lang.Object obj) {
+ if (obj == this) {
+ return true;
+ }
+ if (!(obj
+ instanceof com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)) {
+ return super.equals(obj);
+ }
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN other =
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN) obj;
+
+ if (getTreeDepth() != other.getTreeDepth()) return false;
+ if (getLeafCount() != other.getLeafCount()) return false;
+ if (!getUnknownFields().equals(other.getUnknownFields())) return false;
+ return true;
+ }
+
+ @java.lang.Override
+ public int hashCode() {
+ if (memoizedHashCode != 0) {
+ return memoizedHashCode;
+ }
+ int hash = 41;
+ hash = (19 * hash) + getDescriptor().hashCode();
+ hash = (37 * hash) + TREE_DEPTH_FIELD_NUMBER;
+ hash = (53 * hash) + getTreeDepth();
+ hash = (37 * hash) + LEAF_COUNT_FIELD_NUMBER;
+ hash = (53 * hash) + getLeafCount();
+ hash = (29 * hash) + getUnknownFields().hashCode();
+ memoizedHashCode = hash;
+ return hash;
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(java.nio.ByteBuffer data)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(
+ java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(com.google.protobuf.ByteString data)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(
+ com.google.protobuf.ByteString data,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(byte[] data) throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(
+ java.io.InputStream input,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseDelimitedFrom(java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseDelimitedFrom(
+ java.io.InputStream input,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(com.google.protobuf.CodedInputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ parseFrom(
+ com.google.protobuf.CodedInputStream input,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ @java.lang.Override
+ public Builder newBuilderForType() {
+ return newBuilder();
+ }
+
+ public static Builder newBuilder() {
+ return DEFAULT_INSTANCE.toBuilder();
+ }
+
+ public static Builder newBuilder(
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN prototype) {
+ return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
+ }
+
+ @java.lang.Override
+ public Builder toBuilder() {
+ return this == DEFAULT_INSTANCE ? new Builder() : new Builder().mergeFrom(this);
+ }
+
+ @java.lang.Override
+ protected Builder newBuilderForType(
+ com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
+ Builder builder = new Builder(parent);
+ return builder;
+ }
+
+ /**
+ *
+ *
+ * + * Config for ANN search. + * + * RagManagedDb uses a tree-based structure to partition data and + * facilitate faster searches. As a tradeoff, it requires longer indexing + * time and manual triggering of index rebuild via the ImportRagFiles and + * UpdateRagCorpus API. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN} + */ + public static final class Builder + extends com.google.protobuf.GeneratedMessageV3.Builder
+ * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @return The treeDepth.
+ */
+ @java.lang.Override
+ public int getTreeDepth() {
+ return treeDepth_;
+ }
+
+ /**
+ *
+ *
+ * + * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @param value The treeDepth to set.
+ * @return This builder for chaining.
+ */
+ public Builder setTreeDepth(int value) {
+
+ treeDepth_ = value;
+ bitField0_ |= 0x00000001;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The depth of the tree-based structure. Only depth values of 2 and 3 are + * supported. + * + * Recommended value is 2 if you have if you have O(10K) files in the + * RagCorpus and set this to 3 if more than that. + * + * Default value is 2. + *+ * + *
int32 tree_depth = 1;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearTreeDepth() {
+ bitField0_ = (bitField0_ & ~0x00000001);
+ treeDepth_ = 0;
+ onChanged();
+ return this;
+ }
+
+ private int leafCount_;
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @return The leafCount.
+ */
+ @java.lang.Override
+ public int getLeafCount() {
+ return leafCount_;
+ }
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @param value The leafCount to set.
+ * @return This builder for chaining.
+ */
+ public Builder setLeafCount(int value) {
+
+ leafCount_ = value;
+ bitField0_ |= 0x00000002;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Number of leaf nodes in the tree-based structure. Each leaf node + * contains groups of closely related vectors along with their + * corresponding centroid. + * + * Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus). + * + * Default value is 500. + *+ * + *
int32 leaf_count = 2;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearLeafCount() {
+ bitField0_ = (bitField0_ & ~0x00000002);
+ leafCount_ = 0;
+ onChanged();
+ return this;
+ }
+
+ @java.lang.Override
+ public final Builder setUnknownFields(
+ final com.google.protobuf.UnknownFieldSet unknownFields) {
+ return super.setUnknownFields(unknownFields);
+ }
+
+ @java.lang.Override
+ public final Builder mergeUnknownFields(
+ final com.google.protobuf.UnknownFieldSet unknownFields) {
+ return super.mergeUnknownFields(unknownFields);
+ }
+
+ // @@protoc_insertion_point(builder_scope:google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ }
+
+ // @@protoc_insertion_point(class_scope:google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ private static final com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ DEFAULT_INSTANCE;
+
+ static {
+ DEFAULT_INSTANCE =
+ new com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN();
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ getDefaultInstance() {
+ return DEFAULT_INSTANCE;
+ }
+
+ private static final com.google.protobuf.Parser+ * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return Whether the knn field is set.
+ */
+ @java.lang.Override
+ public boolean hasKnn() {
+ return retrievalStrategyCase_ == 1;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return The knn.
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN getKnn() {
+ if (retrievalStrategyCase_ == 1) {
+ return (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder
+ getKnnOrBuilder() {
+ if (retrievalStrategyCase_ == 1) {
+ return (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ }
+
+ public static final int ANN_FIELD_NUMBER = 2;
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return Whether the ann field is set.
+ */
+ @java.lang.Override
+ public boolean hasAnn() {
+ return retrievalStrategyCase_ == 2;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return The ann.
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN getAnn() {
+ if (retrievalStrategyCase_ == 2) {
+ return (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder
+ getAnnOrBuilder() {
+ if (retrievalStrategyCase_ == 2) {
+ return (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ }
+
+ private byte memoizedIsInitialized = -1;
+
+ @java.lang.Override
+ public final boolean isInitialized() {
+ byte isInitialized = memoizedIsInitialized;
+ if (isInitialized == 1) return true;
+ if (isInitialized == 0) return false;
+
+ memoizedIsInitialized = 1;
+ return true;
+ }
+
+ @java.lang.Override
+ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException {
+ if (retrievalStrategyCase_ == 1) {
+ output.writeMessage(
+ 1,
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_);
+ }
+ if (retrievalStrategyCase_ == 2) {
+ output.writeMessage(
+ 2,
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_);
+ }
+ getUnknownFields().writeTo(output);
+ }
+
+ @java.lang.Override
+ public int getSerializedSize() {
+ int size = memoizedSize;
+ if (size != -1) return size;
+
+ size = 0;
+ if (retrievalStrategyCase_ == 1) {
+ size +=
+ com.google.protobuf.CodedOutputStream.computeMessageSize(
+ 1,
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_);
+ }
+ if (retrievalStrategyCase_ == 2) {
+ size +=
+ com.google.protobuf.CodedOutputStream.computeMessageSize(
+ 2,
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_);
+ }
+ size += getUnknownFields().getSerializedSize();
+ memoizedSize = size;
+ return size;
+ }
+
+ @java.lang.Override
+ public boolean equals(final java.lang.Object obj) {
+ if (obj == this) {
+ return true;
+ }
+ if (!(obj instanceof com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb)) {
+ return super.equals(obj);
+ }
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb other =
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb) obj;
+
+ if (!getRetrievalStrategyCase().equals(other.getRetrievalStrategyCase())) return false;
+ switch (retrievalStrategyCase_) {
+ case 1:
+ if (!getKnn().equals(other.getKnn())) return false;
+ break;
+ case 2:
+ if (!getAnn().equals(other.getAnn())) return false;
+ break;
+ case 0:
+ default:
+ }
+ if (!getUnknownFields().equals(other.getUnknownFields())) return false;
+ return true;
+ }
+
+ @java.lang.Override
+ public int hashCode() {
+ if (memoizedHashCode != 0) {
+ return memoizedHashCode;
+ }
+ int hash = 41;
+ hash = (19 * hash) + getDescriptor().hashCode();
+ switch (retrievalStrategyCase_) {
+ case 1:
+ hash = (37 * hash) + KNN_FIELD_NUMBER;
+ hash = (53 * hash) + getKnn().hashCode();
+ break;
+ case 2:
+ hash = (37 * hash) + ANN_FIELD_NUMBER;
+ hash = (53 * hash) + getAnn().hashCode();
+ break;
+ case 0:
+ default:
+ }
+ hash = (29 * hash) + getUnknownFields().hashCode();
+ memoizedHashCode = hash;
+ return hash;
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ java.nio.ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ com.google.protobuf.ByteString data)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ com.google.protobuf.ByteString data,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ byte[] data) throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb
+ parseDelimitedFrom(java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb
+ parseDelimitedFrom(
+ java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ com.google.protobuf.CodedInputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb parseFrom(
+ com.google.protobuf.CodedInputStream input,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ @java.lang.Override
+ public Builder newBuilderForType() {
+ return newBuilder();
+ }
+
+ public static Builder newBuilder() {
+ return DEFAULT_INSTANCE.toBuilder();
+ }
+
+ public static Builder newBuilder(
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb prototype) {
+ return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
+ }
+
+ @java.lang.Override
+ public Builder toBuilder() {
+ return this == DEFAULT_INSTANCE ? new Builder() : new Builder().mergeFrom(this);
+ }
+
+ @java.lang.Override
+ protected Builder newBuilderForType(
+ com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
+ Builder builder = new Builder(parent);
+ return builder;
+ }
+
+ /**
+ *
+ *
+ * + * The config for the default RAG-managed Vector DB. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb} + */ + public static final class Builder + extends com.google.protobuf.GeneratedMessageV3.Builder
+ * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return Whether the knn field is set.
+ */
+ @java.lang.Override
+ public boolean hasKnn() {
+ return retrievalStrategyCase_ == 1;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ *
+ * @return The knn.
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN getKnn() {
+ if (knnBuilder_ == null) {
+ if (retrievalStrategyCase_ == 1) {
+ return (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ } else {
+ if (retrievalStrategyCase_ == 1) {
+ return knnBuilder_.getMessage();
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public Builder setKnn(
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN value) {
+ if (knnBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ retrievalStrategy_ = value;
+ onChanged();
+ } else {
+ knnBuilder_.setMessage(value);
+ }
+ retrievalStrategyCase_ = 1;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public Builder setKnn(
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN.Builder
+ builderForValue) {
+ if (knnBuilder_ == null) {
+ retrievalStrategy_ = builderForValue.build();
+ onChanged();
+ } else {
+ knnBuilder_.setMessage(builderForValue.build());
+ }
+ retrievalStrategyCase_ = 1;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public Builder mergeKnn(
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN value) {
+ if (knnBuilder_ == null) {
+ if (retrievalStrategyCase_ == 1
+ && retrievalStrategy_
+ != com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance()) {
+ retrievalStrategy_ =
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN
+ .newBuilder(
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_)
+ .mergeFrom(value)
+ .buildPartial();
+ } else {
+ retrievalStrategy_ = value;
+ }
+ onChanged();
+ } else {
+ if (retrievalStrategyCase_ == 1) {
+ knnBuilder_.mergeFrom(value);
+ } else {
+ knnBuilder_.setMessage(value);
+ }
+ }
+ retrievalStrategyCase_ = 1;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public Builder clearKnn() {
+ if (knnBuilder_ == null) {
+ if (retrievalStrategyCase_ == 1) {
+ retrievalStrategyCase_ = 0;
+ retrievalStrategy_ = null;
+ onChanged();
+ }
+ } else {
+ if (retrievalStrategyCase_ == 1) {
+ retrievalStrategyCase_ = 0;
+ retrievalStrategy_ = null;
+ }
+ knnBuilder_.clear();
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN.Builder
+ getKnnBuilder() {
+ return getKnnFieldBuilder().getBuilder();
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder
+ getKnnOrBuilder() {
+ if ((retrievalStrategyCase_ == 1) && (knnBuilder_ != null)) {
+ return knnBuilder_.getMessageOrBuilder();
+ } else {
+ if (retrievalStrategyCase_ == 1) {
+ return (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Performs a KNN search on RagCorpus. + * Default choice if not specified. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN knn = 1;
+ */
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN.Builder,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder>
+ getKnnFieldBuilder() {
+ if (knnBuilder_ == null) {
+ if (!(retrievalStrategyCase_ == 1)) {
+ retrievalStrategy_ =
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN
+ .getDefaultInstance();
+ }
+ knnBuilder_ =
+ new com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN.Builder,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNNOrBuilder>(
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNN)
+ retrievalStrategy_,
+ getParentForChildren(),
+ isClean());
+ retrievalStrategy_ = null;
+ }
+ retrievalStrategyCase_ = 1;
+ onChanged();
+ return knnBuilder_;
+ }
+
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN.Builder,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder>
+ annBuilder_;
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return Whether the ann field is set.
+ */
+ @java.lang.Override
+ public boolean hasAnn() {
+ return retrievalStrategyCase_ == 2;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ *
+ * @return The ann.
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN getAnn() {
+ if (annBuilder_ == null) {
+ if (retrievalStrategyCase_ == 2) {
+ return (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ } else {
+ if (retrievalStrategyCase_ == 2) {
+ return annBuilder_.getMessage();
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public Builder setAnn(
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN value) {
+ if (annBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ retrievalStrategy_ = value;
+ onChanged();
+ } else {
+ annBuilder_.setMessage(value);
+ }
+ retrievalStrategyCase_ = 2;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public Builder setAnn(
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN.Builder
+ builderForValue) {
+ if (annBuilder_ == null) {
+ retrievalStrategy_ = builderForValue.build();
+ onChanged();
+ } else {
+ annBuilder_.setMessage(builderForValue.build());
+ }
+ retrievalStrategyCase_ = 2;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public Builder mergeAnn(
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN value) {
+ if (annBuilder_ == null) {
+ if (retrievalStrategyCase_ == 2
+ && retrievalStrategy_
+ != com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance()) {
+ retrievalStrategy_ =
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ .newBuilder(
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_)
+ .mergeFrom(value)
+ .buildPartial();
+ } else {
+ retrievalStrategy_ = value;
+ }
+ onChanged();
+ } else {
+ if (retrievalStrategyCase_ == 2) {
+ annBuilder_.mergeFrom(value);
+ } else {
+ annBuilder_.setMessage(value);
+ }
+ }
+ retrievalStrategyCase_ = 2;
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public Builder clearAnn() {
+ if (annBuilder_ == null) {
+ if (retrievalStrategyCase_ == 2) {
+ retrievalStrategyCase_ = 0;
+ retrievalStrategy_ = null;
+ onChanged();
+ }
+ } else {
+ if (retrievalStrategyCase_ == 2) {
+ retrievalStrategyCase_ = 0;
+ retrievalStrategy_ = null;
+ }
+ annBuilder_.clear();
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN.Builder
+ getAnnBuilder() {
+ return getAnnFieldBuilder().getBuilder();
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder
+ getAnnOrBuilder() {
+ if ((retrievalStrategyCase_ == 2) && (annBuilder_ != null)) {
+ return annBuilder_.getMessageOrBuilder();
+ } else {
+ if (retrievalStrategyCase_ == 2) {
+ return (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_;
+ }
+ return com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Performs an ANN search on RagCorpus. Use this if you have a lot of + * files (> 10K) in your RagCorpus and want to reduce the search latency. + *+ * + *
.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN ann = 2;
+ */
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN.Builder,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder>
+ getAnnFieldBuilder() {
+ if (annBuilder_ == null) {
+ if (!(retrievalStrategyCase_ == 2)) {
+ retrievalStrategy_ =
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN
+ .getDefaultInstance();
+ }
+ annBuilder_ =
+ new com.google.protobuf.SingleFieldBuilderV3<
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN.Builder,
+ com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANNOrBuilder>(
+ (com.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANN)
+ retrievalStrategy_,
+ getParentForChildren(),
+ isClean());
+ retrievalStrategy_ = null;
+ }
+ retrievalStrategyCase_ = 2;
+ onChanged();
+ return annBuilder_;
+ }
+
@java.lang.Override
public final Builder setUnknownFields(
final com.google.protobuf.UnknownFieldSet unknownFields) {
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/Schema.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/Schema.java
index 0a5d0e4b7c3d..0063ecf0334c 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/Schema.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/Schema.java
@@ -1127,6 +1127,68 @@ public com.google.cloud.aiplatform.v1beta1.SchemaOrBuilder getAnyOfOrBuilder(int
return anyOf_.get(index);
}
+ public static final int ADDITIONAL_PROPERTIES_FIELD_NUMBER = 26;
+ private com.google.protobuf.Value additionalProperties_;
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return Whether the additionalProperties field is set.
+ */
+ @java.lang.Override
+ public boolean hasAdditionalProperties() {
+ return ((bitField0_ & 0x00000008) != 0);
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return The additionalProperties.
+ */
+ @java.lang.Override
+ public com.google.protobuf.Value getAdditionalProperties() {
+ return additionalProperties_ == null
+ ? com.google.protobuf.Value.getDefaultInstance()
+ : additionalProperties_;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ @java.lang.Override
+ public com.google.protobuf.ValueOrBuilder getAdditionalPropertiesOrBuilder() {
+ return additionalProperties_ == null
+ ? com.google.protobuf.Value.getDefaultInstance()
+ : additionalProperties_;
+ }
+
public static final int REF_FIELD_NUMBER = 27;
@SuppressWarnings("serial")
@@ -1425,6 +1487,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
for (int i = 0; i < propertyOrdering_.size(); i++) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 25, propertyOrdering_.getRaw(i));
}
+ if (((bitField0_ & 0x00000008) != 0)) {
+ output.writeMessage(26, getAdditionalProperties());
+ }
if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(ref_)) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 27, ref_);
}
@@ -1528,6 +1593,10 @@ public int getSerializedSize() {
size += dataSize;
size += 2 * getPropertyOrderingList().size();
}
+ if (((bitField0_ & 0x00000008) != 0)) {
+ size +=
+ com.google.protobuf.CodedOutputStream.computeMessageSize(26, getAdditionalProperties());
+ }
if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(ref_)) {
size += com.google.protobuf.GeneratedMessageV3.computeStringSize(27, ref_);
}
@@ -1591,6 +1660,10 @@ public boolean equals(final java.lang.Object obj) {
if (!getExample().equals(other.getExample())) return false;
}
if (!getAnyOfList().equals(other.getAnyOfList())) return false;
+ if (hasAdditionalProperties() != other.hasAdditionalProperties()) return false;
+ if (hasAdditionalProperties()) {
+ if (!getAdditionalProperties().equals(other.getAdditionalProperties())) return false;
+ }
if (!getRef().equals(other.getRef())) return false;
if (!internalGetDefs().equals(other.internalGetDefs())) return false;
if (!getUnknownFields().equals(other.getUnknownFields())) return false;
@@ -1670,6 +1743,10 @@ public int hashCode() {
hash = (37 * hash) + ANY_OF_FIELD_NUMBER;
hash = (53 * hash) + getAnyOfList().hashCode();
}
+ if (hasAdditionalProperties()) {
+ hash = (37 * hash) + ADDITIONAL_PROPERTIES_FIELD_NUMBER;
+ hash = (53 * hash) + getAdditionalProperties().hashCode();
+ }
hash = (37 * hash) + REF_FIELD_NUMBER;
hash = (53 * hash) + getRef().hashCode();
if (!internalGetDefs().getMap().isEmpty()) {
@@ -1850,6 +1927,7 @@ private void maybeForceBuilderInitialization() {
getItemsFieldBuilder();
getExampleFieldBuilder();
getAnyOfFieldBuilder();
+ getAdditionalPropertiesFieldBuilder();
}
}
@@ -1897,6 +1975,11 @@ public Builder clear() {
anyOfBuilder_.clear();
}
bitField0_ = (bitField0_ & ~0x00200000);
+ additionalProperties_ = null;
+ if (additionalPropertiesBuilder_ != null) {
+ additionalPropertiesBuilder_.dispose();
+ additionalPropertiesBuilder_ = null;
+ }
ref_ = "";
internalGetMutableDefs().clear();
return this;
@@ -2020,9 +2103,16 @@ private void buildPartial0(com.google.cloud.aiplatform.v1beta1.Schema result) {
to_bitField0_ |= 0x00000004;
}
if (((from_bitField0_ & 0x00400000) != 0)) {
- result.ref_ = ref_;
+ result.additionalProperties_ =
+ additionalPropertiesBuilder_ == null
+ ? additionalProperties_
+ : additionalPropertiesBuilder_.build();
+ to_bitField0_ |= 0x00000008;
}
if (((from_bitField0_ & 0x00800000) != 0)) {
+ result.ref_ = ref_;
+ }
+ if (((from_bitField0_ & 0x01000000) != 0)) {
result.defs_ = internalGetDefs().build(DefsDefaultEntryHolder.defaultEntry);
}
result.bitField0_ |= to_bitField0_;
@@ -2191,13 +2281,16 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1beta1.Schema other) {
}
}
}
+ if (other.hasAdditionalProperties()) {
+ mergeAdditionalProperties(other.getAdditionalProperties());
+ }
if (!other.getRef().isEmpty()) {
ref_ = other.ref_;
- bitField0_ |= 0x00400000;
+ bitField0_ |= 0x00800000;
onChanged();
}
internalGetMutableDefs().mergeFrom(other.internalGetDefs());
- bitField0_ |= 0x00800000;
+ bitField0_ |= 0x01000000;
this.mergeUnknownFields(other.getUnknownFields());
onChanged();
return this;
@@ -2374,10 +2467,17 @@ public Builder mergeFrom(
propertyOrdering_.add(s);
break;
} // case 202
+ case 210:
+ {
+ input.readMessage(
+ getAdditionalPropertiesFieldBuilder().getBuilder(), extensionRegistry);
+ bitField0_ |= 0x00400000;
+ break;
+ } // case 210
case 218:
{
ref_ = input.readStringRequireUtf8();
- bitField0_ |= 0x00400000;
+ bitField0_ |= 0x00800000;
break;
} // case 218
case 226:
@@ -2389,7 +2489,7 @@ public Builder mergeFrom(
DefsDefaultEntryHolder.defaultEntry.getParserForType(),
extensionRegistry);
internalGetMutableDefs().ensureBuilderMap().put(defs__.getKey(), defs__.getValue());
- bitField0_ |= 0x00800000;
+ bitField0_ |= 0x01000000;
break;
} // case 226
default:
@@ -5389,6 +5489,227 @@ public com.google.cloud.aiplatform.v1beta1.Schema.Builder addAnyOfBuilder(int in
return anyOfBuilder_;
}
+ private com.google.protobuf.Value additionalProperties_;
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.protobuf.Value,
+ com.google.protobuf.Value.Builder,
+ com.google.protobuf.ValueOrBuilder>
+ additionalPropertiesBuilder_;
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return Whether the additionalProperties field is set.
+ */
+ public boolean hasAdditionalProperties() {
+ return ((bitField0_ & 0x00400000) != 0);
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return The additionalProperties.
+ */
+ public com.google.protobuf.Value getAdditionalProperties() {
+ if (additionalPropertiesBuilder_ == null) {
+ return additionalProperties_ == null
+ ? com.google.protobuf.Value.getDefaultInstance()
+ : additionalProperties_;
+ } else {
+ return additionalPropertiesBuilder_.getMessage();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder setAdditionalProperties(com.google.protobuf.Value value) {
+ if (additionalPropertiesBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ additionalProperties_ = value;
+ } else {
+ additionalPropertiesBuilder_.setMessage(value);
+ }
+ bitField0_ |= 0x00400000;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder setAdditionalProperties(com.google.protobuf.Value.Builder builderForValue) {
+ if (additionalPropertiesBuilder_ == null) {
+ additionalProperties_ = builderForValue.build();
+ } else {
+ additionalPropertiesBuilder_.setMessage(builderForValue.build());
+ }
+ bitField0_ |= 0x00400000;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder mergeAdditionalProperties(com.google.protobuf.Value value) {
+ if (additionalPropertiesBuilder_ == null) {
+ if (((bitField0_ & 0x00400000) != 0)
+ && additionalProperties_ != null
+ && additionalProperties_ != com.google.protobuf.Value.getDefaultInstance()) {
+ getAdditionalPropertiesBuilder().mergeFrom(value);
+ } else {
+ additionalProperties_ = value;
+ }
+ } else {
+ additionalPropertiesBuilder_.mergeFrom(value);
+ }
+ if (additionalProperties_ != null) {
+ bitField0_ |= 0x00400000;
+ onChanged();
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public Builder clearAdditionalProperties() {
+ bitField0_ = (bitField0_ & ~0x00400000);
+ additionalProperties_ = null;
+ if (additionalPropertiesBuilder_ != null) {
+ additionalPropertiesBuilder_.dispose();
+ additionalPropertiesBuilder_ = null;
+ }
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public com.google.protobuf.Value.Builder getAdditionalPropertiesBuilder() {
+ bitField0_ |= 0x00400000;
+ onChanged();
+ return getAdditionalPropertiesFieldBuilder().getBuilder();
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ public com.google.protobuf.ValueOrBuilder getAdditionalPropertiesOrBuilder() {
+ if (additionalPropertiesBuilder_ != null) {
+ return additionalPropertiesBuilder_.getMessageOrBuilder();
+ } else {
+ return additionalProperties_ == null
+ ? com.google.protobuf.Value.getDefaultInstance()
+ : additionalProperties_;
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ private com.google.protobuf.SingleFieldBuilderV3<
+ com.google.protobuf.Value,
+ com.google.protobuf.Value.Builder,
+ com.google.protobuf.ValueOrBuilder>
+ getAdditionalPropertiesFieldBuilder() {
+ if (additionalPropertiesBuilder_ == null) {
+ additionalPropertiesBuilder_ =
+ new com.google.protobuf.SingleFieldBuilderV3<
+ com.google.protobuf.Value,
+ com.google.protobuf.Value.Builder,
+ com.google.protobuf.ValueOrBuilder>(
+ getAdditionalProperties(), getParentForChildren(), isClean());
+ additionalProperties_ = null;
+ }
+ return additionalPropertiesBuilder_;
+ }
+
private java.lang.Object ref_ = "";
/**
@@ -5514,7 +5835,7 @@ public Builder setRef(java.lang.String value) {
throw new NullPointerException();
}
ref_ = value;
- bitField0_ |= 0x00400000;
+ bitField0_ |= 0x00800000;
onChanged();
return this;
}
@@ -5552,7 +5873,7 @@ public Builder setRef(java.lang.String value) {
*/
public Builder clearRef() {
ref_ = getDefaultInstance().getRef();
- bitField0_ = (bitField0_ & ~0x00400000);
+ bitField0_ = (bitField0_ & ~0x00800000);
onChanged();
return this;
}
@@ -5595,7 +5916,7 @@ public Builder setRefBytes(com.google.protobuf.ByteString value) {
}
checkByteStringIsUtf8(value);
ref_ = value;
- bitField0_ |= 0x00400000;
+ bitField0_ |= 0x00800000;
onChanged();
return this;
}
@@ -5653,7 +5974,7 @@ public com.google.cloud.aiplatform.v1beta1.Schema build(
if (defs_ == null) {
defs_ = new com.google.protobuf.MapFieldBuilder<>(defsConverter);
}
- bitField0_ |= 0x00800000;
+ bitField0_ |= 0x01000000;
onChanged();
return defs_;
}
@@ -5758,7 +6079,7 @@ public com.google.cloud.aiplatform.v1beta1.Schema getDefsOrThrow(java.lang.Strin
}
public Builder clearDefs() {
- bitField0_ = (bitField0_ & ~0x00800000);
+ bitField0_ = (bitField0_ & ~0x01000000);
internalGetMutableDefs().clear();
return this;
}
@@ -5787,7 +6108,7 @@ public Builder removeDefs(java.lang.String key) {
@java.lang.Deprecated
public java.util.Map+ * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return Whether the additionalProperties field is set.
+ */
+ boolean hasAdditionalProperties();
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ *
+ * @return The additionalProperties.
+ */
+ com.google.protobuf.Value getAdditionalProperties();
+
+ /**
+ *
+ *
+ * + * Optional. Can either be a boolean or an object; controls the presence of + * additional properties. + *+ * + *
+ * .google.protobuf.Value additional_properties = 26 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ */
+ com.google.protobuf.ValueOrBuilder getAdditionalPropertiesOrBuilder();
+
/**
*
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedHyperParameters.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedHyperParameters.java
index 025fe361b6b3..d6d06b1c74d3 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedHyperParameters.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedHyperParameters.java
@@ -94,6 +94,16 @@ public enum AdapterSize implements com.google.protobuf.ProtocolMessageEnum {
* ADAPTER_SIZE_ONE = 1;
*/
ADAPTER_SIZE_ONE(1),
+ /**
+ *
+ *
+ * + * Adapter size 2. + *+ * + *
ADAPTER_SIZE_TWO = 6;
+ */
+ ADAPTER_SIZE_TWO(6),
/**
*
*
@@ -124,6 +134,16 @@ public enum AdapterSize implements com.google.protobuf.ProtocolMessageEnum {
* ADAPTER_SIZE_SIXTEEN = 4;
*/
ADAPTER_SIZE_SIXTEEN(4),
+ /**
+ *
+ *
+ * + * Adapter size 32. + *+ * + *
ADAPTER_SIZE_THIRTY_TWO = 5;
+ */
+ ADAPTER_SIZE_THIRTY_TWO(5),
UNRECOGNIZED(-1),
;
@@ -149,6 +169,17 @@ public enum AdapterSize implements com.google.protobuf.ProtocolMessageEnum {
*/
public static final int ADAPTER_SIZE_ONE_VALUE = 1;
+ /**
+ *
+ *
+ * + * Adapter size 2. + *+ * + *
ADAPTER_SIZE_TWO = 6;
+ */
+ public static final int ADAPTER_SIZE_TWO_VALUE = 6;
+
/**
*
*
@@ -182,6 +213,17 @@ public enum AdapterSize implements com.google.protobuf.ProtocolMessageEnum {
*/
public static final int ADAPTER_SIZE_SIXTEEN_VALUE = 4;
+ /**
+ *
+ *
+ * + * Adapter size 32. + *+ * + *
ADAPTER_SIZE_THIRTY_TWO = 5;
+ */
+ public static final int ADAPTER_SIZE_THIRTY_TWO_VALUE = 5;
+
public final int getNumber() {
if (this == UNRECOGNIZED) {
throw new java.lang.IllegalArgumentException(
@@ -210,12 +252,16 @@ public static AdapterSize forNumber(int value) {
return ADAPTER_SIZE_UNSPECIFIED;
case 1:
return ADAPTER_SIZE_ONE;
+ case 6:
+ return ADAPTER_SIZE_TWO;
case 2:
return ADAPTER_SIZE_FOUR;
case 3:
return ADAPTER_SIZE_EIGHT;
case 4:
return ADAPTER_SIZE_SIXTEEN;
+ case 5:
+ return ADAPTER_SIZE_THIRTY_TWO;
default:
return null;
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningDataStats.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningDataStats.java
index 9f1a766729b7..dc25fca95ae9 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningDataStats.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningDataStats.java
@@ -42,6 +42,7 @@ private SupervisedTuningDataStats(com.google.protobuf.GeneratedMessageV3.Builder
private SupervisedTuningDataStats() {
userDatasetExamples_ = java.util.Collections.emptyList();
truncatedExampleIndices_ = emptyLongList();
+ droppedExampleReasons_ = com.google.protobuf.LazyStringArrayList.emptyList();
}
@java.lang.Override
@@ -122,7 +123,7 @@ public long getTotalTuningCharacterCount() {
*
* @deprecated
* google.cloud.aiplatform.v1beta1.SupervisedTuningDataStats.total_billable_character_count is
- * deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=238
+ * deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=245
* @return The totalBillableCharacterCount.
*/
@java.lang.Override
@@ -455,11 +456,13 @@ public com.google.cloud.aiplatform.v1beta1.ContentOrBuilder getUserDatasetExampl
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The totalTruncatedExampleCount.
*/
@@ -477,11 +480,13 @@ public long getTotalTruncatedExampleCount() {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return A list containing the truncatedExampleIndices.
*/
@@ -494,11 +499,13 @@ public java.util.List
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The count of truncatedExampleIndices.
*/
@@ -510,11 +517,13 @@ public int getTruncatedExampleIndicesCount() {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param index The index of the element to return.
* @return The truncatedExampleIndices at the given index.
@@ -525,6 +534,86 @@ public long getTruncatedExampleIndices(int index) {
private int truncatedExampleIndicesMemoizedSerializedSize = -1;
+ public static final int DROPPED_EXAMPLE_REASONS_FIELD_NUMBER = 12;
+
+ @SuppressWarnings("serial")
+ private com.google.protobuf.LazyStringArrayList droppedExampleReasons_ =
+ com.google.protobuf.LazyStringArrayList.emptyList();
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return A list containing the droppedExampleReasons.
+ */
+ public com.google.protobuf.ProtocolStringList getDroppedExampleReasonsList() {
+ return droppedExampleReasons_;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return The count of droppedExampleReasons.
+ */
+ public int getDroppedExampleReasonsCount() {
+ return droppedExampleReasons_.size();
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the element to return.
+ * @return The droppedExampleReasons at the given index.
+ */
+ public java.lang.String getDroppedExampleReasons(int index) {
+ return droppedExampleReasons_.get(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the value to return.
+ * @return The bytes of the droppedExampleReasons at the given index.
+ */
+ public com.google.protobuf.ByteString getDroppedExampleReasonsBytes(int index) {
+ return droppedExampleReasons_.getByteString(index);
+ }
+
private byte memoizedIsInitialized = -1;
@java.lang.Override
@@ -577,6 +666,10 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
for (int i = 0; i < truncatedExampleIndices_.size(); i++) {
output.writeInt64NoTag(truncatedExampleIndices_.getLong(i));
}
+ for (int i = 0; i < droppedExampleReasons_.size(); i++) {
+ com.google.protobuf.GeneratedMessageV3.writeString(
+ output, 12, droppedExampleReasons_.getRaw(i));
+ }
getUnknownFields().writeTo(output);
}
@@ -639,6 +732,14 @@ public int getSerializedSize() {
}
truncatedExampleIndicesMemoizedSerializedSize = dataSize;
}
+ {
+ int dataSize = 0;
+ for (int i = 0; i < droppedExampleReasons_.size(); i++) {
+ dataSize += computeStringSizeNoTag(droppedExampleReasons_.getRaw(i));
+ }
+ size += dataSize;
+ size += 1 * getDroppedExampleReasonsList().size();
+ }
size += getUnknownFields().getSerializedSize();
memoizedSize = size;
return size;
@@ -680,6 +781,7 @@ public boolean equals(final java.lang.Object obj) {
if (getTotalTruncatedExampleCount() != other.getTotalTruncatedExampleCount()) return false;
if (!getTruncatedExampleIndicesList().equals(other.getTruncatedExampleIndicesList()))
return false;
+ if (!getDroppedExampleReasonsList().equals(other.getDroppedExampleReasonsList())) return false;
if (!getUnknownFields().equals(other.getUnknownFields())) return false;
return true;
}
@@ -723,6 +825,10 @@ public int hashCode() {
hash = (37 * hash) + TRUNCATED_EXAMPLE_INDICES_FIELD_NUMBER;
hash = (53 * hash) + getTruncatedExampleIndicesList().hashCode();
}
+ if (getDroppedExampleReasonsCount() > 0) {
+ hash = (37 * hash) + DROPPED_EXAMPLE_REASONS_FIELD_NUMBER;
+ hash = (53 * hash) + getDroppedExampleReasonsList().hashCode();
+ }
hash = (29 * hash) + getUnknownFields().hashCode();
memoizedHashCode = hash;
return hash;
@@ -905,6 +1011,7 @@ public Builder clear() {
bitField0_ = (bitField0_ & ~0x00000100);
totalTruncatedExampleCount_ = 0L;
truncatedExampleIndices_ = emptyLongList();
+ droppedExampleReasons_ = com.google.protobuf.LazyStringArrayList.emptyList();
return this;
}
@@ -1001,6 +1108,10 @@ private void buildPartial0(
truncatedExampleIndices_.makeImmutable();
result.truncatedExampleIndices_ = truncatedExampleIndices_;
}
+ if (((from_bitField0_ & 0x00000800) != 0)) {
+ droppedExampleReasons_.makeImmutable();
+ result.droppedExampleReasons_ = droppedExampleReasons_;
+ }
result.bitField0_ |= to_bitField0_;
}
@@ -1116,6 +1227,16 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1beta1.SupervisedTuningDat
}
onChanged();
}
+ if (!other.droppedExampleReasons_.isEmpty()) {
+ if (droppedExampleReasons_.isEmpty()) {
+ droppedExampleReasons_ = other.droppedExampleReasons_;
+ bitField0_ |= 0x00000800;
+ } else {
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.addAll(other.droppedExampleReasons_);
+ }
+ onChanged();
+ }
this.mergeUnknownFields(other.getUnknownFields());
onChanged();
return this;
@@ -1231,6 +1352,13 @@ public Builder mergeFrom(
input.popLimit(limit);
break;
} // case 90
+ case 98:
+ {
+ java.lang.String s = input.readStringRequireUtf8();
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.add(s);
+ break;
+ } // case 98
default:
{
if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -1383,7 +1511,7 @@ public Builder clearTotalTuningCharacterCount() {
*
* @deprecated
* google.cloud.aiplatform.v1beta1.SupervisedTuningDataStats.total_billable_character_count
- * is deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=238
+ * is deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=245
* @return The totalBillableCharacterCount.
*/
@java.lang.Override
@@ -1405,7 +1533,7 @@ public long getTotalBillableCharacterCount() {
*
* @deprecated
* google.cloud.aiplatform.v1beta1.SupervisedTuningDataStats.total_billable_character_count
- * is deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=238
+ * is deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=245
* @param value The totalBillableCharacterCount to set.
* @return This builder for chaining.
*/
@@ -1431,7 +1559,7 @@ public Builder setTotalBillableCharacterCount(long value) {
*
* @deprecated
* google.cloud.aiplatform.v1beta1.SupervisedTuningDataStats.total_billable_character_count
- * is deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=238
+ * is deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=245
* @return This builder for chaining.
*/
@java.lang.Deprecated
@@ -2649,11 +2777,13 @@ public com.google.cloud.aiplatform.v1beta1.Content.Builder addUserDatasetExample
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The totalTruncatedExampleCount.
*/
@@ -2666,11 +2796,13 @@ public long getTotalTruncatedExampleCount() {
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param value The totalTruncatedExampleCount to set.
* @return This builder for chaining.
@@ -2687,11 +2819,13 @@ public Builder setTotalTruncatedExampleCount(long value) {
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return This builder for chaining.
*/
@@ -2715,11 +2849,13 @@ private void ensureTruncatedExampleIndicesIsMutable() {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return A list containing the truncatedExampleIndices.
*/
@@ -2732,11 +2868,13 @@ public java.util.List
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The count of truncatedExampleIndices.
*/
@@ -2748,11 +2886,13 @@ public int getTruncatedExampleIndicesCount() {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param index The index of the element to return.
* @return The truncatedExampleIndices at the given index.
@@ -2765,11 +2905,13 @@ public long getTruncatedExampleIndices(int index) {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param index The index to set the value at.
* @param value The truncatedExampleIndices to set.
@@ -2788,11 +2930,13 @@ public Builder setTruncatedExampleIndices(int index, long value) {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param value The truncatedExampleIndices to add.
* @return This builder for chaining.
@@ -2810,11 +2954,13 @@ public Builder addTruncatedExampleIndices(long value) {
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param values The truncatedExampleIndices to add.
* @return This builder for chaining.
@@ -2832,11 +2978,13 @@ public Builder addAllTruncatedExampleIndices(
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return This builder for chaining.
*/
@@ -2847,6 +2995,217 @@ public Builder clearTruncatedExampleIndices() {
return this;
}
+ private com.google.protobuf.LazyStringArrayList droppedExampleReasons_ =
+ com.google.protobuf.LazyStringArrayList.emptyList();
+
+ private void ensureDroppedExampleReasonsIsMutable() {
+ if (!droppedExampleReasons_.isModifiable()) {
+ droppedExampleReasons_ =
+ new com.google.protobuf.LazyStringArrayList(droppedExampleReasons_);
+ }
+ bitField0_ |= 0x00000800;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return A list containing the droppedExampleReasons.
+ */
+ public com.google.protobuf.ProtocolStringList getDroppedExampleReasonsList() {
+ droppedExampleReasons_.makeImmutable();
+ return droppedExampleReasons_;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return The count of droppedExampleReasons.
+ */
+ public int getDroppedExampleReasonsCount() {
+ return droppedExampleReasons_.size();
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the element to return.
+ * @return The droppedExampleReasons at the given index.
+ */
+ public java.lang.String getDroppedExampleReasons(int index) {
+ return droppedExampleReasons_.get(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the value to return.
+ * @return The bytes of the droppedExampleReasons at the given index.
+ */
+ public com.google.protobuf.ByteString getDroppedExampleReasonsBytes(int index) {
+ return droppedExampleReasons_.getByteString(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index to set the value at.
+ * @param value The droppedExampleReasons to set.
+ * @return This builder for chaining.
+ */
+ public Builder setDroppedExampleReasons(int index, java.lang.String value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.set(index, value);
+ bitField0_ |= 0x00000800;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param value The droppedExampleReasons to add.
+ * @return This builder for chaining.
+ */
+ public Builder addDroppedExampleReasons(java.lang.String value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.add(value);
+ bitField0_ |= 0x00000800;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param values The droppedExampleReasons to add.
+ * @return This builder for chaining.
+ */
+ public Builder addAllDroppedExampleReasons(java.lang.Iterable+ * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearDroppedExampleReasons() {
+ droppedExampleReasons_ = com.google.protobuf.LazyStringArrayList.emptyList();
+ bitField0_ = (bitField0_ & ~0x00000800);
+ ;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param value The bytes of the droppedExampleReasons to add.
+ * @return This builder for chaining.
+ */
+ public Builder addDroppedExampleReasonsBytes(com.google.protobuf.ByteString value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ checkByteStringIsUtf8(value);
+ ensureDroppedExampleReasonsIsMutable();
+ droppedExampleReasons_.add(value);
+ bitField0_ |= 0x00000800;
+ onChanged();
+ return this;
+ }
+
@java.lang.Override
public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.setUnknownFields(unknownFields);
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningDataStatsOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningDataStatsOrBuilder.java
index 7b99232e38dc..0b68898dd3bc 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningDataStatsOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningDataStatsOrBuilder.java
@@ -65,7 +65,7 @@ public interface SupervisedTuningDataStatsOrBuilder
*
* @deprecated
* google.cloud.aiplatform.v1beta1.SupervisedTuningDataStats.total_billable_character_count is
- * deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=238
+ * deprecated. See google/cloud/aiplatform/v1beta1/tuning_job.proto;l=245
* @return The totalBillableCharacterCount.
*/
@java.lang.Deprecated
@@ -302,11 +302,13 @@ public interface SupervisedTuningDataStatsOrBuilder
*
*
*
- * The number of examples in the dataset that have been truncated by any
- * amount.
+ * Output only. The number of examples in the dataset that have been dropped.
+ * An example can be dropped for reasons including: too many tokens, contains
+ * an invalid image, contains too many images, etc.
*
*
- * int64 total_truncated_example_count = 10;
+ * int64 total_truncated_example_count = 10 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The totalTruncatedExampleCount.
*/
@@ -316,11 +318,13 @@ public interface SupervisedTuningDataStatsOrBuilder
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return A list containing the truncatedExampleIndices.
*/
@@ -330,11 +334,13 @@ public interface SupervisedTuningDataStatsOrBuilder
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @return The count of truncatedExampleIndices.
*/
@@ -344,14 +350,82 @@ public interface SupervisedTuningDataStatsOrBuilder
*
*
*
- * A partial sample of the indices (starting from 1) of the truncated
- * examples.
+ * Output only. A partial sample of the indices (starting from 1) of the
+ * dropped examples.
*
*
- * repeated int64 truncated_example_indices = 11;
+ *
+ * repeated int64 truncated_example_indices = 11 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
*
* @param index The index of the element to return.
* @return The truncatedExampleIndices at the given index.
*/
long getTruncatedExampleIndices(int index);
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return A list containing the droppedExampleReasons.
+ */
+ java.util.List+ * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @return The count of droppedExampleReasons.
+ */
+ int getDroppedExampleReasonsCount();
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the element to return.
+ * @return The droppedExampleReasons at the given index.
+ */
+ java.lang.String getDroppedExampleReasons(int index);
+
+ /**
+ *
+ *
+ * + * Output only. For each index in `truncated_example_indices`, the user-facing + * reason why the example was dropped. + *+ * + *
+ * repeated string dropped_example_reasons = 12 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ *
+ * @param index The index of the value to return.
+ * @return The bytes of the droppedExampleReasons at the given index.
+ */
+ com.google.protobuf.ByteString getDroppedExampleReasonsBytes(int index);
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningSpec.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningSpec.java
index 70b210426dfb..d91bd9199382 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningSpec.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningSpec.java
@@ -236,6 +236,27 @@ public com.google.cloud.aiplatform.v1beta1.SupervisedHyperParameters getHyperPar
: hyperParameters_;
}
+ public static final int EXPORT_LAST_CHECKPOINT_ONLY_FIELD_NUMBER = 6;
+ private boolean exportLastCheckpointOnly_ = false;
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @return The exportLastCheckpointOnly.
+ */
+ @java.lang.Override
+ public boolean getExportLastCheckpointOnly() {
+ return exportLastCheckpointOnly_;
+ }
+
private byte memoizedIsInitialized = -1;
@java.lang.Override
@@ -259,6 +280,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
if (((bitField0_ & 0x00000001) != 0)) {
output.writeMessage(3, getHyperParameters());
}
+ if (exportLastCheckpointOnly_ != false) {
+ output.writeBool(6, exportLastCheckpointOnly_);
+ }
getUnknownFields().writeTo(output);
}
@@ -277,6 +301,9 @@ public int getSerializedSize() {
if (((bitField0_ & 0x00000001) != 0)) {
size += com.google.protobuf.CodedOutputStream.computeMessageSize(3, getHyperParameters());
}
+ if (exportLastCheckpointOnly_ != false) {
+ size += com.google.protobuf.CodedOutputStream.computeBoolSize(6, exportLastCheckpointOnly_);
+ }
size += getUnknownFields().getSerializedSize();
memoizedSize = size;
return size;
@@ -299,6 +326,7 @@ public boolean equals(final java.lang.Object obj) {
if (hasHyperParameters()) {
if (!getHyperParameters().equals(other.getHyperParameters())) return false;
}
+ if (getExportLastCheckpointOnly() != other.getExportLastCheckpointOnly()) return false;
if (!getUnknownFields().equals(other.getUnknownFields())) return false;
return true;
}
@@ -318,6 +346,8 @@ public int hashCode() {
hash = (37 * hash) + HYPER_PARAMETERS_FIELD_NUMBER;
hash = (53 * hash) + getHyperParameters().hashCode();
}
+ hash = (37 * hash) + EXPORT_LAST_CHECKPOINT_ONLY_FIELD_NUMBER;
+ hash = (53 * hash) + com.google.protobuf.Internal.hashBoolean(getExportLastCheckpointOnly());
hash = (29 * hash) + getUnknownFields().hashCode();
memoizedHashCode = hash;
return hash;
@@ -475,6 +505,7 @@ public Builder clear() {
hyperParametersBuilder_.dispose();
hyperParametersBuilder_ = null;
}
+ exportLastCheckpointOnly_ = false;
return this;
}
@@ -523,6 +554,9 @@ private void buildPartial0(com.google.cloud.aiplatform.v1beta1.SupervisedTuningS
hyperParametersBuilder_ == null ? hyperParameters_ : hyperParametersBuilder_.build();
to_bitField0_ |= 0x00000001;
}
+ if (((from_bitField0_ & 0x00000008) != 0)) {
+ result.exportLastCheckpointOnly_ = exportLastCheckpointOnly_;
+ }
result.bitField0_ |= to_bitField0_;
}
@@ -585,6 +619,9 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1beta1.SupervisedTuningSpe
if (other.hasHyperParameters()) {
mergeHyperParameters(other.getHyperParameters());
}
+ if (other.getExportLastCheckpointOnly() != false) {
+ setExportLastCheckpointOnly(other.getExportLastCheckpointOnly());
+ }
this.mergeUnknownFields(other.getUnknownFields());
onChanged();
return this;
@@ -629,6 +666,12 @@ public Builder mergeFrom(
bitField0_ |= 0x00000004;
break;
} // case 26
+ case 48:
+ {
+ exportLastCheckpointOnly_ = input.readBool();
+ bitField0_ |= 0x00000008;
+ break;
+ } // case 48
default:
{
if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -1099,6 +1142,68 @@ public Builder clearHyperParameters() {
return hyperParametersBuilder_;
}
+ private boolean exportLastCheckpointOnly_;
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @return The exportLastCheckpointOnly.
+ */
+ @java.lang.Override
+ public boolean getExportLastCheckpointOnly() {
+ return exportLastCheckpointOnly_;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @param value The exportLastCheckpointOnly to set.
+ * @return This builder for chaining.
+ */
+ public Builder setExportLastCheckpointOnly(boolean value) {
+
+ exportLastCheckpointOnly_ = value;
+ bitField0_ |= 0x00000008;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearExportLastCheckpointOnly() {
+ bitField0_ = (bitField0_ & ~0x00000008);
+ exportLastCheckpointOnly_ = false;
+ onChanged();
+ return this;
+ }
+
@java.lang.Override
public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.setUnknownFields(unknownFields);
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningSpecOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningSpecOrBuilder.java
index a5c29f835a61..388ecb7cf619 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningSpecOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/SupervisedTuningSpecOrBuilder.java
@@ -123,4 +123,19 @@ public interface SupervisedTuningSpecOrBuilder
*/
com.google.cloud.aiplatform.v1beta1.SupervisedHyperParametersOrBuilder
getHyperParametersOrBuilder();
+
+ /**
+ *
+ *
+ * + * Optional. If set to true, disable intermediate checkpoints for SFT and only + * the last checkpoint will be exported. Otherwise, enable intermediate + * checkpoints for SFT. Default is false. + *+ * + *
bool export_last_checkpoint_only = 6 [(.google.api.field_behavior) = OPTIONAL];
+ *
+ * @return The exportLastCheckpointOnly.
+ */
+ boolean getExportLastCheckpointOnly();
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TunedModel.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TunedModel.java
index 22628796d676..89ae2427c6c2 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TunedModel.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TunedModel.java
@@ -23,7 +23,7 @@
*
*
* - * The Model Registry Model and Online Prediction Endpoint assiociated with + * The Model Registry Model and Online Prediction Endpoint associated with * this [TuningJob][google.cloud.aiplatform.v1.TuningJob]. ** @@ -43,6 +43,7 @@ private TunedModel(com.google.protobuf.GeneratedMessageV3.Builder builder) { private TunedModel() { model_ = ""; endpoint_ = ""; + checkpoints_ = java.util.Collections.emptyList(); } @java.lang.Override @@ -184,6 +185,104 @@ public com.google.protobuf.ByteString getEndpointBytes() { } } + public static final int CHECKPOINTS_FIELD_NUMBER = 3; + + @SuppressWarnings("serial") + private java.util.List
+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public java.util.List+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public java.util.List
+ getCheckpointsOrBuilderList() {
+ return checkpoints_;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public int getCheckpointsCount() {
+ return checkpoints_.size();
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint getCheckpoints(int index) {
+ return checkpoints_.get(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ @java.lang.Override
+ public com.google.cloud.aiplatform.v1beta1.TunedModelCheckpointOrBuilder getCheckpointsOrBuilder(
+ int index) {
+ return checkpoints_.get(index);
+ }
+
private byte memoizedIsInitialized = -1;
@java.lang.Override
@@ -204,6 +303,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(endpoint_)) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 2, endpoint_);
}
+ for (int i = 0; i < checkpoints_.size(); i++) {
+ output.writeMessage(3, checkpoints_.get(i));
+ }
getUnknownFields().writeTo(output);
}
@@ -219,6 +321,9 @@ public int getSerializedSize() {
if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(endpoint_)) {
size += com.google.protobuf.GeneratedMessageV3.computeStringSize(2, endpoint_);
}
+ for (int i = 0; i < checkpoints_.size(); i++) {
+ size += com.google.protobuf.CodedOutputStream.computeMessageSize(3, checkpoints_.get(i));
+ }
size += getUnknownFields().getSerializedSize();
memoizedSize = size;
return size;
@@ -237,6 +342,7 @@ public boolean equals(final java.lang.Object obj) {
if (!getModel().equals(other.getModel())) return false;
if (!getEndpoint().equals(other.getEndpoint())) return false;
+ if (!getCheckpointsList().equals(other.getCheckpointsList())) return false;
if (!getUnknownFields().equals(other.getUnknownFields())) return false;
return true;
}
@@ -252,6 +358,10 @@ public int hashCode() {
hash = (53 * hash) + getModel().hashCode();
hash = (37 * hash) + ENDPOINT_FIELD_NUMBER;
hash = (53 * hash) + getEndpoint().hashCode();
+ if (getCheckpointsCount() > 0) {
+ hash = (37 * hash) + CHECKPOINTS_FIELD_NUMBER;
+ hash = (53 * hash) + getCheckpointsList().hashCode();
+ }
hash = (29 * hash) + getUnknownFields().hashCode();
memoizedHashCode = hash;
return hash;
@@ -357,7 +467,7 @@ protected Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.Build
*
*
*
- * The Model Registry Model and Online Prediction Endpoint assiociated with
+ * The Model Registry Model and Online Prediction Endpoint associated with
* this [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -395,6 +505,13 @@ public Builder clear() {
bitField0_ = 0;
model_ = "";
endpoint_ = "";
+ if (checkpointsBuilder_ == null) {
+ checkpoints_ = java.util.Collections.emptyList();
+ } else {
+ checkpoints_ = null;
+ checkpointsBuilder_.clear();
+ }
+ bitField0_ = (bitField0_ & ~0x00000004);
return this;
}
@@ -422,6 +539,7 @@ public com.google.cloud.aiplatform.v1beta1.TunedModel build() {
public com.google.cloud.aiplatform.v1beta1.TunedModel buildPartial() {
com.google.cloud.aiplatform.v1beta1.TunedModel result =
new com.google.cloud.aiplatform.v1beta1.TunedModel(this);
+ buildPartialRepeatedFields(result);
if (bitField0_ != 0) {
buildPartial0(result);
}
@@ -429,6 +547,18 @@ public com.google.cloud.aiplatform.v1beta1.TunedModel buildPartial() {
return result;
}
+ private void buildPartialRepeatedFields(com.google.cloud.aiplatform.v1beta1.TunedModel result) {
+ if (checkpointsBuilder_ == null) {
+ if (((bitField0_ & 0x00000004) != 0)) {
+ checkpoints_ = java.util.Collections.unmodifiableList(checkpoints_);
+ bitField0_ = (bitField0_ & ~0x00000004);
+ }
+ result.checkpoints_ = checkpoints_;
+ } else {
+ result.checkpoints_ = checkpointsBuilder_.build();
+ }
+ }
+
private void buildPartial0(com.google.cloud.aiplatform.v1beta1.TunedModel result) {
int from_bitField0_ = bitField0_;
if (((from_bitField0_ & 0x00000001) != 0)) {
@@ -494,6 +624,33 @@ public Builder mergeFrom(com.google.cloud.aiplatform.v1beta1.TunedModel other) {
bitField0_ |= 0x00000002;
onChanged();
}
+ if (checkpointsBuilder_ == null) {
+ if (!other.checkpoints_.isEmpty()) {
+ if (checkpoints_.isEmpty()) {
+ checkpoints_ = other.checkpoints_;
+ bitField0_ = (bitField0_ & ~0x00000004);
+ } else {
+ ensureCheckpointsIsMutable();
+ checkpoints_.addAll(other.checkpoints_);
+ }
+ onChanged();
+ }
+ } else {
+ if (!other.checkpoints_.isEmpty()) {
+ if (checkpointsBuilder_.isEmpty()) {
+ checkpointsBuilder_.dispose();
+ checkpointsBuilder_ = null;
+ checkpoints_ = other.checkpoints_;
+ bitField0_ = (bitField0_ & ~0x00000004);
+ checkpointsBuilder_ =
+ com.google.protobuf.GeneratedMessageV3.alwaysUseFieldBuilders
+ ? getCheckpointsFieldBuilder()
+ : null;
+ } else {
+ checkpointsBuilder_.addAllMessages(other.checkpoints_);
+ }
+ }
+ }
this.mergeUnknownFields(other.getUnknownFields());
onChanged();
return this;
@@ -532,6 +689,20 @@ public Builder mergeFrom(
bitField0_ |= 0x00000002;
break;
} // case 18
+ case 26:
+ {
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint m =
+ input.readMessage(
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.parser(),
+ extensionRegistry);
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(m);
+ } else {
+ checkpointsBuilder_.addMessage(m);
+ }
+ break;
+ } // case 26
default:
{
if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -803,6 +974,460 @@ public Builder setEndpointBytes(com.google.protobuf.ByteString value) {
return this;
}
+ private java.util.List+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public java.util.List+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public int getCheckpointsCount() {
+ if (checkpointsBuilder_ == null) {
+ return checkpoints_.size();
+ } else {
+ return checkpointsBuilder_.getCount();
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint getCheckpoints(int index) {
+ if (checkpointsBuilder_ == null) {
+ return checkpoints_.get(index);
+ } else {
+ return checkpointsBuilder_.getMessage(index);
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder setCheckpoints(
+ int index, com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint value) {
+ if (checkpointsBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureCheckpointsIsMutable();
+ checkpoints_.set(index, value);
+ onChanged();
+ } else {
+ checkpointsBuilder_.setMessage(index, value);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder setCheckpoints(
+ int index,
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.Builder builderForValue) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.set(index, builderForValue.build());
+ onChanged();
+ } else {
+ checkpointsBuilder_.setMessage(index, builderForValue.build());
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addCheckpoints(com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint value) {
+ if (checkpointsBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(value);
+ onChanged();
+ } else {
+ checkpointsBuilder_.addMessage(value);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addCheckpoints(
+ int index, com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint value) {
+ if (checkpointsBuilder_ == null) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(index, value);
+ onChanged();
+ } else {
+ checkpointsBuilder_.addMessage(index, value);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addCheckpoints(
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.Builder builderForValue) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(builderForValue.build());
+ onChanged();
+ } else {
+ checkpointsBuilder_.addMessage(builderForValue.build());
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addCheckpoints(
+ int index,
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.Builder builderForValue) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.add(index, builderForValue.build());
+ onChanged();
+ } else {
+ checkpointsBuilder_.addMessage(index, builderForValue.build());
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder addAllCheckpoints(
+ java.lang.Iterable
+ values) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ com.google.protobuf.AbstractMessageLite.Builder.addAll(values, checkpoints_);
+ onChanged();
+ } else {
+ checkpointsBuilder_.addAllMessages(values);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder clearCheckpoints() {
+ if (checkpointsBuilder_ == null) {
+ checkpoints_ = java.util.Collections.emptyList();
+ bitField0_ = (bitField0_ & ~0x00000004);
+ onChanged();
+ } else {
+ checkpointsBuilder_.clear();
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public Builder removeCheckpoints(int index) {
+ if (checkpointsBuilder_ == null) {
+ ensureCheckpointsIsMutable();
+ checkpoints_.remove(index);
+ onChanged();
+ } else {
+ checkpointsBuilder_.remove(index);
+ }
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.Builder getCheckpointsBuilder(
+ int index) {
+ return getCheckpointsFieldBuilder().getBuilder(index);
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1beta1.TunedModelCheckpointOrBuilder
+ getCheckpointsOrBuilder(int index) {
+ if (checkpointsBuilder_ == null) {
+ return checkpoints_.get(index);
+ } else {
+ return checkpointsBuilder_.getMessageOrBuilder(index);
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public java.util.List<
+ ? extends com.google.cloud.aiplatform.v1beta1.TunedModelCheckpointOrBuilder>
+ getCheckpointsOrBuilderList() {
+ if (checkpointsBuilder_ != null) {
+ return checkpointsBuilder_.getMessageOrBuilderList();
+ } else {
+ return java.util.Collections.unmodifiableList(checkpoints_);
+ }
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.Builder
+ addCheckpointsBuilder() {
+ return getCheckpointsFieldBuilder()
+ .addBuilder(
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.getDefaultInstance());
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.Builder addCheckpointsBuilder(
+ int index) {
+ return getCheckpointsFieldBuilder()
+ .addBuilder(
+ index, com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.getDefaultInstance());
+ }
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ public java.util.List+ * TunedModelCheckpoint for the Tuned Model of a Tuning Job. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1beta1.TunedModelCheckpoint} + */ +public final class TunedModelCheckpoint extends com.google.protobuf.GeneratedMessageV3 + implements + // @@protoc_insertion_point(message_implements:google.cloud.aiplatform.v1beta1.TunedModelCheckpoint) + TunedModelCheckpointOrBuilder { + private static final long serialVersionUID = 0L; + + // Use TunedModelCheckpoint.newBuilder() to construct. + private TunedModelCheckpoint(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + + private TunedModelCheckpoint() { + checkpointId_ = ""; + endpoint_ = ""; + } + + @java.lang.Override + @SuppressWarnings({"unused"}) + protected java.lang.Object newInstance(UnusedPrivateParameter unused) { + return new TunedModelCheckpoint(); + } + + public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() { + return com.google.cloud.aiplatform.v1beta1.TuningJobProto + .internal_static_google_cloud_aiplatform_v1beta1_TunedModelCheckpoint_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return com.google.cloud.aiplatform.v1beta1.TuningJobProto + .internal_static_google_cloud_aiplatform_v1beta1_TunedModelCheckpoint_fieldAccessorTable + .ensureFieldAccessorsInitialized( + com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.class, + com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint.Builder.class); + } + + public static final int CHECKPOINT_ID_FIELD_NUMBER = 1; + + @SuppressWarnings("serial") + private volatile java.lang.Object checkpointId_ = ""; + + /** + * + * + *
+ * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The checkpointId.
+ */
+ @java.lang.Override
+ public java.lang.String getCheckpointId() {
+ java.lang.Object ref = checkpointId_;
+ if (ref instanceof java.lang.String) {
+ return (java.lang.String) ref;
+ } else {
+ com.google.protobuf.ByteString bs = (com.google.protobuf.ByteString) ref;
+ java.lang.String s = bs.toStringUtf8();
+ checkpointId_ = s;
+ return s;
+ }
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The bytes for checkpointId.
+ */
+ @java.lang.Override
+ public com.google.protobuf.ByteString getCheckpointIdBytes() {
+ java.lang.Object ref = checkpointId_;
+ if (ref instanceof java.lang.String) {
+ com.google.protobuf.ByteString b =
+ com.google.protobuf.ByteString.copyFromUtf8((java.lang.String) ref);
+ checkpointId_ = b;
+ return b;
+ } else {
+ return (com.google.protobuf.ByteString) ref;
+ }
+ }
+
+ public static final int EPOCH_FIELD_NUMBER = 2;
+ private long epoch_ = 0L;
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @return The epoch.
+ */
+ @java.lang.Override
+ public long getEpoch() {
+ return epoch_;
+ }
+
+ public static final int STEP_FIELD_NUMBER = 3;
+ private long step_ = 0L;
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @return The step.
+ */
+ @java.lang.Override
+ public long getStep() {
+ return step_;
+ }
+
+ public static final int ENDPOINT_FIELD_NUMBER = 4;
+
+ @SuppressWarnings("serial")
+ private volatile java.lang.Object endpoint_ = "";
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The endpoint.
+ */
+ @java.lang.Override
+ public java.lang.String getEndpoint() {
+ java.lang.Object ref = endpoint_;
+ if (ref instanceof java.lang.String) {
+ return (java.lang.String) ref;
+ } else {
+ com.google.protobuf.ByteString bs = (com.google.protobuf.ByteString) ref;
+ java.lang.String s = bs.toStringUtf8();
+ endpoint_ = s;
+ return s;
+ }
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The bytes for endpoint.
+ */
+ @java.lang.Override
+ public com.google.protobuf.ByteString getEndpointBytes() {
+ java.lang.Object ref = endpoint_;
+ if (ref instanceof java.lang.String) {
+ com.google.protobuf.ByteString b =
+ com.google.protobuf.ByteString.copyFromUtf8((java.lang.String) ref);
+ endpoint_ = b;
+ return b;
+ } else {
+ return (com.google.protobuf.ByteString) ref;
+ }
+ }
+
+ private byte memoizedIsInitialized = -1;
+
+ @java.lang.Override
+ public final boolean isInitialized() {
+ byte isInitialized = memoizedIsInitialized;
+ if (isInitialized == 1) return true;
+ if (isInitialized == 0) return false;
+
+ memoizedIsInitialized = 1;
+ return true;
+ }
+
+ @java.lang.Override
+ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException {
+ if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(checkpointId_)) {
+ com.google.protobuf.GeneratedMessageV3.writeString(output, 1, checkpointId_);
+ }
+ if (epoch_ != 0L) {
+ output.writeInt64(2, epoch_);
+ }
+ if (step_ != 0L) {
+ output.writeInt64(3, step_);
+ }
+ if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(endpoint_)) {
+ com.google.protobuf.GeneratedMessageV3.writeString(output, 4, endpoint_);
+ }
+ getUnknownFields().writeTo(output);
+ }
+
+ @java.lang.Override
+ public int getSerializedSize() {
+ int size = memoizedSize;
+ if (size != -1) return size;
+
+ size = 0;
+ if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(checkpointId_)) {
+ size += com.google.protobuf.GeneratedMessageV3.computeStringSize(1, checkpointId_);
+ }
+ if (epoch_ != 0L) {
+ size += com.google.protobuf.CodedOutputStream.computeInt64Size(2, epoch_);
+ }
+ if (step_ != 0L) {
+ size += com.google.protobuf.CodedOutputStream.computeInt64Size(3, step_);
+ }
+ if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(endpoint_)) {
+ size += com.google.protobuf.GeneratedMessageV3.computeStringSize(4, endpoint_);
+ }
+ size += getUnknownFields().getSerializedSize();
+ memoizedSize = size;
+ return size;
+ }
+
+ @java.lang.Override
+ public boolean equals(final java.lang.Object obj) {
+ if (obj == this) {
+ return true;
+ }
+ if (!(obj instanceof com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint)) {
+ return super.equals(obj);
+ }
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint other =
+ (com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint) obj;
+
+ if (!getCheckpointId().equals(other.getCheckpointId())) return false;
+ if (getEpoch() != other.getEpoch()) return false;
+ if (getStep() != other.getStep()) return false;
+ if (!getEndpoint().equals(other.getEndpoint())) return false;
+ if (!getUnknownFields().equals(other.getUnknownFields())) return false;
+ return true;
+ }
+
+ @java.lang.Override
+ public int hashCode() {
+ if (memoizedHashCode != 0) {
+ return memoizedHashCode;
+ }
+ int hash = 41;
+ hash = (19 * hash) + getDescriptor().hashCode();
+ hash = (37 * hash) + CHECKPOINT_ID_FIELD_NUMBER;
+ hash = (53 * hash) + getCheckpointId().hashCode();
+ hash = (37 * hash) + EPOCH_FIELD_NUMBER;
+ hash = (53 * hash) + com.google.protobuf.Internal.hashLong(getEpoch());
+ hash = (37 * hash) + STEP_FIELD_NUMBER;
+ hash = (53 * hash) + com.google.protobuf.Internal.hashLong(getStep());
+ hash = (37 * hash) + ENDPOINT_FIELD_NUMBER;
+ hash = (53 * hash) + getEndpoint().hashCode();
+ hash = (29 * hash) + getUnknownFields().hashCode();
+ memoizedHashCode = hash;
+ return hash;
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(
+ java.nio.ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(
+ java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(
+ com.google.protobuf.ByteString data)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(
+ com.google.protobuf.ByteString data,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(byte[] data)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(
+ byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws com.google.protobuf.InvalidProtocolBufferException {
+ return PARSER.parseFrom(data, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(
+ java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(
+ java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseDelimitedFrom(
+ java.io.InputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseDelimitedFrom(
+ java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseDelimitedWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(
+ com.google.protobuf.CodedInputStream input) throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(PARSER, input);
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint parseFrom(
+ com.google.protobuf.CodedInputStream input,
+ com.google.protobuf.ExtensionRegistryLite extensionRegistry)
+ throws java.io.IOException {
+ return com.google.protobuf.GeneratedMessageV3.parseWithIOException(
+ PARSER, input, extensionRegistry);
+ }
+
+ @java.lang.Override
+ public Builder newBuilderForType() {
+ return newBuilder();
+ }
+
+ public static Builder newBuilder() {
+ return DEFAULT_INSTANCE.toBuilder();
+ }
+
+ public static Builder newBuilder(
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint prototype) {
+ return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
+ }
+
+ @java.lang.Override
+ public Builder toBuilder() {
+ return this == DEFAULT_INSTANCE ? new Builder() : new Builder().mergeFrom(this);
+ }
+
+ @java.lang.Override
+ protected Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
+ Builder builder = new Builder(parent);
+ return builder;
+ }
+
+ /**
+ *
+ *
+ * + * TunedModelCheckpoint for the Tuned Model of a Tuning Job. + *+ * + * Protobuf type {@code google.cloud.aiplatform.v1beta1.TunedModelCheckpoint} + */ + public static final class Builder extends com.google.protobuf.GeneratedMessageV3.Builder
+ * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The checkpointId.
+ */
+ public java.lang.String getCheckpointId() {
+ java.lang.Object ref = checkpointId_;
+ if (!(ref instanceof java.lang.String)) {
+ com.google.protobuf.ByteString bs = (com.google.protobuf.ByteString) ref;
+ java.lang.String s = bs.toStringUtf8();
+ checkpointId_ = s;
+ return s;
+ } else {
+ return (java.lang.String) ref;
+ }
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The bytes for checkpointId.
+ */
+ public com.google.protobuf.ByteString getCheckpointIdBytes() {
+ java.lang.Object ref = checkpointId_;
+ if (ref instanceof String) {
+ com.google.protobuf.ByteString b =
+ com.google.protobuf.ByteString.copyFromUtf8((java.lang.String) ref);
+ checkpointId_ = b;
+ return b;
+ } else {
+ return (com.google.protobuf.ByteString) ref;
+ }
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @param value The checkpointId to set.
+ * @return This builder for chaining.
+ */
+ public Builder setCheckpointId(java.lang.String value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ checkpointId_ = value;
+ bitField0_ |= 0x00000001;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearCheckpointId() {
+ checkpointId_ = getDefaultInstance().getCheckpointId();
+ bitField0_ = (bitField0_ & ~0x00000001);
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @param value The bytes for checkpointId to set.
+ * @return This builder for chaining.
+ */
+ public Builder setCheckpointIdBytes(com.google.protobuf.ByteString value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ checkByteStringIsUtf8(value);
+ checkpointId_ = value;
+ bitField0_ |= 0x00000001;
+ onChanged();
+ return this;
+ }
+
+ private long epoch_;
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @return The epoch.
+ */
+ @java.lang.Override
+ public long getEpoch() {
+ return epoch_;
+ }
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @param value The epoch to set.
+ * @return This builder for chaining.
+ */
+ public Builder setEpoch(long value) {
+
+ epoch_ = value;
+ bitField0_ |= 0x00000002;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearEpoch() {
+ bitField0_ = (bitField0_ & ~0x00000002);
+ epoch_ = 0L;
+ onChanged();
+ return this;
+ }
+
+ private long step_;
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @return The step.
+ */
+ @java.lang.Override
+ public long getStep() {
+ return step_;
+ }
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @param value The step to set.
+ * @return This builder for chaining.
+ */
+ public Builder setStep(long value) {
+
+ step_ = value;
+ bitField0_ |= 0x00000004;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearStep() {
+ bitField0_ = (bitField0_ & ~0x00000004);
+ step_ = 0L;
+ onChanged();
+ return this;
+ }
+
+ private java.lang.Object endpoint_ = "";
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The endpoint.
+ */
+ public java.lang.String getEndpoint() {
+ java.lang.Object ref = endpoint_;
+ if (!(ref instanceof java.lang.String)) {
+ com.google.protobuf.ByteString bs = (com.google.protobuf.ByteString) ref;
+ java.lang.String s = bs.toStringUtf8();
+ endpoint_ = s;
+ return s;
+ } else {
+ return (java.lang.String) ref;
+ }
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The bytes for endpoint.
+ */
+ public com.google.protobuf.ByteString getEndpointBytes() {
+ java.lang.Object ref = endpoint_;
+ if (ref instanceof String) {
+ com.google.protobuf.ByteString b =
+ com.google.protobuf.ByteString.copyFromUtf8((java.lang.String) ref);
+ endpoint_ = b;
+ return b;
+ } else {
+ return (com.google.protobuf.ByteString) ref;
+ }
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @param value The endpoint to set.
+ * @return This builder for chaining.
+ */
+ public Builder setEndpoint(java.lang.String value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ endpoint_ = value;
+ bitField0_ |= 0x00000008;
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return This builder for chaining.
+ */
+ public Builder clearEndpoint() {
+ endpoint_ = getDefaultInstance().getEndpoint();
+ bitField0_ = (bitField0_ & ~0x00000008);
+ onChanged();
+ return this;
+ }
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @param value The bytes for endpoint to set.
+ * @return This builder for chaining.
+ */
+ public Builder setEndpointBytes(com.google.protobuf.ByteString value) {
+ if (value == null) {
+ throw new NullPointerException();
+ }
+ checkByteStringIsUtf8(value);
+ endpoint_ = value;
+ bitField0_ |= 0x00000008;
+ onChanged();
+ return this;
+ }
+
+ @java.lang.Override
+ public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
+ return super.setUnknownFields(unknownFields);
+ }
+
+ @java.lang.Override
+ public final Builder mergeUnknownFields(
+ final com.google.protobuf.UnknownFieldSet unknownFields) {
+ return super.mergeUnknownFields(unknownFields);
+ }
+
+ // @@protoc_insertion_point(builder_scope:google.cloud.aiplatform.v1beta1.TunedModelCheckpoint)
+ }
+
+ // @@protoc_insertion_point(class_scope:google.cloud.aiplatform.v1beta1.TunedModelCheckpoint)
+ private static final com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint DEFAULT_INSTANCE;
+
+ static {
+ DEFAULT_INSTANCE = new com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint();
+ }
+
+ public static com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint getDefaultInstance() {
+ return DEFAULT_INSTANCE;
+ }
+
+ private static final com.google.protobuf.Parser+ * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The checkpointId.
+ */
+ java.lang.String getCheckpointId();
+
+ /**
+ *
+ *
+ * + * The ID of the checkpoint. + *+ * + *
string checkpoint_id = 1;
+ *
+ * @return The bytes for checkpointId.
+ */
+ com.google.protobuf.ByteString getCheckpointIdBytes();
+
+ /**
+ *
+ *
+ * + * The epoch of the checkpoint. + *+ * + *
int64 epoch = 2;
+ *
+ * @return The epoch.
+ */
+ long getEpoch();
+
+ /**
+ *
+ *
+ * + * The step of the checkpoint. + *+ * + *
int64 step = 3;
+ *
+ * @return The step.
+ */
+ long getStep();
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The endpoint.
+ */
+ java.lang.String getEndpoint();
+
+ /**
+ *
+ *
+ *
+ * The Endpoint resource name that the checkpoint is deployed to. Format:
+ * `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ *
+ *
+ * string endpoint = 4;
+ *
+ * @return The bytes for endpoint.
+ */
+ com.google.protobuf.ByteString getEndpointBytes();
+}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TunedModelOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TunedModelOrBuilder.java
index 1361e07cbf72..80f2b77874fc 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TunedModelOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TunedModelOrBuilder.java
@@ -87,4 +87,81 @@ public interface TunedModelOrBuilder
* @return The bytes for endpoint.
*/
com.google.protobuf.ByteString getEndpointBytes();
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ java.util.List+ * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpoint getCheckpoints(int index);
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ int getCheckpointsCount();
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ java.util.List
+ getCheckpointsOrBuilderList();
+
+ /**
+ *
+ *
+ * + * Output only. The checkpoints associated with this TunedModel. + * This field is only populated for tuning jobs that enable intermediate + * checkpoints. + *+ * + *
+ * repeated .google.cloud.aiplatform.v1beta1.TunedModelCheckpoint checkpoints = 3 [(.google.api.field_behavior) = OUTPUT_ONLY];
+ *
+ */
+ com.google.cloud.aiplatform.v1beta1.TunedModelCheckpointOrBuilder getCheckpointsOrBuilder(
+ int index);
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJob.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJob.java
index 4f2026d370cc..a5ad96dd7ebc 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJob.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJob.java
@@ -189,7 +189,8 @@ public TuningSpecCase getTuningSpecCase() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -204,7 +205,8 @@ public boolean hasBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -232,7 +234,8 @@ public java.lang.String getBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -372,7 +375,7 @@ public com.google.cloud.aiplatform.v1beta1.DistillationSpec getDistillationSpec(
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -389,7 +392,7 @@ public boolean hasPartnerModelTuningSpec() {
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -409,7 +412,7 @@ public com.google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec getPartnerMode
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -1126,7 +1129,7 @@ public com.google.protobuf.ByteString getExperimentBytes() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -1145,7 +1148,7 @@ public boolean hasTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -1166,7 +1169,7 @@ public com.google.cloud.aiplatform.v1beta1.TunedModel getTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -2449,7 +2452,8 @@ public Builder clearTuningSpec() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2465,7 +2469,8 @@ public boolean hasBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2494,7 +2499,8 @@ public java.lang.String getBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2523,7 +2529,8 @@ public com.google.protobuf.ByteString getBaseModelBytes() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2545,7 +2552,8 @@ public Builder setBaseModel(java.lang.String value) {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -2565,7 +2573,8 @@ public Builder clearBaseModel() {
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -3049,7 +3058,7 @@ public Builder clearDistillationSpec() {
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -3066,7 +3075,7 @@ public boolean hasPartnerModelTuningSpec() {
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -3093,7 +3102,7 @@ public com.google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec getPartnerMode
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -3118,7 +3127,7 @@ public Builder setPartnerModelTuningSpec(
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -3140,7 +3149,7 @@ public Builder setPartnerModelTuningSpec(
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -3177,7 +3186,7 @@ public Builder mergePartnerModelTuningSpec(
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -3204,7 +3213,7 @@ public Builder clearPartnerModelTuningSpec() {
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -3219,7 +3228,7 @@ public Builder clearPartnerModelTuningSpec() {
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -3242,7 +3251,7 @@ public Builder clearPartnerModelTuningSpec() {
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -5192,7 +5201,7 @@ public Builder setExperimentBytes(com.google.protobuf.ByteString value) {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -5210,7 +5219,7 @@ public boolean hasTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -5234,7 +5243,7 @@ public com.google.cloud.aiplatform.v1beta1.TunedModel getTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -5260,7 +5269,7 @@ public Builder setTunedModel(com.google.cloud.aiplatform.v1beta1.TunedModel valu
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -5284,7 +5293,7 @@ public Builder setTunedModel(
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -5315,7 +5324,7 @@ public Builder mergeTunedModel(com.google.cloud.aiplatform.v1beta1.TunedModel va
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -5338,7 +5347,7 @@ public Builder clearTunedModel() {
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -5356,7 +5365,7 @@ public com.google.cloud.aiplatform.v1beta1.TunedModel.Builder getTunedModelBuild
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -5378,7 +5387,7 @@ public com.google.cloud.aiplatform.v1beta1.TunedModelOrBuilder getTunedModelOrBu
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJobOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJobOrBuilder.java
index 23d9247a1fcd..6a5f78720a5b 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJobOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJobOrBuilder.java
@@ -28,7 +28,8 @@ public interface TuningJobOrBuilder
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -41,7 +42,8 @@ public interface TuningJobOrBuilder
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -54,7 +56,8 @@ public interface TuningJobOrBuilder
*
*
*
- * The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ * The base model that is being tuned. See [Supported
+ * models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
*
*
* string base_model = 4;
@@ -142,7 +145,7 @@ public interface TuningJobOrBuilder
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -156,7 +159,7 @@ public interface TuningJobOrBuilder
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -170,7 +173,7 @@ public interface TuningJobOrBuilder
*
*
*
- * Tuning Spec for open sourced and third party partner models.
+ * Tuning Spec for open sourced and third party Partner models.
*
*
* .google.cloud.aiplatform.v1beta1.PartnerModelTuningSpec partner_model_tuning_spec = 21;
@@ -664,7 +667,7 @@ java.lang.String getLabelsOrDefault(
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -680,7 +683,7 @@ java.lang.String getLabelsOrDefault(
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
@@ -696,7 +699,7 @@ java.lang.String getLabelsOrDefault(
*
*
*
- * Output only. The tuned model resources assiociated with this
+ * Output only. The tuned model resources associated with this
* [TuningJob][google.cloud.aiplatform.v1.TuningJob].
*
*
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJobProto.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJobProto.java
index c40a7e5e9262..a8292adf2b64 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJobProto.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/TuningJobProto.java
@@ -100,6 +100,10 @@ public static void registerAllExtensions(com.google.protobuf.ExtensionRegistry r
internal_static_google_cloud_aiplatform_v1beta1_TunedModelRef_descriptor;
static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
internal_static_google_cloud_aiplatform_v1beta1_TunedModelRef_fieldAccessorTable;
+ static final com.google.protobuf.Descriptors.Descriptor
+ internal_static_google_cloud_aiplatform_v1beta1_TunedModelCheckpoint_descriptor;
+ static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
+ internal_static_google_cloud_aiplatform_v1beta1_TunedModelCheckpoint_fieldAccessorTable;
public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
return descriptor;
@@ -157,12 +161,14 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "ects/{project}/locations/{location}/tuningJobs/{tuning_job}*\n"
+ "tuningJobs2\ttuningJobB\016\n"
+ "\014source_modelB\r\n"
- + "\013tuning_spec\"\202\001\n\n"
+ + "\013tuning_spec\"\323\001\n\n"
+ "TunedModel\0226\n"
+ "\005model\030\001 \001(\tB\'\340A\003\372A!\n"
+ "\037aiplatform.googleapis.com/Model\022<\n"
+ "\010endpoint\030\002 \001(\tB*\340A\003\372A$\n"
- + "\"aiplatform.googleapis.com/Endpoint\"\367\002\n"
+ + "\"aiplatform.googleapis.com/Endpoint\022O\n"
+ + "\013checkpoints\030\003 \003(\01325.google.clo"
+ + "ud.aiplatform.v1beta1.TunedModelCheckpointB\003\340A\003\"\367\002\n"
+ "#SupervisedTuningDatasetDistribution\022\020\n"
+ "\003sum\030\001 \001(\003B\003\340A\003\022\031\n"
+ "\014billable_sum\030\t \001(\003B\003\340A\003\022\020\n"
@@ -172,29 +178,30 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\006median\030\005 \001(\001B\003\340A\003\022\017\n"
+ "\002p5\030\006 \001(\001B\003\340A\003\022\020\n"
+ "\003p95\030\007 \001(\001B\003\340A\003\022h\n"
- + "\007buckets\030\010 \003(\0132R.google.cloud.aipla"
- + "tform.v1beta1.SupervisedTuningDatasetDistribution.DatasetBucketB\003\340A\003\032J\n\r"
+ + "\007buckets\030\010 \003(\0132R.google.cloud.aipl"
+ + "atform.v1beta1.SupervisedTuningDatasetDistribution.DatasetBucketB\003\340A\003\032J\n\r"
+ "DatasetBucket\022\022\n"
+ "\005count\030\001 \001(\001B\003\340A\003\022\021\n"
+ "\004left\030\002 \001(\001B\003\340A\003\022\022\n"
- + "\005right\030\003 \001(\001B\003\340A\003\"\340\005\n"
+ + "\005right\030\003 \001(\001B\003\340A\003\"\220\006\n"
+ "\031SupervisedTuningDataStats\022)\n"
+ "\034tuning_dataset_example_count\030\001 \001(\003B\003\340A\003\022)\n"
+ "\034total_tuning_character_count\030\002 \001(\003B\003\340A\003\022-\n"
+ "\036total_billable_character_count\030\003 \001(\003B\005\030\001\340A\003\022\'\n"
+ "\032total_billable_token_count\030\t \001(\003B\003\340A\003\022\036\n"
+ "\021tuning_step_count\030\004 \001(\003B\003\340A\003\022p\n"
- + "\035user_input_token_distribution\030\005 \001(\0132D.google.cloud.aipla"
- + "tform.v1beta1.SupervisedTuningDatasetDistributionB\003\340A\003\022q\n"
- + "\036user_output_token_distribution\030\006 \001(\0132D.google.cloud.aiplatform"
- + ".v1beta1.SupervisedTuningDatasetDistributionB\003\340A\003\022x\n"
- + "%user_message_per_example_distribution\030\007 \001(\0132D.google.cloud.aiplatfo"
- + "rm.v1beta1.SupervisedTuningDatasetDistributionB\003\340A\003\022L\n"
- + "\025user_dataset_examples\030\010 \003"
- + "(\0132(.google.cloud.aiplatform.v1beta1.ContentB\003\340A\003\022%\n"
+ + "\035user_input_token_distribution\030\005 \001(\0132D.google.cloud.aipl"
+ + "atform.v1beta1.SupervisedTuningDatasetDistributionB\003\340A\003\022q\n"
+ + "\036user_output_token_distribution\030\006 \001(\0132D.google.cloud.aiplatfor"
+ + "m.v1beta1.SupervisedTuningDatasetDistributionB\003\340A\003\022x\n"
+ + "%user_message_per_example_distribution\030\007 \001(\0132D.google.cloud.aiplatf"
+ + "orm.v1beta1.SupervisedTuningDatasetDistributionB\003\340A\003\022L\n"
+ + "\025user_dataset_examples\030\010 "
+ + "\003(\0132(.google.cloud.aiplatform.v1beta1.ContentB\003\340A\003\022*\n"
+ "\035total_truncated_example_count\030\n"
- + " \001(\003\022!\n"
- + "\031truncated_example_indices\030\013 \003(\003\"\306\002\n"
+ + " \001(\003B\003\340A\003\022&\n"
+ + "\031truncated_example_indices\030\013 \003(\003B\003\340A\003\022$\n"
+ + "\027dropped_example_reasons\030\014 \003(\tB\003\340A\003\"\306\002\n"
+ "\023DatasetDistribution\022\020\n"
+ "\003sum\030\001 \001(\001B\003\340A\003\022\020\n"
+ "\003min\030\002 \001(\001B\003\340A\003\022\020\n"
@@ -203,8 +210,8 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\006median\030\005 \001(\001B\003\340A\003\022\017\n"
+ "\002p5\030\006 \001(\001B\003\340A\003\022\020\n"
+ "\003p95\030\007 \001(\001B\003\340A\003\022]\n"
- + "\007buckets\030\010 \003(\0132G.google.cloud.aiplatform."
- + "v1beta1.DatasetDistribution.DistributionBucketB\003\340A\003\032O\n"
+ + "\007buckets\030\010 \003(\0132G.google.cloud.ai"
+ + "platform.v1beta1.DatasetDistribution.DistributionBucketB\003\340A\003\032O\n"
+ "\022DistributionBucket\022\022\n"
+ "\005count\030\001 \001(\003B\003\340A\003\022\021\n"
+ "\004left\030\002 \001(\001B\003\340A\003\022\022\n"
@@ -214,48 +221,51 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\034total_tuning_character_count\030\002 \001(\003B\003\340A\003\022+\n"
+ "\036total_billable_character_count\030\003 \001(\003B\003\340A\003\022\036\n"
+ "\021tuning_step_count\030\004 \001(\003B\003\340A\003\022`\n"
- + "\035user_input_token_distribution\030\005 \001(\01324.google"
- + ".cloud.aiplatform.v1beta1.DatasetDistributionB\003\340A\003\022f\n"
- + "\036user_output_token_distribution\030\006"
- + " \001(\01324.google.cloud.aiplatform.v1beta1.DatasetDistributionB\003\340A\003H\000\210\001\001\022h\n"
- + "%user_message_per_example_distribution\030\007 \001("
- + "\01324.google.cloud.aiplatform.v1beta1.DatasetDistributionB\003\340A\003\022L\n"
+ + "\035user_input_token_distribution\030\005 \001(\013"
+ + "24.google.cloud.aiplatform.v1beta1.DatasetDistributionB\003\340A\003\022f\n"
+ + "\036user_output_token_distribution\030\006 \001(\01324.google.cloud.aipla"
+ + "tform.v1beta1.DatasetDistributionB\003\340A\003H\000\210\001\001\022h\n"
+ + "%user_message_per_example_distribution\030\007"
+ + " \001(\01324.google.cloud.aiplatform.v1beta1.DatasetDistributionB\003\340A\003\022L\n"
+ "\025user_dataset_examples\030\010"
+ " \003(\0132(.google.cloud.aiplatform.v1beta1.ContentB\003\340A\003B!\n"
+ "\037_user_output_token_distribution\"k\n"
+ "\025DistillationDataStats\022R\n"
- + "\026training_dataset_stats\030\001 \001(\0132-.google."
- + "cloud.aiplatform.v1beta1.DatasetStatsB\003\340A\003\"\352\001\n"
+ + "\026training_dataset_stats\030\001 \001(\0132"
+ + "-.google.cloud.aiplatform.v1beta1.DatasetStatsB\003\340A\003\"\352\001\n"
+ "\017TuningDataStats\022b\n"
- + "\034supervised_tuning_data_stats\030\001 \001(\0132:.google.cloud.aipl"
- + "atform.v1beta1.SupervisedTuningDataStatsH\000\022^\n"
- + "\027distillation_data_stats\030\003 \001(\01326.go"
- + "ogle.cloud.aiplatform.v1beta1.DistillationDataStatsB\003\340A\003H\000B\023\n"
- + "\021tuning_data_stats\"\314\002\n"
+ + "\034supervised_tuning_data_stats\030\001 \001(\0132:.google.c"
+ + "loud.aiplatform.v1beta1.SupervisedTuningDataStatsH\000\022^\n"
+ + "\027distillation_data_stats\030\003"
+ + " \001(\01326.google.cloud.aiplatform.v1beta1.DistillationDataStatsB\003\340A\003H\000B\023\n"
+ + "\021tuning_data_stats\"\377\002\n"
+ "\031SupervisedHyperParameters\022\030\n"
+ "\013epoch_count\030\001 \001(\003B\003\340A\001\022%\n"
+ "\030learning_rate_multiplier\030\002 \001(\001B\003\340A\001\022a\n"
- + "\014adapter_size\030\003 \001(\0162F.g"
- + "oogle.cloud.aiplatform.v1beta1.SupervisedHyperParameters.AdapterSizeB\003\340A\001\"\212\001\n"
+ + "\014adapter_size\030\003 \001(\0162F.google.cloud.aiplatform.v1beta1."
+ + "SupervisedHyperParameters.AdapterSizeB\003\340A\001\"\275\001\n"
+ "\013AdapterSize\022\034\n"
+ "\030ADAPTER_SIZE_UNSPECIFIED\020\000\022\024\n"
- + "\020ADAPTER_SIZE_ONE\020\001\022\025\n"
+ + "\020ADAPTER_SIZE_ONE\020\001\022\024\n"
+ + "\020ADAPTER_SIZE_TWO\020\006\022\025\n"
+ "\021ADAPTER_SIZE_FOUR\020\002\022\026\n"
+ "\022ADAPTER_SIZE_EIGHT\020\003\022\030\n"
- + "\024ADAPTER_SIZE_SIXTEEN\020\004\"\271\001\n"
+ + "\024ADAPTER_SIZE_SIXTEEN\020\004\022\033\n"
+ + "\027ADAPTER_SIZE_THIRTY_TWO\020\005\"\343\001\n"
+ "\024SupervisedTuningSpec\022!\n"
+ "\024training_dataset_uri\030\001 \001(\tB\003\340A\002\022#\n"
+ "\026validation_dataset_uri\030\002 \001(\tB\003\340A\001\022Y\n"
- + "\020hyper_parameters\030\003 \001(\0132:.google.cloud.aiplatf"
- + "orm.v1beta1.SupervisedHyperParametersB\003\340A\001\"\217\003\n"
+ + "\020hyper_parameters\030\003 \001(\0132:."
+ + "google.cloud.aiplatform.v1beta1.SupervisedHyperParametersB\003\340A\001\022(\n"
+ + "\033export_last_checkpoint_only\030\006 \001(\010B\003\340A\001\"\217\003\n"
+ "\020DistillationSpec\022\034\n"
+ "\022base_teacher_model\030\005 \001(\tH\000\022J\n"
+ "\032tuned_teacher_model_source\030\006 \001(\tB$\372A!\n"
+ "\037aiplatform.googleapis.com/ModelH\000\022!\n"
+ "\024training_dataset_uri\030\001 \001(\tB\003\340A\002\022(\n"
+ "\026validation_dataset_uri\030\002 \001(\tB\003\340A\001H\001\210\001\001\022[\n"
- + "\020hyper_parameters\030\003 \001(\0132<.googl"
- + "e.cloud.aiplatform.v1beta1.DistillationHyperParametersB\003\340A\001\022\025\n\r"
+ + "\020hyper_parameters\030\003 \001(\0132<.google.cloud.aiplatform"
+ + ".v1beta1.DistillationHyperParametersB\003\340A\001\022\025\n\r"
+ "student_model\030\004 \001(\t\022$\n"
+ "\027pipeline_root_directory\030\007 \001(\tB\003\340A\002B\017\n\r"
+ "teacher_modelB\031\n"
@@ -263,15 +273,15 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\033DistillationHyperParameters\022\035\n"
+ "\013epoch_count\030\001 \001(\003B\003\340A\001H\000\210\001\001\022*\n"
+ "\030learning_rate_multiplier\030\002 \001(\001B\003\340A\001H\001\210\001\001\022a\n"
- + "\014adapter_size\030\003 \001(\0162F.google.cloud.aiplatform."
- + "v1beta1.SupervisedHyperParameters.AdapterSizeB\003\340A\001B\016\n"
+ + "\014adapter_size\030\003 \001(\0162F.go"
+ + "ogle.cloud.aiplatform.v1beta1.SupervisedHyperParameters.AdapterSizeB\003\340A\001B\016\n"
+ "\014_epoch_countB\033\n"
+ "\031_learning_rate_multiplier\"\230\002\n"
+ "\026PartnerModelTuningSpec\022!\n"
+ "\024training_dataset_uri\030\001 \001(\tB\003\340A\002\022#\n"
+ "\026validation_dataset_uri\030\002 \001(\tB\003\340A\001\022f\n"
- + "\020hyper_parameters\030\003 \003(\0132L.google.cloud.aipl"
- + "atform.v1beta1.PartnerModelTuningSpec.HyperParametersEntry\032N\n"
+ + "\020hyper_parameters\030\003 \003(\0132L.google.cloud.aiplatform.v1beta1.Par"
+ + "tnerModelTuningSpec.HyperParametersEntry\032N\n"
+ "\024HyperParametersEntry\022\013\n"
+ "\003key\030\001 \001(\t\022%\n"
+ "\005value\030\002 \001(\0132\026.google.protobuf.Value:\0028\001\"\343\001\n\r"
@@ -282,12 +292,17 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "#aiplatform.googleapis.com/TuningJobH\000\022B\n"
+ "\014pipeline_job\030\003 \001(\tB*\372A\'\n"
+ "%aiplatform.googleapis.com/PipelineJobH\000B\021\n"
- + "\017tuned_model_refB\345\001\n"
- + "#com.google.cloud.aiplatform.v1beta1B\016TuningJobProtoP\001ZCcloud.google"
- + ".com/go/aiplatform/apiv1beta1/aiplatform"
- + "pb;aiplatformpb\252\002\037Google.Cloud.AIPlatfor"
- + "m.V1Beta1\312\002\037Google\\Cloud\\AIPlatform\\V1be"
- + "ta1\352\002\"Google::Cloud::AIPlatform::V1beta1b\006proto3"
+ + "\017tuned_model_ref\"\\\n"
+ + "\024TunedModelCheckpoint\022\025\n\r"
+ + "checkpoint_id\030\001 \001(\t\022\r\n"
+ + "\005epoch\030\002 \001(\003\022\014\n"
+ + "\004step\030\003 \001(\003\022\020\n"
+ + "\010endpoint\030\004 \001(\tB\345\001\n"
+ + "#com.google.cloud.aiplatform.v1beta1B\016TuningJobProtoP\001ZCcloud.google.com"
+ + "/go/aiplatform/apiv1beta1/aiplatformpb;a"
+ + "iplatformpb\252\002\037Google.Cloud.AIPlatform.V1"
+ + "Beta1\312\002\037Google\\Cloud\\AIPlatform\\V1beta1\352"
+ + "\002\"Google::Cloud::AIPlatform::V1beta1b\006proto3"
};
descriptor =
com.google.protobuf.Descriptors.FileDescriptor.internalBuildGeneratedFileFrom(
@@ -347,7 +362,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
internal_static_google_cloud_aiplatform_v1beta1_TunedModel_descriptor,
new java.lang.String[] {
- "Model", "Endpoint",
+ "Model", "Endpoint", "Checkpoints",
});
internal_static_google_cloud_aiplatform_v1beta1_SupervisedTuningDatasetDistribution_descriptor =
getDescriptor().getMessageTypes().get(2);
@@ -384,6 +399,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
"UserDatasetExamples",
"TotalTruncatedExampleCount",
"TruncatedExampleIndices",
+ "DroppedExampleReasons",
});
internal_static_google_cloud_aiplatform_v1beta1_DatasetDistribution_descriptor =
getDescriptor().getMessageTypes().get(4);
@@ -448,7 +464,10 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
internal_static_google_cloud_aiplatform_v1beta1_SupervisedTuningSpec_descriptor,
new java.lang.String[] {
- "TrainingDatasetUri", "ValidationDatasetUri", "HyperParameters",
+ "TrainingDatasetUri",
+ "ValidationDatasetUri",
+ "HyperParameters",
+ "ExportLastCheckpointOnly",
});
internal_static_google_cloud_aiplatform_v1beta1_DistillationSpec_descriptor =
getDescriptor().getMessageTypes().get(10);
@@ -499,6 +518,14 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
new java.lang.String[] {
"TunedModel", "TuningJob", "PipelineJob", "TunedModelRef",
});
+ internal_static_google_cloud_aiplatform_v1beta1_TunedModelCheckpoint_descriptor =
+ getDescriptor().getMessageTypes().get(14);
+ internal_static_google_cloud_aiplatform_v1beta1_TunedModelCheckpoint_fieldAccessorTable =
+ new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
+ internal_static_google_cloud_aiplatform_v1beta1_TunedModelCheckpoint_descriptor,
+ new java.lang.String[] {
+ "CheckpointId", "Epoch", "Step", "Endpoint",
+ });
com.google.protobuf.ExtensionRegistry registry =
com.google.protobuf.ExtensionRegistry.newInstance();
registry.add(com.google.api.FieldBehaviorProto.fieldBehavior);
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/UploadRagFileConfig.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/UploadRagFileConfig.java
index 6293efd9e172..620e457c66a8 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/UploadRagFileConfig.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/UploadRagFileConfig.java
@@ -78,7 +78,7 @@ public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.UploadRagFileConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=520
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=570
* @return Whether the ragFileChunkingConfig field is set.
*/
@java.lang.Override
@@ -99,7 +99,7 @@ public boolean hasRagFileChunkingConfig() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.UploadRagFileConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=520
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=570
* @return The ragFileChunkingConfig.
*/
@java.lang.Override
@@ -619,7 +619,7 @@ public Builder mergeFrom(
*
*
* @deprecated google.cloud.aiplatform.v1beta1.UploadRagFileConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=520
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=570
* @return Whether the ragFileChunkingConfig field is set.
*/
@java.lang.Deprecated
@@ -639,7 +639,7 @@ public boolean hasRagFileChunkingConfig() {
*
*
* @deprecated google.cloud.aiplatform.v1beta1.UploadRagFileConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=520
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=570
* @return The ragFileChunkingConfig.
*/
@java.lang.Deprecated
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/UploadRagFileConfigOrBuilder.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/UploadRagFileConfigOrBuilder.java
index b40ef798b7c1..07af7ba745ef 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/UploadRagFileConfigOrBuilder.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/UploadRagFileConfigOrBuilder.java
@@ -36,7 +36,7 @@ public interface UploadRagFileConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.UploadRagFileConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=520
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=570
* @return Whether the ragFileChunkingConfig field is set.
*/
@java.lang.Deprecated
@@ -54,7 +54,7 @@ public interface UploadRagFileConfigOrBuilder
*
*
* @deprecated google.cloud.aiplatform.v1beta1.UploadRagFileConfig.rag_file_chunking_config is
- * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=520
+ * deprecated. See google/cloud/aiplatform/v1beta1/vertex_rag_data.proto;l=570
* @return The ragFileChunkingConfig.
*/
@java.lang.Deprecated
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/VertexRagDataProto.java b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/VertexRagDataProto.java
index 28f3b5a4d8c5..d541a82bc67d 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/VertexRagDataProto.java
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/VertexRagDataProto.java
@@ -56,6 +56,14 @@ public static void registerAllExtensions(com.google.protobuf.ExtensionRegistry r
internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_descriptor;
static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_fieldAccessorTable;
+ static final com.google.protobuf.Descriptors.Descriptor
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_KNN_descriptor;
+ static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_KNN_fieldAccessorTable;
+ static final com.google.protobuf.Descriptors.Descriptor
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_ANN_descriptor;
+ static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_ANN_fieldAccessorTable;
static final com.google.protobuf.Descriptors.Descriptor
internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_Weaviate_descriptor;
static final com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
@@ -165,13 +173,14 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "5google/cloud/aiplatform/v1beta1/vertex_rag_data.proto\022\037google.cloud.aiplatform"
+ ".v1beta1\032\037google/api/field_behavior.prot"
+ "o\032\031google/api/resource.proto\032.google/clo"
- + "ud/aiplatform/v1beta1/api_auth.proto\032(go"
- + "ogle/cloud/aiplatform/v1beta1/io.proto\032\037google/protobuf/timestamp.proto\"\275\007\n"
+ + "ud/aiplatform/v1beta1/api_auth.proto\0325google/cloud/aiplatform/v1beta1/encryption"
+ + "_spec.proto\032(google/cloud/aiplatform/v1b"
+ + "eta1/io.proto\032\037google/protobuf/timestamp.proto\"\275\007\n"
+ "\027RagEmbeddingModelConfig\022w\n"
- + "\032vertex_prediction_endpoint\030\001 \001(\0132Q.google.cloud.aiplatfor"
- + "m.v1beta1.RagEmbeddingModelConfig.VertexPredictionEndpointH\000\022k\n"
- + "\024hybrid_search_config\030\002 \001(\0132K.google.cloud.aiplatform.v1b"
- + "eta1.RagEmbeddingModelConfig.HybridSearchConfigH\000\032\257\001\n"
+ + "\032vertex_prediction_endpoint\030\001 \001(\0132Q.google."
+ + "cloud.aiplatform.v1beta1.RagEmbeddingModelConfig.VertexPredictionEndpointH\000\022k\n"
+ + "\024hybrid_search_config\030\002 \001(\0132K.google.cloud"
+ + ".aiplatform.v1beta1.RagEmbeddingModelConfig.HybridSearchConfigH\000\032\257\001\n"
+ "\030VertexPredictionEndpoint\022<\n"
+ "\010endpoint\030\001 \001(\tB*\340A\002\372A$\n"
+ "\"aiplatform.googleapis.com/Endpoint\0226\n"
@@ -179,8 +188,8 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\037aiplatform.googleapis.com/Model\022\035\n"
+ "\020model_version_id\030\003 \001(\tB\003\340A\003\032\340\001\n"
+ "\025SparseEmbeddingConfig\022c\n"
- + "\004bm25\030\001 \001(\0132S.google.clo"
- + "ud.aiplatform.v1beta1.RagEmbeddingModelConfig.SparseEmbeddingConfig.Bm25H\000\032Y\n"
+ + "\004bm25\030\001 \001(\0132S.google.cloud.aiplatform.v1beta1.Rag"
+ + "EmbeddingModelConfig.SparseEmbeddingConfig.Bm25H\000\032Y\n"
+ "\004Bm25\022\031\n"
+ "\014multilingual\030\001 \001(\010B\003\340A\001\022\024\n"
+ "\002k1\030\002 \001(\002B\003\340A\001H\000\210\001\001\022\023\n"
@@ -189,27 +198,35 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\002_bB\007\n"
+ "\005model\032\226\002\n"
+ "\022HybridSearchConfig\022t\n"
- + "\027sparse_embedding_config\030\001 \001(\0132N.google."
- + "cloud.aiplatform.v1beta1.RagEmbeddingModelConfig.SparseEmbeddingConfigB\003\340A\001\022\211\001\n"
- + ")dense_embedding_model_prediction_endpoint\030\002"
- + " \001(\0132Q.google.cloud.aiplatform.v1beta"
- + "1.RagEmbeddingModelConfig.VertexPredictionEndpointB\003\340A\002B\016\n"
- + "\014model_config\"\360\006\n"
+ + "\027sparse_embedding_config\030\001 \001(\0132N.google.cloud.aiplatform.v1beta1."
+ + "RagEmbeddingModelConfig.SparseEmbeddingConfigB\003\340A\001\022\211\001\n"
+ + ")dense_embedding_model_prediction_endpoint\030\002 \001(\0132Q.google.cloud.ai"
+ + "platform.v1beta1.RagEmbeddingModelConfig.VertexPredictionEndpointB\003\340A\002B\016\n"
+ + "\014model_config\"\345\010\n"
+ "\021RagVectorDbConfig\022Y\n"
- + "\016rag_managed_db\030\001 \001(\0132?."
- + "google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDbH\000\022O\n"
- + "\010weaviate\030\002 \001"
- + "(\0132;.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.WeaviateH\000\022O\n"
- + "\010pinecone\030\003 "
- + "\001(\0132;.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.PineconeH\000\022e\n"
- + "\024vertex_feature_store\030\004 \001(\0132E.google.cloud.aiplatfo"
- + "rm.v1beta1.RagVectorDbConfig.VertexFeatureStoreH\000\022e\n"
- + "\024vertex_vector_search\030\006 \001(\0132"
- + "E.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.VertexVectorSearchH\000\022:\n"
+ + "\016rag_managed_db\030\001 \001(\0132?.google.cloud.aiplatform.v"
+ + "1beta1.RagVectorDbConfig.RagManagedDbH\000\022O\n"
+ + "\010weaviate\030\002 \001(\0132;.google.cloud.aiplatf"
+ + "orm.v1beta1.RagVectorDbConfig.WeaviateH\000\022O\n"
+ + "\010pinecone\030\003 \001(\0132;.google.cloud.aiplat"
+ + "form.v1beta1.RagVectorDbConfig.PineconeH\000\022e\n"
+ + "\024vertex_feature_store\030\004 \001(\0132E.google"
+ + ".cloud.aiplatform.v1beta1.RagVectorDbConfig.VertexFeatureStoreH\000\022e\n"
+ + "\024vertex_vector_search\030\006 \001(\0132E.google.cloud.aiplatform"
+ + ".v1beta1.RagVectorDbConfig.VertexVectorSearchH\000\022:\n"
+ "\010api_auth\030\005 \001(\0132(.google.cloud.aiplatform.v1beta1.ApiAuth\022d\n"
- + "\032rag_embedding_model_config\030\007"
- + " \001(\01328.google.cloud.aiplatform.v1beta1.RagEmbeddingModelConfigB\006\340A\001\340A\005\032\016\n"
- + "\014RagManagedDb\032:\n"
+ + "\032rag_embedding_model_config\030\007 \001(\01328.google.cloud.a"
+ + "iplatform.v1beta1.RagEmbeddingModelConfigB\006\340A\001\340A\005\032\202\002\n"
+ + "\014RagManagedDb\022R\n"
+ + "\003knn\030\001 \001(\0132"
+ + "C.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.KNNH\000\022R\n"
+ + "\003ann\030\002 "
+ + "\001(\0132C.google.cloud.aiplatform.v1beta1.RagVectorDbConfig.RagManagedDb.ANNH\000\032\005\n"
+ + "\003KNN\032-\n"
+ + "\003ANN\022\022\n\n"
+ + "tree_depth\030\001 \001(\005\022\022\n\n"
+ + "leaf_count\030\002 \001(\005B\024\n"
+ + "\022retrieval_strategy\032:\n"
+ "\010Weaviate\022\025\n\r"
+ "http_endpoint\030\001 \001(\t\022\027\n"
+ "\017collection_name\030\002 \001(\t\032\036\n"
@@ -222,8 +239,8 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\005index\030\002 \001(\tB\013\n"
+ "\tvector_db\"\245\001\n\n"
+ "FileStatus\022E\n"
- + "\005state\030\001 \001(\01621.google.cloud."
- + "aiplatform.v1beta1.FileStatus.StateB\003\340A\003\022\031\n"
+ + "\005state\030\001"
+ + " \001(\01621.google.cloud.aiplatform.v1beta1.FileStatus.StateB\003\340A\003\022\031\n"
+ "\014error_status\030\002 \001(\tB\003\340A\003\"5\n"
+ "\005State\022\025\n"
+ "\021STATE_UNSPECIFIED\020\000\022\n\n"
@@ -232,56 +249,59 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\024VertexAiSearchConfig\022\026\n"
+ "\016serving_config\030\001 \001(\t\"\260\001\n"
+ "\014CorpusStatus\022G\n"
- + "\005state\030\001 \001"
- + "(\01623.google.cloud.aiplatform.v1beta1.CorpusStatus.StateB\003\340A\003\022\031\n"
+ + "\005state\030\001 \001(\01623.google.cloud.ai"
+ + "platform.v1beta1.CorpusStatus.StateB\003\340A\003\022\031\n"
+ "\014error_status\030\002 \001(\tB\003\340A\003\"<\n"
+ "\005State\022\013\n"
+ "\007UNKNOWN\020\000\022\017\n"
+ "\013INITIALIZED\020\001\022\n\n"
+ "\006ACTIVE\020\002\022\t\n"
- + "\005ERROR\020\003\"\273\006\n"
+ + "\005ERROR\020\003\"\215\007\n"
+ "\tRagCorpus\022V\n"
- + "\020vector_db_config\030\t \001(\01322.google.c"
- + "loud.aiplatform.v1beta1.RagVectorDbConfigB\006\340A\001\340A\005H\000\022`\n"
+ + "\020vector_db_config\030\t"
+ + " \001(\01322.google.cloud.aiplatform.v1beta1.RagVectorDbConfigB\006\340A\001\340A\005H\000\022`\n"
+ "\027vertex_ai_search_config\030\n"
- + " \001(\01325.google.cloud.aiplatform.v1beta1.VertexAiSearchConfigB\006\340A\001\340A\005H\000\022\021\n"
+ + " \001(\01325.google.cloud."
+ + "aiplatform.v1beta1.VertexAiSearchConfigB\006\340A\001\340A\005H\000\022\021\n"
+ "\004name\030\001 \001(\tB\003\340A\003\022\031\n"
+ "\014display_name\030\002 \001(\tB\003\340A\002\022\030\n"
+ "\013description\030\003 \001(\tB\003\340A\001\022f\n"
- + "\032rag_embedding_model_config\030\006 \001(\01328.google.cloud.aiplatf"
- + "orm.v1beta1.RagEmbeddingModelConfigB\010\030\001\340A\001\340A\005\022Z\n"
- + "\024rag_vector_db_config\030\007 \001(\01322.go"
- + "ogle.cloud.aiplatform.v1beta1.RagVectorDbConfigB\010\030\001\340A\001\340A\005\0224\n"
+ + "\032rag_embedding_model_config\030\006 \001(\01328."
+ + "google.cloud.aiplatform.v1beta1.RagEmbeddingModelConfigB\010\030\001\340A\001\340A\005\022Z\n"
+ + "\024rag_vector_db_config\030\007"
+ + " \001(\01322.google.cloud.aiplatform.v1beta1.RagVectorDbConfigB\010\030\001\340A\001\340A\005\0224\n"
+ "\013create_time\030\004 \001(\0132\032.google.protobuf.TimestampB\003\340A\003\0224\n"
+ "\013update_time\030\005"
+ " \001(\0132\032.google.protobuf.TimestampB\003\340A\003\022I\n\r"
- + "corpus_status\030\010 \001(\0132-.google.cl"
- + "oud.aiplatform.v1beta1.CorpusStatusB\003\340A\003\022\034\n"
- + "\017rag_files_count\030\013 \001(\005B\003\340A\003:\200\001\352A}\n"
- + "#aiplatform.googleapis.com/RagCorpus\022?proje"
- + "cts/{project}/locations/{location}/ragCorpora/{rag_corpus}*\n"
+ + "corpus_status\030\010"
+ + " \001(\0132-.google.cloud.aiplatform.v1beta1.CorpusStatusB\003\340A\003\022\034\n"
+ + "\017rag_files_count\030\013 \001(\005B\003\340A\003\022P\n"
+ + "\017encryption_spec\030\014 \001(\0132/.go"
+ + "ogle.cloud.aiplatform.v1beta1.EncryptionSpecB\006\340A\005\340A\001:\200\001\352A}\n"
+ + "#aiplatform.googleapis.com/RagCorpus\022?projects/{project}/loca"
+ + "tions/{location}/ragCorpora/{rag_corpus}*\n"
+ "ragCorpora2\tragCorpusB\020\n"
+ "\016backend_config\"\306\010\n"
+ "\007RagFile\022E\n\n"
- + "gcs_source\030\010"
- + " \001(\0132*.google.cloud.aiplatform.v1beta1.GcsSourceB\003\340A\003H\000\022V\n"
- + "\023google_drive_source\030\t"
- + " \001(\01322.google.cloud.aiplatform.v1beta1.GoogleDriveSourceB\003\340A\003H\000\022X\n"
+ + "gcs_source\030\010 \001(\0132*.goog"
+ + "le.cloud.aiplatform.v1beta1.GcsSourceB\003\340A\003H\000\022V\n"
+ + "\023google_drive_source\030\t \001(\01322.goog"
+ + "le.cloud.aiplatform.v1beta1.GoogleDriveSourceB\003\340A\003H\000\022X\n"
+ "\024direct_upload_source\030\n"
- + " \001(\01323.google.cloud.aipl"
- + "atform.v1beta1.DirectUploadSourceB\003\340A\003H\000\022D\n"
+ + " \001"
+ + "(\01323.google.cloud.aiplatform.v1beta1.DirectUploadSourceB\003\340A\003H\000\022D\n"
+ "\014slack_source\030\013"
+ " \001(\0132,.google.cloud.aiplatform.v1beta1.SlackSourceH\000\022B\n"
- + "\013jira_source\030\014"
- + " \001(\0132+.google.cloud.aiplatform.v1beta1.JiraSourceH\000\022Q\n"
- + "\023share_point_sources\030\016"
- + " \001(\01322.google.cloud.aiplatform.v1beta1.SharePointSourcesH\000\022\021\n"
+ + "\013jira_source\030\014 \001(\0132+.goog"
+ + "le.cloud.aiplatform.v1beta1.JiraSourceH\000\022Q\n"
+ + "\023share_point_sources\030\016 \001(\01322.google.c"
+ + "loud.aiplatform.v1beta1.SharePointSourcesH\000\022\021\n"
+ "\004name\030\001 \001(\tB\003\340A\003\022\031\n"
+ "\014display_name\030\002 \001(\tB\003\340A\002\022\030\n"
+ "\013description\030\003 \001(\tB\003\340A\001\022\027\n\n"
+ "size_bytes\030\004 \001(\003B\003\340A\003\022P\n\r"
- + "rag_file_type\030\005 \001(\01624.google.cloud.aip"
- + "latform.v1beta1.RagFile.RagFileTypeB\003\340A\003\0224\n"
+ + "rag_file_type\030\005 "
+ + "\001(\01624.google.cloud.aiplatform.v1beta1.RagFile.RagFileTypeB\003\340A\003\0224\n"
+ "\013create_time\030\006 \001(\0132\032.google.protobuf.TimestampB\003\340A\003\0224\n"
+ "\013update_time\030\007 \001(\0132\032.google.protobuf.TimestampB\003\340A\003\022E\n"
+ "\013file_status\030\r"
@@ -290,20 +310,21 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\031RAG_FILE_TYPE_UNSPECIFIED\020\000\022\025\n"
+ "\021RAG_FILE_TYPE_TXT\020\001\022\025\n"
+ "\021RAG_FILE_TYPE_PDF\020\002:\217\001\352A\213\001\n"
- + "!aiplatform.googleapis.com/RagFile\022Sprojects/{project}/locations/{location}/ragCo"
- + "rpora/{rag_corpus}/ragFiles/{rag_file}*\010ragFiles2\007ragFileB\021\n"
+ + "!aiplatform.googleapis.com/RagFile\022Sprojects/{project}/loca"
+ + "tions/{location}/ragCorpora/{rag_corpus}"
+ + "/ragFiles/{rag_file}*\010ragFiles2\007ragFileB\021\n"
+ "\017rag_file_source\"\245\001\n"
+ "\010RagChunk\022\014\n"
+ "\004text\030\001 \001(\t\022J\n"
- + "\tpage_span\030\002 \001"
- + "(\01322.google.cloud.aiplatform.v1beta1.RagChunk.PageSpanH\000\210\001\001\0321\n"
+ + "\tpage_span\030\002 \001(\01322.google.cloud."
+ + "aiplatform.v1beta1.RagChunk.PageSpanH\000\210\001\001\0321\n"
+ "\010PageSpan\022\022\n\n"
+ "first_page\030\001 \001(\005\022\021\n"
+ "\tlast_page\030\002 \001(\005B\014\n\n"
+ "_page_span\"\214\002\n"
+ "\025RagFileChunkingConfig\022k\n"
- + "\025fixed_length_chunking\030\003 \001(\0132J.google.cloud.aip"
- + "latform.v1beta1.RagFileChunkingConfig.FixedLengthChunkingH\000\022\026\n\n"
+ + "\025fixed_length_chunking\030\003 \001(\0132J.google.cloud.aiplatform.v1beta1.Ra"
+ + "gFileChunkingConfig.FixedLengthChunkingH\000\022\026\n\n"
+ "chunk_size\030\001 \001(\005B\002\030\001\022\031\n\r"
+ "chunk_overlap\030\002 \001(\005B\002\030\001\032@\n"
+ "\023FixedLengthChunking\022\022\n\n"
@@ -311,15 +332,15 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "chunk_overlap\030\002 \001(\005B\021\n"
+ "\017chunking_config\"w\n"
+ "\033RagFileTransformationConfig\022X\n"
- + "\030rag_file_chunking_config\030\001"
- + " \001(\01326.google.cloud.aiplatform.v1beta1.RagFileChunkingConfig\"\236\005\n"
+ + "\030rag_file_chunking_config\030\001 \001"
+ + "(\01326.google.cloud.aiplatform.v1beta1.RagFileChunkingConfig\"\236\005\n"
+ "\024RagFileParsingConfig\022_\n"
- + "\017advanced_parser\030\003 \001(\0132D.google.cloud.aiplatform.v1beta1"
- + ".RagFileParsingConfig.AdvancedParserH\000\022[\n\r"
- + "layout_parser\030\004 \001(\0132B.google.cloud.aip"
- + "latform.v1beta1.RagFileParsingConfig.LayoutParserH\000\022U\n\n"
- + "llm_parser\030\005 \001(\0132?.google"
- + ".cloud.aiplatform.v1beta1.RagFileParsingConfig.LlmParserH\000\022$\n"
+ + "\017advanced_parser\030\003 \001(\0132D.google.cl"
+ + "oud.aiplatform.v1beta1.RagFileParsingConfig.AdvancedParserH\000\022[\n\r"
+ + "layout_parser\030\004 "
+ + "\001(\0132B.google.cloud.aiplatform.v1beta1.RagFileParsingConfig.LayoutParserH\000\022U\n\n"
+ + "llm_parser\030\005 \001(\0132?.google.cloud.aiplatform."
+ + "v1beta1.RagFileParsingConfig.LlmParserH\000\022$\n"
+ "\030use_advanced_pdf_parsing\030\002 \001(\010B\002\030\001\0322\n"
+ "\016AdvancedParser\022 \n"
+ "\030use_advanced_pdf_parsing\030\001 \001(\010\032y\n"
@@ -334,58 +355,59 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
+ "\025custom_parsing_prompt\030\003 \001(\tB\010\n"
+ "\006parser\"\331\001\n"
+ "\023UploadRagFileConfig\022\\\n"
- + "\030rag_file_chunking_config\030\001"
- + " \001(\01326.google.cloud.aiplatform.v1beta1.RagFileChunkingConfigB\002\030\001\022d\n"
- + "\036rag_file_transformation_config\030\003 \001(\0132<.google.clo"
- + "ud.aiplatform.v1beta1.RagFileTransformationConfig\"\265\t\n"
+ + "\030rag_file_chunking_config\030\001 \001(\01326.google.c"
+ + "loud.aiplatform.v1beta1.RagFileChunkingConfigB\002\030\001\022d\n"
+ + "\036rag_file_transformation_config\030\003"
+ + " \001(\0132<.google.cloud.aiplatform.v1beta1.RagFileTransformationConfig\"\320\t\n"
+ "\024ImportRagFilesConfig\022@\n\n"
+ "gcs_source\030\002 \001(\0132*.google.cloud.aiplatform.v1beta1.GcsSourceH\000\022Q\n"
- + "\023google_drive_source\030\003"
- + " \001(\01322.google.cloud.aiplatform.v1beta1.GoogleDriveSourceH\000\022D\n"
+ + "\023google_drive_source\030\003 \001(\01322.google"
+ + ".cloud.aiplatform.v1beta1.GoogleDriveSourceH\000\022D\n"
+ "\014slack_source\030\006"
+ " \001(\0132,.google.cloud.aiplatform.v1beta1.SlackSourceH\000\022B\n"
- + "\013jira_source\030\007 \001(\0132+.goo"
- + "gle.cloud.aiplatform.v1beta1.JiraSourceH\000\022Q\n"
+ + "\013jira_source\030\007"
+ + " \001(\0132+.google.cloud.aiplatform.v1beta1.JiraSourceH\000\022Q\n"
+ "\023share_point_sources\030\r"
+ " \001(\01322.google.cloud.aiplatform.v1beta1.SharePointSourcesH\000\022W\n"
- + "\030partial_failure_gcs_sink\030\013 \001(\0132/"
- + ".google.cloud.aiplatform.v1beta1.GcsDestinationB\002\030\001H\001\022a\n"
- + "\035partial_failure_bigquery_sink\030\014"
- + " \001(\01324.google.cloud.aiplatform.v1beta1.BigQueryDestinationB\002\030\001H\001\022Q\n"
+ + "\030partial_failure_gcs_sink\030\013"
+ + " \001(\0132/.google.cloud.aiplatform.v1beta1.GcsDestinationB\002\030\001H\001\022a\n"
+ + "\035partial_failure_bigquery_sink\030\014 \001(\01324.goo"
+ + "gle.cloud.aiplatform.v1beta1.BigQueryDestinationB\002\030\001H\001\022Q\n"
+ "\026import_result_gcs_sink\030\016"
+ " \001(\0132/.google.cloud.aiplatform.v1beta1.GcsDestinationH\002\022[\n"
- + "\033import_result_bigquery_sink\030\017 \001(\01324.googl"
- + "e.cloud.aiplatform.v1beta1.BigQueryDestinationH\002\022\\\n"
- + "\030rag_file_chunking_config\030\004 \001"
- + "(\01326.google.cloud.aiplatform.v1beta1.RagFileChunkingConfigB\002\030\001\022d\n"
- + "\036rag_file_transformation_config\030\020 \001(\0132<.google.cloud.ai"
- + "platform.v1beta1.RagFileTransformationConfig\022[\n"
- + "\027rag_file_parsing_config\030\010 \001(\01325."
- + "google.cloud.aiplatform.v1beta1.RagFileParsingConfigB\003\340A\001\022+\n"
+ + "\033import_result_bigquery_sink\030\017"
+ + " \001(\01324.google.cloud.aiplatform.v1beta1.BigQueryDestinationH\002\022\\\n"
+ + "\030rag_file_chunking_config\030\004 \001(\01326.google.cloud."
+ + "aiplatform.v1beta1.RagFileChunkingConfigB\002\030\001\022d\n"
+ + "\036rag_file_transformation_config\030\020"
+ + " \001(\0132<.google.cloud.aiplatform.v1beta1.RagFileTransformationConfig\022[\n"
+ + "\027rag_file_parsing_config\030\010 \001(\01325.google.cloud.aipla"
+ + "tform.v1beta1.RagFileParsingConfigB\003\340A\001\022+\n"
+ "\036max_embedding_requests_per_min\030\005 \001(\005B\003\340A\001\0222\n"
- + "%global_max_embedding_requests_per_min\030\022 \001(\005B\003\340A\001B\017\n\r"
+ + "%global_max_embedding_requests_per_min\030\022 \001(\005B\003\340A\001\022\031\n"
+ + "\021rebuild_ann_index\030\023 \001(\010B\017\n\r"
+ "import_sourceB\026\n"
+ "\024partial_failure_sinkB\024\n"
+ "\022import_result_sink\"\325\001\n"
+ "\022RagManagedDbConfig\022T\n\n"
- + "enterprise\030\001 \001(\0132>.google.cloud.aipl"
- + "atform.v1beta1.RagManagedDbConfig.EnterpriseH\000\022J\n"
- + "\005basic\030\002 \001(\01329.google.cloud.aip"
- + "latform.v1beta1.RagManagedDbConfig.BasicH\000\032\014\n\n"
+ + "enterprise\030\001 \001(\0132>.google.c"
+ + "loud.aiplatform.v1beta1.RagManagedDbConfig.EnterpriseH\000\022J\n"
+ + "\005basic\030\002 \001(\01329.google."
+ + "cloud.aiplatform.v1beta1.RagManagedDbConfig.BasicH\000\032\014\n\n"
+ "Enterprise\032\007\n"
+ "\005BasicB\006\n"
+ "\004tier\"\206\002\n"
+ "\017RagEngineConfig\022\021\n"
+ "\004name\030\001 \001(\tB\003\340A\010\022R\n"
- + "\025rag_managed_db_config\030\002 \001(\01323.google.cloud.a"
- + "iplatform.v1beta1.RagManagedDbConfig:\213\001\352A\207\001\n"
- + ")aiplatform.googleapis.com/RagEngineConfig\0227projects/{project}/locations/{lo"
- + "cation}/ragEngineConfig*\020ragEngineConfigs2\017ragEngineConfigB\351\001\n"
- + "#com.google.cloud.aiplatform.v1beta1B\022VertexRagDataProtoP\001"
- + "ZCcloud.google.com/go/aiplatform/apiv1be"
- + "ta1/aiplatformpb;aiplatformpb\252\002\037Google.C"
- + "loud.AIPlatform.V1Beta1\312\002\037Google\\Cloud\\A"
- + "IPlatform\\V1beta1\352\002\"Google::Cloud::AIPlatform::V1beta1b\006proto3"
+ + "\025rag_managed_db_config\030\002 \001(\01323.googl"
+ + "e.cloud.aiplatform.v1beta1.RagManagedDbConfig:\213\001\352A\207\001\n"
+ + ")aiplatform.googleapis.com/RagEngineConfig\0227projects/{project}/loca"
+ + "tions/{location}/ragEngineConfig*\020ragEngineConfigs2\017ragEngineConfigB\351\001\n"
+ + "#com.google.cloud.aiplatform.v1beta1B\022VertexRagDa"
+ + "taProtoP\001ZCcloud.google.com/go/aiplatfor"
+ + "m/apiv1beta1/aiplatformpb;aiplatformpb\252\002"
+ + "\037Google.Cloud.AIPlatform.V1Beta1\312\002\037Googl"
+ + "e\\Cloud\\AIPlatform\\V1beta1\352\002\"Google::Cloud::AIPlatform::V1beta1b\006proto3"
};
descriptor =
com.google.protobuf.Descriptors.FileDescriptor.internalBuildGeneratedFileFrom(
@@ -394,6 +416,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
com.google.api.FieldBehaviorProto.getDescriptor(),
com.google.api.ResourceProto.getDescriptor(),
com.google.cloud.aiplatform.v1beta1.ApiAuthProto.getDescriptor(),
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpecProto.getDescriptor(),
com.google.cloud.aiplatform.v1beta1.IoProto.getDescriptor(),
com.google.protobuf.TimestampProto.getDescriptor(),
});
@@ -467,7 +490,27 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_fieldAccessorTable =
new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_descriptor,
+ new java.lang.String[] {
+ "Knn", "Ann", "RetrievalStrategy",
+ });
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_KNN_descriptor =
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_descriptor
+ .getNestedTypes()
+ .get(0);
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_KNN_fieldAccessorTable =
+ new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_KNN_descriptor,
new java.lang.String[] {});
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_ANN_descriptor =
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_descriptor
+ .getNestedTypes()
+ .get(1);
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_ANN_fieldAccessorTable =
+ new com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
+ internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_RagManagedDb_ANN_descriptor,
+ new java.lang.String[] {
+ "TreeDepth", "LeafCount",
+ });
internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_Weaviate_descriptor =
internal_static_google_cloud_aiplatform_v1beta1_RagVectorDbConfig_descriptor
.getNestedTypes()
@@ -549,6 +592,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
"UpdateTime",
"CorpusStatus",
"RagFilesCount",
+ "EncryptionSpec",
"BackendConfig",
});
internal_static_google_cloud_aiplatform_v1beta1_RagFile_descriptor =
@@ -684,6 +728,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
"RagFileParsingConfig",
"MaxEmbeddingRequestsPerMin",
"GlobalMaxEmbeddingRequestsPerMin",
+ "RebuildAnnIndex",
"ImportSource",
"PartialFailureSink",
"ImportResultSink",
@@ -730,6 +775,7 @@ public static com.google.protobuf.Descriptors.FileDescriptor getDescriptor() {
com.google.api.FieldBehaviorProto.getDescriptor();
com.google.api.ResourceProto.getDescriptor();
com.google.cloud.aiplatform.v1beta1.ApiAuthProto.getDescriptor();
+ com.google.cloud.aiplatform.v1beta1.EncryptionSpecProto.getDescriptor();
com.google.cloud.aiplatform.v1beta1.IoProto.getDescriptor();
com.google.protobuf.TimestampProto.getDescriptor();
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/accelerator_type.proto b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/accelerator_type.proto
index bdf4fa4ae850..6da519c5b076 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/accelerator_type.proto
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/accelerator_type.proto
@@ -24,6 +24,7 @@ option java_package = "com.google.cloud.aiplatform.v1beta1";
option php_namespace = "Google\\Cloud\\AIPlatform\\V1beta1";
option ruby_package = "Google::Cloud::AIPlatform::V1beta1";
+// LINT: LEGACY_NAMES
// Represents a hardware accelerator type.
enum AcceleratorType {
// Unspecified accelerator type, which means no accelerator.
@@ -60,6 +61,12 @@ enum AcceleratorType {
// Nvidia H100 Mega 80Gb GPU.
NVIDIA_H100_MEGA_80GB = 14;
+ // Nvidia H200 141Gb GPU.
+ NVIDIA_H200_141GB = 15;
+
+ // Nvidia B200 GPU.
+ NVIDIA_B200 = 16;
+
// TPU v2.
TPU_V2 = 6;
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/notebook_software_config.proto b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/notebook_software_config.proto
index 743bd4285f73..38967027e7fe 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/notebook_software_config.proto
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/notebook_software_config.proto
@@ -27,43 +27,29 @@ option java_package = "com.google.cloud.aiplatform.v1beta1";
option php_namespace = "Google\\Cloud\\AIPlatform\\V1beta1";
option ruby_package = "Google::Cloud::AIPlatform::V1beta1";
-// Post startup script config.
message PostStartupScriptConfig {
- // Represents a notebook runtime post startup script behavior.
enum PostStartupScriptBehavior {
- // Unspecified post startup script behavior.
POST_STARTUP_SCRIPT_BEHAVIOR_UNSPECIFIED = 0;
- // Run post startup script after runtime is started.
RUN_ONCE = 1;
- // Run post startup script after runtime is stopped.
RUN_EVERY_START = 2;
- // Download and run post startup script every time runtime is started.
DOWNLOAD_AND_RUN_EVERY_START = 3;
}
- // Optional. Post startup script to run after runtime is started.
string post_startup_script = 1 [(google.api.field_behavior) = OPTIONAL];
- // Optional. Post startup script url to download. Example:
- // https://bucket/script.sh
string post_startup_script_url = 2 [(google.api.field_behavior) = OPTIONAL];
- // Optional. Post startup script behavior that defines download and execution
- // behavior.
PostStartupScriptBehavior post_startup_script_behavior = 3
[(google.api.field_behavior) = OPTIONAL];
}
-// Notebook Software Config.
message NotebookSoftwareConfig {
// Optional. Environment variables to be passed to the container.
// Maximum limit is 100.
repeated EnvVar env = 1 [(google.api.field_behavior) = OPTIONAL];
-
- // Optional. Post startup script config.
PostStartupScriptConfig post_startup_script_config = 2
[(google.api.field_behavior) = OPTIONAL];
}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/openapi.proto b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/openapi.proto
index e7a7bac5b086..7439e94d8d12 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/openapi.proto
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/openapi.proto
@@ -139,6 +139,11 @@ message Schema {
// subschemas in the list.
repeated Schema any_of = 11 [(google.api.field_behavior) = OPTIONAL];
+ // Optional. Can either be a boolean or an object; controls the presence of
+ // additional properties.
+ google.protobuf.Value additional_properties = 26
+ [(google.api.field_behavior) = OPTIONAL];
+
// Optional. Allows indirect references between schema nodes. The value should
// be a valid reference to a child of the root `defs`.
//
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/tuning_job.proto b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/tuning_job.proto
index 7ff746aa6e52..46ed03ba34aa 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/tuning_job.proto
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/tuning_job.proto
@@ -43,7 +43,8 @@ message TuningJob {
};
oneof source_model {
- // The base model that is being tuned, e.g., "gemini-1.0-pro-002".
+ // The base model that is being tuned. See [Supported
+ // models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
string base_model = 4;
}
@@ -54,7 +55,7 @@ message TuningJob {
// Tuning Spec for Distillation.
DistillationSpec distillation_spec = 17;
- // Tuning Spec for open sourced and third party partner models.
+ // Tuning Spec for open sourced and third party Partner models.
PartnerModelTuningSpec partner_model_tuning_spec = 21;
}
@@ -125,7 +126,7 @@ message TuningJob {
}
];
- // Output only. The tuned model resources assiociated with this
+ // Output only. The tuned model resources associated with this
// [TuningJob][google.cloud.aiplatform.v1.TuningJob].
TunedModel tuned_model = 14 [(google.api.field_behavior) = OUTPUT_ONLY];
@@ -159,7 +160,7 @@ message TuningJob {
string service_account = 22;
}
-// The Model Registry Model and Online Prediction Endpoint assiociated with
+// The Model Registry Model and Online Prediction Endpoint associated with
// this [TuningJob][google.cloud.aiplatform.v1.TuningJob].
message TunedModel {
// Output only. The resource name of the TunedModel. Format:
@@ -179,6 +180,12 @@ message TunedModel {
type: "aiplatform.googleapis.com/Endpoint"
}
];
+
+ // Output only. The checkpoints associated with this TunedModel.
+ // This field is only populated for tuning jobs that enable intermediate
+ // checkpoints.
+ repeated TunedModelCheckpoint checkpoints = 3
+ [(google.api.field_behavior) = OUTPUT_ONLY];
}
// Dataset distribution for Supervised Tuning.
@@ -262,13 +269,21 @@ message SupervisedTuningDataStats {
repeated Content user_dataset_examples = 8
[(google.api.field_behavior) = OUTPUT_ONLY];
- // The number of examples in the dataset that have been truncated by any
- // amount.
- int64 total_truncated_example_count = 10;
+ // Output only. The number of examples in the dataset that have been dropped.
+ // An example can be dropped for reasons including: too many tokens, contains
+ // an invalid image, contains too many images, etc.
+ int64 total_truncated_example_count = 10
+ [(google.api.field_behavior) = OUTPUT_ONLY];
+
+ // Output only. A partial sample of the indices (starting from 1) of the
+ // dropped examples.
+ repeated int64 truncated_example_indices = 11
+ [(google.api.field_behavior) = OUTPUT_ONLY];
- // A partial sample of the indices (starting from 1) of the truncated
- // examples.
- repeated int64 truncated_example_indices = 11;
+ // Output only. For each index in `truncated_example_indices`, the user-facing
+ // reason why the example was dropped.
+ repeated string dropped_example_reasons = 12
+ [(google.api.field_behavior) = OUTPUT_ONLY];
}
// Distribution computed over a tuning dataset.
@@ -376,6 +391,9 @@ message SupervisedHyperParameters {
// Adapter size 1.
ADAPTER_SIZE_ONE = 1;
+ // Adapter size 2.
+ ADAPTER_SIZE_TWO = 6;
+
// Adapter size 4.
ADAPTER_SIZE_FOUR = 2;
@@ -384,6 +402,9 @@ message SupervisedHyperParameters {
// Adapter size 16.
ADAPTER_SIZE_SIXTEEN = 4;
+
+ // Adapter size 32.
+ ADAPTER_SIZE_THIRTY_TWO = 5;
}
// Optional. Number of complete passes the model makes over the entire
@@ -410,14 +431,20 @@ message SupervisedTuningSpec {
// Optional. Hyperparameters for SFT.
SupervisedHyperParameters hyper_parameters = 3
[(google.api.field_behavior) = OPTIONAL];
+
+ // Optional. If set to true, disable intermediate checkpoints for SFT and only
+ // the last checkpoint will be exported. Otherwise, enable intermediate
+ // checkpoints for SFT. Default is false.
+ bool export_last_checkpoint_only = 6 [(google.api.field_behavior) = OPTIONAL];
}
// Tuning Spec for Distillation.
message DistillationSpec {
- // The teacher model that is being distilled from, e.g., "gemini-1.0-pro-002".
+ // The teacher model that is being distilled from. See [Supported
+ // models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
oneof teacher_model {
- // The base teacher model that is being distilled, e.g.,
- // "gemini-1.0-pro-002".
+ // The base teacher model that is being distilled. See [Supported
+ // models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
string base_teacher_model = 5;
// The resource name of the Tuned teacher model. Format:
@@ -501,3 +528,19 @@ message TunedModelRef {
}];
}
}
+
+// TunedModelCheckpoint for the Tuned Model of a Tuning Job.
+message TunedModelCheckpoint {
+ // The ID of the checkpoint.
+ string checkpoint_id = 1;
+
+ // The epoch of the checkpoint.
+ int64 epoch = 2;
+
+ // The step of the checkpoint.
+ int64 step = 3;
+
+ // The Endpoint resource name that the checkpoint is deployed to. Format:
+ // `projects/{project}/locations/{location}/endpoints/{endpoint}`.
+ string endpoint = 4;
+}
diff --git a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto
index 2262bd435e81..a289d40e92c2 100644
--- a/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto
+++ b/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto
@@ -19,6 +19,7 @@ package google.cloud.aiplatform.v1beta1;
import "google/api/field_behavior.proto";
import "google/api/resource.proto";
import "google/cloud/aiplatform/v1beta1/api_auth.proto";
+import "google/cloud/aiplatform/v1beta1/encryption_spec.proto";
import "google/cloud/aiplatform/v1beta1/io.proto";
import "google/protobuf/timestamp.proto";
@@ -118,7 +119,47 @@ message RagEmbeddingModelConfig {
// Config for the Vector DB to use for RAG.
message RagVectorDbConfig {
// The config for the default RAG-managed Vector DB.
- message RagManagedDb {}
+ message RagManagedDb {
+ // Config for KNN search.
+ message KNN {}
+
+ // Config for ANN search.
+ //
+ // RagManagedDb uses a tree-based structure to partition data and
+ // facilitate faster searches. As a tradeoff, it requires longer indexing
+ // time and manual triggering of index rebuild via the ImportRagFiles and
+ // UpdateRagCorpus API.
+ message ANN {
+ // The depth of the tree-based structure. Only depth values of 2 and 3 are
+ // supported.
+ //
+ // Recommended value is 2 if you have if you have O(10K) files in the
+ // RagCorpus and set this to 3 if more than that.
+ //
+ // Default value is 2.
+ int32 tree_depth = 1;
+
+ // Number of leaf nodes in the tree-based structure. Each leaf node
+ // contains groups of closely related vectors along with their
+ // corresponding centroid.
+ //
+ // Recommended value is 10 * sqrt(num of RagFiles in your RagCorpus).
+ //
+ // Default value is 500.
+ int32 leaf_count = 2;
+ }
+
+ // Choice of retrieval strategy.
+ oneof retrieval_strategy {
+ // Performs a KNN search on RagCorpus.
+ // Default choice if not specified.
+ KNN knn = 1;
+
+ // Performs an ANN search on RagCorpus. Use this if you have a lot of
+ // files (> 10K) in your RagCorpus and want to reduce the search latency.
+ ANN ann = 2;
+ }
+ }
// The config for the Weaviate.
message Weaviate {
@@ -308,6 +349,15 @@ message RagCorpus {
// Output only. Number of RagFiles in the RagCorpus.
int32 rag_files_count = 11 [(google.api.field_behavior) = OUTPUT_ONLY];
+
+ // Optional. Immutable. The CMEK key name used to encrypt at-rest data related
+ // to this Corpus. Only applicable to RagManagedDb option for Vector DB. This
+ // field can only be set at corpus creation time, and cannot be updated or
+ // deleted.
+ EncryptionSpec encryption_spec = 12 [
+ (google.api.field_behavior) = IMMUTABLE,
+ (google.api.field_behavior) = OPTIONAL
+ ];
}
// A RagFile contains user data for chunking, embedding and indexing.
@@ -607,6 +657,15 @@ message ImportRagFilesConfig {
// global limit.
int32 global_max_embedding_requests_per_min = 18
[(google.api.field_behavior) = OPTIONAL];
+
+ // Rebuilds the ANN index to optimize for recall on the imported data.
+ // Only applicable for RagCorpora running on RagManagedDb with
+ // `retrieval_strategy` set to `ANN`. The rebuild will be performed using the
+ // existing ANN config set on the RagCorpus. To change the ANN config, please
+ // use the UpdateRagCorpus API.
+ //
+ // Default is false, i.e., index is not rebuilt.
+ bool rebuild_ann_index = 19;
}
// Configuration message for RagManagedDb used by RagEngine.
diff --git a/java-alloydb-connectors/README.md b/java-alloydb-connectors/README.md
index 96b3d86c1c29..e57f2cc80d87 100644
--- a/java-alloydb-connectors/README.md
+++ b/java-alloydb-connectors/README.md
@@ -20,7 +20,7 @@ If you are using Maven with [BOM][libraries-bom], add this to your pom.xml file:
com.google.cloud
libraries-bom
- 26.59.0
+ 26.60.0
pom
import
@@ -42,20 +42,20 @@ If you are using Maven without the BOM, add this to your dependencies:
com.google.cloud
google-cloud-alloydb-connectors
- 0.39.0
+ 0.40.0
```
If you are using Gradle without BOM, add this to your dependencies:
```Groovy
-implementation 'com.google.cloud:google-cloud-alloydb-connectors:0.39.0'
+implementation 'com.google.cloud:google-cloud-alloydb-connectors:0.40.0'
```
If you are using SBT, add this to your dependencies:
```Scala
-libraryDependencies += "com.google.cloud" % "google-cloud-alloydb-connectors" % "0.39.0"
+libraryDependencies += "com.google.cloud" % "google-cloud-alloydb-connectors" % "0.40.0"
```
## Authentication
@@ -194,7 +194,7 @@ Java is a registered trademark of Oracle and/or its affiliates.
[kokoro-badge-link-5]: http://storage.googleapis.com/cloud-devrel-public/java/badges/google-cloud-java/java11.html
[stability-image]: https://img.shields.io/badge/stability-stable-green
[maven-version-image]: https://img.shields.io/maven-central/v/com.google.cloud/google-cloud-alloydb-connectors.svg
-[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-alloydb-connectors/0.39.0
+[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-alloydb-connectors/0.40.0
[authentication]: https://github.com/googleapis/google-cloud-java#authentication
[auth-scopes]: https://developers.google.com/identity/protocols/oauth2/scopes
[predefined-iam-roles]: https://cloud.google.com/iam/docs/understanding-roles#predefined_roles
diff --git a/java-alloydb/README.md b/java-alloydb/README.md
index b3c04ade912c..29ab592c7d86 100644
--- a/java-alloydb/README.md
+++ b/java-alloydb/README.md
@@ -20,7 +20,7 @@ If you are using Maven with [BOM][libraries-bom], add this to your pom.xml file:
com.google.cloud
libraries-bom
- 26.59.0
+ 26.60.0
pom
import
@@ -42,20 +42,20 @@ If you are using Maven without the BOM, add this to your dependencies:
com.google.cloud
google-cloud-alloydb
- 0.50.0
+ 0.51.0
```
If you are using Gradle without BOM, add this to your dependencies:
```Groovy
-implementation 'com.google.cloud:google-cloud-alloydb:0.50.0'
+implementation 'com.google.cloud:google-cloud-alloydb:0.51.0'
```
If you are using SBT, add this to your dependencies:
```Scala
-libraryDependencies += "com.google.cloud" % "google-cloud-alloydb" % "0.50.0"
+libraryDependencies += "com.google.cloud" % "google-cloud-alloydb" % "0.51.0"
```
## Authentication
@@ -194,7 +194,7 @@ Java is a registered trademark of Oracle and/or its affiliates.
[kokoro-badge-link-5]: http://storage.googleapis.com/cloud-devrel-public/java/badges/google-cloud-java/java11.html
[stability-image]: https://img.shields.io/badge/stability-stable-green
[maven-version-image]: https://img.shields.io/maven-central/v/com.google.cloud/google-cloud-alloydb.svg
-[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-alloydb/0.50.0
+[maven-version-link]: https://central.sonatype.com/artifact/com.google.cloud/google-cloud-alloydb/0.51.0
[authentication]: https://github.com/googleapis/google-cloud-java#authentication
[auth-scopes]: https://developers.google.com/identity/protocols/oauth2/scopes
[predefined-iam-roles]: https://cloud.google.com/iam/docs/understanding-roles#predefined_roles
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBAdminClient.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBAdminClient.java
index 2254aefd5234..485f9482e54b 100644
--- a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBAdminClient.java
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBAdminClient.java
@@ -151,6 +151,66 @@
*
*
*
+ * ExportCluster
Exports data from the cluster. Imperative only.
+ * Request object method variants only take one parameter, a request object, which must be constructed before the call.
+ *
+ * exportClusterAsync(ExportClusterRequest request)
+ *
+ * Methods that return long-running operations have "Async" method variants that return `OperationFuture`, which is used to track polling of the service.
+ *
+ * exportClusterAsync(ClusterName name, GcsDestination gcsDestination, String database, ExportClusterRequest.CsvExportOptions csvExportOptions, ExportClusterRequest.SqlExportOptions sqlExportOptions)
+ *
exportClusterAsync(String name, GcsDestination gcsDestination, String database, ExportClusterRequest.CsvExportOptions csvExportOptions, ExportClusterRequest.SqlExportOptions sqlExportOptions)
+ *
+ * Callable method variants take no parameters and return an immutable API callable object, which can be used to initiate calls to the service.
+ *
+ * exportClusterOperationCallable()
+ *
exportClusterCallable()
+ *
+ *
+ *
+ *
+ * ImportCluster
Imports data to the cluster. Imperative only.
+ * Request object method variants only take one parameter, a request object, which must be constructed before the call.
+ *
+ * importClusterAsync(ImportClusterRequest request)
+ *
+ * Methods that return long-running operations have "Async" method variants that return `OperationFuture`, which is used to track polling of the service.
+ *
+ * importClusterAsync(ClusterName name, String gcsUri, String database, String user)
+ *
importClusterAsync(String name, String gcsUri, String database, String user)
+ *
+ * Callable method variants take no parameters and return an immutable API callable object, which can be used to initiate calls to the service.
+ *
+ * importClusterOperationCallable()
+ *
importClusterCallable()
+ *
+ *
+ *
+ *
+ * UpgradeCluster
Upgrades a single Cluster. Imperative only.
+ * Request object method variants only take one parameter, a request object, which must be constructed before the call.
+ *
+ * upgradeClusterAsync(UpgradeClusterRequest request)
+ *
+ * Methods that return long-running operations have "Async" method variants that return `OperationFuture`, which is used to track polling of the service.
+ *
+ * upgradeClusterAsync(ClusterName name, DatabaseVersion version)
+ *
upgradeClusterAsync(String name, DatabaseVersion version)
+ *
+ * Callable method variants take no parameters and return an immutable API callable object, which can be used to initiate calls to the service.
+ *
+ * upgradeClusterOperationCallable()
+ *
upgradeClusterCallable()
+ *
+ *
+ *
+ *
* DeleteCluster
Deletes a single Cluster.
@@ -1472,6 +1532,550 @@ public final UnaryCallable updateClusterCallabl
return stub.updateClusterCallable();
}
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Exports data from the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * ClusterName name = ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]");
+ * GcsDestination gcsDestination = GcsDestination.newBuilder().build();
+ * String database = "database1789464955";
+ * ExportClusterRequest.CsvExportOptions csvExportOptions =
+ * ExportClusterRequest.CsvExportOptions.newBuilder().build();
+ * ExportClusterRequest.SqlExportOptions sqlExportOptions =
+ * ExportClusterRequest.SqlExportOptions.newBuilder().build();
+ * ExportClusterResponse response =
+ * alloyDBAdminClient
+ * .exportClusterAsync(
+ * name, gcsDestination, database, csvExportOptions, sqlExportOptions)
+ * .get();
+ * }
+ * }
+ *
+ * @param name Required. The resource name of the cluster.
+ * @param gcsDestination Required. Option to export data to cloud storage.
+ * @param database Required. Name of the database where the export command will be executed. Note
+ * - Value provided should be the same as expected from `SELECT current_database();` and NOT
+ * as a resource reference.
+ * @param csvExportOptions Options for exporting data in CSV format. Required field to be set for
+ * CSV file type.
+ * @param sqlExportOptions Options for exporting data in SQL format. Required field to be set for
+ * SQL file type.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture exportClusterAsync(
+ ClusterName name,
+ GcsDestination gcsDestination,
+ String database,
+ ExportClusterRequest.CsvExportOptions csvExportOptions,
+ ExportClusterRequest.SqlExportOptions sqlExportOptions) {
+ ExportClusterRequest request =
+ ExportClusterRequest.newBuilder()
+ .setName(name == null ? null : name.toString())
+ .setGcsDestination(gcsDestination)
+ .setDatabase(database)
+ .setCsvExportOptions(csvExportOptions)
+ .setSqlExportOptions(sqlExportOptions)
+ .build();
+ return exportClusterAsync(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Exports data from the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * String name = ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString();
+ * GcsDestination gcsDestination = GcsDestination.newBuilder().build();
+ * String database = "database1789464955";
+ * ExportClusterRequest.CsvExportOptions csvExportOptions =
+ * ExportClusterRequest.CsvExportOptions.newBuilder().build();
+ * ExportClusterRequest.SqlExportOptions sqlExportOptions =
+ * ExportClusterRequest.SqlExportOptions.newBuilder().build();
+ * ExportClusterResponse response =
+ * alloyDBAdminClient
+ * .exportClusterAsync(
+ * name, gcsDestination, database, csvExportOptions, sqlExportOptions)
+ * .get();
+ * }
+ * }
+ *
+ * @param name Required. The resource name of the cluster.
+ * @param gcsDestination Required. Option to export data to cloud storage.
+ * @param database Required. Name of the database where the export command will be executed. Note
+ * - Value provided should be the same as expected from `SELECT current_database();` and NOT
+ * as a resource reference.
+ * @param csvExportOptions Options for exporting data in CSV format. Required field to be set for
+ * CSV file type.
+ * @param sqlExportOptions Options for exporting data in SQL format. Required field to be set for
+ * SQL file type.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture exportClusterAsync(
+ String name,
+ GcsDestination gcsDestination,
+ String database,
+ ExportClusterRequest.CsvExportOptions csvExportOptions,
+ ExportClusterRequest.SqlExportOptions sqlExportOptions) {
+ ExportClusterRequest request =
+ ExportClusterRequest.newBuilder()
+ .setName(name)
+ .setGcsDestination(gcsDestination)
+ .setDatabase(database)
+ .setCsvExportOptions(csvExportOptions)
+ .setSqlExportOptions(sqlExportOptions)
+ .build();
+ return exportClusterAsync(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Exports data from the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * ExportClusterRequest request =
+ * ExportClusterRequest.newBuilder()
+ * .setName(ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString())
+ * .setDatabase("database1789464955")
+ * .build();
+ * ExportClusterResponse response = alloyDBAdminClient.exportClusterAsync(request).get();
+ * }
+ * }
+ *
+ * @param request The request object containing all of the parameters for the API call.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture exportClusterAsync(
+ ExportClusterRequest request) {
+ return exportClusterOperationCallable().futureCall(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Exports data from the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * ExportClusterRequest request =
+ * ExportClusterRequest.newBuilder()
+ * .setName(ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString())
+ * .setDatabase("database1789464955")
+ * .build();
+ * OperationFuture future =
+ * alloyDBAdminClient.exportClusterOperationCallable().futureCall(request);
+ * // Do something.
+ * ExportClusterResponse response = future.get();
+ * }
+ * }
+ */
+ public final OperationCallable
+ exportClusterOperationCallable() {
+ return stub.exportClusterOperationCallable();
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Exports data from the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * ExportClusterRequest request =
+ * ExportClusterRequest.newBuilder()
+ * .setName(ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString())
+ * .setDatabase("database1789464955")
+ * .build();
+ * ApiFuture future = alloyDBAdminClient.exportClusterCallable().futureCall(request);
+ * // Do something.
+ * Operation response = future.get();
+ * }
+ * }
+ */
+ public final UnaryCallable exportClusterCallable() {
+ return stub.exportClusterCallable();
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Imports data to the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * ClusterName name = ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]");
+ * String gcsUri = "gcsUri-1251224875";
+ * String database = "database1789464955";
+ * String user = "user3599307";
+ * ImportClusterResponse response =
+ * alloyDBAdminClient.importClusterAsync(name, gcsUri, database, user).get();
+ * }
+ * }
+ *
+ * @param name Required. The resource name of the cluster.
+ * @param gcsUri Required. The path to the file in Google Cloud Storage where the source file for
+ * import will be stored. The URI is in the form `gs://bucketName/fileName`.
+ * @param database Optional. Name of the database to which the import will be done. For import
+ * from SQL file, this is required only if the file does not specify a database. Note - Value
+ * provided should be the same as expected from `SELECT current_database();` and NOT as a
+ * resource reference.
+ * @param user Optional. Database user to be used for importing the data. Note - Value provided
+ * should be the same as expected from `SELECT current_user;` and NOT as a resource reference.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture importClusterAsync(
+ ClusterName name, String gcsUri, String database, String user) {
+ ImportClusterRequest request =
+ ImportClusterRequest.newBuilder()
+ .setName(name == null ? null : name.toString())
+ .setGcsUri(gcsUri)
+ .setDatabase(database)
+ .setUser(user)
+ .build();
+ return importClusterAsync(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Imports data to the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * String name = ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString();
+ * String gcsUri = "gcsUri-1251224875";
+ * String database = "database1789464955";
+ * String user = "user3599307";
+ * ImportClusterResponse response =
+ * alloyDBAdminClient.importClusterAsync(name, gcsUri, database, user).get();
+ * }
+ * }
+ *
+ * @param name Required. The resource name of the cluster.
+ * @param gcsUri Required. The path to the file in Google Cloud Storage where the source file for
+ * import will be stored. The URI is in the form `gs://bucketName/fileName`.
+ * @param database Optional. Name of the database to which the import will be done. For import
+ * from SQL file, this is required only if the file does not specify a database. Note - Value
+ * provided should be the same as expected from `SELECT current_database();` and NOT as a
+ * resource reference.
+ * @param user Optional. Database user to be used for importing the data. Note - Value provided
+ * should be the same as expected from `SELECT current_user;` and NOT as a resource reference.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture importClusterAsync(
+ String name, String gcsUri, String database, String user) {
+ ImportClusterRequest request =
+ ImportClusterRequest.newBuilder()
+ .setName(name)
+ .setGcsUri(gcsUri)
+ .setDatabase(database)
+ .setUser(user)
+ .build();
+ return importClusterAsync(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Imports data to the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * ImportClusterRequest request =
+ * ImportClusterRequest.newBuilder()
+ * .setName(ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString())
+ * .setGcsUri("gcsUri-1251224875")
+ * .setDatabase("database1789464955")
+ * .setUser("user3599307")
+ * .build();
+ * ImportClusterResponse response = alloyDBAdminClient.importClusterAsync(request).get();
+ * }
+ * }
+ *
+ * @param request The request object containing all of the parameters for the API call.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture importClusterAsync(
+ ImportClusterRequest request) {
+ return importClusterOperationCallable().futureCall(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Imports data to the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * ImportClusterRequest request =
+ * ImportClusterRequest.newBuilder()
+ * .setName(ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString())
+ * .setGcsUri("gcsUri-1251224875")
+ * .setDatabase("database1789464955")
+ * .setUser("user3599307")
+ * .build();
+ * OperationFuture future =
+ * alloyDBAdminClient.importClusterOperationCallable().futureCall(request);
+ * // Do something.
+ * ImportClusterResponse response = future.get();
+ * }
+ * }
+ */
+ public final OperationCallable
+ importClusterOperationCallable() {
+ return stub.importClusterOperationCallable();
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Imports data to the cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * ImportClusterRequest request =
+ * ImportClusterRequest.newBuilder()
+ * .setName(ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString())
+ * .setGcsUri("gcsUri-1251224875")
+ * .setDatabase("database1789464955")
+ * .setUser("user3599307")
+ * .build();
+ * ApiFuture future = alloyDBAdminClient.importClusterCallable().futureCall(request);
+ * // Do something.
+ * Operation response = future.get();
+ * }
+ * }
+ */
+ public final UnaryCallable importClusterCallable() {
+ return stub.importClusterCallable();
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Upgrades a single Cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * ClusterName name = ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]");
+ * DatabaseVersion version = DatabaseVersion.forNumber(0);
+ * UpgradeClusterResponse response = alloyDBAdminClient.upgradeClusterAsync(name, version).get();
+ * }
+ * }
+ *
+ * @param name Required. The resource name of the cluster.
+ * @param version Required. The version the cluster is going to be upgraded to.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture upgradeClusterAsync(
+ ClusterName name, DatabaseVersion version) {
+ UpgradeClusterRequest request =
+ UpgradeClusterRequest.newBuilder()
+ .setName(name == null ? null : name.toString())
+ .setVersion(version)
+ .build();
+ return upgradeClusterAsync(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Upgrades a single Cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * String name = ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString();
+ * DatabaseVersion version = DatabaseVersion.forNumber(0);
+ * UpgradeClusterResponse response = alloyDBAdminClient.upgradeClusterAsync(name, version).get();
+ * }
+ * }
+ *
+ * @param name Required. The resource name of the cluster.
+ * @param version Required. The version the cluster is going to be upgraded to.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture upgradeClusterAsync(
+ String name, DatabaseVersion version) {
+ UpgradeClusterRequest request =
+ UpgradeClusterRequest.newBuilder().setName(name).setVersion(version).build();
+ return upgradeClusterAsync(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Upgrades a single Cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * UpgradeClusterRequest request =
+ * UpgradeClusterRequest.newBuilder()
+ * .setName(ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString())
+ * .setVersion(DatabaseVersion.forNumber(0))
+ * .setRequestId("requestId693933066")
+ * .setValidateOnly(true)
+ * .setEtag("etag3123477")
+ * .build();
+ * UpgradeClusterResponse response = alloyDBAdminClient.upgradeClusterAsync(request).get();
+ * }
+ * }
+ *
+ * @param request The request object containing all of the parameters for the API call.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture upgradeClusterAsync(
+ UpgradeClusterRequest request) {
+ return upgradeClusterOperationCallable().futureCall(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Upgrades a single Cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * UpgradeClusterRequest request =
+ * UpgradeClusterRequest.newBuilder()
+ * .setName(ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString())
+ * .setVersion(DatabaseVersion.forNumber(0))
+ * .setRequestId("requestId693933066")
+ * .setValidateOnly(true)
+ * .setEtag("etag3123477")
+ * .build();
+ * OperationFuture future =
+ * alloyDBAdminClient.upgradeClusterOperationCallable().futureCall(request);
+ * // Do something.
+ * UpgradeClusterResponse response = future.get();
+ * }
+ * }
+ */
+ public final OperationCallable
+ upgradeClusterOperationCallable() {
+ return stub.upgradeClusterOperationCallable();
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Upgrades a single Cluster. Imperative only.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBAdminClient alloyDBAdminClient = AlloyDBAdminClient.create()) {
+ * UpgradeClusterRequest request =
+ * UpgradeClusterRequest.newBuilder()
+ * .setName(ClusterName.of("[PROJECT]", "[LOCATION]", "[CLUSTER]").toString())
+ * .setVersion(DatabaseVersion.forNumber(0))
+ * .setRequestId("requestId693933066")
+ * .setValidateOnly(true)
+ * .setEtag("etag3123477")
+ * .build();
+ * ApiFuture future = alloyDBAdminClient.upgradeClusterCallable().futureCall(request);
+ * // Do something.
+ * Operation response = future.get();
+ * }
+ * }
+ */
+ public final UnaryCallable upgradeClusterCallable() {
+ return stub.upgradeClusterCallable();
+ }
+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
/**
* Deletes a single Cluster.
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBAdminSettings.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBAdminSettings.java
index c232fd5f5836..06ba80a169a7 100644
--- a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBAdminSettings.java
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBAdminSettings.java
@@ -158,6 +158,39 @@ public UnaryCallSettings updateClusterSettings(
return ((AlloyDBAdminStubSettings) getStubSettings()).updateClusterOperationSettings();
}
+ /** Returns the object with the settings used for calls to exportCluster. */
+ public UnaryCallSettings exportClusterSettings() {
+ return ((AlloyDBAdminStubSettings) getStubSettings()).exportClusterSettings();
+ }
+
+ /** Returns the object with the settings used for calls to exportCluster. */
+ public OperationCallSettings
+ exportClusterOperationSettings() {
+ return ((AlloyDBAdminStubSettings) getStubSettings()).exportClusterOperationSettings();
+ }
+
+ /** Returns the object with the settings used for calls to importCluster. */
+ public UnaryCallSettings importClusterSettings() {
+ return ((AlloyDBAdminStubSettings) getStubSettings()).importClusterSettings();
+ }
+
+ /** Returns the object with the settings used for calls to importCluster. */
+ public OperationCallSettings
+ importClusterOperationSettings() {
+ return ((AlloyDBAdminStubSettings) getStubSettings()).importClusterOperationSettings();
+ }
+
+ /** Returns the object with the settings used for calls to upgradeCluster. */
+ public UnaryCallSettings upgradeClusterSettings() {
+ return ((AlloyDBAdminStubSettings) getStubSettings()).upgradeClusterSettings();
+ }
+
+ /** Returns the object with the settings used for calls to upgradeCluster. */
+ public OperationCallSettings
+ upgradeClusterOperationSettings() {
+ return ((AlloyDBAdminStubSettings) getStubSettings()).upgradeClusterOperationSettings();
+ }
+
/** Returns the object with the settings used for calls to deleteCluster. */
public UnaryCallSettings deleteClusterSettings() {
return ((AlloyDBAdminStubSettings) getStubSettings()).deleteClusterSettings();
@@ -574,6 +607,42 @@ public UnaryCallSettings.Builder updateClusterS
return getStubSettingsBuilder().updateClusterOperationSettings();
}
+ /** Returns the builder for the settings used for calls to exportCluster. */
+ public UnaryCallSettings.Builder exportClusterSettings() {
+ return getStubSettingsBuilder().exportClusterSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to exportCluster. */
+ public OperationCallSettings.Builder<
+ ExportClusterRequest, ExportClusterResponse, OperationMetadata>
+ exportClusterOperationSettings() {
+ return getStubSettingsBuilder().exportClusterOperationSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to importCluster. */
+ public UnaryCallSettings.Builder importClusterSettings() {
+ return getStubSettingsBuilder().importClusterSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to importCluster. */
+ public OperationCallSettings.Builder<
+ ImportClusterRequest, ImportClusterResponse, OperationMetadata>
+ importClusterOperationSettings() {
+ return getStubSettingsBuilder().importClusterOperationSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to upgradeCluster. */
+ public UnaryCallSettings.Builder upgradeClusterSettings() {
+ return getStubSettingsBuilder().upgradeClusterSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to upgradeCluster. */
+ public OperationCallSettings.Builder<
+ UpgradeClusterRequest, UpgradeClusterResponse, OperationMetadata>
+ upgradeClusterOperationSettings() {
+ return getStubSettingsBuilder().upgradeClusterOperationSettings();
+ }
+
/** Returns the builder for the settings used for calls to deleteCluster. */
public UnaryCallSettings.Builder deleteClusterSettings() {
return getStubSettingsBuilder().deleteClusterSettings();
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBCSQLAdminClient.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBCSQLAdminClient.java
new file mode 100644
index 000000000000..b79cad96443a
--- /dev/null
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBCSQLAdminClient.java
@@ -0,0 +1,674 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.cloud.alloydb.v1;
+
+import com.google.api.core.ApiFuture;
+import com.google.api.core.ApiFutures;
+import com.google.api.core.BetaApi;
+import com.google.api.gax.core.BackgroundResource;
+import com.google.api.gax.httpjson.longrunning.OperationsClient;
+import com.google.api.gax.longrunning.OperationFuture;
+import com.google.api.gax.paging.AbstractFixedSizeCollection;
+import com.google.api.gax.paging.AbstractPage;
+import com.google.api.gax.paging.AbstractPagedListResponse;
+import com.google.api.gax.rpc.OperationCallable;
+import com.google.api.gax.rpc.PageContext;
+import com.google.api.gax.rpc.UnaryCallable;
+import com.google.cloud.alloydb.v1.stub.AlloyDBCSQLAdminStub;
+import com.google.cloud.alloydb.v1.stub.AlloyDBCSQLAdminStubSettings;
+import com.google.cloud.location.GetLocationRequest;
+import com.google.cloud.location.ListLocationsRequest;
+import com.google.cloud.location.ListLocationsResponse;
+import com.google.cloud.location.Location;
+import com.google.common.util.concurrent.MoreExecutors;
+import com.google.longrunning.Operation;
+import java.io.IOException;
+import java.util.List;
+import java.util.concurrent.TimeUnit;
+import javax.annotation.Generated;
+
+// AUTO-GENERATED DOCUMENTATION AND CLASS.
+/**
+ * Service Description: Service for interactions with CloudSQL.
+ *
+ * This class provides the ability to make remote calls to the backing service through method
+ * calls that map to API methods. Sample code to get started:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * GetLocationRequest request = GetLocationRequest.newBuilder().setName("name3373707").build();
+ * Location response = alloyDBCSQLAdminClient.getLocation(request);
+ * }
+ * }
+ *
+ * Note: close() needs to be called on the AlloyDBCSQLAdminClient object to clean up resources
+ * such as threads. In the example above, try-with-resources is used, which automatically calls
+ * close().
+ *
+ *
+ * Methods
+ *
+ * Method
+ * Description
+ * Method Variants
+ *
+ *
+ * RestoreFromCloudSQL
Restores an AlloyDB cluster from a CloudSQL resource.
+ * Request object method variants only take one parameter, a request object, which must be constructed before the call.
+ *
+ * restoreFromCloudSQLAsync(RestoreFromCloudSQLRequest request)
+ *
+ * Methods that return long-running operations have "Async" method variants that return `OperationFuture`, which is used to track polling of the service.
+ *
+ * restoreFromCloudSQLAsync(LocationName parent, String clusterId)
+ *
restoreFromCloudSQLAsync(String parent, String clusterId)
+ *
+ * Callable method variants take no parameters and return an immutable API callable object, which can be used to initiate calls to the service.
+ *
+ * restoreFromCloudSQLOperationCallable()
+ *
restoreFromCloudSQLCallable()
+ *
+ *
+ *
+ *
+ * ListLocations
Lists information about the supported locations for this service.
+ * Request object method variants only take one parameter, a request object, which must be constructed before the call.
+ *
+ * listLocations(ListLocationsRequest request)
+ *
+ * Callable method variants take no parameters and return an immutable API callable object, which can be used to initiate calls to the service.
+ *
+ * listLocationsPagedCallable()
+ *
listLocationsCallable()
+ *
+ *
+ *
+ *
+ * GetLocation
Gets information about a location.
+ * Request object method variants only take one parameter, a request object, which must be constructed before the call.
+ *
+ * getLocation(GetLocationRequest request)
+ *
+ * Callable method variants take no parameters and return an immutable API callable object, which can be used to initiate calls to the service.
+ *
+ * getLocationCallable()
+ *
+ *
+ *
+ *
+ *
+ * See the individual methods for example code.
+ *
+ *
Many parameters require resource names to be formatted in a particular way. To assist with
+ * these names, this class includes a format method for each type of name, and additionally a parse
+ * method to extract the individual identifiers contained within names that are returned.
+ *
+ *
This class can be customized by passing in a custom instance of AlloyDBCSQLAdminSettings to
+ * create(). For example:
+ *
+ *
To customize credentials:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * AlloyDBCSQLAdminSettings alloyDBCSQLAdminSettings =
+ * AlloyDBCSQLAdminSettings.newBuilder()
+ * .setCredentialsProvider(FixedCredentialsProvider.create(myCredentials))
+ * .build();
+ * AlloyDBCSQLAdminClient alloyDBCSQLAdminClient =
+ * AlloyDBCSQLAdminClient.create(alloyDBCSQLAdminSettings);
+ * }
+ *
+ * To customize the endpoint:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * AlloyDBCSQLAdminSettings alloyDBCSQLAdminSettings =
+ * AlloyDBCSQLAdminSettings.newBuilder().setEndpoint(myEndpoint).build();
+ * AlloyDBCSQLAdminClient alloyDBCSQLAdminClient =
+ * AlloyDBCSQLAdminClient.create(alloyDBCSQLAdminSettings);
+ * }
+ *
+ * To use REST (HTTP1.1/JSON) transport (instead of gRPC) for sending and receiving requests over
+ * the wire:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * AlloyDBCSQLAdminSettings alloyDBCSQLAdminSettings =
+ * AlloyDBCSQLAdminSettings.newHttpJsonBuilder().build();
+ * AlloyDBCSQLAdminClient alloyDBCSQLAdminClient =
+ * AlloyDBCSQLAdminClient.create(alloyDBCSQLAdminSettings);
+ * }
+ *
+ * Please refer to the GitHub repository's samples for more quickstart code snippets.
+ */
+@Generated("by gapic-generator-java")
+public class AlloyDBCSQLAdminClient implements BackgroundResource {
+ private final AlloyDBCSQLAdminSettings settings;
+ private final AlloyDBCSQLAdminStub stub;
+ private final OperationsClient httpJsonOperationsClient;
+ private final com.google.longrunning.OperationsClient operationsClient;
+
+ /** Constructs an instance of AlloyDBCSQLAdminClient with default settings. */
+ public static final AlloyDBCSQLAdminClient create() throws IOException {
+ return create(AlloyDBCSQLAdminSettings.newBuilder().build());
+ }
+
+ /**
+ * Constructs an instance of AlloyDBCSQLAdminClient, using the given settings. The channels are
+ * created based on the settings passed in, or defaults for any settings that are not set.
+ */
+ public static final AlloyDBCSQLAdminClient create(AlloyDBCSQLAdminSettings settings)
+ throws IOException {
+ return new AlloyDBCSQLAdminClient(settings);
+ }
+
+ /**
+ * Constructs an instance of AlloyDBCSQLAdminClient, using the given stub for making calls. This
+ * is for advanced usage - prefer using create(AlloyDBCSQLAdminSettings).
+ */
+ public static final AlloyDBCSQLAdminClient create(AlloyDBCSQLAdminStub stub) {
+ return new AlloyDBCSQLAdminClient(stub);
+ }
+
+ /**
+ * Constructs an instance of AlloyDBCSQLAdminClient, using the given settings. This is protected
+ * so that it is easy to make a subclass, but otherwise, the static factory methods should be
+ * preferred.
+ */
+ protected AlloyDBCSQLAdminClient(AlloyDBCSQLAdminSettings settings) throws IOException {
+ this.settings = settings;
+ this.stub = ((AlloyDBCSQLAdminStubSettings) settings.getStubSettings()).createStub();
+ this.operationsClient =
+ com.google.longrunning.OperationsClient.create(this.stub.getOperationsStub());
+ this.httpJsonOperationsClient = OperationsClient.create(this.stub.getHttpJsonOperationsStub());
+ }
+
+ protected AlloyDBCSQLAdminClient(AlloyDBCSQLAdminStub stub) {
+ this.settings = null;
+ this.stub = stub;
+ this.operationsClient =
+ com.google.longrunning.OperationsClient.create(this.stub.getOperationsStub());
+ this.httpJsonOperationsClient = OperationsClient.create(this.stub.getHttpJsonOperationsStub());
+ }
+
+ public final AlloyDBCSQLAdminSettings getSettings() {
+ return settings;
+ }
+
+ public AlloyDBCSQLAdminStub getStub() {
+ return stub;
+ }
+
+ /**
+ * Returns the OperationsClient that can be used to query the status of a long-running operation
+ * returned by another API method call.
+ */
+ public final com.google.longrunning.OperationsClient getOperationsClient() {
+ return operationsClient;
+ }
+
+ /**
+ * Returns the OperationsClient that can be used to query the status of a long-running operation
+ * returned by another API method call.
+ */
+ @BetaApi
+ public final OperationsClient getHttpJsonOperationsClient() {
+ return httpJsonOperationsClient;
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Restores an AlloyDB cluster from a CloudSQL resource.
+ *
+ *
Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
+ * String clusterId = "clusterId561939637";
+ * Cluster response = alloyDBCSQLAdminClient.restoreFromCloudSQLAsync(parent, clusterId).get();
+ * }
+ * }
+ *
+ * @param parent Required. The location of the new cluster. For the required format, see the
+ * comment on Cluster.name field.
+ * @param clusterId Required. ID of the requesting object.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture restoreFromCloudSQLAsync(
+ LocationName parent, String clusterId) {
+ RestoreFromCloudSQLRequest request =
+ RestoreFromCloudSQLRequest.newBuilder()
+ .setParent(parent == null ? null : parent.toString())
+ .setClusterId(clusterId)
+ .build();
+ return restoreFromCloudSQLAsync(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Restores an AlloyDB cluster from a CloudSQL resource.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * String parent = LocationName.of("[PROJECT]", "[LOCATION]").toString();
+ * String clusterId = "clusterId561939637";
+ * Cluster response = alloyDBCSQLAdminClient.restoreFromCloudSQLAsync(parent, clusterId).get();
+ * }
+ * }
+ *
+ * @param parent Required. The location of the new cluster. For the required format, see the
+ * comment on Cluster.name field.
+ * @param clusterId Required. ID of the requesting object.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture restoreFromCloudSQLAsync(
+ String parent, String clusterId) {
+ RestoreFromCloudSQLRequest request =
+ RestoreFromCloudSQLRequest.newBuilder().setParent(parent).setClusterId(clusterId).build();
+ return restoreFromCloudSQLAsync(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Restores an AlloyDB cluster from a CloudSQL resource.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * RestoreFromCloudSQLRequest request =
+ * RestoreFromCloudSQLRequest.newBuilder()
+ * .setParent(LocationName.of("[PROJECT]", "[LOCATION]").toString())
+ * .setClusterId("clusterId561939637")
+ * .setCluster(Cluster.newBuilder().build())
+ * .build();
+ * Cluster response = alloyDBCSQLAdminClient.restoreFromCloudSQLAsync(request).get();
+ * }
+ * }
+ *
+ * @param request The request object containing all of the parameters for the API call.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final OperationFuture restoreFromCloudSQLAsync(
+ RestoreFromCloudSQLRequest request) {
+ return restoreFromCloudSQLOperationCallable().futureCall(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Restores an AlloyDB cluster from a CloudSQL resource.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * RestoreFromCloudSQLRequest request =
+ * RestoreFromCloudSQLRequest.newBuilder()
+ * .setParent(LocationName.of("[PROJECT]", "[LOCATION]").toString())
+ * .setClusterId("clusterId561939637")
+ * .setCluster(Cluster.newBuilder().build())
+ * .build();
+ * OperationFuture future =
+ * alloyDBCSQLAdminClient.restoreFromCloudSQLOperationCallable().futureCall(request);
+ * // Do something.
+ * Cluster response = future.get();
+ * }
+ * }
+ */
+ public final OperationCallable
+ restoreFromCloudSQLOperationCallable() {
+ return stub.restoreFromCloudSQLOperationCallable();
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Restores an AlloyDB cluster from a CloudSQL resource.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * RestoreFromCloudSQLRequest request =
+ * RestoreFromCloudSQLRequest.newBuilder()
+ * .setParent(LocationName.of("[PROJECT]", "[LOCATION]").toString())
+ * .setClusterId("clusterId561939637")
+ * .setCluster(Cluster.newBuilder().build())
+ * .build();
+ * ApiFuture future =
+ * alloyDBCSQLAdminClient.restoreFromCloudSQLCallable().futureCall(request);
+ * // Do something.
+ * Operation response = future.get();
+ * }
+ * }
+ */
+ public final UnaryCallable restoreFromCloudSQLCallable() {
+ return stub.restoreFromCloudSQLCallable();
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Lists information about the supported locations for this service.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * ListLocationsRequest request =
+ * ListLocationsRequest.newBuilder()
+ * .setName("name3373707")
+ * .setFilter("filter-1274492040")
+ * .setPageSize(883849137)
+ * .setPageToken("pageToken873572522")
+ * .build();
+ * for (Location element : alloyDBCSQLAdminClient.listLocations(request).iterateAll()) {
+ * // doThingsWith(element);
+ * }
+ * }
+ * }
+ *
+ * @param request The request object containing all of the parameters for the API call.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final ListLocationsPagedResponse listLocations(ListLocationsRequest request) {
+ return listLocationsPagedCallable().call(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Lists information about the supported locations for this service.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * ListLocationsRequest request =
+ * ListLocationsRequest.newBuilder()
+ * .setName("name3373707")
+ * .setFilter("filter-1274492040")
+ * .setPageSize(883849137)
+ * .setPageToken("pageToken873572522")
+ * .build();
+ * ApiFuture future =
+ * alloyDBCSQLAdminClient.listLocationsPagedCallable().futureCall(request);
+ * // Do something.
+ * for (Location element : future.get().iterateAll()) {
+ * // doThingsWith(element);
+ * }
+ * }
+ * }
+ */
+ public final UnaryCallable
+ listLocationsPagedCallable() {
+ return stub.listLocationsPagedCallable();
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Lists information about the supported locations for this service.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * ListLocationsRequest request =
+ * ListLocationsRequest.newBuilder()
+ * .setName("name3373707")
+ * .setFilter("filter-1274492040")
+ * .setPageSize(883849137)
+ * .setPageToken("pageToken873572522")
+ * .build();
+ * while (true) {
+ * ListLocationsResponse response =
+ * alloyDBCSQLAdminClient.listLocationsCallable().call(request);
+ * for (Location element : response.getLocationsList()) {
+ * // doThingsWith(element);
+ * }
+ * String nextPageToken = response.getNextPageToken();
+ * if (!Strings.isNullOrEmpty(nextPageToken)) {
+ * request = request.toBuilder().setPageToken(nextPageToken).build();
+ * } else {
+ * break;
+ * }
+ * }
+ * }
+ * }
+ */
+ public final UnaryCallable listLocationsCallable() {
+ return stub.listLocationsCallable();
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Gets information about a location.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * GetLocationRequest request = GetLocationRequest.newBuilder().setName("name3373707").build();
+ * Location response = alloyDBCSQLAdminClient.getLocation(request);
+ * }
+ * }
+ *
+ * @param request The request object containing all of the parameters for the API call.
+ * @throws com.google.api.gax.rpc.ApiException if the remote call fails
+ */
+ public final Location getLocation(GetLocationRequest request) {
+ return getLocationCallable().call(request);
+ }
+
+ // AUTO-GENERATED DOCUMENTATION AND METHOD.
+ /**
+ * Gets information about a location.
+ *
+ * Sample code:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * GetLocationRequest request = GetLocationRequest.newBuilder().setName("name3373707").build();
+ * ApiFuture future = alloyDBCSQLAdminClient.getLocationCallable().futureCall(request);
+ * // Do something.
+ * Location response = future.get();
+ * }
+ * }
+ */
+ public final UnaryCallable getLocationCallable() {
+ return stub.getLocationCallable();
+ }
+
+ @Override
+ public final void close() {
+ stub.close();
+ }
+
+ @Override
+ public void shutdown() {
+ stub.shutdown();
+ }
+
+ @Override
+ public boolean isShutdown() {
+ return stub.isShutdown();
+ }
+
+ @Override
+ public boolean isTerminated() {
+ return stub.isTerminated();
+ }
+
+ @Override
+ public void shutdownNow() {
+ stub.shutdownNow();
+ }
+
+ @Override
+ public boolean awaitTermination(long duration, TimeUnit unit) throws InterruptedException {
+ return stub.awaitTermination(duration, unit);
+ }
+
+ public static class ListLocationsPagedResponse
+ extends AbstractPagedListResponse<
+ ListLocationsRequest,
+ ListLocationsResponse,
+ Location,
+ ListLocationsPage,
+ ListLocationsFixedSizeCollection> {
+
+ public static ApiFuture createAsync(
+ PageContext context,
+ ApiFuture futureResponse) {
+ ApiFuture futurePage =
+ ListLocationsPage.createEmptyPage().createPageAsync(context, futureResponse);
+ return ApiFutures.transform(
+ futurePage,
+ input -> new ListLocationsPagedResponse(input),
+ MoreExecutors.directExecutor());
+ }
+
+ private ListLocationsPagedResponse(ListLocationsPage page) {
+ super(page, ListLocationsFixedSizeCollection.createEmptyCollection());
+ }
+ }
+
+ public static class ListLocationsPage
+ extends AbstractPage<
+ ListLocationsRequest, ListLocationsResponse, Location, ListLocationsPage> {
+
+ private ListLocationsPage(
+ PageContext context,
+ ListLocationsResponse response) {
+ super(context, response);
+ }
+
+ private static ListLocationsPage createEmptyPage() {
+ return new ListLocationsPage(null, null);
+ }
+
+ @Override
+ protected ListLocationsPage createPage(
+ PageContext context,
+ ListLocationsResponse response) {
+ return new ListLocationsPage(context, response);
+ }
+
+ @Override
+ public ApiFuture createPageAsync(
+ PageContext context,
+ ApiFuture futureResponse) {
+ return super.createPageAsync(context, futureResponse);
+ }
+ }
+
+ public static class ListLocationsFixedSizeCollection
+ extends AbstractFixedSizeCollection<
+ ListLocationsRequest,
+ ListLocationsResponse,
+ Location,
+ ListLocationsPage,
+ ListLocationsFixedSizeCollection> {
+
+ private ListLocationsFixedSizeCollection(List pages, int collectionSize) {
+ super(pages, collectionSize);
+ }
+
+ private static ListLocationsFixedSizeCollection createEmptyCollection() {
+ return new ListLocationsFixedSizeCollection(null, 0);
+ }
+
+ @Override
+ protected ListLocationsFixedSizeCollection createCollection(
+ List pages, int collectionSize) {
+ return new ListLocationsFixedSizeCollection(pages, collectionSize);
+ }
+ }
+}
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBCSQLAdminSettings.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBCSQLAdminSettings.java
new file mode 100644
index 000000000000..f3e1d7103428
--- /dev/null
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/AlloyDBCSQLAdminSettings.java
@@ -0,0 +1,287 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.cloud.alloydb.v1;
+
+import static com.google.cloud.alloydb.v1.AlloyDBCSQLAdminClient.ListLocationsPagedResponse;
+
+import com.google.api.core.ApiFunction;
+import com.google.api.core.BetaApi;
+import com.google.api.gax.core.GoogleCredentialsProvider;
+import com.google.api.gax.core.InstantiatingExecutorProvider;
+import com.google.api.gax.grpc.InstantiatingGrpcChannelProvider;
+import com.google.api.gax.httpjson.InstantiatingHttpJsonChannelProvider;
+import com.google.api.gax.rpc.ApiClientHeaderProvider;
+import com.google.api.gax.rpc.ClientContext;
+import com.google.api.gax.rpc.ClientSettings;
+import com.google.api.gax.rpc.OperationCallSettings;
+import com.google.api.gax.rpc.PagedCallSettings;
+import com.google.api.gax.rpc.TransportChannelProvider;
+import com.google.api.gax.rpc.UnaryCallSettings;
+import com.google.cloud.alloydb.v1.stub.AlloyDBCSQLAdminStubSettings;
+import com.google.cloud.location.GetLocationRequest;
+import com.google.cloud.location.ListLocationsRequest;
+import com.google.cloud.location.ListLocationsResponse;
+import com.google.cloud.location.Location;
+import com.google.longrunning.Operation;
+import java.io.IOException;
+import java.util.List;
+import javax.annotation.Generated;
+
+// AUTO-GENERATED DOCUMENTATION AND CLASS.
+/**
+ * Settings class to configure an instance of {@link AlloyDBCSQLAdminClient}.
+ *
+ * The default instance has everything set to sensible defaults:
+ *
+ *
+ * - The default service address (alloydb.googleapis.com) and default port (443) are used.
+ *
- Credentials are acquired automatically through Application Default Credentials.
+ *
- Retries are configured for idempotent methods but not for non-idempotent methods.
+ *
+ *
+ * The builder of this class is recursive, so contained classes are themselves builders. When
+ * build() is called, the tree of builders is called to create the complete settings object.
+ *
+ *
For example, to set the
+ * [RetrySettings](https://cloud.google.com/java/docs/reference/gax/latest/com.google.api.gax.retrying.RetrySettings)
+ * of getLocation:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * AlloyDBCSQLAdminSettings.Builder alloyDBCSQLAdminSettingsBuilder =
+ * AlloyDBCSQLAdminSettings.newBuilder();
+ * alloyDBCSQLAdminSettingsBuilder
+ * .getLocationSettings()
+ * .setRetrySettings(
+ * alloyDBCSQLAdminSettingsBuilder
+ * .getLocationSettings()
+ * .getRetrySettings()
+ * .toBuilder()
+ * .setInitialRetryDelayDuration(Duration.ofSeconds(1))
+ * .setInitialRpcTimeoutDuration(Duration.ofSeconds(5))
+ * .setMaxAttempts(5)
+ * .setMaxRetryDelayDuration(Duration.ofSeconds(30))
+ * .setMaxRpcTimeoutDuration(Duration.ofSeconds(60))
+ * .setRetryDelayMultiplier(1.3)
+ * .setRpcTimeoutMultiplier(1.5)
+ * .setTotalTimeoutDuration(Duration.ofSeconds(300))
+ * .build());
+ * AlloyDBCSQLAdminSettings alloyDBCSQLAdminSettings = alloyDBCSQLAdminSettingsBuilder.build();
+ * }
+ *
+ * Please refer to the [Client Side Retry
+ * Guide](https://github.com/googleapis/google-cloud-java/blob/main/docs/client_retries.md) for
+ * additional support in setting retries.
+ *
+ * To configure the RetrySettings of a Long Running Operation method, create an
+ * OperationTimedPollAlgorithm object and update the RPC's polling algorithm. For example, to
+ * configure the RetrySettings for restoreFromCloudSQL:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * AlloyDBCSQLAdminSettings.Builder alloyDBCSQLAdminSettingsBuilder =
+ * AlloyDBCSQLAdminSettings.newBuilder();
+ * TimedRetryAlgorithm timedRetryAlgorithm =
+ * OperationalTimedPollAlgorithm.create(
+ * RetrySettings.newBuilder()
+ * .setInitialRetryDelayDuration(Duration.ofMillis(500))
+ * .setRetryDelayMultiplier(1.5)
+ * .setMaxRetryDelayDuration(Duration.ofMillis(5000))
+ * .setTotalTimeoutDuration(Duration.ofHours(24))
+ * .build());
+ * alloyDBCSQLAdminSettingsBuilder
+ * .createClusterOperationSettings()
+ * .setPollingAlgorithm(timedRetryAlgorithm)
+ * .build();
+ * }
+ */
+@Generated("by gapic-generator-java")
+public class AlloyDBCSQLAdminSettings extends ClientSettings {
+
+ /** Returns the object with the settings used for calls to restoreFromCloudSQL. */
+ public UnaryCallSettings restoreFromCloudSQLSettings() {
+ return ((AlloyDBCSQLAdminStubSettings) getStubSettings()).restoreFromCloudSQLSettings();
+ }
+
+ /** Returns the object with the settings used for calls to restoreFromCloudSQL. */
+ public OperationCallSettings
+ restoreFromCloudSQLOperationSettings() {
+ return ((AlloyDBCSQLAdminStubSettings) getStubSettings())
+ .restoreFromCloudSQLOperationSettings();
+ }
+
+ /** Returns the object with the settings used for calls to listLocations. */
+ public PagedCallSettings
+ listLocationsSettings() {
+ return ((AlloyDBCSQLAdminStubSettings) getStubSettings()).listLocationsSettings();
+ }
+
+ /** Returns the object with the settings used for calls to getLocation. */
+ public UnaryCallSettings getLocationSettings() {
+ return ((AlloyDBCSQLAdminStubSettings) getStubSettings()).getLocationSettings();
+ }
+
+ public static final AlloyDBCSQLAdminSettings create(AlloyDBCSQLAdminStubSettings stub)
+ throws IOException {
+ return new AlloyDBCSQLAdminSettings.Builder(stub.toBuilder()).build();
+ }
+
+ /** Returns a builder for the default ExecutorProvider for this service. */
+ public static InstantiatingExecutorProvider.Builder defaultExecutorProviderBuilder() {
+ return AlloyDBCSQLAdminStubSettings.defaultExecutorProviderBuilder();
+ }
+
+ /** Returns the default service endpoint. */
+ public static String getDefaultEndpoint() {
+ return AlloyDBCSQLAdminStubSettings.getDefaultEndpoint();
+ }
+
+ /** Returns the default service scopes. */
+ public static List getDefaultServiceScopes() {
+ return AlloyDBCSQLAdminStubSettings.getDefaultServiceScopes();
+ }
+
+ /** Returns a builder for the default credentials for this service. */
+ public static GoogleCredentialsProvider.Builder defaultCredentialsProviderBuilder() {
+ return AlloyDBCSQLAdminStubSettings.defaultCredentialsProviderBuilder();
+ }
+
+ /** Returns a builder for the default gRPC ChannelProvider for this service. */
+ public static InstantiatingGrpcChannelProvider.Builder defaultGrpcTransportProviderBuilder() {
+ return AlloyDBCSQLAdminStubSettings.defaultGrpcTransportProviderBuilder();
+ }
+
+ /** Returns a builder for the default REST ChannelProvider for this service. */
+ @BetaApi
+ public static InstantiatingHttpJsonChannelProvider.Builder
+ defaultHttpJsonTransportProviderBuilder() {
+ return AlloyDBCSQLAdminStubSettings.defaultHttpJsonTransportProviderBuilder();
+ }
+
+ public static TransportChannelProvider defaultTransportChannelProvider() {
+ return AlloyDBCSQLAdminStubSettings.defaultTransportChannelProvider();
+ }
+
+ public static ApiClientHeaderProvider.Builder defaultApiClientHeaderProviderBuilder() {
+ return AlloyDBCSQLAdminStubSettings.defaultApiClientHeaderProviderBuilder();
+ }
+
+ /** Returns a new gRPC builder for this class. */
+ public static Builder newBuilder() {
+ return Builder.createDefault();
+ }
+
+ /** Returns a new REST builder for this class. */
+ public static Builder newHttpJsonBuilder() {
+ return Builder.createHttpJsonDefault();
+ }
+
+ /** Returns a new builder for this class. */
+ public static Builder newBuilder(ClientContext clientContext) {
+ return new Builder(clientContext);
+ }
+
+ /** Returns a builder containing all the values of this settings class. */
+ public Builder toBuilder() {
+ return new Builder(this);
+ }
+
+ protected AlloyDBCSQLAdminSettings(Builder settingsBuilder) throws IOException {
+ super(settingsBuilder);
+ }
+
+ /** Builder for AlloyDBCSQLAdminSettings. */
+ public static class Builder extends ClientSettings.Builder {
+
+ protected Builder() throws IOException {
+ this(((ClientContext) null));
+ }
+
+ protected Builder(ClientContext clientContext) {
+ super(AlloyDBCSQLAdminStubSettings.newBuilder(clientContext));
+ }
+
+ protected Builder(AlloyDBCSQLAdminSettings settings) {
+ super(settings.getStubSettings().toBuilder());
+ }
+
+ protected Builder(AlloyDBCSQLAdminStubSettings.Builder stubSettings) {
+ super(stubSettings);
+ }
+
+ private static Builder createDefault() {
+ return new Builder(AlloyDBCSQLAdminStubSettings.newBuilder());
+ }
+
+ private static Builder createHttpJsonDefault() {
+ return new Builder(AlloyDBCSQLAdminStubSettings.newHttpJsonBuilder());
+ }
+
+ public AlloyDBCSQLAdminStubSettings.Builder getStubSettingsBuilder() {
+ return ((AlloyDBCSQLAdminStubSettings.Builder) getStubSettings());
+ }
+
+ /**
+ * Applies the given settings updater function to all of the unary API methods in this service.
+ *
+ * Note: This method does not support applying settings to streaming methods.
+ */
+ public Builder applyToAllUnaryMethods(
+ ApiFunction, Void> settingsUpdater) {
+ super.applyToAllUnaryMethods(
+ getStubSettingsBuilder().unaryMethodSettingsBuilders(), settingsUpdater);
+ return this;
+ }
+
+ /** Returns the builder for the settings used for calls to restoreFromCloudSQL. */
+ public UnaryCallSettings.Builder
+ restoreFromCloudSQLSettings() {
+ return getStubSettingsBuilder().restoreFromCloudSQLSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to restoreFromCloudSQL. */
+ public OperationCallSettings.Builder
+ restoreFromCloudSQLOperationSettings() {
+ return getStubSettingsBuilder().restoreFromCloudSQLOperationSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to listLocations. */
+ public PagedCallSettings.Builder<
+ ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
+ listLocationsSettings() {
+ return getStubSettingsBuilder().listLocationsSettings();
+ }
+
+ /** Returns the builder for the settings used for calls to getLocation. */
+ public UnaryCallSettings.Builder getLocationSettings() {
+ return getStubSettingsBuilder().getLocationSettings();
+ }
+
+ @Override
+ public AlloyDBCSQLAdminSettings build() throws IOException {
+ return new AlloyDBCSQLAdminSettings(this);
+ }
+ }
+}
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/gapic_metadata.json b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/gapic_metadata.json
index 506253f1e827..8611dc76cff8 100644
--- a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/gapic_metadata.json
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/gapic_metadata.json
@@ -5,6 +5,24 @@
"protoPackage": "google.cloud.alloydb.v1",
"libraryPackage": "com.google.cloud.alloydb.v1",
"services": {
+ "AlloyDBCSQLAdmin": {
+ "clients": {
+ "grpc": {
+ "libraryClient": "AlloyDBCSQLAdminClient",
+ "rpcs": {
+ "GetLocation": {
+ "methods": ["getLocation", "getLocationCallable"]
+ },
+ "ListLocations": {
+ "methods": ["listLocations", "listLocationsPagedCallable", "listLocationsCallable"]
+ },
+ "RestoreFromCloudSQL": {
+ "methods": ["restoreFromCloudSQLAsync", "restoreFromCloudSQLAsync", "restoreFromCloudSQLAsync", "restoreFromCloudSQLOperationCallable", "restoreFromCloudSQLCallable"]
+ }
+ }
+ }
+ }
+ },
"AlloyDBAdmin": {
"clients": {
"grpc": {
@@ -46,6 +64,9 @@
"ExecuteSql": {
"methods": ["executeSql", "executeSql", "executeSql", "executeSqlCallable"]
},
+ "ExportCluster": {
+ "methods": ["exportClusterAsync", "exportClusterAsync", "exportClusterAsync", "exportClusterOperationCallable", "exportClusterCallable"]
+ },
"FailoverInstance": {
"methods": ["failoverInstanceAsync", "failoverInstanceAsync", "failoverInstanceAsync", "failoverInstanceOperationCallable", "failoverInstanceCallable"]
},
@@ -70,6 +91,9 @@
"GetUser": {
"methods": ["getUser", "getUser", "getUser", "getUserCallable"]
},
+ "ImportCluster": {
+ "methods": ["importClusterAsync", "importClusterAsync", "importClusterAsync", "importClusterOperationCallable", "importClusterCallable"]
+ },
"InjectFault": {
"methods": ["injectFaultAsync", "injectFaultAsync", "injectFaultAsync", "injectFaultOperationCallable", "injectFaultCallable"]
},
@@ -117,6 +141,9 @@
},
"UpdateUser": {
"methods": ["updateUser", "updateUser", "updateUserCallable"]
+ },
+ "UpgradeCluster": {
+ "methods": ["upgradeClusterAsync", "upgradeClusterAsync", "upgradeClusterAsync", "upgradeClusterOperationCallable", "upgradeClusterCallable"]
}
}
}
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/package-info.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/package-info.java
index af49cf2d89a3..38f9aa96438a 100644
--- a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/package-info.java
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/package-info.java
@@ -19,6 +19,24 @@
*
* The interfaces provided are listed below, along with usage samples.
*
+ *
======================= AlloyDBCSQLAdminClient =======================
+ *
+ *
Service Description: Service for interactions with CloudSQL.
+ *
+ *
Sample for AlloyDBCSQLAdminClient:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * try (AlloyDBCSQLAdminClient alloyDBCSQLAdminClient = AlloyDBCSQLAdminClient.create()) {
+ * GetLocationRequest request = GetLocationRequest.newBuilder().setName("name3373707").build();
+ * Location response = alloyDBCSQLAdminClient.getLocation(request);
+ * }
+ * }
+ *
* ======================= AlloyDBAdminClient =======================
*
*
Service Description: Service describing handlers for resources
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBAdminStub.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBAdminStub.java
index 335183af5b26..3af12e923ff1 100644
--- a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBAdminStub.java
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBAdminStub.java
@@ -44,6 +44,8 @@
import com.google.cloud.alloydb.v1.DeleteUserRequest;
import com.google.cloud.alloydb.v1.ExecuteSqlRequest;
import com.google.cloud.alloydb.v1.ExecuteSqlResponse;
+import com.google.cloud.alloydb.v1.ExportClusterRequest;
+import com.google.cloud.alloydb.v1.ExportClusterResponse;
import com.google.cloud.alloydb.v1.FailoverInstanceRequest;
import com.google.cloud.alloydb.v1.GenerateClientCertificateRequest;
import com.google.cloud.alloydb.v1.GenerateClientCertificateResponse;
@@ -52,6 +54,8 @@
import com.google.cloud.alloydb.v1.GetConnectionInfoRequest;
import com.google.cloud.alloydb.v1.GetInstanceRequest;
import com.google.cloud.alloydb.v1.GetUserRequest;
+import com.google.cloud.alloydb.v1.ImportClusterRequest;
+import com.google.cloud.alloydb.v1.ImportClusterResponse;
import com.google.cloud.alloydb.v1.InjectFaultRequest;
import com.google.cloud.alloydb.v1.Instance;
import com.google.cloud.alloydb.v1.ListBackupsRequest;
@@ -75,6 +79,8 @@
import com.google.cloud.alloydb.v1.UpdateClusterRequest;
import com.google.cloud.alloydb.v1.UpdateInstanceRequest;
import com.google.cloud.alloydb.v1.UpdateUserRequest;
+import com.google.cloud.alloydb.v1.UpgradeClusterRequest;
+import com.google.cloud.alloydb.v1.UpgradeClusterResponse;
import com.google.cloud.alloydb.v1.User;
import com.google.cloud.location.GetLocationRequest;
import com.google.cloud.location.ListLocationsRequest;
@@ -132,6 +138,33 @@ public UnaryCallable updateClusterCallable() {
throw new UnsupportedOperationException("Not implemented: updateClusterCallable()");
}
+ public OperationCallable
+ exportClusterOperationCallable() {
+ throw new UnsupportedOperationException("Not implemented: exportClusterOperationCallable()");
+ }
+
+ public UnaryCallable exportClusterCallable() {
+ throw new UnsupportedOperationException("Not implemented: exportClusterCallable()");
+ }
+
+ public OperationCallable
+ importClusterOperationCallable() {
+ throw new UnsupportedOperationException("Not implemented: importClusterOperationCallable()");
+ }
+
+ public UnaryCallable importClusterCallable() {
+ throw new UnsupportedOperationException("Not implemented: importClusterCallable()");
+ }
+
+ public OperationCallable
+ upgradeClusterOperationCallable() {
+ throw new UnsupportedOperationException("Not implemented: upgradeClusterOperationCallable()");
+ }
+
+ public UnaryCallable upgradeClusterCallable() {
+ throw new UnsupportedOperationException("Not implemented: upgradeClusterCallable()");
+ }
+
public OperationCallable
deleteClusterOperationCallable() {
throw new UnsupportedOperationException("Not implemented: deleteClusterOperationCallable()");
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBAdminStubSettings.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBAdminStubSettings.java
index 817530f64b6a..904fc7c94b24 100644
--- a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBAdminStubSettings.java
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBAdminStubSettings.java
@@ -72,6 +72,8 @@
import com.google.cloud.alloydb.v1.DeleteUserRequest;
import com.google.cloud.alloydb.v1.ExecuteSqlRequest;
import com.google.cloud.alloydb.v1.ExecuteSqlResponse;
+import com.google.cloud.alloydb.v1.ExportClusterRequest;
+import com.google.cloud.alloydb.v1.ExportClusterResponse;
import com.google.cloud.alloydb.v1.FailoverInstanceRequest;
import com.google.cloud.alloydb.v1.GenerateClientCertificateRequest;
import com.google.cloud.alloydb.v1.GenerateClientCertificateResponse;
@@ -80,6 +82,8 @@
import com.google.cloud.alloydb.v1.GetConnectionInfoRequest;
import com.google.cloud.alloydb.v1.GetInstanceRequest;
import com.google.cloud.alloydb.v1.GetUserRequest;
+import com.google.cloud.alloydb.v1.ImportClusterRequest;
+import com.google.cloud.alloydb.v1.ImportClusterResponse;
import com.google.cloud.alloydb.v1.InjectFaultRequest;
import com.google.cloud.alloydb.v1.Instance;
import com.google.cloud.alloydb.v1.ListBackupsRequest;
@@ -104,6 +108,8 @@
import com.google.cloud.alloydb.v1.UpdateClusterRequest;
import com.google.cloud.alloydb.v1.UpdateInstanceRequest;
import com.google.cloud.alloydb.v1.UpdateUserRequest;
+import com.google.cloud.alloydb.v1.UpgradeClusterRequest;
+import com.google.cloud.alloydb.v1.UpgradeClusterResponse;
import com.google.cloud.alloydb.v1.User;
import com.google.cloud.location.GetLocationRequest;
import com.google.cloud.location.ListLocationsRequest;
@@ -212,6 +218,18 @@ public class AlloyDBAdminStubSettings extends StubSettings updateClusterSettings;
private final OperationCallSettings
updateClusterOperationSettings;
+ private final UnaryCallSettings exportClusterSettings;
+ private final OperationCallSettings<
+ ExportClusterRequest, ExportClusterResponse, OperationMetadata>
+ exportClusterOperationSettings;
+ private final UnaryCallSettings importClusterSettings;
+ private final OperationCallSettings<
+ ImportClusterRequest, ImportClusterResponse, OperationMetadata>
+ importClusterOperationSettings;
+ private final UnaryCallSettings upgradeClusterSettings;
+ private final OperationCallSettings<
+ UpgradeClusterRequest, UpgradeClusterResponse, OperationMetadata>
+ upgradeClusterOperationSettings;
private final UnaryCallSettings deleteClusterSettings;
private final OperationCallSettings
deleteClusterOperationSettings;
@@ -710,6 +728,39 @@ public UnaryCallSettings updateClusterSettings(
return updateClusterOperationSettings;
}
+ /** Returns the object with the settings used for calls to exportCluster. */
+ public UnaryCallSettings exportClusterSettings() {
+ return exportClusterSettings;
+ }
+
+ /** Returns the object with the settings used for calls to exportCluster. */
+ public OperationCallSettings
+ exportClusterOperationSettings() {
+ return exportClusterOperationSettings;
+ }
+
+ /** Returns the object with the settings used for calls to importCluster. */
+ public UnaryCallSettings importClusterSettings() {
+ return importClusterSettings;
+ }
+
+ /** Returns the object with the settings used for calls to importCluster. */
+ public OperationCallSettings
+ importClusterOperationSettings() {
+ return importClusterOperationSettings;
+ }
+
+ /** Returns the object with the settings used for calls to upgradeCluster. */
+ public UnaryCallSettings upgradeClusterSettings() {
+ return upgradeClusterSettings;
+ }
+
+ /** Returns the object with the settings used for calls to upgradeCluster. */
+ public OperationCallSettings
+ upgradeClusterOperationSettings() {
+ return upgradeClusterOperationSettings;
+ }
+
/** Returns the object with the settings used for calls to deleteCluster. */
public UnaryCallSettings deleteClusterSettings() {
return deleteClusterSettings;
@@ -1096,6 +1147,12 @@ protected AlloyDBAdminStubSettings(Builder settingsBuilder) throws IOException {
createClusterOperationSettings = settingsBuilder.createClusterOperationSettings().build();
updateClusterSettings = settingsBuilder.updateClusterSettings().build();
updateClusterOperationSettings = settingsBuilder.updateClusterOperationSettings().build();
+ exportClusterSettings = settingsBuilder.exportClusterSettings().build();
+ exportClusterOperationSettings = settingsBuilder.exportClusterOperationSettings().build();
+ importClusterSettings = settingsBuilder.importClusterSettings().build();
+ importClusterOperationSettings = settingsBuilder.importClusterOperationSettings().build();
+ upgradeClusterSettings = settingsBuilder.upgradeClusterSettings().build();
+ upgradeClusterOperationSettings = settingsBuilder.upgradeClusterOperationSettings().build();
deleteClusterSettings = settingsBuilder.deleteClusterSettings().build();
deleteClusterOperationSettings = settingsBuilder.deleteClusterOperationSettings().build();
promoteClusterSettings = settingsBuilder.promoteClusterSettings().build();
@@ -1164,6 +1221,19 @@ public static class Builder extends StubSettings.Builder updateClusterSettings;
private final OperationCallSettings.Builder
updateClusterOperationSettings;
+ private final UnaryCallSettings.Builder exportClusterSettings;
+ private final OperationCallSettings.Builder<
+ ExportClusterRequest, ExportClusterResponse, OperationMetadata>
+ exportClusterOperationSettings;
+ private final UnaryCallSettings.Builder importClusterSettings;
+ private final OperationCallSettings.Builder<
+ ImportClusterRequest, ImportClusterResponse, OperationMetadata>
+ importClusterOperationSettings;
+ private final UnaryCallSettings.Builder
+ upgradeClusterSettings;
+ private final OperationCallSettings.Builder<
+ UpgradeClusterRequest, UpgradeClusterResponse, OperationMetadata>
+ upgradeClusterOperationSettings;
private final UnaryCallSettings.Builder deleteClusterSettings;
private final OperationCallSettings.Builder
deleteClusterOperationSettings;
@@ -1316,6 +1386,12 @@ protected Builder(ClientContext clientContext) {
createClusterOperationSettings = OperationCallSettings.newBuilder();
updateClusterSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
updateClusterOperationSettings = OperationCallSettings.newBuilder();
+ exportClusterSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
+ exportClusterOperationSettings = OperationCallSettings.newBuilder();
+ importClusterSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
+ importClusterOperationSettings = OperationCallSettings.newBuilder();
+ upgradeClusterSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
+ upgradeClusterOperationSettings = OperationCallSettings.newBuilder();
deleteClusterSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
deleteClusterOperationSettings = OperationCallSettings.newBuilder();
promoteClusterSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
@@ -1372,6 +1448,9 @@ protected Builder(ClientContext clientContext) {
getClusterSettings,
createClusterSettings,
updateClusterSettings,
+ exportClusterSettings,
+ importClusterSettings,
+ upgradeClusterSettings,
deleteClusterSettings,
promoteClusterSettings,
switchoverClusterSettings,
@@ -1416,6 +1495,12 @@ protected Builder(AlloyDBAdminStubSettings settings) {
createClusterOperationSettings = settings.createClusterOperationSettings.toBuilder();
updateClusterSettings = settings.updateClusterSettings.toBuilder();
updateClusterOperationSettings = settings.updateClusterOperationSettings.toBuilder();
+ exportClusterSettings = settings.exportClusterSettings.toBuilder();
+ exportClusterOperationSettings = settings.exportClusterOperationSettings.toBuilder();
+ importClusterSettings = settings.importClusterSettings.toBuilder();
+ importClusterOperationSettings = settings.importClusterOperationSettings.toBuilder();
+ upgradeClusterSettings = settings.upgradeClusterSettings.toBuilder();
+ upgradeClusterOperationSettings = settings.upgradeClusterOperationSettings.toBuilder();
deleteClusterSettings = settings.deleteClusterSettings.toBuilder();
deleteClusterOperationSettings = settings.deleteClusterOperationSettings.toBuilder();
promoteClusterSettings = settings.promoteClusterSettings.toBuilder();
@@ -1474,6 +1559,9 @@ protected Builder(AlloyDBAdminStubSettings settings) {
getClusterSettings,
createClusterSettings,
updateClusterSettings,
+ exportClusterSettings,
+ importClusterSettings,
+ upgradeClusterSettings,
deleteClusterSettings,
promoteClusterSettings,
switchoverClusterSettings,
@@ -1553,6 +1641,21 @@ private static Builder initDefaults(Builder builder) {
.setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_1_codes"))
.setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_1_params"));
+ builder
+ .exportClusterSettings()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_1_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_1_params"));
+
+ builder
+ .importClusterSettings()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_1_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_1_params"));
+
+ builder
+ .upgradeClusterSettings()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_1_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_1_params"));
+
builder
.deleteClusterSettings()
.setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_1_codes"))
@@ -1761,6 +1864,78 @@ private static Builder initDefaults(Builder builder) {
.setTotalTimeoutDuration(Duration.ofMillis(300000L))
.build()));
+ builder
+ .exportClusterOperationSettings()
+ .setInitialCallSettings(
+ UnaryCallSettings
+ .newUnaryCallSettingsBuilder()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_1_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_1_params"))
+ .build())
+ .setResponseTransformer(
+ ProtoOperationTransformers.ResponseTransformer.create(ExportClusterResponse.class))
+ .setMetadataTransformer(
+ ProtoOperationTransformers.MetadataTransformer.create(OperationMetadata.class))
+ .setPollingAlgorithm(
+ OperationTimedPollAlgorithm.create(
+ RetrySettings.newBuilder()
+ .setInitialRetryDelayDuration(Duration.ofMillis(5000L))
+ .setRetryDelayMultiplier(1.5)
+ .setMaxRetryDelayDuration(Duration.ofMillis(45000L))
+ .setInitialRpcTimeoutDuration(Duration.ZERO)
+ .setRpcTimeoutMultiplier(1.0)
+ .setMaxRpcTimeoutDuration(Duration.ZERO)
+ .setTotalTimeoutDuration(Duration.ofMillis(300000L))
+ .build()));
+
+ builder
+ .importClusterOperationSettings()
+ .setInitialCallSettings(
+ UnaryCallSettings
+ .newUnaryCallSettingsBuilder()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_1_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_1_params"))
+ .build())
+ .setResponseTransformer(
+ ProtoOperationTransformers.ResponseTransformer.create(ImportClusterResponse.class))
+ .setMetadataTransformer(
+ ProtoOperationTransformers.MetadataTransformer.create(OperationMetadata.class))
+ .setPollingAlgorithm(
+ OperationTimedPollAlgorithm.create(
+ RetrySettings.newBuilder()
+ .setInitialRetryDelayDuration(Duration.ofMillis(5000L))
+ .setRetryDelayMultiplier(1.5)
+ .setMaxRetryDelayDuration(Duration.ofMillis(45000L))
+ .setInitialRpcTimeoutDuration(Duration.ZERO)
+ .setRpcTimeoutMultiplier(1.0)
+ .setMaxRpcTimeoutDuration(Duration.ZERO)
+ .setTotalTimeoutDuration(Duration.ofMillis(300000L))
+ .build()));
+
+ builder
+ .upgradeClusterOperationSettings()
+ .setInitialCallSettings(
+ UnaryCallSettings
+ .newUnaryCallSettingsBuilder()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_1_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_1_params"))
+ .build())
+ .setResponseTransformer(
+ ProtoOperationTransformers.ResponseTransformer.create(UpgradeClusterResponse.class))
+ .setMetadataTransformer(
+ ProtoOperationTransformers.MetadataTransformer.create(OperationMetadata.class))
+ .setPollingAlgorithm(
+ OperationTimedPollAlgorithm.create(
+ RetrySettings.newBuilder()
+ .setInitialRetryDelayDuration(Duration.ofMillis(5000L))
+ .setRetryDelayMultiplier(1.5)
+ .setMaxRetryDelayDuration(Duration.ofMillis(45000L))
+ .setInitialRpcTimeoutDuration(Duration.ZERO)
+ .setRpcTimeoutMultiplier(1.0)
+ .setMaxRpcTimeoutDuration(Duration.ZERO)
+ .setTotalTimeoutDuration(Duration.ofMillis(300000L))
+ .build()));
+
builder
.deleteClusterOperationSettings()
.setInitialCallSettings(
@@ -2197,6 +2372,42 @@ public UnaryCallSettings.Builder updateClusterS
return updateClusterOperationSettings;
}
+ /** Returns the builder for the settings used for calls to exportCluster. */
+ public UnaryCallSettings.Builder exportClusterSettings() {
+ return exportClusterSettings;
+ }
+
+ /** Returns the builder for the settings used for calls to exportCluster. */
+ public OperationCallSettings.Builder<
+ ExportClusterRequest, ExportClusterResponse, OperationMetadata>
+ exportClusterOperationSettings() {
+ return exportClusterOperationSettings;
+ }
+
+ /** Returns the builder for the settings used for calls to importCluster. */
+ public UnaryCallSettings.Builder importClusterSettings() {
+ return importClusterSettings;
+ }
+
+ /** Returns the builder for the settings used for calls to importCluster. */
+ public OperationCallSettings.Builder<
+ ImportClusterRequest, ImportClusterResponse, OperationMetadata>
+ importClusterOperationSettings() {
+ return importClusterOperationSettings;
+ }
+
+ /** Returns the builder for the settings used for calls to upgradeCluster. */
+ public UnaryCallSettings.Builder upgradeClusterSettings() {
+ return upgradeClusterSettings;
+ }
+
+ /** Returns the builder for the settings used for calls to upgradeCluster. */
+ public OperationCallSettings.Builder<
+ UpgradeClusterRequest, UpgradeClusterResponse, OperationMetadata>
+ upgradeClusterOperationSettings() {
+ return upgradeClusterOperationSettings;
+ }
+
/** Returns the builder for the settings used for calls to deleteCluster. */
public UnaryCallSettings.Builder deleteClusterSettings() {
return deleteClusterSettings;
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBCSQLAdminStub.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBCSQLAdminStub.java
new file mode 100644
index 000000000000..7bdd829fae75
--- /dev/null
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBCSQLAdminStub.java
@@ -0,0 +1,77 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.cloud.alloydb.v1.stub;
+
+import static com.google.cloud.alloydb.v1.AlloyDBCSQLAdminClient.ListLocationsPagedResponse;
+
+import com.google.api.gax.core.BackgroundResource;
+import com.google.api.gax.rpc.OperationCallable;
+import com.google.api.gax.rpc.UnaryCallable;
+import com.google.cloud.alloydb.v1.Cluster;
+import com.google.cloud.alloydb.v1.OperationMetadata;
+import com.google.cloud.alloydb.v1.RestoreFromCloudSQLRequest;
+import com.google.cloud.location.GetLocationRequest;
+import com.google.cloud.location.ListLocationsRequest;
+import com.google.cloud.location.ListLocationsResponse;
+import com.google.cloud.location.Location;
+import com.google.longrunning.Operation;
+import com.google.longrunning.stub.OperationsStub;
+import javax.annotation.Generated;
+
+// AUTO-GENERATED DOCUMENTATION AND CLASS.
+/**
+ * Base stub class for the AlloyDBCSQLAdmin service API.
+ *
+ * This class is for advanced usage and reflects the underlying API directly.
+ */
+@Generated("by gapic-generator-java")
+public abstract class AlloyDBCSQLAdminStub implements BackgroundResource {
+
+ public OperationsStub getOperationsStub() {
+ return null;
+ }
+
+ public com.google.api.gax.httpjson.longrunning.stub.OperationsStub getHttpJsonOperationsStub() {
+ return null;
+ }
+
+ public OperationCallable
+ restoreFromCloudSQLOperationCallable() {
+ throw new UnsupportedOperationException(
+ "Not implemented: restoreFromCloudSQLOperationCallable()");
+ }
+
+ public UnaryCallable restoreFromCloudSQLCallable() {
+ throw new UnsupportedOperationException("Not implemented: restoreFromCloudSQLCallable()");
+ }
+
+ public UnaryCallable
+ listLocationsPagedCallable() {
+ throw new UnsupportedOperationException("Not implemented: listLocationsPagedCallable()");
+ }
+
+ public UnaryCallable listLocationsCallable() {
+ throw new UnsupportedOperationException("Not implemented: listLocationsCallable()");
+ }
+
+ public UnaryCallable getLocationCallable() {
+ throw new UnsupportedOperationException("Not implemented: getLocationCallable()");
+ }
+
+ @Override
+ public abstract void close();
+}
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBCSQLAdminStubSettings.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBCSQLAdminStubSettings.java
new file mode 100644
index 000000000000..6e6654b1c582
--- /dev/null
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/AlloyDBCSQLAdminStubSettings.java
@@ -0,0 +1,525 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.cloud.alloydb.v1.stub;
+
+import static com.google.cloud.alloydb.v1.AlloyDBCSQLAdminClient.ListLocationsPagedResponse;
+
+import com.google.api.core.ApiFunction;
+import com.google.api.core.ApiFuture;
+import com.google.api.core.BetaApi;
+import com.google.api.core.ObsoleteApi;
+import com.google.api.gax.core.GaxProperties;
+import com.google.api.gax.core.GoogleCredentialsProvider;
+import com.google.api.gax.core.InstantiatingExecutorProvider;
+import com.google.api.gax.grpc.GaxGrpcProperties;
+import com.google.api.gax.grpc.GrpcTransportChannel;
+import com.google.api.gax.grpc.InstantiatingGrpcChannelProvider;
+import com.google.api.gax.grpc.ProtoOperationTransformers;
+import com.google.api.gax.httpjson.GaxHttpJsonProperties;
+import com.google.api.gax.httpjson.HttpJsonTransportChannel;
+import com.google.api.gax.httpjson.InstantiatingHttpJsonChannelProvider;
+import com.google.api.gax.longrunning.OperationSnapshot;
+import com.google.api.gax.longrunning.OperationTimedPollAlgorithm;
+import com.google.api.gax.retrying.RetrySettings;
+import com.google.api.gax.rpc.ApiCallContext;
+import com.google.api.gax.rpc.ApiClientHeaderProvider;
+import com.google.api.gax.rpc.ClientContext;
+import com.google.api.gax.rpc.OperationCallSettings;
+import com.google.api.gax.rpc.PageContext;
+import com.google.api.gax.rpc.PagedCallSettings;
+import com.google.api.gax.rpc.PagedListDescriptor;
+import com.google.api.gax.rpc.PagedListResponseFactory;
+import com.google.api.gax.rpc.StatusCode;
+import com.google.api.gax.rpc.StubSettings;
+import com.google.api.gax.rpc.TransportChannelProvider;
+import com.google.api.gax.rpc.UnaryCallSettings;
+import com.google.api.gax.rpc.UnaryCallable;
+import com.google.cloud.alloydb.v1.Cluster;
+import com.google.cloud.alloydb.v1.OperationMetadata;
+import com.google.cloud.alloydb.v1.RestoreFromCloudSQLRequest;
+import com.google.cloud.location.GetLocationRequest;
+import com.google.cloud.location.ListLocationsRequest;
+import com.google.cloud.location.ListLocationsResponse;
+import com.google.cloud.location.Location;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.ImmutableSet;
+import com.google.common.collect.Lists;
+import com.google.longrunning.Operation;
+import java.io.IOException;
+import java.time.Duration;
+import java.util.List;
+import javax.annotation.Generated;
+
+// AUTO-GENERATED DOCUMENTATION AND CLASS.
+/**
+ * Settings class to configure an instance of {@link AlloyDBCSQLAdminStub}.
+ *
+ * The default instance has everything set to sensible defaults:
+ *
+ *
+ * - The default service address (alloydb.googleapis.com) and default port (443) are used.
+ *
- Credentials are acquired automatically through Application Default Credentials.
+ *
- Retries are configured for idempotent methods but not for non-idempotent methods.
+ *
+ *
+ * The builder of this class is recursive, so contained classes are themselves builders. When
+ * build() is called, the tree of builders is called to create the complete settings object.
+ *
+ *
For example, to set the
+ * [RetrySettings](https://cloud.google.com/java/docs/reference/gax/latest/com.google.api.gax.retrying.RetrySettings)
+ * of getLocation:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * AlloyDBCSQLAdminStubSettings.Builder alloyDBCSQLAdminSettingsBuilder =
+ * AlloyDBCSQLAdminStubSettings.newBuilder();
+ * alloyDBCSQLAdminSettingsBuilder
+ * .getLocationSettings()
+ * .setRetrySettings(
+ * alloyDBCSQLAdminSettingsBuilder
+ * .getLocationSettings()
+ * .getRetrySettings()
+ * .toBuilder()
+ * .setInitialRetryDelayDuration(Duration.ofSeconds(1))
+ * .setInitialRpcTimeoutDuration(Duration.ofSeconds(5))
+ * .setMaxAttempts(5)
+ * .setMaxRetryDelayDuration(Duration.ofSeconds(30))
+ * .setMaxRpcTimeoutDuration(Duration.ofSeconds(60))
+ * .setRetryDelayMultiplier(1.3)
+ * .setRpcTimeoutMultiplier(1.5)
+ * .setTotalTimeoutDuration(Duration.ofSeconds(300))
+ * .build());
+ * AlloyDBCSQLAdminStubSettings alloyDBCSQLAdminSettings = alloyDBCSQLAdminSettingsBuilder.build();
+ * }
+ *
+ * Please refer to the [Client Side Retry
+ * Guide](https://github.com/googleapis/google-cloud-java/blob/main/docs/client_retries.md) for
+ * additional support in setting retries.
+ *
+ * To configure the RetrySettings of a Long Running Operation method, create an
+ * OperationTimedPollAlgorithm object and update the RPC's polling algorithm. For example, to
+ * configure the RetrySettings for restoreFromCloudSQL:
+ *
+ *
{@code
+ * // This snippet has been automatically generated and should be regarded as a code template only.
+ * // It will require modifications to work:
+ * // - It may require correct/in-range values for request initialization.
+ * // - It may require specifying regional endpoints when creating the service client as shown in
+ * // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
+ * AlloyDBCSQLAdminStubSettings.Builder alloyDBCSQLAdminSettingsBuilder =
+ * AlloyDBCSQLAdminStubSettings.newBuilder();
+ * TimedRetryAlgorithm timedRetryAlgorithm =
+ * OperationalTimedPollAlgorithm.create(
+ * RetrySettings.newBuilder()
+ * .setInitialRetryDelayDuration(Duration.ofMillis(500))
+ * .setRetryDelayMultiplier(1.5)
+ * .setMaxRetryDelayDuration(Duration.ofMillis(5000))
+ * .setTotalTimeoutDuration(Duration.ofHours(24))
+ * .build());
+ * alloyDBCSQLAdminSettingsBuilder
+ * .createClusterOperationSettings()
+ * .setPollingAlgorithm(timedRetryAlgorithm)
+ * .build();
+ * }
+ */
+@Generated("by gapic-generator-java")
+public class AlloyDBCSQLAdminStubSettings extends StubSettings {
+ /** The default scopes of the service. */
+ private static final ImmutableList DEFAULT_SERVICE_SCOPES =
+ ImmutableList.builder().add("https://www.googleapis.com/auth/cloud-platform").build();
+
+ private final UnaryCallSettings
+ restoreFromCloudSQLSettings;
+ private final OperationCallSettings
+ restoreFromCloudSQLOperationSettings;
+ private final PagedCallSettings<
+ ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
+ listLocationsSettings;
+ private final UnaryCallSettings getLocationSettings;
+
+ private static final PagedListDescriptor
+ LIST_LOCATIONS_PAGE_STR_DESC =
+ new PagedListDescriptor() {
+ @Override
+ public String emptyToken() {
+ return "";
+ }
+
+ @Override
+ public ListLocationsRequest injectToken(ListLocationsRequest payload, String token) {
+ return ListLocationsRequest.newBuilder(payload).setPageToken(token).build();
+ }
+
+ @Override
+ public ListLocationsRequest injectPageSize(ListLocationsRequest payload, int pageSize) {
+ return ListLocationsRequest.newBuilder(payload).setPageSize(pageSize).build();
+ }
+
+ @Override
+ public Integer extractPageSize(ListLocationsRequest payload) {
+ return payload.getPageSize();
+ }
+
+ @Override
+ public String extractNextToken(ListLocationsResponse payload) {
+ return payload.getNextPageToken();
+ }
+
+ @Override
+ public Iterable extractResources(ListLocationsResponse payload) {
+ return payload.getLocationsList();
+ }
+ };
+
+ private static final PagedListResponseFactory<
+ ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
+ LIST_LOCATIONS_PAGE_STR_FACT =
+ new PagedListResponseFactory<
+ ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>() {
+ @Override
+ public ApiFuture getFuturePagedResponse(
+ UnaryCallable callable,
+ ListLocationsRequest request,
+ ApiCallContext context,
+ ApiFuture futureResponse) {
+ PageContext pageContext =
+ PageContext.create(callable, LIST_LOCATIONS_PAGE_STR_DESC, request, context);
+ return ListLocationsPagedResponse.createAsync(pageContext, futureResponse);
+ }
+ };
+
+ /** Returns the object with the settings used for calls to restoreFromCloudSQL. */
+ public UnaryCallSettings restoreFromCloudSQLSettings() {
+ return restoreFromCloudSQLSettings;
+ }
+
+ /** Returns the object with the settings used for calls to restoreFromCloudSQL. */
+ public OperationCallSettings
+ restoreFromCloudSQLOperationSettings() {
+ return restoreFromCloudSQLOperationSettings;
+ }
+
+ /** Returns the object with the settings used for calls to listLocations. */
+ public PagedCallSettings
+ listLocationsSettings() {
+ return listLocationsSettings;
+ }
+
+ /** Returns the object with the settings used for calls to getLocation. */
+ public UnaryCallSettings getLocationSettings() {
+ return getLocationSettings;
+ }
+
+ public AlloyDBCSQLAdminStub createStub() throws IOException {
+ if (getTransportChannelProvider()
+ .getTransportName()
+ .equals(GrpcTransportChannel.getGrpcTransportName())) {
+ return GrpcAlloyDBCSQLAdminStub.create(this);
+ }
+ if (getTransportChannelProvider()
+ .getTransportName()
+ .equals(HttpJsonTransportChannel.getHttpJsonTransportName())) {
+ return HttpJsonAlloyDBCSQLAdminStub.create(this);
+ }
+ throw new UnsupportedOperationException(
+ String.format(
+ "Transport not supported: %s", getTransportChannelProvider().getTransportName()));
+ }
+
+ /** Returns the default service name. */
+ @Override
+ public String getServiceName() {
+ return "alloydb";
+ }
+
+ /** Returns a builder for the default ExecutorProvider for this service. */
+ public static InstantiatingExecutorProvider.Builder defaultExecutorProviderBuilder() {
+ return InstantiatingExecutorProvider.newBuilder();
+ }
+
+ /** Returns the default service endpoint. */
+ @ObsoleteApi("Use getEndpoint() instead")
+ public static String getDefaultEndpoint() {
+ return "alloydb.googleapis.com:443";
+ }
+
+ /** Returns the default mTLS service endpoint. */
+ public static String getDefaultMtlsEndpoint() {
+ return "alloydb.mtls.googleapis.com:443";
+ }
+
+ /** Returns the default service scopes. */
+ public static List getDefaultServiceScopes() {
+ return DEFAULT_SERVICE_SCOPES;
+ }
+
+ /** Returns a builder for the default credentials for this service. */
+ public static GoogleCredentialsProvider.Builder defaultCredentialsProviderBuilder() {
+ return GoogleCredentialsProvider.newBuilder()
+ .setScopesToApply(DEFAULT_SERVICE_SCOPES)
+ .setUseJwtAccessWithScope(true);
+ }
+
+ /** Returns a builder for the default gRPC ChannelProvider for this service. */
+ public static InstantiatingGrpcChannelProvider.Builder defaultGrpcTransportProviderBuilder() {
+ return InstantiatingGrpcChannelProvider.newBuilder()
+ .setMaxInboundMessageSize(Integer.MAX_VALUE);
+ }
+
+ /** Returns a builder for the default REST ChannelProvider for this service. */
+ @BetaApi
+ public static InstantiatingHttpJsonChannelProvider.Builder
+ defaultHttpJsonTransportProviderBuilder() {
+ return InstantiatingHttpJsonChannelProvider.newBuilder();
+ }
+
+ public static TransportChannelProvider defaultTransportChannelProvider() {
+ return defaultGrpcTransportProviderBuilder().build();
+ }
+
+ public static ApiClientHeaderProvider.Builder defaultGrpcApiClientHeaderProviderBuilder() {
+ return ApiClientHeaderProvider.newBuilder()
+ .setGeneratedLibToken(
+ "gapic", GaxProperties.getLibraryVersion(AlloyDBCSQLAdminStubSettings.class))
+ .setTransportToken(
+ GaxGrpcProperties.getGrpcTokenName(), GaxGrpcProperties.getGrpcVersion());
+ }
+
+ public static ApiClientHeaderProvider.Builder defaultHttpJsonApiClientHeaderProviderBuilder() {
+ return ApiClientHeaderProvider.newBuilder()
+ .setGeneratedLibToken(
+ "gapic", GaxProperties.getLibraryVersion(AlloyDBCSQLAdminStubSettings.class))
+ .setTransportToken(
+ GaxHttpJsonProperties.getHttpJsonTokenName(),
+ GaxHttpJsonProperties.getHttpJsonVersion());
+ }
+
+ public static ApiClientHeaderProvider.Builder defaultApiClientHeaderProviderBuilder() {
+ return AlloyDBCSQLAdminStubSettings.defaultGrpcApiClientHeaderProviderBuilder();
+ }
+
+ /** Returns a new gRPC builder for this class. */
+ public static Builder newBuilder() {
+ return Builder.createDefault();
+ }
+
+ /** Returns a new REST builder for this class. */
+ public static Builder newHttpJsonBuilder() {
+ return Builder.createHttpJsonDefault();
+ }
+
+ /** Returns a new builder for this class. */
+ public static Builder newBuilder(ClientContext clientContext) {
+ return new Builder(clientContext);
+ }
+
+ /** Returns a builder containing all the values of this settings class. */
+ public Builder toBuilder() {
+ return new Builder(this);
+ }
+
+ protected AlloyDBCSQLAdminStubSettings(Builder settingsBuilder) throws IOException {
+ super(settingsBuilder);
+
+ restoreFromCloudSQLSettings = settingsBuilder.restoreFromCloudSQLSettings().build();
+ restoreFromCloudSQLOperationSettings =
+ settingsBuilder.restoreFromCloudSQLOperationSettings().build();
+ listLocationsSettings = settingsBuilder.listLocationsSettings().build();
+ getLocationSettings = settingsBuilder.getLocationSettings().build();
+ }
+
+ /** Builder for AlloyDBCSQLAdminStubSettings. */
+ public static class Builder extends StubSettings.Builder {
+ private final ImmutableList> unaryMethodSettingsBuilders;
+ private final UnaryCallSettings.Builder
+ restoreFromCloudSQLSettings;
+ private final OperationCallSettings.Builder<
+ RestoreFromCloudSQLRequest, Cluster, OperationMetadata>
+ restoreFromCloudSQLOperationSettings;
+ private final PagedCallSettings.Builder<
+ ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
+ listLocationsSettings;
+ private final UnaryCallSettings.Builder getLocationSettings;
+ private static final ImmutableMap>
+ RETRYABLE_CODE_DEFINITIONS;
+
+ static {
+ ImmutableMap.Builder> definitions =
+ ImmutableMap.builder();
+ definitions.put("no_retry_codes", ImmutableSet.copyOf(Lists.newArrayList()));
+ RETRYABLE_CODE_DEFINITIONS = definitions.build();
+ }
+
+ private static final ImmutableMap RETRY_PARAM_DEFINITIONS;
+
+ static {
+ ImmutableMap.Builder definitions = ImmutableMap.builder();
+ RetrySettings settings = null;
+ settings = RetrySettings.newBuilder().setRpcTimeoutMultiplier(1.0).build();
+ definitions.put("no_retry_params", settings);
+ RETRY_PARAM_DEFINITIONS = definitions.build();
+ }
+
+ protected Builder() {
+ this(((ClientContext) null));
+ }
+
+ protected Builder(ClientContext clientContext) {
+ super(clientContext);
+
+ restoreFromCloudSQLSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
+ restoreFromCloudSQLOperationSettings = OperationCallSettings.newBuilder();
+ listLocationsSettings = PagedCallSettings.newBuilder(LIST_LOCATIONS_PAGE_STR_FACT);
+ getLocationSettings = UnaryCallSettings.newUnaryCallSettingsBuilder();
+
+ unaryMethodSettingsBuilders =
+ ImmutableList.>of(
+ restoreFromCloudSQLSettings, listLocationsSettings, getLocationSettings);
+ initDefaults(this);
+ }
+
+ protected Builder(AlloyDBCSQLAdminStubSettings settings) {
+ super(settings);
+
+ restoreFromCloudSQLSettings = settings.restoreFromCloudSQLSettings.toBuilder();
+ restoreFromCloudSQLOperationSettings =
+ settings.restoreFromCloudSQLOperationSettings.toBuilder();
+ listLocationsSettings = settings.listLocationsSettings.toBuilder();
+ getLocationSettings = settings.getLocationSettings.toBuilder();
+
+ unaryMethodSettingsBuilders =
+ ImmutableList.>of(
+ restoreFromCloudSQLSettings, listLocationsSettings, getLocationSettings);
+ }
+
+ private static Builder createDefault() {
+ Builder builder = new Builder(((ClientContext) null));
+
+ builder.setTransportChannelProvider(defaultTransportChannelProvider());
+ builder.setCredentialsProvider(defaultCredentialsProviderBuilder().build());
+ builder.setInternalHeaderProvider(defaultApiClientHeaderProviderBuilder().build());
+ builder.setMtlsEndpoint(getDefaultMtlsEndpoint());
+ builder.setSwitchToMtlsEndpointAllowed(true);
+
+ return initDefaults(builder);
+ }
+
+ private static Builder createHttpJsonDefault() {
+ Builder builder = new Builder(((ClientContext) null));
+
+ builder.setTransportChannelProvider(defaultHttpJsonTransportProviderBuilder().build());
+ builder.setCredentialsProvider(defaultCredentialsProviderBuilder().build());
+ builder.setInternalHeaderProvider(defaultHttpJsonApiClientHeaderProviderBuilder().build());
+ builder.setMtlsEndpoint(getDefaultMtlsEndpoint());
+ builder.setSwitchToMtlsEndpointAllowed(true);
+
+ return initDefaults(builder);
+ }
+
+ private static Builder initDefaults(Builder builder) {
+ builder
+ .restoreFromCloudSQLSettings()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_params"));
+
+ builder
+ .listLocationsSettings()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_params"));
+
+ builder
+ .getLocationSettings()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_params"));
+
+ builder
+ .restoreFromCloudSQLOperationSettings()
+ .setInitialCallSettings(
+ UnaryCallSettings
+ .newUnaryCallSettingsBuilder()
+ .setRetryableCodes(RETRYABLE_CODE_DEFINITIONS.get("no_retry_codes"))
+ .setRetrySettings(RETRY_PARAM_DEFINITIONS.get("no_retry_params"))
+ .build())
+ .setResponseTransformer(
+ ProtoOperationTransformers.ResponseTransformer.create(Cluster.class))
+ .setMetadataTransformer(
+ ProtoOperationTransformers.MetadataTransformer.create(OperationMetadata.class))
+ .setPollingAlgorithm(
+ OperationTimedPollAlgorithm.create(
+ RetrySettings.newBuilder()
+ .setInitialRetryDelayDuration(Duration.ofMillis(5000L))
+ .setRetryDelayMultiplier(1.5)
+ .setMaxRetryDelayDuration(Duration.ofMillis(45000L))
+ .setInitialRpcTimeoutDuration(Duration.ZERO)
+ .setRpcTimeoutMultiplier(1.0)
+ .setMaxRpcTimeoutDuration(Duration.ZERO)
+ .setTotalTimeoutDuration(Duration.ofMillis(300000L))
+ .build()));
+
+ return builder;
+ }
+
+ /**
+ * Applies the given settings updater function to all of the unary API methods in this service.
+ *
+ * Note: This method does not support applying settings to streaming methods.
+ */
+ public Builder applyToAllUnaryMethods(
+ ApiFunction, Void> settingsUpdater) {
+ super.applyToAllUnaryMethods(unaryMethodSettingsBuilders, settingsUpdater);
+ return this;
+ }
+
+ public ImmutableList> unaryMethodSettingsBuilders() {
+ return unaryMethodSettingsBuilders;
+ }
+
+ /** Returns the builder for the settings used for calls to restoreFromCloudSQL. */
+ public UnaryCallSettings.Builder
+ restoreFromCloudSQLSettings() {
+ return restoreFromCloudSQLSettings;
+ }
+
+ /** Returns the builder for the settings used for calls to restoreFromCloudSQL. */
+ public OperationCallSettings.Builder
+ restoreFromCloudSQLOperationSettings() {
+ return restoreFromCloudSQLOperationSettings;
+ }
+
+ /** Returns the builder for the settings used for calls to listLocations. */
+ public PagedCallSettings.Builder<
+ ListLocationsRequest, ListLocationsResponse, ListLocationsPagedResponse>
+ listLocationsSettings() {
+ return listLocationsSettings;
+ }
+
+ /** Returns the builder for the settings used for calls to getLocation. */
+ public UnaryCallSettings.Builder getLocationSettings() {
+ return getLocationSettings;
+ }
+
+ @Override
+ public AlloyDBCSQLAdminStubSettings build() throws IOException {
+ return new AlloyDBCSQLAdminStubSettings(this);
+ }
+ }
+}
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/GrpcAlloyDBAdminStub.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/GrpcAlloyDBAdminStub.java
index dc54b28e7967..87edac788f2e 100644
--- a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/GrpcAlloyDBAdminStub.java
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/GrpcAlloyDBAdminStub.java
@@ -49,6 +49,8 @@
import com.google.cloud.alloydb.v1.DeleteUserRequest;
import com.google.cloud.alloydb.v1.ExecuteSqlRequest;
import com.google.cloud.alloydb.v1.ExecuteSqlResponse;
+import com.google.cloud.alloydb.v1.ExportClusterRequest;
+import com.google.cloud.alloydb.v1.ExportClusterResponse;
import com.google.cloud.alloydb.v1.FailoverInstanceRequest;
import com.google.cloud.alloydb.v1.GenerateClientCertificateRequest;
import com.google.cloud.alloydb.v1.GenerateClientCertificateResponse;
@@ -57,6 +59,8 @@
import com.google.cloud.alloydb.v1.GetConnectionInfoRequest;
import com.google.cloud.alloydb.v1.GetInstanceRequest;
import com.google.cloud.alloydb.v1.GetUserRequest;
+import com.google.cloud.alloydb.v1.ImportClusterRequest;
+import com.google.cloud.alloydb.v1.ImportClusterResponse;
import com.google.cloud.alloydb.v1.InjectFaultRequest;
import com.google.cloud.alloydb.v1.Instance;
import com.google.cloud.alloydb.v1.ListBackupsRequest;
@@ -80,6 +84,8 @@
import com.google.cloud.alloydb.v1.UpdateClusterRequest;
import com.google.cloud.alloydb.v1.UpdateInstanceRequest;
import com.google.cloud.alloydb.v1.UpdateUserRequest;
+import com.google.cloud.alloydb.v1.UpgradeClusterRequest;
+import com.google.cloud.alloydb.v1.UpgradeClusterResponse;
import com.google.cloud.alloydb.v1.User;
import com.google.cloud.location.GetLocationRequest;
import com.google.cloud.location.ListLocationsRequest;
@@ -140,6 +146,36 @@ public class GrpcAlloyDBAdminStub extends AlloyDBAdminStub {
.setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
.build();
+ private static final MethodDescriptor
+ exportClusterMethodDescriptor =
+ MethodDescriptor.newBuilder()
+ .setType(MethodDescriptor.MethodType.UNARY)
+ .setFullMethodName("google.cloud.alloydb.v1.AlloyDBAdmin/ExportCluster")
+ .setRequestMarshaller(
+ ProtoUtils.marshaller(ExportClusterRequest.getDefaultInstance()))
+ .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
+ .build();
+
+ private static final MethodDescriptor
+ importClusterMethodDescriptor =
+ MethodDescriptor.newBuilder()
+ .setType(MethodDescriptor.MethodType.UNARY)
+ .setFullMethodName("google.cloud.alloydb.v1.AlloyDBAdmin/ImportCluster")
+ .setRequestMarshaller(
+ ProtoUtils.marshaller(ImportClusterRequest.getDefaultInstance()))
+ .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
+ .build();
+
+ private static final MethodDescriptor
+ upgradeClusterMethodDescriptor =
+ MethodDescriptor.newBuilder()
+ .setType(MethodDescriptor.MethodType.UNARY)
+ .setFullMethodName("google.cloud.alloydb.v1.AlloyDBAdmin/UpgradeCluster")
+ .setRequestMarshaller(
+ ProtoUtils.marshaller(UpgradeClusterRequest.getDefaultInstance()))
+ .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
+ .build();
+
private static final MethodDescriptor
deleteClusterMethodDescriptor =
MethodDescriptor.newBuilder()
@@ -458,6 +494,15 @@ public class GrpcAlloyDBAdminStub extends AlloyDBAdminStub {
private final UnaryCallable updateClusterCallable;
private final OperationCallable
updateClusterOperationCallable;
+ private final UnaryCallable exportClusterCallable;
+ private final OperationCallable
+ exportClusterOperationCallable;
+ private final UnaryCallable importClusterCallable;
+ private final OperationCallable
+ importClusterOperationCallable;
+ private final UnaryCallable upgradeClusterCallable;
+ private final OperationCallable
+ upgradeClusterOperationCallable;
private final UnaryCallable deleteClusterCallable;
private final OperationCallable
deleteClusterOperationCallable;
@@ -622,6 +667,36 @@ protected GrpcAlloyDBAdminStub(
return builder.build();
})
.build();
+ GrpcCallSettings exportClusterTransportSettings =
+ GrpcCallSettings.newBuilder()
+ .setMethodDescriptor(exportClusterMethodDescriptor)
+ .setParamsExtractor(
+ request -> {
+ RequestParamsBuilder builder = RequestParamsBuilder.create();
+ builder.add("name", String.valueOf(request.getName()));
+ return builder.build();
+ })
+ .build();
+ GrpcCallSettings importClusterTransportSettings =
+ GrpcCallSettings.newBuilder()
+ .setMethodDescriptor(importClusterMethodDescriptor)
+ .setParamsExtractor(
+ request -> {
+ RequestParamsBuilder builder = RequestParamsBuilder.create();
+ builder.add("name", String.valueOf(request.getName()));
+ return builder.build();
+ })
+ .build();
+ GrpcCallSettings upgradeClusterTransportSettings =
+ GrpcCallSettings.newBuilder()
+ .setMethodDescriptor(upgradeClusterMethodDescriptor)
+ .setParamsExtractor(
+ request -> {
+ RequestParamsBuilder builder = RequestParamsBuilder.create();
+ builder.add("name", String.valueOf(request.getName()));
+ return builder.build();
+ })
+ .build();
GrpcCallSettings deleteClusterTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(deleteClusterMethodDescriptor)
@@ -976,6 +1051,33 @@ protected GrpcAlloyDBAdminStub(
settings.updateClusterOperationSettings(),
clientContext,
operationsStub);
+ this.exportClusterCallable =
+ callableFactory.createUnaryCallable(
+ exportClusterTransportSettings, settings.exportClusterSettings(), clientContext);
+ this.exportClusterOperationCallable =
+ callableFactory.createOperationCallable(
+ exportClusterTransportSettings,
+ settings.exportClusterOperationSettings(),
+ clientContext,
+ operationsStub);
+ this.importClusterCallable =
+ callableFactory.createUnaryCallable(
+ importClusterTransportSettings, settings.importClusterSettings(), clientContext);
+ this.importClusterOperationCallable =
+ callableFactory.createOperationCallable(
+ importClusterTransportSettings,
+ settings.importClusterOperationSettings(),
+ clientContext,
+ operationsStub);
+ this.upgradeClusterCallable =
+ callableFactory.createUnaryCallable(
+ upgradeClusterTransportSettings, settings.upgradeClusterSettings(), clientContext);
+ this.upgradeClusterOperationCallable =
+ callableFactory.createOperationCallable(
+ upgradeClusterTransportSettings,
+ settings.upgradeClusterOperationSettings(),
+ clientContext,
+ operationsStub);
this.deleteClusterCallable =
callableFactory.createUnaryCallable(
deleteClusterTransportSettings, settings.deleteClusterSettings(), clientContext);
@@ -1248,6 +1350,39 @@ public UnaryCallable updateClusterCallable() {
return updateClusterOperationCallable;
}
+ @Override
+ public UnaryCallable exportClusterCallable() {
+ return exportClusterCallable;
+ }
+
+ @Override
+ public OperationCallable
+ exportClusterOperationCallable() {
+ return exportClusterOperationCallable;
+ }
+
+ @Override
+ public UnaryCallable importClusterCallable() {
+ return importClusterCallable;
+ }
+
+ @Override
+ public OperationCallable
+ importClusterOperationCallable() {
+ return importClusterOperationCallable;
+ }
+
+ @Override
+ public UnaryCallable upgradeClusterCallable() {
+ return upgradeClusterCallable;
+ }
+
+ @Override
+ public OperationCallable
+ upgradeClusterOperationCallable() {
+ return upgradeClusterOperationCallable;
+ }
+
@Override
public UnaryCallable deleteClusterCallable() {
return deleteClusterCallable;
diff --git a/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/GrpcAlloyDBCSQLAdminCallableFactory.java b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/GrpcAlloyDBCSQLAdminCallableFactory.java
new file mode 100644
index 000000000000..e450f46633c6
--- /dev/null
+++ b/java-alloydb/google-cloud-alloydb/src/main/java/com/google/cloud/alloydb/v1/stub/GrpcAlloyDBCSQLAdminCallableFactory.java
@@ -0,0 +1,113 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.cloud.alloydb.v1.stub;
+
+import com.google.api.gax.grpc.GrpcCallSettings;
+import com.google.api.gax.grpc.GrpcCallableFactory;
+import com.google.api.gax.grpc.GrpcStubCallableFactory;
+import com.google.api.gax.rpc.BatchingCallSettings;
+import com.google.api.gax.rpc.BidiStreamingCallable;
+import com.google.api.gax.rpc.ClientContext;
+import com.google.api.gax.rpc.ClientStreamingCallable;
+import com.google.api.gax.rpc.OperationCallSettings;
+import com.google.api.gax.rpc.OperationCallable;
+import com.google.api.gax.rpc.PagedCallSettings;
+import com.google.api.gax.rpc.ServerStreamingCallSettings;
+import com.google.api.gax.rpc.ServerStreamingCallable;
+import com.google.api.gax.rpc.StreamingCallSettings;
+import com.google.api.gax.rpc.UnaryCallSettings;
+import com.google.api.gax.rpc.UnaryCallable;
+import com.google.longrunning.Operation;
+import com.google.longrunning.stub.OperationsStub;
+import javax.annotation.Generated;
+
+// AUTO-GENERATED DOCUMENTATION AND CLASS.
+/**
+ * gRPC callable factory implementation for the AlloyDBCSQLAdmin service API.
+ *
+ * This class is for advanced usage.
+ */
+@Generated("by gapic-generator-java")
+public class GrpcAlloyDBCSQLAdminCallableFactory implements GrpcStubCallableFactory {
+
+ @Override
+ public UnaryCallable createUnaryCallable(
+ GrpcCallSettings grpcCallSettings,
+ UnaryCallSettings callSettings,
+ ClientContext clientContext) {
+ return GrpcCallableFactory.createUnaryCallable(grpcCallSettings, callSettings, clientContext);
+ }
+
+ @Override
+ public
+ UnaryCallable