`i5ddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZmZmZddlmZmZmZmZmZddlmZddlmZddlmZddlmZdd l m!Z!Gd d Z"dS) N)Enum)Any AwaitableCallableListOptionalSequenceUnioncast)CONTENT_TYPE_LATESTREGISTRYCollectorRegistrygenerate_latest multiprocess) Starlette)Request)Response)metrics)"PrometheusInstrumentatorMiddlewareceZdZdddddddggdddddfdeded ed ed ed ed edeedeedededededeedfddfdZ d+de dededede ee efde ee efddfd Z d,de d"ed#ed$ed%eeeeefd&eddfd'Zd(eeejgededffddfd)Zdefd*ZdS)-PrometheusFastApiInstrumentatorTFENABLE_METRICShttp_requests_inprogressNshould_group_status_codesshould_ignore_untemplatedshould_group_untemplatedshould_round_latency_decimalsshould_respect_env_var%should_instrument_requests_inprogress!should_exclude_streaming_durationexcluded_handlers body_handlersround_latency_decimals env_var_nameinprogress_nameinprogress_labelsregistryreturnc~||_||_||_||_||_||_||_| |_| |_| |_ | |_ d|D|_ d| D|_ g|_ g|_dtjvrGdtjvr9tjdtjd<t#jdt&|r||_n t*|_dtjvrDtjd}tj|st1d|ddSdS) a Create a Prometheus FastAPI (and Starlette) Instrumentator. Args: should_group_status_codes (bool): Should status codes be grouped into `2xx`, `3xx` and so on? Defaults to `True`. should_ignore_untemplated (bool): Should requests without a matching template be ignored? Defaults to `False`. This means that by default a request like `curl -X GET localhost:80/doesnotexist` will be ignored. should_group_untemplated (bool): Should requests without a matching template be grouped to handler `none`? Defaults to `True`. should_round_latency_decimals: Should recorded latencies be rounded to a certain number of decimals? should_respect_env_var (bool): Should the instrumentator only work - for example the methods `instrument()` and `expose()` - if a certain environment variable is set to `true`? Usecase: A base FastAPI app that is used by multiple distinct apps. The apps only have to set the variable to be instrumented. Defaults to `False`. should_instrument_requests_inprogress (bool): Enables a gauge that shows the inprogress requests. See also the related args starting with `inprogress`. Defaults to `False`. should_exclude_streaming_duration: Should the streaming duration be excluded? Only relevant if default metrics are used. Defaults to `False`. excluded_handlers (List[str]): List of strings that will be compiled to regex patterns. All matches will be skipped and not instrumented. Defaults to `[]`. body_handlers (List[str]): List of strings that will be compiled to regex patterns to match handlers for the middleware to pass through response bodies to instrumentations. So only relevant for instrumentations that access `info.response.body`. Note that this has a noticeable negative impact on performance with responses larger than a few MBs. Defaults to `[]`. round_latency_decimals (int): Number of decimals latencies should be rounded to. Ignored unless `should_round_latency_decimals` is `True`. Defaults to `4`. env_var_name (str): Any valid os environment variable name that will be checked for existence before instrumentation. Ignored unless `should_respect_env_var` is `True`. Defaults to `"ENABLE_METRICS"`. inprogress_name (str): Name of the gauge. Defaults to `http_requests_inprogress`. Ignored unless `should_instrument_requests_inprogress` is `True`. inprogress_labels (bool): Should labels `method` and `handler` be part of the inprogress label? Ignored unless `should_instrument_requests_inprogress` is `True`. Defaults to `False`. registry (CollectorRegistry): A custom Prometheus registry to use. If not provided, the default `REGISTRY` will be used. This can be useful if you need to run multiple apps at the same time, with their own registries, for example during testing. Raises: ValueError: If `PROMETHEUS_MULTIPROC_DIR` env var is found but doesn't point to a valid directory. c6g|]}tj|Srecompile.0paths /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/prometheus_fastapi_instrumentator/instrumentation.py z.s"!Q!Q!Qt"*T"2"2!Q!Q!Qc6g|]}tj|Sr,r-r0s r3r4z.s"III4bj..IIIr5prometheus_multiproc_dirPROMETHEUS_MULTIPROC_DIRzpprometheus_multiproc_dir variable has been deprecated in favor of the upper case naming PROMETHEUS_MULTIPROC_DIRz"Env var PROMETHEUS_MULTIPROC_DIR='z' not a directory.N)rrrrrr r!r$r%r&r'r"r#instrumentationsasync_instrumentationsosenvironwarningswarnDeprecationWarningr(r r2isdir ValueError)selfrrrrrr r!r"r#r$r%r&r'r(pmds r3__init__z(PrometheusFastApiInstrumentator.__init__sil*C&)B&(@%-J*&<#5Z21R.&<#(.!2!Q!Q?P!Q!Q!QII=IIIFHWY# '"* 4 4*"*<<57Z*6BJ1 2 MC"     %$DMM$DM % 3 3*78C7==%%  PPPP 4 3  r5g{Gz?g?g?g333333?皙?g??g?g?g@g @rg@g@ <rGrHrIappmetric_namespacemetric_subsystem!should_only_respect_2xx_for_highrlatency_highr_bucketslatency_lowr_bucketsc@|jr|s|S|t|j|j|j|j|j|j|j |j |j |j |j |j|j|j|j||||||j|S)aPerforms the instrumentation by adding middleware. The middleware iterates through all `instrumentations` and executes them. Args: app: Starlette app instance. Note that every FastAPI app is a Starlette app. Raises: e: Only raised if app itself throws an exception. Returns: self: Instrumentator. Builder Pattern. )rrrrrr r!r$r%r&r'r9r:r"r#rRrSrTrUrVr()r_should_instrumentateadd_middlewarerrrrrr r!r$r%r&r'r9r:r"r#r()rBrQrRrSrTrUrVs r3 instrumentz*PrometheusFastApiInstrumentator.instruments\  & t/I/I/K/K K  .&*&D&*&D%)%B*.*L#'#>262\.2.T#'#>* 0"4!2#'#>"4,--.O"7!5]-    0 r5/metrics should_gzipendpointinclude_in_schematagskwargsc LjrsSdtdtffd }d}tjdr4ddlm} t|| r|} | j |f||d||d }|s| ||| S) aExposes endpoint for metrics. Args: app: App instance. Endpoint will be added to this app. This can be a Starlette app or a FastAPI app. If it is a Starlette app, `tags` `kwargs` will be ignored. should_gzip: Should the endpoint return compressed data? It will also check for `gzip` in the `Accept-Encoding` header. Compression consumes more CPU cycles. In most cases it's best to just leave this option off since network bandwidth is usually cheaper than CPU cycles. Defaults to `False`. endpoint: Endpoint on which metrics should be exposed. include_in_schema: Should the endpoint show up in the documentation? tags (List[str], optional): If you manage your routes with tags. Defaults to None. Only passed to FastAPI app. kwargs: Will be passed to app. Only passed to FastAPI app. Returns: self: Instrumentator. Builder Pattern. requestr)cj}dtjvr"t}t j|rfd|jddvrIttj t|}t|jd<d|jd<n,tt|}t|jd<|S)z(Endpoint that serves Prometheus metrics.r8gzipzAccept-EncodingrE)contentz Content-TypezContent-Encoding) r(r;r<rrMultiProcessCollectorheadersgetrrdcompressrr )rbephemeral_registryresprBr\s r3rz7PrometheusFastApiInstrumentator.expose..metricss"& )RZ77%6%8%8"23EFFF Cv)<)<=NPR)S)SSS M/:L*M*MNN0C ^,39 /008J(K(KLLL/B ^,Kr5Ffastapir)FastAPI)r^r_T)r2router^) rrXrr importlibutil find_specrlrm isinstancerh add_route) rBrQr\r]r^r_r`rroute_configuredrm fastapi_apps ` ` r3exposez&PrometheusFastApiInstrumentator.exposesF  & t/I/I/K/K K W        (! > # #I . . ( ' ' ' ' ' '#w'' ('*  0APV$(   MMW@Q     r5instrumentation_functionc T|D]}|rtj|rL|jt t t jgtdf|d|j t t t jgdf||S)avAdds function to list of instrumentations. Args: instrumentation_function: Function that will be executed during every request handler call (if not excluded). See above for detailed information on the interface of the function. Returns: self: Instrumentator. Builder Pattern. N) asyncioiscoroutinefunctionr:appendr rrInforr9)rBrwfuncs r3addz#PrometheusFastApiInstrumentator.add3s$-  D .t44 /66$gl^Yt_%DE )00Xw|nd&:;TBB r5c^tj|jddvS)z?Checks if instrumentation should be performed based on env var.False)true1)r;getenvr%lower)rBs r3rXz5PrometheusFastApiInstrumentator._should_instrumentateUs*y*G44::<< MMr5)rErEFrFrP)Fr[TN)__name__ __module__ __qualname__boolrstrintr rrDrr floatrZrrrrvrrr|rr~rXr,r5r3rrs+/*/)-.3',6;27')#%&',9"'37#'$(#'  (,  !% 04,0 9Cy!$ )4/0 !H!# "27> .=J;II II I ,0 I (eSj(9: I:'uUCZ'89;I< +=IIII\"""&15 II II I  I tE#t),-. II +IIIIV #+ gl^U44+@%AA B$   +    DNtNNNNNNr5r)#ryrdimportlib.utilror;r.r=enumrtypingrrrrrr r r prometheus_clientr r rrrstarlette.applicationsrstarlette.requestsrstarlette.responsesr!prometheus_fastapi_instrumentatorr,prometheus_fastapi_instrumentator.middlewarerrr,r5r3rs  RRRRRRRRRRRRRRRRRRRR-,,,,,&&&&&&((((((555555 }N}N}N}N}N}N}N}N}N}Nr5