`ijtdZddlmZmZmZmZmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZGddZded ed ed eeeeeffd Zd ed efdZdddddddde je if dededededed ed ededeeeefde ded eeegdffdZddddddde if dededededed ed ede ded eeegdffd Zd!d"ddddde if dededededed ed ede ded eeegdffd#Zd$d%ddddde if dededededed ed ede ded eeegdffd&Zd'd(ddddde if dededededed ed ede ded eeegdffd)Zd*e d e!efd+Z"ddddd,d-e ifdeded.eded/eeeefd0eeeefde ded eeegdffd1Z#dS)2a This module contains ready-to-use functions that can be passed on to the instrumentator instance with the `add()` method. The idea behind this is to make the types of metrics you want to export with the instrumentation easily customizable. The default instrumentation function `default` can also be found here. If your requirements are really specific or very extensive it makes sense to create your own instrumentation function instead of combining several functions from this module. )CallableListOptionalSequenceTupleUnion)REGISTRYCollectorRegistryCounter HistogramSummary)Request)ResponsecBeZdZ d dedeedededededefd Zd S) Inforequestresponsemethodmodified_handlermodified_statusmodified_duration#modified_duration_without_streamingch||_||_||_||_||_||_||_dS)aCreates Info object that is used for instrumentation functions. This is the only argument that is passed to the instrumentation functions. Args: request (Request): Python Requests request object. response (Response or None): Python Requests response object. method (str): Unmodified method of the request. modified_handler (str): Handler representation after processing by instrumentator. For example grouped to `none` if not templated. modified_status (str): Status code representation after processing by instrumentator. For example grouping into `2xx`, `3xx` and so on. modified_duration (float): Latency representation after processing by instrumentator. For example rounding of decimals. Seconds. modified_duration_without_streaming (float): Latency between request arrival and response starts (i.e. first chunk duration). Excluding the streaming duration. Defaults to 0. N)rrrrrrr)selfrrrrrrrs }/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/prometheus_fastapi_instrumentator/metrics.py__init__z Info.__init__s?8    0.!23V000N)r) __name__ __module__ __qualname__rrrstrfloatrrrrrs69"W"W"W8$"W "W  "W  "W!"W.3"W"W"W"W"W"Wrrshould_include_handlershould_include_methodshould_include_statusreturncg}g}|r*|d|d|r*|d|d|r*|d|d||fS)aXBuilds up tuple with to be used label and attribute names. Args: should_include_handler (bool): Should the `handler` label be part of the metric? should_include_method (bool): Should the `method` label be part of the metric? should_include_status (bool): Should the `status` label be part of the metric? Returns: Tuple with two list elements. First element: List with all labels to be used. Second element: List with all attribute names to be used from the `Info` object. Done like this to enable dynamic on / off of labels. handlerrrstatusrappend)r%r&r' label_namesinfo_attribute_namess r_build_label_attribute_namesr0;s(K89%%%##$6777.8$$$##H---78$$$##$5666 , ,,rerrorc`tt|jdjddgS)Nrz+Duplicated timeseries in CollectorRegistry:z,Duplicated time series in CollectorRegistry:)anymapargs __contains__)r1s r_is_duplicated_time_seriesr7as8  JqM &=>     rhttp_request_duration_secondsz$Duration of HTTP requests in secondsTF metric_name metric_docmetric_namespacemetric_subsystem!should_exclude_streaming_durationbucketsregistry custom_labelsNc |dtdkrg|td}t|||\| D],} | | - rt|||||| nt|||||| dtddffd } | S#t $r} t | s| Yd} ~ nd} ~ wwxYwdS) aDefault metric for the Prometheus Starlette Instrumentator. Args: metric_name (str, optional): Name of the metric to be created. Must be unique. Defaults to "http_request_duration_seconds". metric_doc (str, optional): Documentation of the metric. Defaults to "Duration of HTTP requests in seconds". metric_namespace (str, optional): Namespace of all metrics in this metric function. Defaults to "". metric_subsystem (str, optional): Subsystem of all metrics in this metric function. Defaults to "". should_include_handler: Should the `handler` label be part of the metric? Defaults to `True`. should_include_method: Should the `method` label be part of the metric? Defaults to `True`. should_include_status: Should the `status` label be part of the metric? Defaults to `True`. should_exclude_streaming_duration: Should the streaming duration be excluded? Defaults to `False`. buckets: Buckets for the histogram. Defaults to Prometheus default. Defaults to default buckets from Prometheus client library. Returns: Function that takes a single parameter `Info`. inf) labelnamesr? namespace subsystemr@)r?rFrGr@infor(Ncj}rj}nj}r-fdD}j||dS|dS)Nc0g|]}t|Sr$getattr.0attribute_namerHs r z4latency..instrumentation..3   &D.11   r)rrlabelsobserve)rHduration label_valuesMETRICr/r.r>s` rinstrumentationz latency..instrumentations-H0 2C1 )    *>     |,44X>>>>>x(((((r)r#r0r-r r ValueErrorr7)r:r;r<r=r%r&r'r>r?r@rAkeyrWerVr/r.s ` @@@rlatencyr[qs^r{eEll""*G*U5\\*(D 57L))%K%))3##C(((()  &**!FF**! F )$ )4 ) ) ) ) ) ) ) ) )" )!,, G      4s5AB77 CCChttp_request_size_byteszContent bytes of requests.c b t|||\ |D],} |  | - rt||||| nt||||| dtddf fd } | S#t$r} t | s| Yd} ~ nd} ~ wwxYwdS)aRecord the content length of incoming requests. If content length is missing 0 will be assumed. Args: metric_name (str, optional): Name of the metric to be created. Must be unique. Defaults to "http_request_size_bytes". metric_doc (str, optional): Documentation of the metric. Defaults to "Content bytes of requests.". metric_namespace (str, optional): Namespace of all metrics in this metric function. Defaults to "". metric_subsystem (str, optional): Subsystem of all metrics in this metric function. Defaults to "". should_include_handler: Should the `handler` label be part of the metric? Defaults to `True`. should_include_method: Should the `method` label be part of the metric? Defaults to `True`. should_include_status: Should the `status` label be part of the metric? Defaults to `True`. Returns: Function that takes a single parameter `Info`. rErFrGr@rFrGr@rHr(Ncjjdd}r:fdD}j|t |dSt |dS)NContent-Lengthrc0g|]}t|Sr$rKrMs rrPz9request_size..instrumentation..&rQr)rheadersgetrRrSintrHcontent_lengthrUrVr/r.s` rrWz%request_size..instrumentation#s!\1556FJJN 4    *>     |,44S5H5HIIIIIs>2233333rr0r-r rrXr7r:r;r<r=r%r&r'r@rArYrWrZrVr/r.s @@@r request_sizerjsOF)E 57L))%K%))3##C(((("  &**! FF**! F 4$ 44 4 4 4 4 4 4 4 4 )!,, G      4>B B,B''B,http_response_size_byteszContent bytes of responses.c b t|||\ |D],} |  | - rt||||| nt||||| dtddf fd } | S#t$r} t | s| Yd} ~ nd} ~ wwxYwdS)aRecord the content length of outgoing responses. If content length is missing 0 will be assumed. Args: metric_name (str, optional): Name of the metric to be created. Must be unique. Defaults to "http_response_size_bytes". metric_doc (str, optional): Documentation of the metric. Defaults to "Content bytes of responses.". metric_namespace (str, optional): Namespace of all metrics in this metric function. Defaults to "". metric_subsystem (str, optional): Subsystem of all metrics in this metric function. Defaults to "". should_include_handler: Should the `handler` label be part of the metric? Defaults to `True`. should_include_method: Should the `method` label be part of the metric? Defaults to `True`. should_include_status: Should the `status` label be part of the metric? Defaults to `True`. Returns: Function that takes a single parameter `Info`. r^r_rHr(NcDjr6tjdr!jjdd}nd}r:fdD}j|t |dSt |dS)Nrcrarc0g|]}t|Sr$rKrMs rrPz:response_size..instrumentation..rQr)rhasattrrcrdrRrSrerfs` rrWz&response_size..instrumentations} # !B!B #!%!6!:!:;KQ!O!O!" 4    *>     |,44S5H5HIIIIIs>2233333rrhris @@@r response_sizerq7sOR)E 57L))%K%))3##C((((&  &**! FF**! F 4$ 44 4 4 4 4 4 4 4 4  )!,, G      4rkhttp_combined_size_bytesz(Content bytes of requests and responses.c b t|||\ |D],} |  | - rt||||| nt||||| dtddf fd } | S#t$r} t | s| Yd} ~ nd} ~ wwxYwdS)a!Record the combined content length of requests and responses. If content length is missing 0 will be assumed. Args: metric_name (str, optional): Name of the metric to be created. Must be unique. Defaults to "http_combined_size_bytes". metric_doc (str, optional): Documentation of the metric. Defaults to "Content bytes of requests and responses.". metric_namespace (str, optional): Namespace of all metrics in this metric function. Defaults to "". metric_subsystem (str, optional): Subsystem of all metrics in this metric function. Defaults to "". should_include_handler: Should the `handler` label be part of the metric? Defaults to `True`. should_include_method: Should the `method` label be part of the metric? Defaults to `True`. should_include_status: Should the `status` label be part of the metric? Defaults to `True`. Returns: Function that takes a single parameter `Info`. r^r_rHr(Ncjjdd}jr6t jdr!jjdd}nd}t |t |z}r:fdD}j|t |dSt |dS)Nrarrcc0g|]}t|Sr$rKrMs rrPz:combined_size..instrumentation..rQr)rrcrdrrprerRrS)rH request_cl response_clrgrUrVr/r.s` rrWz&combined_size..instrumentations-112BAFFJ}  !B!B "m3778H!LL   __s;/?/??N 4    *>     |,44S5H5HIIIIIs>2233333rrhris @@@r combined_sizerxsOR)E 57L))%K%))3##C((((*  &**! FF**! F 4$ 44 4 4 4 4 4 4 4 4( )!,, G      4rkhttp_requests_total7Total number of requests by method, status and handler.c b t|||\ |D],} |  | - rt||||| nt||||| dtddf fd } | S#t$r} t | s| Yd} ~ nd} ~ wwxYwdS)a Record the number of requests. Args: metric_name (str, optional): Name of the metric to be created. Must be unique. Defaults to "http_requests_total". metric_doc (str, optional): Documentation of the metric. Defaults to "Total number of requests by method, status and handler.". metric_namespace (str, optional): Namespace of all metrics in this metric function. Defaults to "". metric_subsystem (str, optional): Subsystem of all metrics in this metric function. Defaults to "". should_include_handler (bool, optional): Should the `handler` label be part of the metric? Defaults to `True`. should_include_method (bool, optional): Should the `method` label be part of the metric? Defaults to `True`. should_include_status (bool, optional): Should the `status` label be part of the metric? Defaults to `True`. Returns: Function that takes a single parameter `Info`. r^r_rHr(Ncr,fdD}j|dSdS)Nc0g|]}t|Sr$rKrMs rrPz5requests..instrumentation..HrQr)rRinc)rHrUrVr/r.s` rrWz!requests..instrumentationFsh     *>     |,0022222 r)r0r-r rrXr7ris @@@rrequestsrsON)E 57L))%K%))3##C((((!  &**! FF**! F $ 4         )!,, G      4rk label_namecg}dddd}|D]7}||vr|||"||8|S)Nrrr)r*r+rTr,)rattribute_namesmappingitems r_map_label_name_valuerYsnO%#'G )) 7??  " "74= 1 1 1 1  " "4 ( ( ( ( r)g{Gz?g?g?g333333?皙?g??g?g?g@g @g@g@ <)rrr!should_only_respect_2xx_for_highrlatency_highr_bucketslatency_lowr_bucketsc  |dtdkrg|td}|dtdkrg|td}tdD} dtdd|z|||dtd d |z||| dtd d |z|||t d d|||| dt dd||z||| dt ddf f d } | S#t $r} t| s| Yd} ~ nd} ~ wwxYwdS)aContains multiple metrics to cover multiple things. Combines several metrics into a single function. Also more efficient than multiple separate instrumentation functions that do more or less the same. You get the following: * `http_requests_total` (`handler`, `status`, `method`): Total number of requests by handler, status and method. * `http_request_size_bytes` (`handler`): Total number of incoming content length bytes by handler. * `http_response_size_bytes` (`handler`): Total number of outgoing content length bytes by handler. * `http_request_duration_highr_seconds` (no labels): High number of buckets leading to more accurate calculation of percentiles. * `http_request_duration_seconds` (`handler`, `method`): Kepp the bucket count very low. Only put in SLIs. Args: metric_namespace (str, optional): Namespace of all metrics in this metric function. Defaults to "". metric_subsystem (str, optional): Subsystem of all metrics in this metric function. Defaults to "". should_only_respect_2xx_for_highr (str, optional): Should the metric `http_request_duration_highr_seconds` only include latencies of requests / responses that have a status code starting with `2`? Defaults to `False`. should_exclude_streaming_duration: Should the streaming duration be excluded? Defaults to `False`. latency_highr_buckets (tuple[float], optional): Buckets tuple for high res histogram. Can be large because no labels are used. Defaults to (0.01, 0.025, 0.05, 0.075, 0.1, 0.25, 0.5, 0.75, 1, 1.5, 2, 2.5, 3, 3.5, 4, 4.5, 5, 7.5, 10, 30, 60). latency_lowr_buckets (tuple[float], optional): Buckets tuple for low res histogram. Should be very small as all possible labels are included. Defaults to `(0.1, 0.5, 1)`. Returns: Function that takes a single parameter `Info`. rCrDcg|]}|Sr$r$)rNrYs rrPzdefault..s#A#A#ACC#A#A#Ar)rr+r*ryrz)name documentationrErFrGr@)r*r\zContent length of incoming requests by handler. Only value of header is respected. Otherwise ignored. No percentile calculated. rlzContent length of outgoing responses by handler. Only value of header is respected. Otherwise ignored. No percentile calculated. #http_request_duration_highr_secondszfLatency with many buckets but no API specific labels. Made for more accurate percentile calculations. )rrr?rFrGr@)rr*r8zgLatency with only few buckets by handler. Made to be only used if aggregation by handler is important. )rrr?rErFrGr@rHr(Nc j} rj}nj}fdtDtz}j|fdt Dtz}j|tj j ddfdt Dtz}j r^tj drIj|tj j ddnj|d rjdr|fdt Dtz}j||dS) Nc0g|]}t|Sr$rKrMs rrPz4default..instrumentation..3"n--rc0g|]}t|Sr$rKrMs rrPz4default..instrumentation..rrrarc0g|]}t|Sr$rKrMs rrPz4default..instrumentation.. rrrc2c0g|]}t|Sr$rKrMs rrPz4default..instrumentation..0rr)rrrlistvaluesrRr~rSrerrcrdrrpr startswith)rHrTrUIN_SIZE LATENCY_HIGHR LATENCY_LOWROUT_SIZETOTALrA in_size_nameslatency_lower_namesout_size_namesr>rtotal_label_namess` rrWz default..instrumentation s-H0 2C1&;rrr@rAadditional_label_namesrWrZrrrrrrrrrs `` ` @@@@@@@@@rdefaultrhsrZR E%LL00 F"7 Fu FB5<<//D!5DuU||D##A#A=#A#A#ABBw  &S(+AA&&    % *-%'==&&    &+-&(>>&&    "6C*&&     !0P)*-CC&&    ) A$) A4) A) A) A) A) A) A) A) A) A) A) A) A) A) A) A) A) AV )!,, G      4s?BD D>#D99D>)$__doc__typingrrrrrrprometheus_clientr r r r r starlette.requestsrstarlette.responsesrrboolr"r0rXr7DEFAULT_BUCKETSr#dictr[rjrqrxrrrrrr$rrrs  DCCCCCCCCCCCCCCCVVVVVVVVVVVVVV&&&&&&((((((#W#W#W#W#W#W#W#WL#- #-#- #- 49d3i  #-#-#-#-L j T    "7<#'"&"&.3+4+D"*kkkkk k ! k  k k(,keE3J' (k kkhvt|$%kkkk^12#'"&"&"*UUUUU U ! U  U U UUhvt|$%UUUUr23#'"&"&"*_____ _ ! _  _ _ __hvt|$%____F2@#'"&"&"*bbbbb b ! b  b b bbhvt|$%bbbbL-O#'"&"&"*XXXXX X ! X  X X XXhvt|$%XXXXv e S     .3.3:.9F"*=TTTT(,T(, T $E%*$56 T8#5#459T: ;T<=T>hvt|$%?TTTTTTr