feat: Model Registry Model Checkpoint API

Google APIs · copybara-github · commit 7473ce12803b · 2025-04-16T16:20:47.000-07:00
PiperOrigin-RevId: 748459487
diff --git a/google/cloud/aiplatform/v1beta1/model.proto b/google/cloud/aiplatform/v1beta1/model.proto
@@ -437,6 +437,12 @@ message Model {
 
   // Output only. Reserved for future use.
   bool satisfies_pzi = 52 [(google.api.field_behavior) = OUTPUT_ONLY];
+
+  // Optional. Output only. The checkpoints of the model.
+  repeated Checkpoint checkpoints = 57 [
+    (google.api.field_behavior) = OUTPUT_ONLY,
+    (google.api.field_behavior) = OPTIONAL
+  ];
 }
 
 // Contains information about the Large Model.
@@ -938,3 +944,15 @@ message Probe {
   // Maps to Kubernetes probe argument 'initialDelaySeconds'.
   int32 initial_delay_seconds = 9;
 }
+
+// Describes the machine learning model version checkpoint.
+message Checkpoint {
+  // 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;
+}
