@flyteorg/flyteidl 1.2.2 → 1.2.3

package/gen/pb-js/flyteidl.d.ts

@@ -12,7 +12,8 @@ export namespace flyteidl {
             CACHE_HIT = 2,
             CACHE_POPULATED = 3,
             CACHE_LOOKUP_FAILURE = 4,
-            CACHE_PUT_FAILURE = 5
+            CACHE_PUT_FAILURE = 5,
+            CACHE_SKIPPED = 6
         }
 
         /** Properties of a CatalogArtifactTag. */
@@ -9289,6 +9290,9 @@ export namespace flyteidl {
 
             /** ExecutionRelaunchRequest name */
             name?: (string|null);
+
+            /** ExecutionRelaunchRequest overwriteCache */
+            overwriteCache?: (boolean|null);
         }
 
         /** Represents an ExecutionRelaunchRequest. */
@@ -9306,6 +9310,9 @@ export namespace flyteidl {
             /** ExecutionRelaunchRequest name. */
             public name: string;
 
+            /** ExecutionRelaunchRequest overwriteCache. */
+            public overwriteCache: boolean;
+
             /**
              * Creates a new ExecutionRelaunchRequest instance using the specified properties.
              * @param [properties] Properties to set
@@ -10130,6 +10137,9 @@ export namespace flyteidl {
 
             /** ExecutionSpec interruptible */
             interruptible?: (google.protobuf.IBoolValue|null);
+
+            /** ExecutionSpec overwriteCache */
+            overwriteCache?: (boolean|null);
         }
 
         /** Represents an ExecutionSpec. */
