diff --git a/pyinfra/server/server.py b/pyinfra/server/server.py index 301f239..1c9aa8b 100644 --- a/pyinfra/server/server.py +++ b/pyinfra/server/server.py @@ -5,6 +5,7 @@ from flask import Flask, jsonify, request from funcy import merge from prometheus_client import generate_latest, Summary, CollectorRegistry +from pyinfra.config import CONFIG from pyinfra.server.buffering.stream import FlatStreamBuffer from pyinfra.server.stream.queued_stream_function import QueuedStreamFunction from pyinfra.server.stream.rest import LazyRestProcessor @@ -58,7 +59,7 @@ def __set_up_processing_server(operation2function: Dict[str, QueuedStreamFunctio } def make_summary_instance(op: str): - op = op.replace("_pickup", "") + op = op.replace("_pickup", "") if op != "pickup" else CONFIG.service.prometheus_metric_name return Summary(f"redactmanager_{op}_seconds", f"Time spent on {op}.", registry=registry) submit_operation2processor = {submit_suffix(op): prc for op, prc in operation2processor.items()}