@@ -639,41 +639,45 @@ def setup_long_running_tasks(app: FastAPI) -> None:
639639        lrt_namespace = f"{ APP_NAME }  -{ app_settings .DY_SIDECAR_RUN_ID }  " ,
640640    )
641641
642-     shared_store : SharedStore  =  app .state .shared_store 
643-     mounted_volumes : MountedVolumes  =  app .state .mounted_volumes 
644-     outputs_manager : OutputsManager  =  app .state .outputs_manager 
645- 
646-     context_app_store : dict [str , Any ] =  {
647-         "app" : app ,
648-         "shared_store" : shared_store ,
649-     }
650-     context_app_store_volumes : dict [str , Any ] =  {
651-         "app" : app ,
652-         "shared_store" : shared_store ,
653-         "mounted_volumes" : mounted_volumes ,
654-     }
655-     context_app_volumes : dict [str , Any ] =  {
656-         "app" : app ,
657-         "mounted_volumes" : mounted_volumes ,
658-     }
659-     context_app_outputs : dict [str , Any ] =  {
660-         "app" : app ,
661-         "outputs_manager" : outputs_manager ,
662-     }
663- 
664-     task_context : dict [TaskProtocol , dict [str , Any ]] =  {
665-         task_pull_user_servcices_docker_images : context_app_store ,
666-         task_create_service_containers : context_app_store ,
667-         task_runs_docker_compose_down : context_app_store_volumes ,
668-         task_restore_state : context_app_volumes ,
669-         task_save_state : context_app_volumes ,
670-         task_ports_inputs_pull : context_app_volumes ,
671-         task_ports_outputs_pull : context_app_volumes ,
672-         task_ports_outputs_push : context_app_outputs ,
673-         task_containers_restart : context_app_store ,
674-     }
642+     task_context : dict [TaskProtocol , dict [str , Any ]] =  {}
675643
676644    async  def  on_startup () ->  None :
645+         shared_store : SharedStore  =  app .state .shared_store 
646+         mounted_volumes : MountedVolumes  =  app .state .mounted_volumes 
647+         outputs_manager : OutputsManager  =  app .state .outputs_manager 
648+ 
649+         context_app_store : dict [str , Any ] =  {
650+             "app" : app ,
651+             "shared_store" : shared_store ,
652+         }
653+         context_app_store_volumes : dict [str , Any ] =  {
654+             "app" : app ,
655+             "shared_store" : shared_store ,
656+             "mounted_volumes" : mounted_volumes ,
657+         }
658+         context_app_volumes : dict [str , Any ] =  {
659+             "app" : app ,
660+             "mounted_volumes" : mounted_volumes ,
661+         }
662+         context_app_outputs : dict [str , Any ] =  {
663+             "app" : app ,
664+             "outputs_manager" : outputs_manager ,
665+         }
666+ 
667+         task_context .update (
668+             {
669+                 task_pull_user_servcices_docker_images : context_app_store ,
670+                 task_create_service_containers : context_app_store ,
671+                 task_runs_docker_compose_down : context_app_store_volumes ,
672+                 task_restore_state : context_app_volumes ,
673+                 task_save_state : context_app_volumes ,
674+                 task_ports_inputs_pull : context_app_volumes ,
675+                 task_ports_outputs_pull : context_app_volumes ,
676+                 task_ports_outputs_push : context_app_outputs ,
677+                 task_containers_restart : context_app_store ,
678+             }
679+         )
680+ 
677681        for  handler , context  in  task_context .items ():
678682            TaskRegistry .register (handler , ** context )
679683
0 commit comments