google-cloud-ai_platform-v1 0.1.0

data/lib/google/cloud/ai_platform/v1/vizier_service/paths.rb

@@ -0,0 +1,109 @@
+# frozen_string_literal: true
+
+# Copyright 2022 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.
+
+# Auto-generated by gapic-generator-ruby. DO NOT EDIT!
+
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        module VizierService
+          # Path helper methods for the VizierService API.
+          module Paths
+            ##
+            # Create a fully-qualified CustomJob resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}/customJobs/{custom_job}`
+            #
+            # @param project [String]
+            # @param location [String]
+            # @param custom_job [String]
+            #
+            # @return [::String]
+            def custom_job_path project:, location:, custom_job:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+              raise ::ArgumentError, "location cannot contain /" if location.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}/customJobs/#{custom_job}"
+            end
+
+            ##
+            # Create a fully-qualified Location resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}`
+            #
+            # @param project [String]
+            # @param location [String]
+            #
+            # @return [::String]
+            def location_path project:, location:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}"
+            end
+
+            ##
+            # Create a fully-qualified Study resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}/studies/{study}`
+            #
+            # @param project [String]
+            # @param location [String]
+            # @param study [String]
+            #
+            # @return [::String]
+            def study_path project:, location:, study:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+              raise ::ArgumentError, "location cannot contain /" if location.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}/studies/#{study}"
+            end
+
+            ##
+            # Create a fully-qualified Trial resource string.
+            #
+            # The resource will be in the following format:
+            #
+            # `projects/{project}/locations/{location}/studies/{study}/trials/{trial}`
+            #
+            # @param project [String]
+            # @param location [String]
+            # @param study [String]
+            # @param trial [String]
+            #
+            # @return [::String]
+            def trial_path project:, location:, study:, trial:
+              raise ::ArgumentError, "project cannot contain /" if project.to_s.include? "/"
+              raise ::ArgumentError, "location cannot contain /" if location.to_s.include? "/"
+              raise ::ArgumentError, "study cannot contain /" if study.to_s.include? "/"
+
+              "projects/#{project}/locations/#{location}/studies/#{study}/trials/#{trial}"
+            end
+
+            extend self
+          end
+        end
+      end
+    end
+  end
+end

data/lib/google/cloud/ai_platform/v1/vizier_service.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+# Copyright 2022 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.
+
+# Auto-generated by gapic-generator-ruby. DO NOT EDIT!
+
+require "gapic/common"
+require "gapic/config"
+require "gapic/config/method"
+
+require "google/cloud/ai_platform/v1/version"
+
+require "google/cloud/ai_platform/v1/vizier_service/credentials"
+require "google/cloud/ai_platform/v1/vizier_service/paths"
+require "google/cloud/ai_platform/v1/vizier_service/operations"
+require "google/cloud/ai_platform/v1/vizier_service/client"
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        ##
+        # Vertex AI Vizier API.
+        #
+        # Vertex AI Vizier is a service to solve blackbox optimization problems,
+        # such as tuning machine learning hyperparameters and searching over deep
+        # learning architectures.
+        #
+        # To load this service and instantiate a client:
+        #
+        #     require "google/cloud/ai_platform/v1/vizier_service"
+        #     client = ::Google::Cloud::AIPlatform::V1::VizierService::Client.new
+        #
+        module VizierService
+        end
+      end
+    end
+  end
+end
+
+helper_path = ::File.join __dir__, "vizier_service", "helpers.rb"
+require "google/cloud/ai_platform/v1/vizier_service/helpers" if ::File.file? helper_path

