diff --git a/packages/adapters/in_memory/asset_repository.py b/packages/adapters/in_memory/asset_repository.py index 0cc11ba6a..c3fc13e5c 100755 --- a/packages/adapters/in_memory/asset_repository.py +++ b/packages/adapters/in_memory/asset_repository.py @@ -45,9 +45,10 @@ class InMemoryAssetRepository: def batch_delete(self, asset_ids: list[str]) -> int: """批量删除素材(软删除,标记 status=deleted),返回实际影响数量。""" - from packages.domain import AssetStatus from datetime import datetime, timezone + from packages.domain import AssetStatus + count = 0 for aid in asset_ids: asset = self._assets.get(aid) diff --git a/tests/integration/test_assets_api.py b/tests/integration/test_assets_api.py index ccb312d0d..d15ef7df1 100755 --- a/tests/integration/test_assets_api.py +++ b/tests/integration/test_assets_api.py @@ -130,9 +130,10 @@ class StubAssetRepository: def batch_delete(self, asset_ids: list[str]) -> int: """软删除:标记 status=deleted。""" - from packages.domain import AssetStatus from datetime import datetime, timezone + from packages.domain import AssetStatus + count = 0 for aid in asset_ids: asset = self._assets.get(aid)