add state handler

prefeitura-rio · Sep 17, 2024 · 62e55c8 · 62e55c8
1 parent 4fbeca6
commit 62e55c8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pipelines/serpro/flows.py b/pipelines/serpro/flows.py
@@ -2,6 +2,7 @@
 from prefect.run_configs import KubernetesRun
 from prefect.storage import GCS
 from prefeitura_rio.pipelines_utils.custom import Flow
+from prefeitura_rio.pipelines_utils.state_handlers import handler_inject_bd_credentials
 
 from pipelines.constants import constants as smtr_constants
 from pipelines.serpro.tasks import wait_sleeping
@@ -16,4 +17,4 @@
     image=smtr_constants.DOCKER_IMAGE_FEDORA.value,
     labels=[smtr_constants.RJ_SMTR_AGENT_LABEL.value],
 )
-flow.state_handlers = [handler_setup_serpro]
+flow.state_handlers = [handler_setup_serpro, handler_inject_bd_credentials]