data/lib/google/cloud/ai_platform/v1.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+# Copyright 2022 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.
+
+# Auto-generated by gapic-generator-ruby. DO NOT EDIT!
+
+require "google/cloud/ai_platform/v1/dataset_service"
+require "google/cloud/ai_platform/v1/endpoint_service"
+require "google/cloud/ai_platform/v1/featurestore_online_serving_service"
+require "google/cloud/ai_platform/v1/featurestore_service"
+require "google/cloud/ai_platform/v1/index_endpoint_service"
+require "google/cloud/ai_platform/v1/index_service"
+require "google/cloud/ai_platform/v1/job_service"
+require "google/cloud/ai_platform/v1/metadata_service"
+require "google/cloud/ai_platform/v1/migration_service"
+require "google/cloud/ai_platform/v1/model_service"
+require "google/cloud/ai_platform/v1/pipeline_service"
+require "google/cloud/ai_platform/v1/prediction_service"
+require "google/cloud/ai_platform/v1/specialist_pool_service"
+require "google/cloud/ai_platform/v1/tensorboard_service"
+require "google/cloud/ai_platform/v1/vizier_service"
+require "google/cloud/ai_platform/v1/version"
+
+module Google
+  module Cloud
+    module AIPlatform
+      ##
+      # To load this package, including all its services, and instantiate a client:
+      #
+      #     require "google/cloud/ai_platform/v1"
+      #     client = ::Google::Cloud::AIPlatform::V1::DatasetService::Client.new
+      #
+      module V1
+      end
+    end
+  end
+end
+
+helper_path = ::File.join __dir__, "v1", "_helpers.rb"
+require "google/cloud/ai_platform/v1/_helpers" if ::File.file? helper_path

data/lib/google/cloud/aiplatform/v1/accelerator_type_pb.rb

@@ -0,0 +1,29 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/accelerator_type.proto
+
+require 'google/api/annotations_pb'
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/accelerator_type.proto", :syntax => :proto3) do
+    add_enum "google.cloud.aiplatform.v1.AcceleratorType" do
+      value :ACCELERATOR_TYPE_UNSPECIFIED, 0
+      value :NVIDIA_TESLA_K80, 1
+      value :NVIDIA_TESLA_P100, 2
+      value :NVIDIA_TESLA_V100, 3
+      value :NVIDIA_TESLA_P4, 4
+      value :NVIDIA_TESLA_T4, 5
+      value :NVIDIA_TESLA_A100, 8
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        AcceleratorType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.AcceleratorType").enummodule
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/annotation_pb.rb

@@ -0,0 +1,35 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/annotation.proto
+
+require 'google/api/annotations_pb'
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/cloud/aiplatform/v1/user_action_reference_pb'
+require 'google/protobuf/struct_pb'
+require 'google/protobuf/timestamp_pb'
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/annotation.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.Annotation" do
+      optional :name, :string, 1
+      optional :payload_schema_uri, :string, 2
+      optional :payload, :message, 3, "google.protobuf.Value"
+      optional :create_time, :message, 4, "google.protobuf.Timestamp"
+      optional :update_time, :message, 7, "google.protobuf.Timestamp"
+      optional :etag, :string, 8
+      optional :annotation_source, :message, 5, "google.cloud.aiplatform.v1.UserActionReference"
+      map :labels, :string, :string, 6
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        Annotation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Annotation").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/annotation_spec_pb.rb

@@ -0,0 +1,30 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/annotation_spec.proto
+
+require 'google/api/annotations_pb'
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/protobuf/timestamp_pb'
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/annotation_spec.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.AnnotationSpec" do
+      optional :name, :string, 1
+      optional :display_name, :string, 2
+      optional :create_time, :message, 3, "google.protobuf.Timestamp"
+      optional :update_time, :message, 4, "google.protobuf.Timestamp"
+      optional :etag, :string, 5
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        AnnotationSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.AnnotationSpec").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/artifact_pb.rb

