update update classes

Author: alexluck-sift

python/lib/sift_client/sift_types/asset.py

@@ -104,7 +104,6 @@ class AssetUpdate(ModelUpdate[AssetProto]):
     """Model of the Asset Fields that can be updated."""
 
     tags: list[str] | None = None
-    archived_date: datetime | str | None = None
     metadata: dict[str, str | float | bool] | None = None
     is_archived: bool | None = None
 

python/lib/sift_client/sift_types/calculated_channel.py

@@ -201,7 +201,7 @@ class CalculatedChannelUpdate(CalculatedChannelBase, ModelUpdate[CalculatedChann
     """Update model for a Calculated Channel."""
 
     name: str | None = None
-    archived_date: datetime | None = None
+    is_archived: bool | None = None
 
     @model_validator(mode="after")
     def _validate_non_updatable_fields(self):

python/lib/sift_client/sift_types/rule.py

@@ -315,7 +315,8 @@ class RuleVersion(BaseType[RuleVersionProto, "RuleVersion"]):
     created_by_user_id: str
     version_notes: str
     generated_change_message: str
-    deleted_date: datetime | None = None
+    archived_date: datetime | None
+    is_archived: bool
 
     @classmethod
     def _from_proto(
@@ -329,6 +330,7 @@ def _from_proto(
             created_by_user_id=proto.created_by_user_id,
             version_notes=proto.version_notes,
             generated_change_message=proto.generated_change_message,
-            deleted_date=proto.deleted_date.ToDatetime() if proto.deleted_date else None,
+            archived_date=proto.archived_date.ToDatetime() if proto.archived_date else None,
+            is_archived=proto.is_archived,
             _client=sift_client,
         )

python/lib/sift_client/sift_types/run.py

@@ -121,6 +121,17 @@ class RunCreate(RunBase, ModelCreate[CreateRunRequestProto]):
 
     name: str
     client_key: str | None = None
+    tags: list[str] | None = None
+    metadata: dict[str, str | float | bool] | None = None
+    start_time: datetime | None = None
+    stop_time: datetime | None = None
+    organization_id: str | None = None
+
+    _to_proto_helpers: ClassVar = {
+        "metadata": MappingHelper(
+            proto_attr_path="metadata", update_field="metadata", converter=metadata_dict_to_proto
+        ),
+    }
 
     def _get_proto_class(self) -> type[CreateRunRequestProto]:
         return CreateRunRequestProto
@@ -130,14 +141,17 @@ class RunUpdate(RunBase, ModelUpdate[RunProto]):
     """Update model for Run."""
 
     name: str | None = None
+    tags: list[str] | None = None
+    metadata: dict[str, str | float | bool] | None = None
+    start_time: datetime | None = None
+    stop_time: datetime | None = None
     is_archived: bool | None = None
 
-    @model_validator(mode="after")
-    def _validate_non_updatable_fields(self):
-        """Validate that the fields that cannot be updated are not set."""
-        if self.client_key is not None:
-            raise ValueError("Cannot update client key")
-        return self
+    _to_proto_helpers: ClassVar = {
+        "metadata": MappingHelper(
+            proto_attr_path="metadata", update_field="metadata", converter=metadata_dict_to_proto
+        ),
+    }
 
     def _get_proto_class(self) -> type[RunProto]:
         return RunProto