google-cloud-ai_platform-v1 0.58.0 → 0.59.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (91) hide show
  1. checksums.yaml +4 -4
  2. data/lib/google/cloud/ai_platform/v1/dataset_service/operations.rb +3 -10
  3. data/lib/google/cloud/ai_platform/v1/dataset_service/rest/operations.rb +73 -10
  4. data/lib/google/cloud/ai_platform/v1/deployment_resource_pool_service/operations.rb +3 -10
  5. data/lib/google/cloud/ai_platform/v1/deployment_resource_pool_service/rest/operations.rb +73 -10
  6. data/lib/google/cloud/ai_platform/v1/endpoint_service/client.rb +103 -0
  7. data/lib/google/cloud/ai_platform/v1/endpoint_service/operations.rb +3 -10
  8. data/lib/google/cloud/ai_platform/v1/endpoint_service/rest/client.rb +96 -0
  9. data/lib/google/cloud/ai_platform/v1/endpoint_service/rest/operations.rb +73 -10
  10. data/lib/google/cloud/ai_platform/v1/endpoint_service/rest/service_stub.rb +60 -0
  11. data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/operations.rb +3 -10
  12. data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/rest/operations.rb +73 -10
  13. data/lib/google/cloud/ai_platform/v1/feature_registry_service/operations.rb +3 -10
  14. data/lib/google/cloud/ai_platform/v1/feature_registry_service/rest/operations.rb +73 -10
  15. data/lib/google/cloud/ai_platform/v1/featurestore_service/operations.rb +3 -10
  16. data/lib/google/cloud/ai_platform/v1/featurestore_service/rest/operations.rb +73 -10
  17. data/lib/google/cloud/ai_platform/v1/gen_ai_tuning_service/operations.rb +3 -10
  18. data/lib/google/cloud/ai_platform/v1/gen_ai_tuning_service/rest/operations.rb +73 -10
  19. data/lib/google/cloud/ai_platform/v1/index_endpoint_service/operations.rb +3 -10
  20. data/lib/google/cloud/ai_platform/v1/index_endpoint_service/rest/operations.rb +73 -10
  21. data/lib/google/cloud/ai_platform/v1/index_service/operations.rb +3 -10
  22. data/lib/google/cloud/ai_platform/v1/index_service/rest/operations.rb +73 -10
  23. data/lib/google/cloud/ai_platform/v1/job_service/operations.rb +3 -10
  24. data/lib/google/cloud/ai_platform/v1/job_service/rest/operations.rb +73 -10
  25. data/lib/google/cloud/ai_platform/v1/llm_utility_service/paths.rb +19 -0
  26. data/lib/google/cloud/ai_platform/v1/metadata_service/operations.rb +3 -10
  27. data/lib/google/cloud/ai_platform/v1/metadata_service/rest/operations.rb +73 -10
  28. data/lib/google/cloud/ai_platform/v1/migration_service/operations.rb +3 -10
  29. data/lib/google/cloud/ai_platform/v1/migration_service/rest/operations.rb +73 -10
  30. data/lib/google/cloud/ai_platform/v1/model_service/operations.rb +3 -10
  31. data/lib/google/cloud/ai_platform/v1/model_service/rest/operations.rb +73 -10
  32. data/lib/google/cloud/ai_platform/v1/notebook_service/operations.rb +3 -10
  33. data/lib/google/cloud/ai_platform/v1/notebook_service/rest/operations.rb +73 -10
  34. data/lib/google/cloud/ai_platform/v1/persistent_resource_service/operations.rb +3 -10
  35. data/lib/google/cloud/ai_platform/v1/persistent_resource_service/rest/operations.rb +73 -10
  36. data/lib/google/cloud/ai_platform/v1/pipeline_service/operations.rb +3 -10
  37. data/lib/google/cloud/ai_platform/v1/pipeline_service/rest/operations.rb +73 -10
  38. data/lib/google/cloud/ai_platform/v1/prediction_service/paths.rb +19 -0
  39. data/lib/google/cloud/ai_platform/v1/rest.rb +2 -0
  40. data/lib/google/cloud/ai_platform/v1/schedule_service/operations.rb +3 -10
  41. data/lib/google/cloud/ai_platform/v1/schedule_service/paths.rb +38 -0
  42. data/lib/google/cloud/ai_platform/v1/schedule_service/rest/operations.rb +73 -10
  43. data/lib/google/cloud/ai_platform/v1/specialist_pool_service/operations.rb +3 -10
  44. data/lib/google/cloud/ai_platform/v1/specialist_pool_service/rest/operations.rb +73 -10
  45. data/lib/google/cloud/ai_platform/v1/tensorboard_service/operations.rb +3 -10
  46. data/lib/google/cloud/ai_platform/v1/tensorboard_service/rest/operations.rb +73 -10
  47. data/lib/google/cloud/ai_platform/v1/version.rb +1 -1
  48. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/client.rb +1405 -0
  49. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/credentials.rb +47 -0
  50. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/operations.rb +810 -0
  51. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/paths.rb +173 -0
  52. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/client.rb +1318 -0
  53. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/operations.rb +4244 -0
  54. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/service_stub.rb +663 -0
  55. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest.rb +54 -0
  56. data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service.rb +56 -0
  57. data/lib/google/cloud/ai_platform/v1/vertex_rag_service/client.rb +672 -0
  58. data/lib/google/cloud/ai_platform/v1/vertex_rag_service/credentials.rb +47 -0
  59. data/lib/google/cloud/ai_platform/v1/vertex_rag_service/paths.rb +69 -0
  60. data/lib/google/cloud/ai_platform/v1/vertex_rag_service/rest/client.rb +634 -0
  61. data/lib/google/cloud/ai_platform/v1/vertex_rag_service/rest/service_stub.rb +249 -0
  62. data/lib/google/cloud/ai_platform/v1/vertex_rag_service/rest.rb +53 -0
  63. data/lib/google/cloud/ai_platform/v1/vertex_rag_service.rb +55 -0
  64. data/lib/google/cloud/ai_platform/v1/vizier_service/operations.rb +3 -10
  65. data/lib/google/cloud/ai_platform/v1/vizier_service/rest/operations.rb +73 -10
  66. data/lib/google/cloud/ai_platform/v1.rb +2 -0
  67. data/lib/google/cloud/aiplatform/v1/api_auth_pb.rb +46 -0
  68. data/lib/google/cloud/aiplatform/v1/endpoint_pb.rb +5 -1
  69. data/lib/google/cloud/aiplatform/v1/endpoint_service_pb.rb +3 -1
  70. data/lib/google/cloud/aiplatform/v1/endpoint_service_services_pb.rb +2 -0
  71. data/lib/google/cloud/aiplatform/v1/io_pb.rb +16 -1
  72. data/lib/google/cloud/aiplatform/v1/machine_resources_pb.rb +1 -1
  73. data/lib/google/cloud/aiplatform/v1/notebook_execution_job_pb.rb +6 -1
  74. data/lib/google/cloud/aiplatform/v1/tool_pb.rb +5 -1
  75. data/lib/google/cloud/aiplatform/v1/vertex_rag_data_pb.rb +67 -0
  76. data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_pb.rb +71 -0
  77. data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_services_pb.rb +63 -0
  78. data/lib/google/cloud/aiplatform/v1/vertex_rag_service_pb.rb +65 -0
  79. data/lib/google/cloud/aiplatform/v1/vertex_rag_service_services_pb.rb +52 -0
  80. data/proto_docs/google/cloud/aiplatform/v1/api_auth.rb +45 -0
  81. data/proto_docs/google/cloud/aiplatform/v1/endpoint.rb +30 -0
  82. data/proto_docs/google/cloud/aiplatform/v1/endpoint_service.rb +22 -0
  83. data/proto_docs/google/cloud/aiplatform/v1/io.rb +158 -0
  84. data/proto_docs/google/cloud/aiplatform/v1/machine_resources.rb +8 -0
  85. data/proto_docs/google/cloud/aiplatform/v1/notebook_execution_job.rb +18 -0
  86. data/proto_docs/google/cloud/aiplatform/v1/tool.rb +72 -0
  87. data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_data.rb +334 -0
  88. data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_data_service.rb +293 -0
  89. data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_service.rb +279 -0
  90. data/proto_docs/google/longrunning/operations.rb +19 -14
  91. metadata +28 -2
