register job pydantic types for serialization - can run function via celery now

Author: bisgaard-itis

services/api-server/src/simcore_service_api_server/api/dependencies/celery.py

@@ -4,17 +4,10 @@
 from celery_library.task_manager import CeleryTaskManager
 from celery_library.types import register_celery_types, register_pydantic_types
 from fastapi import FastAPI
-from models_library.functions import (
-    RegisteredProjectFunction,
-    RegisteredPythonCodeFunction,
-    RegisteredSolverFunction,
-)
 from servicelib.celery.task_manager import TaskManager
 from settings_library.celery import CelerySettings
 
-from ...api.dependencies.authentication import Identity
-from ...models.api_resources import JobLinks
-from ...models.schemas.jobs import JobPricingSpecification
+from ...celery.worker_tasks.tasks import registered_pydantic_types
 
 ASYNC_JOB_CLIENT_NAME: Final[str] = "API_SERVER"
 
@@ -26,14 +19,7 @@ async def on_startup() -> None:
         )
 
         register_celery_types()
-        register_pydantic_types(
-            Identity,
-            RegisteredProjectFunction,
-            RegisteredPythonCodeFunction,
-            RegisteredSolverFunction,
-            JobPricingSpecification,
-            JobLinks,
-        )
+        register_pydantic_types(*registered_pydantic_types)
 
     app.add_event_handler("startup", on_startup)
 

services/api-server/src/simcore_service_api_server/celery/worker_tasks/tasks.py

@@ -5,8 +5,10 @@
 from celery_library.types import register_celery_types, register_pydantic_types
 from models_library.functions import (
     RegisteredProjectFunction,
+    RegisteredProjectFunctionJob,
     RegisteredPythonCodeFunction,
     RegisteredSolverFunction,
+    RegisteredSolverFunctionJob,
 )
 from servicelib.logging_utils import log_context
 
@@ -17,17 +19,22 @@
 
 _logger = logging.getLogger(__name__)
 
+registered_pydantic_types = (
+    Identity,
+    JobLinks,
+    JobPricingSpecification,
+    RegisteredProjectFunction,
+    RegisteredProjectFunctionJob,
+    RegisteredPythonCodeFunction,
+    RegisteredProjectFunctionJob,
+    RegisteredSolverFunction,
+    RegisteredSolverFunctionJob,
+)
+
 
 def setup_worker_tasks(app: Celery) -> None:
     register_celery_types()
-    register_pydantic_types(
-        Identity,
-        RegisteredProjectFunction,
-        RegisteredPythonCodeFunction,
-        RegisteredSolverFunction,
-        JobPricingSpecification,
-        JobLinks,
-    )
+    register_pydantic_types(*registered_pydantic_types)
 
     with log_context(_logger, logging.INFO, msg="worker task registration"):
         register_task(app, run_function)