Skip to content

Commit 0bca99d

Browse files
gcf-owl-bot[bot]sofisl
gcf-owl-bot[bot]
and
sofisl
authored
fix: improve retry logic for streaming API calls (#4942)
* fix: Fix rpc tensorboard_service.proto definitions for BatchCreateTensorboardTimeSeries and BatchReadTensorboardTimeSeriesData feat: add models to CustomJobSpec feat: add max_wait_duration to Scheduling docs: minor changes to comments PiperOrigin-RevId: 597666793 Source-Link: googleapis/googleapis@84ee19c Source-Link: googleapis/googleapis-gen@8dc3e3e Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6IjhkYzNlM2UzMjA0M2FjZDY4OTNiZWQ5NmE2YjQ1NjZmMDA4NDMzMjkifQ== * fix!: Fix rpc tensorboard_service.proto definitions for BatchCreateTensorboardTimeSeries and BatchReadTensorboardTimeSeriesData feat: add models to CustomJobSpec feat: add max_wait_duration to Scheduling feat: add CountTokens method to llm_utility_service docs: minor changes to comments PiperOrigin-RevId: 597667108 Source-Link: googleapis/googleapis@714ab03 Source-Link: googleapis/googleapis-gen@3ff71b3 Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6IjNmZjcxYjM5ZDJhZDJlZWI1MTEzOTQwYzVkZTU1OTNkYTA1MjdiMjYifQ== * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * fix: improve retry logic for streaming API calls build: update typescript generator version to 4.3.0 The streaming API call retry logic has changed, which in some rare cases may require code changes. Please feel free to reach out to us in the issues if you experience new problems with retrying streaming calls after this update. PiperOrigin-RevId: 599622271 Source-Link: googleapis/googleapis@6239c21 Source-Link: googleapis/googleapis-gen@da13d82 Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6ImRhMTNkODIyMmQzYmEzMzczNDUwMTk5OTg2NDQ1ODY0MGYxNDA1YWUifQ== * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com> Co-authored-by: sofisl <55454395+sofisl@users.noreply.github.com>
1 parent 7fbf16c commit 0bca99d

File tree

73 files changed

+53572
-48020
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

73 files changed

+53572
-48020
lines changed

‎packages/google-cloud-aiplatform/README.md‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -495,6 +495,7 @@ Samples are in the [`samples/`](https://github.com/googleapis/google-cloud-node/
495495
| Job_service.search_model_deployment_monitoring_stats_anomalies | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/job_service.search_model_deployment_monitoring_stats_anomalies.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/job_service.search_model_deployment_monitoring_stats_anomalies.js,packages/google-cloud-aiplatform/samples/README.md) |
496496
| Job_service.update_model_deployment_monitoring_job | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/job_service.update_model_deployment_monitoring_job.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/job_service.update_model_deployment_monitoring_job.js,packages/google-cloud-aiplatform/samples/README.md) |
497497
| Llm_utility_service.compute_tokens | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/llm_utility_service.compute_tokens.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/llm_utility_service.compute_tokens.js,packages/google-cloud-aiplatform/samples/README.md) |
498+
| Llm_utility_service.count_tokens | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/llm_utility_service.count_tokens.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/llm_utility_service.count_tokens.js,packages/google-cloud-aiplatform/samples/README.md) |
498499
| Match_service.find_neighbors | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/match_service.find_neighbors.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/match_service.find_neighbors.js,packages/google-cloud-aiplatform/samples/README.md) |
499500
| Match_service.read_index_datapoints | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/match_service.read_index_datapoints.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/match_service.read_index_datapoints.js,packages/google-cloud-aiplatform/samples/README.md) |
500501
| Metadata_service.add_context_artifacts_and_executions | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/metadata_service.add_context_artifacts_and_executions.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/metadata_service.add_context_artifacts_and_executions.js,packages/google-cloud-aiplatform/samples/README.md) |

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/custom_job.proto‎

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,26 @@ message CustomJobSpec {
242242
type: "aiplatform.googleapis.com/Context"
243243
}
244244
];
245+
246+
// Optional. The name of the Model resources for which to generate a mapping
247+
// to artifact URIs. Applicable only to some of the Google-provided custom
248+
// jobs. Format: `projects/{project}/locations/{location}/models/{model}`
249+
//
250+
// In order to retrieve a specific version of the model, also provide
251+
// the version ID or version alias.
252+
// Example: `projects/{project}/locations/{location}/models/{model}@2`
253+
// or
254+
// `projects/{project}/locations/{location}/models/{model}@golden`
255+
// If no version ID or alias is specified, the "default" version will be
256+
// returned. The "default" version alias is created for the first version of
257+
// the model, and can be moved to other versions later on. There will be
258+
// exactly one default version.
259+
repeated string models = 20 [
260+
(google.api.field_behavior) = OPTIONAL,
261+
(google.api.resource_reference) = {
262+
type: "aiplatform.googleapis.com/Model"
263+
}
264+
];
245265
}
246266

