update asset archive

Author: alexluck-sift

python/lib/sift_client/_internal/low_level_wrappers/assets.py

@@ -3,7 +3,8 @@
 from typing import Any, cast
 
 from sift.assets.v1.assets_pb2 import (
-    DeleteAssetRequest,
+    ArchiveAssetRequest,
+ArchiveAssetResponse,
     GetAssetRequest,
     GetAssetResponse,
     ListAssetsRequest,
@@ -96,6 +97,8 @@ async def update_asset(self, update: AssetUpdate) -> Asset:
         updated_grpc_asset = cast("UpdateAssetResponse", response).asset
         return Asset._from_proto(updated_grpc_asset)
 
-    async def delete_asset(self, asset_id: str, archive_runs: bool = False) -> None:
-        request = DeleteAssetRequest(asset_id=asset_id, archive_runs=archive_runs)
-        await self._grpc_client.get_stub(AssetServiceStub).DeleteAsset(request)
+    async def archive_asset(self, asset_id: str, archive_runs: bool = False) -> list[str] | None:
+        request = ArchiveAssetRequest(asset_id=asset_id, archive_runs=archive_runs)
+        response = await self._grpc_client.get_stub(AssetServiceStub).ArchiveAsset(request)
+        response = cast("ArchiveAssetResponse", response)
+        return response.archived_runs

python/lib/sift_client/resources/assets.py

@@ -192,8 +192,17 @@ async def archive(self, asset: str | Asset, *, archive_runs: bool = False) -> As
         """
         asset_id = asset._id_or_error if isinstance(asset, Asset) else asset
 
-        await self._low_level_client.delete_asset(asset_id, archive_runs=archive_runs)
+        await self._low_level_client.archive_asset(asset_id, archive_runs=archive_runs)
 
         return await self.get(asset_id=asset_id)
 
-    # TODO: unarchive
+    async def unarchive(self, asset: str | Asset) -> Asset:
+        """Unarchive an asset.
+
+        Args:
+             asset: The Asset or asset ID to unarchive.
+
+        Returns:
+             The unarchived Asset.
+        """
+        return await self.update(asset, AssetUpdate(is_archived=False))