added pydantic models to be compatible with celery tasks

Author: sanderegg

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

@@ -1,6 +1,11 @@
+from functools import partial
 from pathlib import Path
+from typing import Any
 
-from kombu.utils.json import register_type  # type: ignore[import-untyped]
+from kombu.utils.json import register_type
+from models_library.api_schemas_storage.storage_schemas import FileUploadCompletionBody
+from pydantic import BaseModel
+from simcore_service_storage.models import FileMetaData
 
 
 def _path_encoder(obj):
@@ -20,10 +25,30 @@ def _class_full_name(clz: type) -> str:
     return ".".join([clz.__module__, clz.__qualname__])
 
 
+def _encoder(obj: BaseModel, *args, **kwargs) -> dict[str, Any]:
+    return obj.model_dump(*args, **kwargs)
+
+
+def _decoder(clz: type[BaseModel], data: dict[str, Any]) -> BaseModel:
+    return clz(**data)
+
+
+def _register_pydantic_types(*models: type[BaseModel]) -> None:
+    for model in models:
+        register_type(
+            model,
+            _class_full_name(model),
+            encoder=_encoder,
+            decoder=partial(_decoder, model),
+        )
+
+
 def register_celery_types() -> None:
     register_type(
         Path,
         _class_full_name(Path),
         _path_encoder,
         _path_decoder,
     )
+    _register_pydantic_types(FileUploadCompletionBody)
+    _register_pydantic_types(FileMetaData)