247267
// Represents the spec of a worker pool in a job.
@@ -329,4 +349,9 @@ message Scheduling {
329349
// job starts running. If true, overrides
330350
// `Scheduling.restart_job_on_worker_restart` to false.
331351
bool disable_retries = 5 [(google.api.field_behavior) = OPTIONAL];
352+
353+
// Optional. This is the maximum time a user will wait in the QRM queue for
354+
// resources. Default is 1 day
355+
google.protobuf.Duration max_wait_duration = 6
356+
[(google.api.field_behavior) = OPTIONAL];
332357
}

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/dataset_service.proto‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -414,8 +414,9 @@ message ExportDataRequest {
414414
// [DatasetService.ExportData][google.cloud.aiplatform.v1.DatasetService.ExportData].
415415
message ExportDataResponse {
416416
// All of the files that are exported in this export operation. For custom
417-
// code training export, only three (training, validation and test) GCS paths
418-
// in wildcard format are populated (e.g., gs://.../training-*).
417+
// code training export, only three (training, validation and test)
418+
// Cloud Storage paths in wildcard format are populated
419+
// (for example, gs://.../training-*).
419420
repeated string exported_files = 1;
420421

421422
// Only present for custom code training export use case. Records data stats,

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/feature.proto‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ message Feature {
167167
[(google.api.field_behavior) = OUTPUT_ONLY];
168168

169169
// Only applicable for Vertex AI Feature Store.
170-
// The name of the BigQuery Table/View columnn hosting data for this version.
170+
// The name of the BigQuery Table/View column hosting data for this version.
171171
// If no value is provided, will use feature_id.
172172
string version_column_name = 106;
173173
}

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/feature_online_store_admin_service.proto‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ service FeatureOnlineStoreAdminService {
196196
message CreateFeatureOnlineStoreRequest {
197197
// Required. The resource name of the Location to create FeatureOnlineStores.
198198
// Format:
199-
// `projects/{project}/locations/{location}'`
199+
// `projects/{project}/locations/{location}`
200200
string parent = 1 [
201201
(google.api.field_behavior) = REQUIRED,
202202
(google.api.resource_reference) = {

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/model_evaluation.proto‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,6 @@ message ModelEvaluation {
115115
// The metadata of the ModelEvaluation.
116116
// For the ModelEvaluation uploaded from Managed Pipeline, metadata contains a
117117
// structured value with keys of "pipeline_job_id", "evaluation_dataset_type",
118-
// "evaluation_dataset_path".
118+
// "evaluation_dataset_path", "row_based_metrics_path".
119119
google.protobuf.Value metadata = 11;
120120
}

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/tensorboard_service.proto‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ service TensorboardService {
238238
rpc BatchCreateTensorboardTimeSeries(BatchCreateTensorboardTimeSeriesRequest)
239239
returns (BatchCreateTensorboardTimeSeriesResponse) {
240240
option (google.api.http) = {
241-
post: "/v1/{parent=projects/*/locations/*/tensorboards/*/experiments/*}/runs/*/timeSeries:batchCreate"
241+
post: "/v1/{parent=projects/*/locations/*/tensorboards/*/experiments/*}:batchCreate"
242242
body: "*"
243243
};
244244
option (google.api.method_signature) = "parent,requests";
@@ -305,7 +305,7 @@ service TensorboardService {
305305
BatchReadTensorboardTimeSeriesDataRequest)
306306
returns (BatchReadTensorboardTimeSeriesDataResponse) {
307307
option (google.api.http) = {
308-
get: "/v1/{tensorboard=projects/*/locations/*/tensorboards/*}/experiments/*/runs/*/timeSeries:batchRead"
308+
get: "/v1/{tensorboard=projects/*/locations/*/tensorboards/*}:batchRead"
309309
};
310310
option (google.api.method_signature) = "tensorboard";
311311
}

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1beta1/custom_job.proto‎

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,26 @@ message CustomJobSpec {
256256
type: "aiplatform.googleapis.com/Context"
257257
}
258258
];
259+
260+
// Optional. The name of the Model resources for which to generate a mapping
261+
// to artifact URIs. Applicable only to some of the Google-provided custom
262+
// jobs. Format: `projects/{project}/locations/{location}/models/{model}`
263+
//
264+
// In order to retrieve a specific version of the model, also provide
265+
// the version ID or version alias.
266+
// Example: `projects/{project}/locations/{location}/models/{model}@2`
267+
// or
268+
// `projects/{project}/locations/{location}/models/{model}@golden`
269+
// If no version ID or alias is specified, the "default" version will be
270+
// returned. The "default" version alias is created for the first version of
271+
// the model, and can be moved to other versions later on. There will be
272+
// exactly one default version.
273+
repeated string models = 20 [
274+
(google.api.field_behavior) = OPTIONAL,
275+
(google.api.resource_reference) = {
276+
type: "aiplatform.googleapis.com/Model"
277+
}
278+
];
259279
}
260280

261281
// Represents the spec of a worker pool in a job.
@@ -343,4 +363,9 @@ message Scheduling {
343363
// job starts running. If true, overrides
344364
// `Scheduling.restart_job_on_worker_restart` to false.
345365
bool disable_retries = 5 [(google.api.field_behavior) = OPTIONAL];
366+
367+
// Optional. This is the maximum time a user will wait in the QRM queue for
368+
// resources. Default is 1 day
369+
google.protobuf.Duration max_wait_duration = 6
370+
[(google.api.field_behavior) = OPTIONAL];
346371
}

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1beta1/dataset_service.proto‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -413,8 +413,9 @@ message ExportDataRequest {
413413
// [DatasetService.ExportData][google.cloud.aiplatform.v1beta1.DatasetService.ExportData].
414414
message ExportDataResponse {
415415
// All of the files that are exported in this export operation. For custom
416-
// code training export, only three (training, validation and test) GCS paths
417-
// in wildcard format are populated (e.g., gs://.../training-*).
416+
// code training export, only three (training, validation and test)
417+
// Cloud Storage paths in wildcard format are populated
418+
// (for example, gs://.../training-*).
418419
repeated string exported_files = 1;
419420
}
420421

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1beta1/feature.proto‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ message Feature {
194194
[(google.api.field_behavior) = OUTPUT_ONLY];
195195

196196
// Only applicable for Vertex AI Feature Store.
197-
// The name of the BigQuery Table/View columnn hosting data for this version.
197+
// The name of the BigQuery Table/View column hosting data for this version.
198198
// If no value is provided, will use feature_id.
199199
string version_column_name = 106;
200200
}

0 commit comments

Comments
 (0)