Skip to content

Commit b02e11a

Browse files
fix: pydantic types registration
1 parent d7d7a3e commit b02e11a

File tree

1 file changed

+8
-1
lines changed
  • services/storage/src/simcore_service_storage/modules/celery

1 file changed

+8
-1
lines changed

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

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,15 @@
11
from celery_library.common import create_app, create_task_manager
22
from celery_library.task_manager import CeleryTaskManager
3-
from celery_library.types import register_celery_types
3+
from celery_library.types import register_celery_types, register_pydantic_types
44
from fastapi import FastAPI
5+
from models_library.api_schemas_storage.storage_schemas import (
6+
FileUploadCompletionBody,
7+
FoldersBody,
8+
)
59
from settings_library.celery import CelerySettings
610

11+
from ...models import FileMetaData
12+
713

814
def setup_task_manager(app: FastAPI, celery_settings: CelerySettings) -> None:
915
async def on_startup() -> None:
@@ -12,6 +18,7 @@ async def on_startup() -> None:
1218
)
1319

1420
register_celery_types()
21+
register_pydantic_types(FileUploadCompletionBody, FileMetaData, FoldersBody)
1522

1623
app.add_event_handler("startup", on_startup)
1724

0 commit comments

Comments
 (0)