@@ -0,0 +1,45 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/artifact.proto
+
+require 'google/api/annotations_pb'
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/cloud/aiplatform/v1/value_pb'
+require 'google/protobuf/struct_pb'
+require 'google/protobuf/timestamp_pb'
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/artifact.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.Artifact" do
+      optional :name, :string, 1
+      optional :display_name, :string, 2
+      optional :uri, :string, 6
+      optional :etag, :string, 9
+      map :labels, :string, :string, 10
+      optional :create_time, :message, 11, "google.protobuf.Timestamp"
+      optional :update_time, :message, 12, "google.protobuf.Timestamp"
+      optional :state, :enum, 13, "google.cloud.aiplatform.v1.Artifact.State"
+      optional :schema_title, :string, 14
+      optional :schema_version, :string, 15
+      optional :metadata, :message, 16, "google.protobuf.Struct"
+      optional :description, :string, 17
+    end
+    add_enum "google.cloud.aiplatform.v1.Artifact.State" do
+      value :STATE_UNSPECIFIED, 0
+      value :PENDING, 1
+      value :LIVE, 2
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        Artifact = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Artifact").msgclass
+        Artifact::State = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Artifact.State").enummodule
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/batch_prediction_job_pb.rb

@@ -0,0 +1,83 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/batch_prediction_job.proto
+
+require 'google/api/annotations_pb'
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/cloud/aiplatform/v1/completion_stats_pb'
+require 'google/cloud/aiplatform/v1/encryption_spec_pb'
+require 'google/cloud/aiplatform/v1/explanation_pb'
+require 'google/cloud/aiplatform/v1/io_pb'
+require 'google/cloud/aiplatform/v1/job_state_pb'
+require 'google/cloud/aiplatform/v1/machine_resources_pb'
+require 'google/cloud/aiplatform/v1/manual_batch_tuning_parameters_pb'
+require 'google/cloud/aiplatform/v1/model_monitoring_pb'
+require 'google/cloud/aiplatform/v1/unmanaged_container_model_pb'
+require 'google/protobuf/struct_pb'
+require 'google/protobuf/timestamp_pb'
+require 'google/rpc/status_pb'
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/batch_prediction_job.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.BatchPredictionJob" do
+      optional :name, :string, 1
+      optional :display_name, :string, 2
+      optional :model, :string, 3
+      optional :unmanaged_container_model, :message, 28, "google.cloud.aiplatform.v1.UnmanagedContainerModel"
+      optional :input_config, :message, 4, "google.cloud.aiplatform.v1.BatchPredictionJob.InputConfig"
+      optional :model_parameters, :message, 5, "google.protobuf.Value"
+      optional :output_config, :message, 6, "google.cloud.aiplatform.v1.BatchPredictionJob.OutputConfig"
+      optional :dedicated_resources, :message, 7, "google.cloud.aiplatform.v1.BatchDedicatedResources"
+      optional :manual_batch_tuning_parameters, :message, 8, "google.cloud.aiplatform.v1.ManualBatchTuningParameters"
+      optional :generate_explanation, :bool, 23
+      optional :explanation_spec, :message, 25, "google.cloud.aiplatform.v1.ExplanationSpec"
+      optional :output_info, :message, 9, "google.cloud.aiplatform.v1.BatchPredictionJob.OutputInfo"
+      optional :state, :enum, 10, "google.cloud.aiplatform.v1.JobState"
+      optional :error, :message, 11, "google.rpc.Status"
+      repeated :partial_failures, :message, 12, "google.rpc.Status"
+      optional :resources_consumed, :message, 13, "google.cloud.aiplatform.v1.ResourcesConsumed"
+      optional :completion_stats, :message, 14, "google.cloud.aiplatform.v1.CompletionStats"
+      optional :create_time, :message, 15, "google.protobuf.Timestamp"
+      optional :start_time, :message, 16, "google.protobuf.Timestamp"
+      optional :end_time, :message, 17, "google.protobuf.Timestamp"
+      optional :update_time, :message, 18, "google.protobuf.Timestamp"
+      map :labels, :string, :string, 19
+      optional :encryption_spec, :message, 24, "google.cloud.aiplatform.v1.EncryptionSpec"
+    end
+    add_message "google.cloud.aiplatform.v1.BatchPredictionJob.InputConfig" do
+      optional :instances_format, :string, 1
+      oneof :source do
+        optional :gcs_source, :message, 2, "google.cloud.aiplatform.v1.GcsSource"
+        optional :bigquery_source, :message, 3, "google.cloud.aiplatform.v1.BigQuerySource"
+      end
+    end
+    add_message "google.cloud.aiplatform.v1.BatchPredictionJob.OutputConfig" do
+      optional :predictions_format, :string, 1
+      oneof :destination do
+        optional :gcs_destination, :message, 2, "google.cloud.aiplatform.v1.GcsDestination"
+        optional :bigquery_destination, :message, 3, "google.cloud.aiplatform.v1.BigQueryDestination"
+      end
+    end
+    add_message "google.cloud.aiplatform.v1.BatchPredictionJob.OutputInfo" do
+      optional :bigquery_output_table, :string, 4
+      oneof :output_location do
+        optional :gcs_output_directory, :string, 1
+        optional :bigquery_output_dataset, :string, 2
+      end
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        BatchPredictionJob = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchPredictionJob").msgclass
+        BatchPredictionJob::InputConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchPredictionJob.InputConfig").msgclass
+        BatchPredictionJob::OutputConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchPredictionJob.OutputConfig").msgclass
+        BatchPredictionJob::OutputInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.BatchPredictionJob.OutputInfo").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/completion_stats_pb.rb