@@ -0,0 +1,45 @@
1
+ # frozen_string_literal: true
2
+
3
+ # Copyright 2024 Google LLC
4
+ #
5
+ # Licensed under the Apache License, Version 2.0 (the "License");
6
+ # you may not use this file except in compliance with the License.
7
+ # You may obtain a copy of the License at
8
+ #
9
+ # https://www.apache.org/licenses/LICENSE-2.0
10
+ #
11
+ # Unless required by applicable law or agreed to in writing, software
12
+ # distributed under the License is distributed on an "AS IS" BASIS,
13
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ # See the License for the specific language governing permissions and
15
+ # limitations under the License.
16
+
17
+ # Auto-generated by gapic-generator-ruby. DO NOT EDIT!
18
+
19
+
20
+ module Google
21
+ module Cloud
22
+ module AIPlatform
23
+ module V1
24
+ # The generic reusable api auth config.
25
+ # @!attribute [rw] api_key_config
26
+ # @return [::Google::Cloud::AIPlatform::V1::ApiAuth::ApiKeyConfig]
27
+ # The API secret.
28
+ class ApiAuth
29
+ include ::Google::Protobuf::MessageExts
30
+ extend ::Google::Protobuf::MessageExts::ClassMethods
31
+
32
+ # The API secret.
33
+ # @!attribute [rw] api_key_secret_version
34
+ # @return [::String]
35
+ # Required. The SecretManager secret version resource name storing API key.
36
+ # e.g. projects/\\{project}/secrets/\\{secret}/versions/\\{version}
37
+ class ApiKeyConfig
38
+ include ::Google::Protobuf::MessageExts
39
+ extend ::Google::Protobuf::MessageExts::ClassMethods
40
+ end
41
+ end
42
+ end
43
+ end
44
+ end
45
+ end
@@ -135,6 +135,9 @@ module Google
135
135
  # dedicated_endpoint_enabled is true.