@@ -10183,6 +10193,9 @@ export namespace flyteidl {
             /** ExecutionSpec interruptible. */
             public interruptible?: (google.protobuf.IBoolValue|null);
 
+            /** ExecutionSpec overwriteCache. */
+            public overwriteCache: boolean;
+
             /** ExecutionSpec notificationOverrides. */
             public notificationOverrides?: ("notifications"|"disableAll");
 
@@ -10953,6 +10966,9 @@ export namespace flyteidl {
 
             /** LaunchPlanSpec interruptible */
             interruptible?: (google.protobuf.IBoolValue|null);
+
+            /** LaunchPlanSpec overwriteCache */
+            overwriteCache?: (boolean|null);
         }
 
         /** Represents a LaunchPlanSpec. */
@@ -11006,6 +11022,9 @@ export namespace flyteidl {
             /** LaunchPlanSpec interruptible. */
             public interruptible?: (google.protobuf.IBoolValue|null);
 
+            /** LaunchPlanSpec overwriteCache. */
+            public overwriteCache: boolean;
+
             /**
              * Creates a new LaunchPlanSpec instance using the specified properties.
              * @param [properties] Properties to set
@@ -12048,6 +12067,9 @@ export namespace flyteidl {
 
             /** WorkflowExecutionConfig interruptible */
             interruptible?: (google.protobuf.IBoolValue|null);
+
+            /** WorkflowExecutionConfig overwriteCache */
+            overwriteCache?: (boolean|null);
         }
 
         /** Represents a WorkflowExecutionConfig. */
@@ -12077,6 +12099,9 @@ export namespace flyteidl {
             /** WorkflowExecutionConfig interruptible. */
             public interruptible?: (google.protobuf.IBoolValue|null);
 
+            /** WorkflowExecutionConfig overwriteCache. */
+            public overwriteCache: boolean;
+
             /**
              * Creates a new WorkflowExecutionConfig instance using the specified properties.
              * @param [properties] Properties to set

package/gen/pb-js/flyteidl.js

@@ -44,6 +44,7 @@
              * @property {number} CACHE_POPULATED=3 CACHE_POPULATED value
              * @property {number} CACHE_LOOKUP_FAILURE=4 CACHE_LOOKUP_FAILURE value
              * @property {number} CACHE_PUT_FAILURE=5 CACHE_PUT_FAILURE value
+             * @property {number} CACHE_SKIPPED=6 CACHE_SKIPPED value
              */
             core.CatalogCacheStatus = (function() {
                 var valuesById = {}, values = Object.create(valuesById);
@@ -53,6 +54,7 @@
                 values[valuesById[3] = "CACHE_POPULATED"] = 3;
                 values[valuesById[4] = "CACHE_LOOKUP_FAILURE"] = 4;
                 values[valuesById[5] = "CACHE_PUT_FAILURE"] = 5;
+                values[valuesById[6] = "CACHE_SKIPPED"] = 6;
                 return values;
             })();
     
@@ -16348,6 +16350,7 @@
                         case 3:
                         case 4:
                         case 5:
+                        case 6:
                             break;
                         }
                     if (message.catalogKey != null && message.hasOwnProperty("catalogKey")) {
@@ -17385,6 +17388,7 @@
                         case 3:
                         case 4:
                         case 5:
+                        case 6:
                             break;
                         }
                     if (message.logs != null && message.hasOwnProperty("logs")) {
@@ -22252,6 +22256,7 @@
                  * @interface IExecutionRelaunchRequest
                  * @property {flyteidl.core.IWorkflowExecutionIdentifier|null} [id] ExecutionRelaunchRequest id
                  * @property {string|null} [name] ExecutionRelaunchRequest name
+                 * @property {boolean|null} [overwriteCache] ExecutionRelaunchRequest overwriteCache
                  */
     
                 /**
@@ -22285,6 +22290,14 @@
                  */
                 ExecutionRelaunchRequest.prototype.name = "";
     
+                /**
+                 * ExecutionRelaunchRequest overwriteCache.
+                 * @member {boolean} overwriteCache
+                 * @memberof flyteidl.admin.ExecutionRelaunchRequest
+                 * @instance
+                 */
+                ExecutionRelaunchRequest.prototype.overwriteCache = false;
+    
                 /**
                  * Creates a new ExecutionRelaunchRequest instance using the specified properties.
                  * @function create
@@ -22313,6 +22326,8 @@
                         $root.flyteidl.core.WorkflowExecutionIdentifier.encode(message.id, writer.uint32(/* id 1, wireType 2 =*/10).fork()).ldelim();
                     if (message.name != null && message.hasOwnProperty("name"))
                         writer.uint32(/* id 3, wireType 2 =*/26).string(message.name);
+                    if (message.overwriteCache != null && message.hasOwnProperty("overwriteCache"))
+                        writer.uint32(/* id 4, wireType 0 =*/32).bool(message.overwriteCache);
                     return writer;
                 };
     
@@ -22340,6 +22355,9 @@
                         case 3:
                             message.name = reader.string();
                             break;
+                        case 4:
+                            message.overwriteCache = reader.bool();
+                            break;
                         default:
                             reader.skipType(tag & 7);
                             break;
@@ -22367,6 +22385,9 @@
                     if (message.name != null && message.hasOwnProperty("name"))
                         if (!$util.isString(message.name))
                             return "name: string expected";
+                    if (message.overwriteCache != null && message.hasOwnProperty("overwriteCache"))
+                        if (typeof message.overwriteCache !== "boolean")
+                            return "overwriteCache: boolean expected";
                     return null;
                 };
     
@@ -24224,6 +24245,7 @@
                  * @property {flyteidl.admin.IRawOutputDataConfig|null} [rawOutputDataConfig] ExecutionSpec rawOutputDataConfig
                  * @property {flyteidl.admin.IClusterAssignment|null} [clusterAssignment] ExecutionSpec clusterAssignment
                  * @property {google.protobuf.IBoolValue|null} [interruptible] ExecutionSpec interruptible
+                 * @property {boolean|null} [overwriteCache] ExecutionSpec overwriteCache
                  */
     
                 /**
@@ -24353,6 +24375,14 @@
                  */
                 ExecutionSpec.prototype.interruptible = null;
     