@@ -0,0 +1,27 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/completion_stats.proto
+
+require 'google/api/annotations_pb'
+require 'google/api/field_behavior_pb'
+require 'google/rpc/status_pb'
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/completion_stats.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.CompletionStats" do
+      optional :successful_count, :int64, 1
+      optional :failed_count, :int64, 2
+      optional :incomplete_count, :int64, 3
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        CompletionStats = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CompletionStats").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/context_pb.rb

@@ -0,0 +1,38 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/context.proto
+
+require 'google/api/annotations_pb'
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/cloud/aiplatform/v1/value_pb'
+require 'google/protobuf/struct_pb'
+require 'google/protobuf/timestamp_pb'
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/context.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.Context" do
+      optional :name, :string, 1
+      optional :display_name, :string, 2
+      optional :etag, :string, 8
+      map :labels, :string, :string, 9
+      optional :create_time, :message, 10, "google.protobuf.Timestamp"
+      optional :update_time, :message, 11, "google.protobuf.Timestamp"
+      repeated :parent_contexts, :string, 12
+      optional :schema_title, :string, 13
+      optional :schema_version, :string, 14
+      optional :metadata, :message, 15, "google.protobuf.Struct"
+      optional :description, :string, 16
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        Context = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Context").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/custom_job_pb.rb

