Skip to content

Commit 1c28009

Browse files
fix: pydantic types
1 parent 4b403b0 commit 1c28009

File tree

1 file changed

+7
-1
lines changed
  • services/storage/src/simcore_service_storage/api/_worker_tasks

1 file changed

+7
-1
lines changed

services/storage/src/simcore_service_storage/api/_worker_tasks/tasks.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,15 @@
22

33
from celery import Celery # type: ignore[import-untyped]
44
from celery_library.task import register_task
5-
from celery_library.types import register_celery_types
5+
from celery_library.types import register_celery_types, register_pydantic_types
66
from models_library.api_schemas_storage.export_data_async_jobs import AccessRightError
7+
from models_library.api_schemas_storage.storage_schemas import (
8+
FileUploadCompletionBody,
9+
FoldersBody,
10+
)
711
from servicelib.logging_utils import log_context
812

13+
from ...models import FileMetaData
914
from ._files import complete_upload_file
1015
from ._paths import compute_path_size, delete_paths
1116
from ._simcore_s3 import deep_copy_files_from_project, export_data
@@ -15,6 +20,7 @@
1520

1621
def setup_worker_tasks(app: Celery) -> None:
1722
register_celery_types()
23+
register_pydantic_types(FileUploadCompletionBody, FileMetaData, FoldersBody)
1824

1925
with log_context(_logger, logging.INFO, msg="worker task registration"):
2026
register_task(app, export_data, dont_autoretry_for=(AccessRightError,))

0 commit comments

Comments
 (0)