+                /**
+                 * ExecutionSpec overwriteCache.
+                 * @member {boolean} overwriteCache
+                 * @memberof flyteidl.admin.ExecutionSpec
+                 * @instance
+                 */
+                ExecutionSpec.prototype.overwriteCache = false;
+    
                 // OneOf field names bound to virtual getters and setters
                 var $oneOfFields;
     
@@ -24419,6 +24449,8 @@
                         $root.flyteidl.admin.ClusterAssignment.encode(message.clusterAssignment, writer.uint32(/* id 20, wireType 2 =*/162).fork()).ldelim();
                     if (message.interruptible != null && message.hasOwnProperty("interruptible"))
                         $root.google.protobuf.BoolValue.encode(message.interruptible, writer.uint32(/* id 21, wireType 2 =*/170).fork()).ldelim();
+                    if (message.overwriteCache != null && message.hasOwnProperty("overwriteCache"))
+                        writer.uint32(/* id 22, wireType 0 =*/176).bool(message.overwriteCache);
                     return writer;
                 };
     
@@ -24482,6 +24514,9 @@
                         case 21:
                             message.interruptible = $root.google.protobuf.BoolValue.decode(reader, reader.uint32());
                             break;
+                        case 22:
+                            message.overwriteCache = reader.bool();
+                            break;
                         default:
                             reader.skipType(tag & 7);
                             break;
@@ -24575,6 +24610,9 @@
                         if (error)
                             return "interruptible." + error;
                     }
+                    if (message.overwriteCache != null && message.hasOwnProperty("overwriteCache"))
+                        if (typeof message.overwriteCache !== "boolean")
+                            return "overwriteCache: boolean expected";
                     return null;
                 };
     
@@ -26148,6 +26186,7 @@
                  * @property {flyteidl.admin.IRawOutputDataConfig|null} [rawOutputDataConfig] LaunchPlanSpec rawOutputDataConfig
                  * @property {number|null} [maxParallelism] LaunchPlanSpec maxParallelism
                  * @property {google.protobuf.IBoolValue|null} [interruptible] LaunchPlanSpec interruptible
+                 * @property {boolean|null} [overwriteCache] LaunchPlanSpec overwriteCache
                  */
     
                 /**
@@ -26277,6 +26316,14 @@
                  */
                 LaunchPlanSpec.prototype.interruptible = null;
     
+                /**
+                 * LaunchPlanSpec overwriteCache.
+                 * @member {boolean} overwriteCache
+                 * @memberof flyteidl.admin.LaunchPlanSpec
+                 * @instance
+                 */
+                LaunchPlanSpec.prototype.overwriteCache = false;
+    
                 /**
                  * Creates a new LaunchPlanSpec instance using the specified properties.
                  * @function create
@@ -26329,6 +26376,8 @@
                         writer.uint32(/* id 18, wireType 0 =*/144).int32(message.maxParallelism);
                     if (message.interruptible != null && message.hasOwnProperty("interruptible"))
                         $root.google.protobuf.BoolValue.encode(message.interruptible, writer.uint32(/* id 19, wireType 2 =*/154).fork()).ldelim();
+                    if (message.overwriteCache != null && message.hasOwnProperty("overwriteCache"))
+                        writer.uint32(/* id 20, wireType 0 =*/160).bool(message.overwriteCache);
                     return writer;
                 };
     
@@ -26392,6 +26441,9 @@
                         case 19:
                             message.interruptible = $root.google.protobuf.BoolValue.decode(reader, reader.uint32());
                             break;
+                        case 20:
+                            message.overwriteCache = reader.bool();
+                            break;
                         default:
                             reader.skipType(tag & 7);
                             break;
@@ -26477,6 +26529,9 @@
                         if (error)
                             return "interruptible." + error;
                     }