136
136
  # Format:
137
137
  # `https://{endpoint_id}.{region}-{project_number}.prediction.vertexai.goog`.
138
+ # @!attribute [rw] client_connection_config
139
+ # @return [::Google::Cloud::AIPlatform::V1::ClientConnectionConfig]
140
+ # Configurations that are applied to the endpoint for online prediction.
138
141
  # @!attribute [r] satisfies_pzs
139
142
  # @return [::Boolean]
140
143
  # Output only. Reserved for future use.
@@ -269,6 +272,9 @@ module Google
269
272
  # @!attribute [rw] faster_deployment_config
270
273
  # @return [::Google::Cloud::AIPlatform::V1::FasterDeploymentConfig]
271
274
  # Configuration for faster model deployment.
275
+ # @!attribute [r] status
276
+ # @return [::Google::Cloud::AIPlatform::V1::DeployedModel::Status]
277
+ # Output only. Runtime status of the deployed model.
272
278
  # @!attribute [rw] system_labels
273
279
  # @return [::Google::Protobuf::Map{::String => ::String}]
274
280
  # System labels to apply to Model Garden deployments.
@@ -277,6 +283,21 @@ module Google
277
283
  include ::Google::Protobuf::MessageExts
278
284
  extend ::Google::Protobuf::MessageExts::ClassMethods
279
285
 
286
+ # Runtime status of the deployed model.
287
+ # @!attribute [r] message
288
+ # @return [::String]
289
+ # Output only. The latest deployed model's status message (if any).
290
+ # @!attribute [r] last_update_time
291
+ # @return [::Google::Protobuf::Timestamp]
292
+ # Output only. The time at which the status was last updated.
293
+ # @!attribute [r] available_replica_count
294
+ # @return [::Integer]
295
+ # Output only. The number of available replicas of the deployed model.
296
+ class Status
297
+ include ::Google::Protobuf::MessageExts
298
+ extend ::Google::Protobuf::MessageExts::ClassMethods
299
+ end
300
+
280
301
  # @!attribute [rw] key
281
302
  # @return [::String]
282
303
  # @!attribute [rw] value
@@ -339,6 +360,15 @@ module Google
339
360
  include ::Google::Protobuf::MessageExts
340
361
  extend ::Google::Protobuf::MessageExts::ClassMethods
341
362
  end
363
+
364
+ # Configurations (e.g. inference timeout) that are applied on your endpoints.
365
+ # @!attribute [rw] inference_timeout
366
+ # @return [::Google::Protobuf::Duration]
367
+ # Customizable online prediction request timeout.
368
+ class ClientConnectionConfig
369
+ include ::Google::Protobuf::MessageExts
370
+ extend ::Google::Protobuf::MessageExts::ClassMethods
371
+ end
342
372
  end