@@ -0,0 +1,84 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/custom_job.proto
+
+require 'google/api/annotations_pb'
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/cloud/aiplatform/v1/encryption_spec_pb'
+require 'google/cloud/aiplatform/v1/env_var_pb'
+require 'google/cloud/aiplatform/v1/io_pb'
+require 'google/cloud/aiplatform/v1/job_state_pb'
+require 'google/cloud/aiplatform/v1/machine_resources_pb'
+require 'google/protobuf/duration_pb'
+require 'google/protobuf/timestamp_pb'
+require 'google/rpc/status_pb'
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/custom_job.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.CustomJob" do
+      optional :name, :string, 1
+      optional :display_name, :string, 2
+      optional :job_spec, :message, 4, "google.cloud.aiplatform.v1.CustomJobSpec"
+      optional :state, :enum, 5, "google.cloud.aiplatform.v1.JobState"
+      optional :create_time, :message, 6, "google.protobuf.Timestamp"
+      optional :start_time, :message, 7, "google.protobuf.Timestamp"
+      optional :end_time, :message, 8, "google.protobuf.Timestamp"
+      optional :update_time, :message, 9, "google.protobuf.Timestamp"
+      optional :error, :message, 10, "google.rpc.Status"
+      map :labels, :string, :string, 11
+      optional :encryption_spec, :message, 12, "google.cloud.aiplatform.v1.EncryptionSpec"
+      map :web_access_uris, :string, :string, 16
+    end
+    add_message "google.cloud.aiplatform.v1.CustomJobSpec" do
+      repeated :worker_pool_specs, :message, 1, "google.cloud.aiplatform.v1.WorkerPoolSpec"
+      optional :scheduling, :message, 3, "google.cloud.aiplatform.v1.Scheduling"
+      optional :service_account, :string, 4
+      optional :network, :string, 5
+      optional :base_output_directory, :message, 6, "google.cloud.aiplatform.v1.GcsDestination"
+      optional :tensorboard, :string, 7
+      optional :enable_web_access, :bool, 10
+    end
+    add_message "google.cloud.aiplatform.v1.WorkerPoolSpec" do
+      optional :machine_spec, :message, 1, "google.cloud.aiplatform.v1.MachineSpec"
+      optional :replica_count, :int64, 2
+      optional :disk_spec, :message, 5, "google.cloud.aiplatform.v1.DiskSpec"
+      oneof :task do
+        optional :container_spec, :message, 6, "google.cloud.aiplatform.v1.ContainerSpec"
+        optional :python_package_spec, :message, 7, "google.cloud.aiplatform.v1.PythonPackageSpec"
+      end
+    end
+    add_message "google.cloud.aiplatform.v1.ContainerSpec" do
+      optional :image_uri, :string, 1
+      repeated :command, :string, 2
+      repeated :args, :string, 3
+      repeated :env, :message, 4, "google.cloud.aiplatform.v1.EnvVar"
+    end
+    add_message "google.cloud.aiplatform.v1.PythonPackageSpec" do
+      optional :executor_image_uri, :string, 1
+      repeated :package_uris, :string, 2
+      optional :python_module, :string, 3
+      repeated :args, :string, 4
+      repeated :env, :message, 5, "google.cloud.aiplatform.v1.EnvVar"
+    end
+    add_message "google.cloud.aiplatform.v1.Scheduling" do
+      optional :timeout, :message, 1, "google.protobuf.Duration"
+      optional :restart_job_on_worker_restart, :bool, 3
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        CustomJob = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CustomJob").msgclass
+        CustomJobSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CustomJobSpec").msgclass
+        WorkerPoolSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.WorkerPoolSpec").msgclass
+        ContainerSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ContainerSpec").msgclass
+        PythonPackageSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PythonPackageSpec").msgclass
+        Scheduling = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Scheduling").msgclass
+      end
+    end
+  end
+end

data/lib/google/cloud/aiplatform/v1/data_item_pb.rb

@@ -0,0 +1,32 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: google/cloud/aiplatform/v1/data_item.proto
+
+require 'google/api/annotations_pb'
+require 'google/api/field_behavior_pb'
+require 'google/api/resource_pb'
+require 'google/protobuf/struct_pb'
+require 'google/protobuf/timestamp_pb'
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("google/cloud/aiplatform/v1/data_item.proto", :syntax => :proto3) do
+    add_message "google.cloud.aiplatform.v1.DataItem" do
+      optional :name, :string, 1
+      optional :create_time, :message, 2, "google.protobuf.Timestamp"
+      optional :update_time, :message, 6, "google.protobuf.Timestamp"
+      map :labels, :string, :string, 3
+      optional :payload, :message, 4, "google.protobuf.Value"
+      optional :etag, :string, 7
+    end
+  end
+end
+
+module Google
+  module Cloud
+    module AIPlatform
+      module V1
+        DataItem = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DataItem").msgclass
+      end
+    end
+  end
+end