feat: Model Registry Model Checkpoint API

Google APIs · copybara-github · commit 245c4f55bc6c · 2025-04-22T13:47:05.000-07:00
PiperOrigin-RevId: 750313610
diff --git a/google/cloud/aiplatform/v1/model.proto b/google/cloud/aiplatform/v1/model.proto
@@ -478,6 +478,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.
@@ -972,3 +978,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;
+}