343
373
  end
344
374
  end
@@ -160,6 +160,28 @@ module Google
160
160
  extend ::Google::Protobuf::MessageExts::ClassMethods
161
161
  end
162
162
 
163
+ # Request message for
164
+ # {::Google::Cloud::AIPlatform::V1::EndpointService::Client#update_endpoint_long_running EndpointService.UpdateEndpointLongRunning}.
165
+ # @!attribute [rw] endpoint
166
+ # @return [::Google::Cloud::AIPlatform::V1::Endpoint]
167
+ # Required. The Endpoint which replaces the resource on the server. Currently
168
+ # we only support updating the `client_connection_config` field, all the
169
+ # other fields' update will be blocked.
170
+ class UpdateEndpointLongRunningRequest
171
+ include ::Google::Protobuf::MessageExts
172
+ extend ::Google::Protobuf::MessageExts::ClassMethods
173
+ end
174
+
175
+ # Runtime operation information for
176
+ # {::Google::Cloud::AIPlatform::V1::EndpointService::Client#update_endpoint_long_running EndpointService.UpdateEndpointLongRunning}.
177
+ # @!attribute [rw] generic_metadata
178
+ # @return [::Google::Cloud::AIPlatform::V1::GenericOperationMetadata]
179
+ # The operation generic information.
180
+ class UpdateEndpointOperationMetadata
181
+ include ::Google::Protobuf::MessageExts
182
+ extend ::Google::Protobuf::MessageExts::ClassMethods
183
+ end
184
+
163
185
  # Request message for
164
186
  # {::Google::Cloud::AIPlatform::V1::EndpointService::Client#delete_endpoint EndpointService.DeleteEndpoint}.
165
187
  # @!attribute [rw] name
@@ -129,6 +129,164 @@ module Google
129
129
  include ::Google::Protobuf::MessageExts
130
130
  extend ::Google::Protobuf::MessageExts::ClassMethods
131
131
  end
