move forget

Author: giancarloromeo

services/storage/src/simcore_service_storage/modules/celery/backends/_redis.py

@@ -2,8 +2,6 @@
 from datetime import timedelta
 from typing import Final
 
-from celery.result import AsyncResult # type: ignore[import-untyped]
-from common_library.async_tools import make_async
 from models_library.progress_bar import ProgressReport
 from servicelib.redis._client import RedisClientSDK
 
@@ -97,14 +95,8 @@ async def list_tasks(self, task_context: TaskContext) -> list[Task]:
             if metadata is not None
         ]
 
-    @make_async()
-    @staticmethod
-    def _forget_task(task_id: TaskID) -> None:
-        AsyncResult(task_id).forget()
-
     async def remove_task(self, task_id: TaskID) -> None:
         await self._redis_client_sdk.redis.delete(_build_key(task_id))
-        await self._forget_task(task_id)
 
     async def set_task_progress(self, task_id: TaskID, report: ProgressReport) -> None:
         await self._redis_client_sdk.redis.hset(

services/storage/src/simcore_service_storage/modules/celery/client.py

@@ -15,6 +15,7 @@
 from .models import (
     Task,
     TaskContext,
+    TaskID,
     TaskInfoStore,
     TaskMetadata,
     TaskState,
@@ -79,6 +80,10 @@ async def abort_task(self, task_context: TaskContext, task_uuid: TaskUUID) -> No
         ):
             await self._abort_task(task_context, task_uuid)
 
+    @make_async()
+    async def _forget_task(self, task_id: TaskID) -> None:
+        AbortableAsyncResult(task_id, app=self._celery_app).forget()
+
     async def get_task_result(
         self, task_context: TaskContext, task_uuid: TaskUUID
     ) -> Any:
@@ -94,6 +99,7 @@ async def get_task_result(
                 task_metadata = await self._task_store.get_task_metadata(task_id)
                 if task_metadata is not None and task_metadata.ephemeral:
                     await self._task_store.remove_task(task_id)
+                    await self._forget_task(task_id)
             return result
 
     async def _get_task_progress_report(