propagate pre_registered_function_job_id properly

Author: bisgaard-itis

services/api-server/src/simcore_service_api_server/_service_function_jobs.py

@@ -368,6 +368,7 @@ async def run_function(
         *,
         job_creation_task_id: TaskID | None,
         function: RegisteredFunction,
+        pre_registered_function_job_id: FunctionJobID,
         job_inputs: JobInputs,
         pricing_spec: JobPricingSpecification | None,
         job_links: JobLinks,
@@ -393,7 +394,7 @@ async def run_function(
             return await self.patch_registered_function_job(
                 user_id=self.user_id,
                 product_name=self.product_name,
-                function_job_id=study_job.id,
+                function_job_id=pre_registered_function_job_id,
                 function_class=FunctionClass.PROJECT,
                 job_creation_task_id=job_creation_task_id,
                 project_job_id=study_job.id,
@@ -418,7 +419,7 @@ async def run_function(
             return await self.patch_registered_function_job(
                 user_id=self.user_id,
                 product_name=self.product_name,
-                function_job_id=solver_job.id,
+                function_job_id=pre_registered_function_job_id,
                 function_class=FunctionClass.SOLVER,
                 job_creation_task_id=job_creation_task_id,
                 solver_job_id=solver_job.id,

services/api-server/src/simcore_service_api_server/api/routes/functions_routes.py

@@ -387,6 +387,7 @@ async def run_function(  # noqa: PLR0913
         task_filter=task_filter,
         user_identity=user_identity,
         function=to_run_function,
+        pre_registered_function_job_id=pre_registered_function_job_id,
         job_inputs=job_inputs,
         pricing_spec=pricing_spec,
         job_links=job_links,

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

@@ -3,7 +3,7 @@
 )
 from celery_library.utils import get_app_server  # pylint: disable=no-name-in-module
 from fastapi import FastAPI
-from models_library.functions import RegisteredFunction
+from models_library.functions import FunctionJobID, RegisteredFunction
 from models_library.projects_nodes_io import NodeID
 from servicelib.celery.models import TaskID
 from simcore_service_api_server._service_function_jobs import FunctionJobService
@@ -94,6 +94,7 @@ async def run_function(
     *,
     user_identity: Identity,
     function: RegisteredFunction,
+    pre_registered_function_job_id: FunctionJobID,
     job_inputs: JobInputs,
     pricing_spec: JobPricingSpecification | None,
     job_links: JobLinks,
@@ -109,6 +110,7 @@ async def run_function(
     return await function_job_service.run_function(
         job_creation_task_id=task_id,
         function=function,
+        pre_registered_function_job_id=pre_registered_function_job_id,
         job_inputs=job_inputs,
         pricing_spec=pricing_spec,
         job_links=job_links,