132
+
133
+ # The Google Drive location for the input content.
134
+ # @!attribute [rw] resource_ids
135
+ # @return [::Array<::Google::Cloud::AIPlatform::V1::GoogleDriveSource::ResourceId>]
136
+ # Required. Google Drive resource IDs.
137
+ class GoogleDriveSource
138
+ include ::Google::Protobuf::MessageExts
139
+ extend ::Google::Protobuf::MessageExts::ClassMethods
140
+
141
+ # The type and ID of the Google Drive resource.
142
+ # @!attribute [rw] resource_type
143
+ # @return [::Google::Cloud::AIPlatform::V1::GoogleDriveSource::ResourceId::ResourceType]
144
+ # Required. The type of the Google Drive resource.
145
+ # @!attribute [rw] resource_id
146
+ # @return [::String]
147
+ # Required. The ID of the Google Drive resource.
148
+ class ResourceId
149
+ include ::Google::Protobuf::MessageExts
150
+ extend ::Google::Protobuf::MessageExts::ClassMethods
151
+
152
+ # The type of the Google Drive resource.
153
+ module ResourceType
154
+ # Unspecified resource type.
155
+ RESOURCE_TYPE_UNSPECIFIED = 0
156
+
157
+ # File resource type.
158
+ RESOURCE_TYPE_FILE = 1
159
+
160
+ # Folder resource type.
161
+ RESOURCE_TYPE_FOLDER = 2
162
+ end
163
+ end
164
+ end
165
+
166
+ # The input content is encapsulated and uploaded in the request.
167
+ class DirectUploadSource
168
+ include ::Google::Protobuf::MessageExts
169
+ extend ::Google::Protobuf::MessageExts::ClassMethods
170
+ end
171
+
172
+ # The Slack source for the ImportRagFilesRequest.
173
+ # @!attribute [rw] channels
174
+ # @return [::Array<::Google::Cloud::AIPlatform::V1::SlackSource::SlackChannels>]
175
+ # Required. The Slack channels.
176
+ class SlackSource
177
+ include ::Google::Protobuf::MessageExts
178
+ extend ::Google::Protobuf::MessageExts::ClassMethods
179
+
180
+ # SlackChannels contains the Slack channels and corresponding access token.
181
+ # @!attribute [rw] channels
182
+ # @return [::Array<::Google::Cloud::AIPlatform::V1::SlackSource::SlackChannels::SlackChannel>]
183
+ # Required. The Slack channel IDs.
184
+ # @!attribute [rw] api_key_config
185
+ # @return [::Google::Cloud::AIPlatform::V1::ApiAuth::ApiKeyConfig]
186
+ # Required. The SecretManager secret version resource name (e.g.
187
+ # projects/\\{project}/secrets/\\{secret}/versions/\\{version}) storing the
188
+ # Slack channel access token that has access to the slack channel IDs.
189
+ # See: https://api.slack.com/tutorials/tracks/getting-a-token.
190
+ class SlackChannels
191
+ include ::Google::Protobuf::MessageExts
192
+ extend ::Google::Protobuf::MessageExts::ClassMethods
193
+
194
+ # SlackChannel contains the Slack channel ID and the time range to import.
195
+ # @!attribute [rw] channel_id
196
+ # @return [::String]
197
+ # Required. The Slack channel ID.
198
+ # @!attribute [rw] start_time
199
+ # @return [::Google::Protobuf::Timestamp]
200
+ # Optional. The starting timestamp for messages to import.
201
+ # @!attribute [rw] end_time
202
+ # @return [::Google::Protobuf::Timestamp]
203
+ # Optional. The ending timestamp for messages to import.
204
+ class SlackChannel
205
+ include ::Google::Protobuf::MessageExts
206
+ extend ::Google::Protobuf::MessageExts::ClassMethods
207
+ end
208
+ end
209
+ end
210
+
211
+ # The Jira source for the ImportRagFilesRequest.
212
+ # @!attribute [rw] jira_queries
213
+ # @return [::Array<::Google::Cloud::AIPlatform::V1::JiraSource::JiraQueries>]
214
+ # Required. The Jira queries.
215
+ class JiraSource
216
+ include ::Google::Protobuf::MessageExts
217
+ extend ::Google::Protobuf::MessageExts::ClassMethods
218
+
219
+ # JiraQueries contains the Jira queries and corresponding authentication.
220
+ # @!attribute [rw] projects
221
+ # @return [::Array<::String>]
222
+ # A list of Jira projects to import in their entirety.
223
+ # @!attribute [rw] custom_queries
224
+ # @return [::Array<::String>]
225
+ # A list of custom Jira queries to import. For information about JQL (Jira
226
+ # Query Language), see
227
+ # https://support.atlassian.com/jira-service-management-cloud/docs/use-advanced-search-with-jira-query-language-jql/
228
+ # @!attribute [rw] email
229
+ # @return [::String]
230
+ # Required. The Jira email address.
231
+ # @!attribute [rw] server_uri
232
+ # @return [::String]
233
+ # Required. The Jira server URI.
234
+ # @!attribute [rw] api_key_config
235
+ # @return [::Google::Cloud::AIPlatform::V1::ApiAuth::ApiKeyConfig]
236
+ # Required. The SecretManager secret version resource name (e.g.
237
+ # projects/\\{project}/secrets/\\{secret}/versions/\\{version}) storing the
238
+ # Jira API key. See [Manage API tokens for your Atlassian
239
+ # account](https://support.atlassian.com/atlassian-account/docs/manage-api-tokens-for-your-atlassian-account/).
240
+ class JiraQueries
241
+ include ::Google::Protobuf::MessageExts
242
+ extend ::Google::Protobuf::MessageExts::ClassMethods
243
+ end
244
+ end
245
+
246
+ # The SharePointSources to pass to ImportRagFiles.
247
+ # @!attribute [rw] share_point_sources
248
+ # @return [::Array<::Google::Cloud::AIPlatform::V1::SharePointSources::SharePointSource>]
249
+ # The SharePoint sources.
250
+ class SharePointSources
251
+ include ::Google::Protobuf::MessageExts
252
+ extend ::Google::Protobuf::MessageExts::ClassMethods
253
+
254
+ # An individual SharePointSource.
255
+ # @!attribute [rw] sharepoint_folder_path
256
+ # @return [::String]
257
+ # The path of the SharePoint folder to download from.
258
+ # @!attribute [rw] sharepoint_folder_id
259
+ # @return [::String]
260
+ # The ID of the SharePoint folder to download from.
261
+ # @!attribute [rw] drive_name
262
+ # @return [::String]
263
+ # The name of the drive to download from.
264
+ # @!attribute [rw] drive_id
265
+ # @return [::String]
266
+ # The ID of the drive to download from.
267
+ # @!attribute [rw] client_id
268
+ # @return [::String]
269
+ # The Application ID for the app registered in Microsoft Azure Portal.
270
+ # The application must also be configured with MS Graph permissions
271
+ # "Files.ReadAll", "Sites.ReadAll" and BrowserSiteLists.Read.All.
272
+ # @!attribute [rw] client_secret
273
+ # @return [::Google::Cloud::AIPlatform::V1::ApiAuth::ApiKeyConfig]
274
+ # The application secret for the app registered in Azure.
275
+ # @!attribute [rw] tenant_id
276
+ # @return [::String]
277
+ # Unique identifier of the Azure Active Directory Instance.
278
+ # @!attribute [rw] sharepoint_site_name
279
+ # @return [::String]
280
+ # The name of the SharePoint site to download from. This can be the site
281
+ # name or the site id.
282
+ # @!attribute [r] file_id
283
+ # @return [::String]
284
+ # Output only. The SharePoint file id. Output only.
285
+ class SharePointSource
286
+ include ::Google::Protobuf::MessageExts
287
+ extend ::Google::Protobuf::MessageExts::ClassMethods
288
+ end
289
+ end
132
290
  end