+                    if (message.overwriteCache != null && message.hasOwnProperty("overwriteCache"))
+                        if (typeof message.overwriteCache !== "boolean")
+                            return "overwriteCache: boolean expected";
                     return null;
                 };
     
@@ -28788,6 +28843,7 @@
                  * @property {flyteidl.admin.ILabels|null} [labels] WorkflowExecutionConfig labels
                  * @property {flyteidl.admin.IAnnotations|null} [annotations] WorkflowExecutionConfig annotations
                  * @property {google.protobuf.IBoolValue|null} [interruptible] WorkflowExecutionConfig interruptible
+                 * @property {boolean|null} [overwriteCache] WorkflowExecutionConfig overwriteCache
                  */
     
                 /**
@@ -28853,6 +28909,14 @@
                  */
                 WorkflowExecutionConfig.prototype.interruptible = null;
     
+                /**
+                 * WorkflowExecutionConfig overwriteCache.
+                 * @member {boolean} overwriteCache
+                 * @memberof flyteidl.admin.WorkflowExecutionConfig
+                 * @instance
+                 */
+                WorkflowExecutionConfig.prototype.overwriteCache = false;
+    
                 /**
                  * Creates a new WorkflowExecutionConfig instance using the specified properties.
                  * @function create
@@ -28889,6 +28953,8 @@
                         $root.flyteidl.admin.Annotations.encode(message.annotations, writer.uint32(/* id 5, wireType 2 =*/42).fork()).ldelim();
                     if (message.interruptible != null && message.hasOwnProperty("interruptible"))
                         $root.google.protobuf.BoolValue.encode(message.interruptible, writer.uint32(/* id 6, wireType 2 =*/50).fork()).ldelim();
+                    if (message.overwriteCache != null && message.hasOwnProperty("overwriteCache"))
+                        writer.uint32(/* id 7, wireType 0 =*/56).bool(message.overwriteCache);
                     return writer;
                 };
     
@@ -28928,6 +28994,9 @@
                         case 6:
                             message.interruptible = $root.google.protobuf.BoolValue.decode(reader, reader.uint32());
                             break;
+                        case 7:
+                            message.overwriteCache = reader.bool();
+                            break;
                         default:
                             reader.skipType(tag & 7);
                             break;
@@ -28975,6 +29044,9 @@
                         if (error)
                             return "interruptible." + error;
                     }
+                    if (message.overwriteCache != null && message.hasOwnProperty("overwriteCache"))
+                        if (typeof message.overwriteCache !== "boolean")
+                            return "overwriteCache: boolean expected";
                     return null;
                 };
     
@@ -31265,6 +31337,7 @@
                         case 3:
                         case 4:
                         case 5:
+                        case 6:
                             break;
                         }
                     if (message.catalogKey != null && message.hasOwnProperty("catalogKey")) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@flyteorg/flyteidl",
