Skip to content

Commit 2f57d5e

Browse files
committed
register task
1 parent ed6ed1e commit 2f57d5e

File tree

1 file changed

+20
-2
lines changed
  • services/api-server/src/simcore_service_api_server/celery/_worker_tasks

1 file changed

+20
-2
lines changed
Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,33 @@
11
import logging
22

33
from celery import Celery # type: ignore[import-untyped]
4+
from celery_library.task import register_task
45
from celery_library.types import register_celery_types, register_pydantic_types
6+
from models_library.functions import (
7+
RegisteredProjectFunction,
8+
RegisteredPythonCodeFunction,
9+
RegisteredSolverFunction,
10+
)
511
from servicelib.logging_utils import log_context
612

13+
from ...api.dependencies.authentication import Identity
14+
from ...models.api_resources import JobLinks
15+
from ...models.schemas.jobs import JobPricingSpecification
16+
from ._functions_tasks import run_function
17+
718
_logger = logging.getLogger(__name__)
819

920

1021
def setup_worker_tasks(app: Celery) -> None:
1122
register_celery_types()
12-
register_pydantic_types()
23+
register_pydantic_types(
24+
Identity,
25+
RegisteredProjectFunction,
26+
RegisteredPythonCodeFunction,
27+
RegisteredSolverFunction,
28+
JobPricingSpecification,
29+
JobLinks,
30+
)
1331

1432
with log_context(_logger, logging.INFO, msg="worker task registration"):
15-
pass
33+
register_task(app, run_function)

0 commit comments

Comments
 (0)