133
291
  end
134
292
  end
@@ -89,6 +89,14 @@ module Google
89
89
  # quotas. Specifically, you will be charged for (max_replica_count *
90
90
  # number of cores in the selected machine type) and (max_replica_count *
91
91
  # number of GPUs per replica in the selected machine type).
92
+ # @!attribute [rw] required_replica_count
93
+ # @return [::Integer]
94
+ # Optional. Number of required available replicas for the deployment to
95
+ # succeed. This field is only needed when partial model deployment/mutation
96
+ # is desired. If set, the model deploy/mutate operation will succeed once
97
+ # available_replica_count reaches required_replica_count, and the rest of
98
+ # the replicas will be retried. If not set, the default
99
+ # required_replica_count will be min_replica_count.
92
100
  # @!attribute [rw] autoscaling_metric_specs
93
101
  # @return [::Array<::Google::Cloud::AIPlatform::V1::AutoscalingMetricSpec>]
94
102
  # Immutable. The metric specifications that overrides a resource
@@ -35,6 +35,9 @@ module Google
35
35
  # @!attribute [rw] notebook_runtime_template_resource_name
36
36
  # @return [::String]
37
37
  # The NotebookRuntimeTemplate to source compute configuration from.
38
+ # @!attribute [rw] custom_environment_spec
39
+ # @return [::Google::Cloud::AIPlatform::V1::NotebookExecutionJob::CustomEnvironmentSpec]
40
+ # The custom compute configuration for an execution job.
38
41
  # @!attribute [rw] gcs_output_uri
39
42
  # @return [::String]
40
43
  # The Cloud Storage location to upload the result to. Format:
@@ -134,6 +137,21 @@ module Google
134
137
  extend ::Google::Protobuf::MessageExts::ClassMethods
135
138
  end
136
139
 
140
+ # Compute configuration to use for an execution job.
141
+ # @!attribute [rw] machine_spec
142
+ # @return [::Google::Cloud::AIPlatform::V1::MachineSpec]
143
+ # The specification of a single machine for the execution job.
144
+ # @!attribute [rw] persistent_disk_spec
145
+ # @return [::Google::Cloud::AIPlatform::V1::PersistentDiskSpec]
146
+ # The specification of a persistent disk to attach for the execution job.
147
+ # @!attribute [rw] network_spec
148
+ # @return [::Google::Cloud::AIPlatform::V1::NetworkSpec]
149
+ # The network configuration to use for the execution job.
150
+ class CustomEnvironmentSpec
151
+ include ::Google::Protobuf::MessageExts
152
+ extend ::Google::Protobuf::MessageExts::ClassMethods
153
+ end
154
+
137
155
  # @!attribute [rw] key
138
156
  # @return [::String]
139
157
  # @!attribute [rw] value
@@ -133,6 +133,10 @@ module Google
133
133
  # @!attribute [rw] vertex_ai_search