-  "version": "1.2.2",
+  "version": "1.2.3",
   "description": "Compiled protocol buffers and gRPC service clients/servers for Flyte IDLs",
   "repository": {
     "type": "git",

package/protos/docs/admin/admin.rst

@@ -1084,6 +1084,7 @@ Request to relaunch the referenced execution.
 
    "id", ":ref:`ref_flyteidl.core.WorkflowExecutionIdentifier`", "", "Identifier of the workflow execution to relaunch. +required"
    "name", ":ref:`ref_string`", "", "User provided value for the relaunched execution. If none is provided the system will generate a unique string. +optional"
+   "overwrite_cache", ":ref:`ref_bool`", "", "Allows for all cached values of a workflow and its tasks to be overwritten for a single execution. If enabled, all calculations are performed even if cached results would be available, overwriting the stored data once execution finishes successfully."
 
 
 
@@ -1119,6 +1120,7 @@ of an execution as it progresses across phase changes.
    "raw_output_data_config", ":ref:`ref_flyteidl.admin.RawOutputDataConfig`", "", "User setting to configure where to store offloaded data (i.e. Blobs, structured datasets, query data, etc.). This should be a prefix like s3://my-bucket/my-data"
    "cluster_assignment", ":ref:`ref_flyteidl.admin.ClusterAssignment`", "", "Controls how to select an available cluster on which this execution should run."
    "interruptible", ":ref:`ref_google.protobuf.BoolValue`", "", "Allows for the interruptible flag of a workflow to be overwritten for a single execution. Omitting this field uses the workflow's value as a default. As we need to distinguish between the field not being provided and its default value false, we have to use a wrapper around the bool field."
+   "overwrite_cache", ":ref:`ref_bool`", "", "Allows for all cached values of a workflow and its tasks to be overwritten for a single execution. If enabled, all calculations are performed even if cached results would be available, overwriting the stored data once execution finishes successfully."
 
 
 
@@ -1655,6 +1657,7 @@ User-provided launch plan definition and configuration values.
    "raw_output_data_config", ":ref:`ref_flyteidl.admin.RawOutputDataConfig`", "", "Encapsulates user settings pertaining to offloaded data (i.e. Blobs, Schema, query data, etc.)."
    "max_parallelism", ":ref:`ref_int32`", "", "Controls the maximum number of tasknodes that can be run in parallel for the entire workflow. This is useful to achieve fairness. Note: MapTasks are regarded as one unit, and parallelism/concurrency of MapTasks is independent from this."
    "interruptible", ":ref:`ref_google.protobuf.BoolValue`", "", "Allows for the interruptible flag of a workflow to be overwritten for a single execution. Omitting this field uses the workflow's value as a default. As we need to distinguish between the field not being provided and its default value false, we have to use a wrapper around the bool field."
+   "overwrite_cache", ":ref:`ref_bool`", "", "Allows for all cached values of a workflow and its tasks to be overwritten for a single execution. If enabled, all calculations are performed even if cached results would be available, overwriting the stored data once execution finishes successfully."
 
 
 
@@ -2042,6 +2045,7 @@ Adds defaults for customizable workflow-execution specifications and overrides.
    "labels", ":ref:`ref_flyteidl.admin.Labels`", "", "Custom labels to be applied to a triggered execution resource."
    "annotations", ":ref:`ref_flyteidl.admin.Annotations`", "", "Custom annotations to be applied to a triggered execution resource."
    "interruptible", ":ref:`ref_google.protobuf.BoolValue`", "", "Allows for the interruptible flag of a workflow to be overwritten for a single execution. Omitting this field uses the workflow's value as a default. As we need to distinguish between the field not being provided and its default value false, we have to use a wrapper around the bool field."
+   "overwrite_cache", ":ref:`ref_bool`", "", "Allows for all cached values of a workflow and its tasks to be overwritten for a single execution. If enabled, all calculations are performed even if cached results would be available, overwriting the stored data once execution finishes successfully."
 
 
 

package/protos/docs/core/core.rst

@@ -94,6 +94,7 @@ Indicates the status of CatalogCaching. The reason why this is not embedded in T
    "CACHE_POPULATED", "3", "used to indicate that the resultant artifact was added to the cache"
    "CACHE_LOOKUP_FAILURE", "4", "Used to indicate that cache lookup failed because of an error"
    "CACHE_PUT_FAILURE", "5", "Used to indicate that cache lookup failed because of an error"
+   "CACHE_SKIPPED", "6", "Used to indicate the cache lookup was skipped"
 
 
 

package/protos/docs/datacatalog/datacatalog.rst

@@ -790,6 +790,51 @@ Tag properties we can filter by
 
 
 
+
+.. _ref_datacatalog.UpdateArtifactRequest:
+
+UpdateArtifactRequest
+------------------------------------------------------------------
+
+Request message for updating an Artifact and overwriting its associated ArtifactData.
+
+
+
+.. csv-table:: UpdateArtifactRequest type fields
+   :header: "Field", "Type", "Label", "Description"
+   :widths: auto
+
+   "dataset", ":ref:`ref_datacatalog.DatasetID`", "", "ID of dataset the artifact is associated with"
+   "artifact_id", ":ref:`ref_string`", "", ""
+   "tag_name", ":ref:`ref_string`", "", ""
+   "data", ":ref:`ref_datacatalog.ArtifactData`", "repeated", "List of data to overwrite stored artifact data with. Must contain ALL data for updated Artifact as any missing ArtifactData entries will be removed from the underlying blob storage and database."
+
+
+
+
+
+
+
+.. _ref_datacatalog.UpdateArtifactResponse:
+
+UpdateArtifactResponse
+------------------------------------------------------------------
+
+Response message for updating an Artifact.
+
+
+
+.. csv-table:: UpdateArtifactResponse type fields
+   :header: "Field", "Type", "Label", "Description"
+   :widths: auto
+
+   "artifact_id", ":ref:`ref_string`", "", "The unique ID of the artifact updated"
+
+
+
+
+
+
 ..
    end messages
 
@@ -869,6 +914,7 @@ Artifacts are associated with a Dataset, and can be tagged for retrieval.
    "AddTag", ":ref:`ref_datacatalog.AddTagRequest`", ":ref:`ref_datacatalog.AddTagResponse`", "Associate a tag with an artifact. Tags are unique within a Dataset."
    "ListArtifacts", ":ref:`ref_datacatalog.ListArtifactsRequest`", ":ref:`ref_datacatalog.ListArtifactsResponse`", "Return a paginated list of artifacts"
    "ListDatasets", ":ref:`ref_datacatalog.ListDatasetsRequest`", ":ref:`ref_datacatalog.ListDatasetsResponse`", "Return a paginated list of datasets"
+   "UpdateArtifact", ":ref:`ref_datacatalog.UpdateArtifactRequest`", ":ref:`ref_datacatalog.UpdateArtifactResponse`", "Updates an existing artifact, overwriting the stored artifact data in the underlying blob storage."
    "GetOrExtendReservation", ":ref:`ref_datacatalog.GetOrExtendReservationRequest`", ":ref:`ref_datacatalog.GetOrExtendReservationResponse`", "Attempts to get or extend a reservation for the corresponding artifact. If one already exists (ie. another entity owns the reservation) then that reservation is retrieved. Once you acquire a reservation, you need to periodically extend the reservation with an identical call. If the reservation is not extended before the defined expiration, it may be acquired by another task. Note: We may have multiple concurrent tasks with the same signature and the same input that try to populate the same artifact at the same time. Thus with reservation, only one task can run at a time, until the reservation expires. Note: If task A does not extend the reservation in time and the reservation expires, another task B may take over the reservation, resulting in two tasks A and B running in parallel. So a third task C may get the Artifact from A or B, whichever writes last."
    "ReleaseReservation", ":ref:`ref_datacatalog.ReleaseReservationRequest`", ":ref:`ref_datacatalog.ReleaseReservationResponse`", "Release the reservation when the task holding the spot fails so that the other tasks can grab the spot."
 

package/protos/flyteidl/admin/execution.proto

@@ -45,10 +45,18 @@ message ExecutionRelaunchRequest {
     // +required
     core.WorkflowExecutionIdentifier id = 1;
 
+    // Deprecated field, do not use.
+    reserved 2;
+
     // User provided value for the relaunched execution.
     // If none is provided the system will generate a unique string.
     // +optional
     string name = 3;
+
+    // Allows for all cached values of a workflow and its tasks to be overwritten for a single execution.
+    // If enabled, all calculations are performed even if cached results would be available, overwriting the stored
+    // data once execution finishes successfully.
+    bool overwrite_cache = 4;
 }
 
 // Request to recover the referenced execution.
@@ -296,6 +304,11 @@ message ExecutionSpec {
     // As we need to distinguish between the field not being provided and its default value false, we have to use a wrapper
     // around the bool field.
     google.protobuf.BoolValue interruptible = 21;
+
+    // Allows for all cached values of a workflow and its tasks to be overwritten for a single execution.
+    // If enabled, all calculations are performed even if cached results would be available, overwriting the stored
+    // data once execution finishes successfully.
+    bool overwrite_cache = 22;
 }
 
 // Request to terminate an in-progress execution.  This action is irreversible.

package/protos/flyteidl/admin/launch_plan.proto

@@ -125,6 +125,11 @@ message LaunchPlanSpec {
     // As we need to distinguish between the field not being provided and its default value false, we have to use a wrapper
     // around the bool field.
     google.protobuf.BoolValue interruptible = 19;
+
+    // Allows for all cached values of a workflow and its tasks to be overwritten for a single execution.
+    // If enabled, all calculations are performed even if cached results would be available, overwriting the stored
+    // data once execution finishes successfully.
+    bool overwrite_cache = 20;
 }
 
 // Values computed by the flyte platform after launch plan registration.

package/protos/flyteidl/admin/matchable_resource.proto

@@ -123,6 +123,11 @@ message WorkflowExecutionConfig {
   // As we need to distinguish between the field not being provided and its default value false, we have to use a wrapper
   // around the bool field.
   google.protobuf.BoolValue interruptible = 6;
+
+  // Allows for all cached values of a workflow and its tasks to be overwritten for a single execution.
+  // If enabled, all calculations are performed even if cached results would be available, overwriting the stored
+  // data once execution finishes successfully.
+  bool overwrite_cache = 7;
 }
 
 // Generic container for encapsulating all types of the above attributes messages.

package/protos/flyteidl/core/catalog.proto

@@ -20,6 +20,8 @@ enum CatalogCacheStatus {
   CACHE_LOOKUP_FAILURE = 4;
   // Used to indicate that cache lookup failed because of an error
   CACHE_PUT_FAILURE = 5;
+  // Used to indicate the cache lookup was skipped
+  CACHE_SKIPPED = 6;
 };
 
 message CatalogArtifactTag {

package/protos/flyteidl/datacatalog/datacatalog.proto

@@ -37,6 +37,9 @@ service DataCatalog {
     // Return a paginated list of datasets
     rpc ListDatasets (ListDatasetsRequest) returns (ListDatasetsResponse);
 
+    // Updates an existing artifact, overwriting the stored artifact data in the underlying blob storage.
+    rpc UpdateArtifact (UpdateArtifactRequest) returns (UpdateArtifactResponse);
+
     // Attempts to get or extend a reservation for the corresponding artifact. If one already exists
     // (ie. another entity owns the reservation) then that reservation is retrieved.
     // Once you acquire a reservation, you need to  periodically extend the reservation with an
@@ -170,6 +173,32 @@ message ListDatasetsResponse {
     string next_token = 2;
 }
 
+/*
+ * Request message for updating an Artifact and overwriting its associated ArtifactData.
+ */
+message UpdateArtifactRequest {
+    // ID of dataset the artifact is associated with
+    DatasetID dataset = 1;
+
+    // Either ID of artifact or name of tag to retrieve existing artifact from
+    oneof query_handle {
+        string artifact_id = 2;
+        string tag_name = 3;
+    }
+
+    // List of data to overwrite stored artifact data with. Must contain ALL data for updated Artifact as any missing
+    // ArtifactData entries will be removed from the underlying blob storage and database.
+    repeated ArtifactData data = 4;
+}
+
+/*
+ * Response message for updating an Artifact.
+ */
+message UpdateArtifactResponse {
+    // The unique ID of the artifact updated
+    string artifact_id = 1;
+}
+
 /*
  * ReservationID message that is composed of several string fields.
  */