tPj$ UdZddlmZddlmZmZmZddlmZmZddl m Z m Z m Z m Z ddlmZmZddlmZddlmZmZmZdd lmZdd lmZmZmZe Zeed <d Zeed <dZ eed<GddeZ!ee"e!dfZ#eed<de"e$e$ffdZ%dee!de"edffdZ&dee!de"edffdZ'ejQddgee dde ddfd eee efd!e$dzd"e$dzdefd#Z)y)$a GATEWAY REQUEST COUNTS (SGR) GET /gateway/daily/activity - successful/failed gateway requests by date and route Source of truth is LiteLLM_DailyGatewayRequests, written at the ASGI edge by BillableRequestMetricsMiddleware. This counts what the proxy answered, so it is independent of whether a request reached litellm's logging callbacks. The table carries no key/user/team dimension, so these totals are deployment-wide and the endpoint is restricted to proxy admin roles. )Sequence)datetime timedeltatimezone) AnnotatedFinal) APIRouterDepends HTTPExceptionQuery) BaseModel TypeAdapter)verbose_proxy_logger)CommonProxyErrorsLitellmUserRolesUserAPIKeyAuth)user_api_key_auth)GatewayRequestActivityResponseGatewayRequestBreakdownEntryGatewayRequestDailyEntryrouter_DEFAULT_LOOKBACK_DAYSa# SELECT date, category, route, SUM(successful_requests)::bigint AS successful_requests, SUM(failed_requests)::bigint AS failed_requests FROM "LiteLLM_DailyGatewayRequests" WHERE date >= $1 AND date <= $2 GROUP BY date, category, route _AGGREGATE_SQLcDeZdZUdZeed<eed<eed<eed<eed<y) _AggregateRowzLValidates one query_raw row so the handler works with typed values, not Any.datecategoryroutesuccessful_requestsfailed_requestsN)__name__ __module__ __qualname____doc__str__annotations__int/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/management_endpoints/gateway_request_endpoints.pyrr/s V IM Jr*r. _ROWS_ADAPTERreturnctjtj}|t t z }|j d|j dfS)N)daysz%Y-%m-%d)rnowrutcrrstrftime)endstarts r+_default_ranger5<sAhll+C(>??E >>* %s||J'? ??r*rowscbttdD}tfd|DS)Nc34K|]}|jywN)r.0rows r+ z _fold_by_date..Cs#=CHH#=c 3K|]6ttfdDtfdD8yw)c3VK|] }|jk(s|j"ywr9)rr r;r<rs r+r=z*_fold_by_date...Gs##`sxx[_O_C$;$;#`))c3VK|] }|jk(s|j"ywr9)rr!rAs r+r=z*_fold_by_date...Hs#XsxxSWGW 3 3XrB)rr r!N)rsum)r;rr6s @r+r=z _fold_by_date..DsD  ! ##`t#` `XtXX  sE    r*cttdD}tfd|D}tt|ddS)Nc3LK|]}|j|jfywr9)rrr:s r+r=z!_fold_by_route..Os#N#S\\399$=#Ns"$c 3K|]<\ttfdDtfdD>yw)c3vK|]0}|jk(s|jk(s#|j2ywr9)rrr r;r<rrs r+r=z+_fold_by_route...Ts6$,/3<<8;SX[XaXaejXj''$ 999c3vK|]0}|jk(s|jk(s#|j2ywr9)rrr!rOs r+r=z+_fold_by_route...Ws2ws||W_G_dgdmdmqvdv 3 3wrP)rrr r!N)rrD)r;rrr6s @@r+r=z!_fold_by_route..PsT  He % #$37$! wtww    sAAc|jSr9r )entrys r+z _fold_by_route..[s 53L3Lr*T)keyreverserE)r6pairsentriess` r+_fold_by_routerZNsI)#N#NNOE  %  G %LVZ[ \\r*z/gateway/daily/activityzBudget & Spend Tracking)tagsresponse_modelNzStart date in YYYY-MM-DD format)default descriptionzEnd date in YYYY-MM-DD formatuser_api_key_dict start_dateend_datecHKddlm}|jtjtj fvr t dd|%t dtjjt\}}|jjt|xs||xs|d{}tj|xsd}t!j"d t%|t't)d |Dt)d |Dt+|t-| S7}w) z Successful and failed gateway requests, counted at the ASGI edge. Deployment-wide: the underlying table has no per-key or per-user dimension, so this is admin-only. r) prisma_clientizLOnly proxy admin roles can view gateway request counts across the deployment) status_codedetailNir)z,/gateway/daily/activity - aggregated %d rowsc34K|]}|jywr9rSr:s r+r=z-get_gateway_daily_activity..s%N#c&=&=%Nr>c34K|]}|jywr9)r!r:s r+r=z-get_gateway_daily_activity..s!F##"5"5!Fr>)total_successful_requeststotal_failed_requestsby_dateby_route)litellm.proxy.proxy_serverrc user_roler PROXY_ADMINPROXY_ADMIN_VIEW_ONLYr rdb_not_connected_errorvaluer5db query_rawrr,validate_pythonrdebuglenrrDrJrZ)r_r`rarc default_start default_endraw_rowsr6s r+get_gateway_daily_activityrz^s  9""$$..+a  4E4\4\4b4bcc!/!1M;),,66#mKH  //B?DMsSWyY )"%%N%N"N!!F!FFd#%  sB D""D #A>D")*r%collections.abcrrrrtypingrrfastapir r r r pydanticr rlitellm._loggingrlitellm.proxy._typesrrr$litellm.proxy.auth.user_api_key_authr$litellm.types.proxy.gateway_requestsrrrrr'rrrrHr,r&r5rJrZgetrzr)r*r+rsh %22#<<+1TTB   ""  I#5);#<= u=@c3h@ / E:RTW:W4X  ]-0 ]U;WY\;\5] ]  # $1 #4=^_ ;Z[( 9J1K!KL(d (Dj($ (  (r*