134
134
  # @return [::Google::Cloud::AIPlatform::V1::VertexAISearch]
135
135
  # Set to use data source powered by Vertex AI Search.
136
+ # @!attribute [rw] vertex_rag_store
137
+ # @return [::Google::Cloud::AIPlatform::V1::VertexRagStore]
138
+ # Set to use data source powered by Vertex RAG store.
139
+ # User data is uploaded via the VertexRagDataService.
136
140
  # @!attribute [rw] disable_attribution
137
141
  # @deprecated This field is deprecated and may be removed in the next major version update.
138
142
  # @return [::Boolean]
@@ -142,6 +146,45 @@ module Google
142
146
  extend ::Google::Protobuf::MessageExts::ClassMethods
143
147
  end
144
148
 
149
+ # Retrieve from Vertex RAG Store for grounding.
150
+ # @!attribute [rw] rag_resources
151
+ # @return [::Array<::Google::Cloud::AIPlatform::V1::VertexRagStore::RagResource>]
152
+ # Optional. The representation of the rag source. It can be used to specify
153
+ # corpus only or ragfiles. Currently only support one corpus or multiple
154
+ # files from one corpus. In the future we may open up multiple corpora
155
+ # support.
156
+ # @!attribute [rw] similarity_top_k
157
+ # @deprecated This field is deprecated and may be removed in the next major version update.
158
+ # @return [::Integer]
159
+ # Optional. Number of top k results to return from the selected corpora.
160
+ # @!attribute [rw] vector_distance_threshold
161
+ # @deprecated This field is deprecated and may be removed in the next major version update.
162
+ # @return [::Float]
163
+ # Optional. Only return results with vector distance smaller than the
164
+ # threshold.
165
+ # @!attribute [rw] rag_retrieval_config
166
+ # @return [::Google::Cloud::AIPlatform::V1::RagRetrievalConfig]
167
+ # Optional. The retrieval config for the Rag query.
168
+ class VertexRagStore
169
+ include ::Google::Protobuf::MessageExts
170
+ extend ::Google::Protobuf::MessageExts::ClassMethods
171
+
172
+ # The definition of the Rag resource.
173
+ # @!attribute [rw] rag_corpus
174
+ # @return [::String]
175
+ # Optional. RagCorpora resource name.
176
+ # Format:
177
+ # `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}`
178
+ # @!attribute [rw] rag_file_ids
179
+ # @return [::Array<::String>]
180
+ # Optional. rag_file_id. The files should be in the same rag_corpus set in
181
+ # rag_corpus field.
182
+ class RagResource
183
+ include ::Google::Protobuf::MessageExts
184
+ extend ::Google::Protobuf::MessageExts::ClassMethods
185
+ end
186
+ end
187
+
145
188
  # Retrieve from Vertex AI Search datastore for grounding.
146
189
  # See https://cloud.google.com/products/agent-builder
147
190
  # @!attribute [rw] datastore
@@ -227,6 +270,35 @@ module Google
227
270
  NONE = 3
228
271
  end
229
272
  end
273
+
274
+ # Specifies the context retrieval config.
275
+ # @!attribute [rw] top_k
276
+ # @return [::Integer]
277
+ # Optional. The number of contexts to retrieve.
278
+ # @!attribute [rw] filter
279
+ # @return [::Google::Cloud::AIPlatform::V1::RagRetrievalConfig::Filter]
280
+ # Optional. Config for filters.
281
+ class RagRetrievalConfig
282
+ include ::Google::Protobuf::MessageExts
283
+ extend ::Google::Protobuf::MessageExts::ClassMethods
284
+
285
+ # Config for filters.
286
+ # @!attribute [rw] vector_distance_threshold
287
+ # @return [::Float]
288
+ # Optional. Only returns contexts with vector distance smaller than the
289
+ # threshold.
290
+ # @!attribute [rw] vector_similarity_threshold
291
+ # @return [::Float]
292
+ # Optional. Only returns contexts with vector similarity larger than the
293
+ # threshold.
294
+ # @!attribute [rw] metadata_filter
295
+ # @return [::String]
296
+ # Optional. String for metadata filtering.
297
+ class Filter
298
+ include ::Google::Protobuf::MessageExts
299
+ extend ::Google::Protobuf::MessageExts::ClassMethods
300
+ end
301
+ end
230
302
  end
231
303
  end
232
304
  end