tPjoUdZddlZddlmZddlmZmZddlmZddl m Z m Z m Z m Z ddlZddlmZddlmZej&j(Zeed <Gd d Zy) z7 Prometheus Auth Middleware - Pure ASGI implementation N)MutableMapping)AnyFinal)Request)ASGIAppReceiveScopeSend)SpecialHeaders)user_api_key_auth_AUTHORIZATION_HEADERc8eZdZdZdeddfdZdedededdfd Z y) PrometheusAuthMiddlewarea Middleware to authenticate requests to the metrics endpoint. By default, auth is run on the metrics endpoint. To allow unauthenticated metrics in proxy_config.yaml: ```yaml litellm_settings: require_auth_for_metrics_endpoint: false ``` appreturnNc||_yN)r)selfrs x/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/middleware/prometheus_auth_middleware.py__init__z!PrometheusAuthMiddleware.__init__"s scopereceivesendc l K|ddk7sd|jddvr|j||d{ytjdurg dtt t ff fd }t||} t||jjtxsd|jjtjjxsd|jjtjj|jjtjj|jjtj jxsd|jjtj"j d{d dtt t ff  fd } |j|| |d{y|j||d{y77\#t$$r}t'|d t |}t)j*d |d j-d }|ddddgdt t/|j-dggdd{7|d|dd{7Yd}~yd}~wwxYw77w)Ntypehttpz/metricspathFrcTKd{}j||S7wr)append)messagebuffered_messagesrs rreceive_for_authz;PrometheusAuthMiddleware.__call__..receive_for_auth2s('.y!((1"1s (&()requestapi_keyazure_api_key_headeranthropic_api_key_headergoogle_ai_studio_api_key_headerazure_apim_headercustom_litellm_key_headerr"z)Unauthorized access to metrics endpoint: z| To allow unauthenticated access, set `litellm_settings.require_auth_for_metrics_endpoint: false` in your proxy_config.yaml.zutf-8zhttp.response.startis content-typesapplication/jsonscontent-lengthascii)rstatusheaderszhttp.response.body)rbodyrcbKtkr }dz |Sd{S7w)N)len)msgr#r replay_idxs rreceive_replayz9PrometheusAuthMiddleware.__call__..receive_replaybs9$5 66!2:!>C!OJJ$Y&s %/-/)getrlitellm!require_auth_for_metrics_endpointrstrrrr r.r r azure_authorizationvalueanthropic_authorizationgoogle_ai_studio_authorizationazure_apim_authorizationcustom_litellm_api_key Exceptiongetattrjsondumpsencoder2) rrrrr$r%e error_messager/r5r#r4s ` @@r__call__z!PrometheusAuthMiddleware.__call__%s{ =F "j &"8M&M((5'40 0 0   4 4E AHJ  N38,D  %U,<=G% '##OO//0EFL")0)<)<^=_=_=e=e)f)ljl-4__-@-@AgAgAmAm-n4;OO4G4G&EEKK5'.oo&9&9.:a:a:g:g&h&nln.5oo.A.A.BgBgBmBm.n   LJ '.c*B '((5.$7 7 7 hhugt,,,M 1$  '.q)SV'D "jj? O12 &/   5"%,.AB.CI0E0Eg0NO$    4 $ 3 J 8 -s3J4HAJ4:D3H -H .H 23J4%J0&J4J2J4 H J-A4J( J  J(J J(#J4(J--J42J4) __name__ __module__ __qualname____doc__rrr rr rGrrrrs@ GI-EI-GI-4I-DI-rr)rKrBcollections.abcrtypingrrfastapirstarlette.typesrrr r r7litellm.proxy._typesr $litellm.proxy.auth.user_api_key_authr openai_authorizationr;r __annotations__rrLrrrUsL *99/B .BBHHuHZ-Z-r