diff --git a/pipelines/serpro/flows.py b/pipelines/serpro/flows.py index 2e0585b2..859573df 100644 --- 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]