Skip to content

Commit

Permalink
feat: Model Registry Checkpoint API
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 720330032

Source-Link: googleapis/googleapis@2c21b3a

Source-Link: googleapis/googleapis-gen@3524e3e
Copy-Tag: eyJwIjoiYXBpcy9Hb29nbGUuQ2xvdWQuQUlQbGF0Zm9ybS5WMUJldGExLy5Pd2xCb3QueWFtbCIsImgiOiIzNTI0ZTNlN2E1Y2YzYTlmM2VkYmE0Nzk1NDQ5ZmNiYzYxMGUzODczIn0=
  • Loading branch information
gcf-owl-bot[bot] committed Jan 28, 2025
1 parent fc7c5dc commit 5e484f0
Show file tree
Hide file tree
Showing 14 changed files with 2,737 additions and 492 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,79 @@
// 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.

// Generated code. DO NOT EDIT!

namespace GoogleCSharpSnippets
{
// [START aiplatform_v1beta1_generated_ModelService_ListModelVersionCheckpoints_async_flattened]
using Google.Api.Gax;
using Google.Cloud.AIPlatform.V1Beta1;
using System;
using System.Linq;
using System.Threading.Tasks;

public sealed partial class GeneratedModelServiceClientSnippets
{
/// <summary>Snippet for ListModelVersionCheckpointsAsync</summary>
/// <remarks>
/// 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/dotnet/docs/reference/help/client-configuration#endpoint.
/// </remarks>
public async Task ListModelVersionCheckpointsAsync()
{
// Create client
ModelServiceClient modelServiceClient = await ModelServiceClient.CreateAsync();
// Initialize request argument(s)
string name = "projects/[PROJECT]/locations/[LOCATION]/models/[MODEL]";
// Make the request
PagedAsyncEnumerable<ListModelVersionCheckpointsResponse, ModelVersionCheckpoint> response = modelServiceClient.ListModelVersionCheckpointsAsync(name);

// Iterate over all response items, lazily performing RPCs as required
await response.ForEachAsync((ModelVersionCheckpoint item) =>
{
// Do something with each item
Console.WriteLine(item);
});

// Or iterate over pages (of server-defined size), performing one RPC per page
await response.AsRawResponses().ForEachAsync((ListModelVersionCheckpointsResponse page) =>
{
// Do something with each page of items
Console.WriteLine("A page of results:");
foreach (ModelVersionCheckpoint item in page)
{
// Do something with each item
Console.WriteLine(item);
}
});

// Or retrieve a single page of known size (unless it's the final page), performing as many RPCs as required
int pageSize = 10;
Page<ModelVersionCheckpoint> singlePage = await response.ReadPageAsync(pageSize);
// Do something with the page of items
Console.WriteLine($"A page of {pageSize} results (unless it's the final page):");
foreach (ModelVersionCheckpoint item in singlePage)
{
// Do something with each item
Console.WriteLine(item);
}
// Store the pageToken, for when the next page is required.
string nextPageToken = singlePage.NextPageToken;
}
}
// [END aiplatform_v1beta1_generated_ModelService_ListModelVersionCheckpoints_async_flattened]
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,82 @@
// 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.

// Generated code. DO NOT EDIT!

namespace GoogleCSharpSnippets
{
// [START aiplatform_v1beta1_generated_ModelService_ListModelVersionCheckpoints_async]
using Google.Api.Gax;
using Google.Cloud.AIPlatform.V1Beta1;
using System;
using System.Linq;
using System.Threading.Tasks;

public sealed partial class GeneratedModelServiceClientSnippets
{
/// <summary>Snippet for ListModelVersionCheckpointsAsync</summary>
/// <remarks>
/// 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/dotnet/docs/reference/help/client-configuration#endpoint.
/// </remarks>
public async Task ListModelVersionCheckpointsRequestObjectAsync()
{
// Create client
ModelServiceClient modelServiceClient = await ModelServiceClient.CreateAsync();
// Initialize request argument(s)
ListModelVersionCheckpointsRequest request = new ListModelVersionCheckpointsRequest
{
ModelName = ModelName.FromProjectLocationModel("[PROJECT]", "[LOCATION]", "[MODEL]"),
};
// Make the request
PagedAsyncEnumerable<ListModelVersionCheckpointsResponse, ModelVersionCheckpoint> response = modelServiceClient.ListModelVersionCheckpointsAsync(request);

// Iterate over all response items, lazily performing RPCs as required
await response.ForEachAsync((ModelVersionCheckpoint item) =>
{
// Do something with each item
Console.WriteLine(item);
});

// Or iterate over pages (of server-defined size), performing one RPC per page
await response.AsRawResponses().ForEachAsync((ListModelVersionCheckpointsResponse page) =>
{
// Do something with each page of items
Console.WriteLine("A page of results:");
foreach (ModelVersionCheckpoint item in page)
{
// Do something with each item
Console.WriteLine(item);
}
});

// Or retrieve a single page of known size (unless it's the final page), performing as many RPCs as required
int pageSize = 10;
Page<ModelVersionCheckpoint> singlePage = await response.ReadPageAsync(pageSize);
// Do something with the page of items
Console.WriteLine($"A page of {pageSize} results (unless it's the final page):");
foreach (ModelVersionCheckpoint item in singlePage)
{
// Do something with each item
Console.WriteLine(item);
}
// Store the pageToken, for when the next page is required.
string nextPageToken = singlePage.NextPageToken;
}
}
// [END aiplatform_v1beta1_generated_ModelService_ListModelVersionCheckpoints_async]
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,80 @@
// 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.

// Generated code. DO NOT EDIT!

namespace GoogleCSharpSnippets
{
// [START aiplatform_v1beta1_generated_ModelService_ListModelVersionCheckpoints_sync]
using Google.Api.Gax;
using Google.Cloud.AIPlatform.V1Beta1;
using System;

public sealed partial class GeneratedModelServiceClientSnippets
{
/// <summary>Snippet for ListModelVersionCheckpoints</summary>
/// <remarks>
/// 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/dotnet/docs/reference/help/client-configuration#endpoint.
/// </remarks>
public void ListModelVersionCheckpointsRequestObject()
{
// Create client
ModelServiceClient modelServiceClient = ModelServiceClient.Create();
// Initialize request argument(s)
ListModelVersionCheckpointsRequest request = new ListModelVersionCheckpointsRequest
{
ModelName = ModelName.FromProjectLocationModel("[PROJECT]", "[LOCATION]", "[MODEL]"),
};
// Make the request
PagedEnumerable<ListModelVersionCheckpointsResponse, ModelVersionCheckpoint> response = modelServiceClient.ListModelVersionCheckpoints(request);

// Iterate over all response items, lazily performing RPCs as required
foreach (ModelVersionCheckpoint item in response)
{
// Do something with each item
Console.WriteLine(item);
}

// Or iterate over pages (of server-defined size), performing one RPC per page
foreach (ListModelVersionCheckpointsResponse page in response.AsRawResponses())
{
// Do something with each page of items
Console.WriteLine("A page of results:");
foreach (ModelVersionCheckpoint item in page)
{
// Do something with each item
Console.WriteLine(item);
}
}

// Or retrieve a single page of known size (unless it's the final page), performing as many RPCs as required
int pageSize = 10;
Page<ModelVersionCheckpoint> singlePage = response.ReadPage(pageSize);
// Do something with the page of items
Console.WriteLine($"A page of {pageSize} results (unless it's the final page):");
foreach (ModelVersionCheckpoint item in singlePage)
{
// Do something with each item
Console.WriteLine(item);
}
// Store the pageToken, for when the next page is required.
string nextPageToken = singlePage.NextPageToken;
}
}
// [END aiplatform_v1beta1_generated_ModelService_ListModelVersionCheckpoints_sync]
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,79 @@
// 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.

// Generated code. DO NOT EDIT!

namespace GoogleCSharpSnippets
{
// [START aiplatform_v1beta1_generated_ModelService_ListModelVersionCheckpoints_async_flattened_resourceNames]
using Google.Api.Gax;
using Google.Cloud.AIPlatform.V1Beta1;
using System;
using System.Linq;
using System.Threading.Tasks;

public sealed partial class GeneratedModelServiceClientSnippets
{
/// <summary>Snippet for ListModelVersionCheckpointsAsync</summary>
/// <remarks>
/// 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/dotnet/docs/reference/help/client-configuration#endpoint.
/// </remarks>
public async Task ListModelVersionCheckpointsResourceNamesAsync()
{
// Create client
ModelServiceClient modelServiceClient = await ModelServiceClient.CreateAsync();
// Initialize request argument(s)
ModelName name = ModelName.FromProjectLocationModel("[PROJECT]", "[LOCATION]", "[MODEL]");
// Make the request
PagedAsyncEnumerable<ListModelVersionCheckpointsResponse, ModelVersionCheckpoint> response = modelServiceClient.ListModelVersionCheckpointsAsync(name);

// Iterate over all response items, lazily performing RPCs as required
await response.ForEachAsync((ModelVersionCheckpoint item) =>
{
// Do something with each item
Console.WriteLine(item);
});

// Or iterate over pages (of server-defined size), performing one RPC per page
await response.AsRawResponses().ForEachAsync((ListModelVersionCheckpointsResponse page) =>
{
// Do something with each page of items
Console.WriteLine("A page of results:");
foreach (ModelVersionCheckpoint item in page)
{
// Do something with each item
Console.WriteLine(item);
}
});

// Or retrieve a single page of known size (unless it's the final page), performing as many RPCs as required
int pageSize = 10;
Page<ModelVersionCheckpoint> singlePage = await response.ReadPageAsync(pageSize);
// Do something with the page of items
Console.WriteLine($"A page of {pageSize} results (unless it's the final page):");
foreach (ModelVersionCheckpoint item in singlePage)
{
// Do something with each item
Console.WriteLine(item);
}
// Store the pageToken, for when the next page is required.
string nextPageToken = singlePage.NextPageToken;
}
}
// [END aiplatform_v1beta1_generated_ModelService_ListModelVersionCheckpoints_async_flattened_resourceNames]
}
Loading

0 comments on commit 5e484f0

Please sign in to comment.