tPj UdZddlmZmZddlmZddlmZddlmZddl m Z m Z ddl m Z mZmZddlmZmZdd lmZdd lmZmZmZdd lmZdd lmZmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/m0Z0e e+Z1e e2d<dZ3e e2d<GddeZ4GddeZ5ee6e4dfZ7e e2d<ee6e5dfZ8e e2d<djsde4jtDZ;e e2d<edd Gd!d"Zeee?e@d()eeAe@d*)eee@d+)d,ZBe ee?efe2d-<e d.e@d/e@d0eBe&d1d2fd3d4e>e=d56 ZCe e e4e4fe2d7<e1jd8d9eefe/e4:d;ed)?z`GET /management/v1/budgets`.)MappingSequence) dataclass)datetime)MappingProxyType) AnnotatedFinal) APIRouterDependsRequest) BaseModel TypeAdapter)verbose_proxy_logger)CommonProxyErrorsUserAPIKeyAuthuser_api_key_has_admin_view)user_api_key_auth)PROBLEM_TYPE_BASEManagementProblem) FilterSpecListSpec Predicate QueryPlanScopeScopeAll ScopeDeniedSortKey handle_list order_by_sql where_sql)MANAGEMENT_V1_PREFIX) PrismaClient) ListResponse ProblemDetail)prefixrouterz"LiteLLM_BudgetTable" BUDGET_TABLEceZdZUdZeed<dZedzed<dZedzed<dZ e dzed<dZ e dzed<dZ edzed<dZ edzed <eed <eed <y) BudgetListItemaDOne budget as the Budgets page reads it, and as it comes back off the table. Validating the raw row through here is what makes `tpm_limit` / `rpm_limit` numbers: they are `BigInt?` in the schema, which the query engine hands back as decimal strings, and a quoted "60000" breaks arithmetic in the dashboard. budget_idN max_budget soft_budget tpm_limit rpm_limitbudget_durationbudget_reset_at created_at updated_at)__name__ __module__ __qualname____doc__str__annotations__r+floatr,r-intr.r/r0r}/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/management_endpoints/management_v1/budgets.pyr)r)0snN#J # $K$ IsTz  IsTz "&OS4Z&'+OX_+r<r)ceZdZUeed<y) _RowCountcountN)r3r4r5r:r8r;r<r=r?r?Cs Jr<r?. _BUDGET_ROWS _ROW_COUNTSz, c#(K|] }d|d yw)"Nr;).0names r= rGJs#XDavQK#XsSELECTED_COLUMNST)frozenslotscLeZdZUdZeed<deedfdefdZ de de e fdZ y ) PrismaBudgetListExecutorzThe database half of the budgets list. Every caller-supplied value is bound to a placeholder by `where_sql`; only the spec's own column names reach the SQL text. prisma_clientwhere.returncKt|\}}dt|rd|ndz}|jjj|g|d{}t j |}|r|djSdS7,w)NzSELECT COUNT(*) AS count FROM  WHERE r)r r'rMdb query_rawrBvalidate_pythonr@)selfrNclausesparamssqlrowscounteds r=r@zPrismaBudgetListExecutor.countTs#E*5l^D_f'RYQZH[lno;D..11;;CI&II$44T:#*wqz11JsA A=A;-A=plancKt|j\}}dtdt|rd|ndzdt |j zdt |dzdt |d zz}|jjj|g||j|jd{}tj|S7w) NzSELECT z FROM rQrRz ORDER BY z LIMIT $z OFFSET $)r rNrHr'rorderlenrMrSrTtakeskiprArU)rVr\rWrXrYrZs r= find_manyz"PrismaBudgetListExecutor.find_many[s#DJJ/&'vl^ <&- "2 7< 345 6Vq)3v;?2CD E 2J 2r<rLrowrOc|S)zNThe row shape is the wire shape: the query selects exactly the columns served.r;)rfs r= _serializerhgs Jr<callerc^t|r tStd|jS)Nz.Only proxy admins can list budgets, your role=)reason)rrr user_role)ris r=_scopermls,"6*z  NvO_O_N`a bbr<)inis_null)typeops)gteltero)rrrs)r/r+r1BUDGET_FILTERSbudgets)r*r+r-r.r1)r*r1)field descending2dr*) resourcesortable searchablefilters default_sortdefault_page_size max_page_sizescope serialize tiebreakerBUDGETS_LIST_SPECz/budgets)zbudget management)tags dependenciesresponse_modelrequestuser_api_key_dictc vK ddlm}|7tttdddt j jttt|||d{S7#t$rt$r:}tjd |tttd d d d d}~wwxYww)a The budgets defined on this proxy, paged, sortable and filterable, for the Budgets page. Readable by a proxy admin or an admin viewer; anyone else is refused 403. The older `/budget/list` answers with the whole table as a bare array and has no way to page, sort or filter it. `sort` takes a comma-separated list of `budget_id`, `max_budget`, `tpm_limit`, `rpm_limit` or `created_at`, each optionally prefixed with `-` for descending, and defaults to `-created_at`. `budget_id` is appended to every sort as the tiebreaker. `q` is a case-insensitive substring match on `budget_id`. `page_size` defaults to 50 and is capped at 100. Filters are `filter[budget_duration][in|is_null]`, `filter[max_budget][gte|lte|is_null]` and `filter[created_at][gte|lte]`. Example curl: ``` curl --location --globoff 'http://0.0.0.0:4000/management/v1/budgets?sort=-max_budget&filter[budget_duration][in]=7d,30d&page_size=25' --header 'Authorization: Bearer sk-1234' ``` r)rMNzdatabase-not-connectedzDatabase not connectedi)rptitlestatusdetail)specexecutorrriz_litellm.proxy.management_endpoints.management_v1.budgets.list_budgets(): Exception occured - %szinternal-server-errorzInternal server errorizFailed to list budgets.)litellm.proxy.proxy_serverrMrr$rrdb_not_connected_errorvaluerrrL Exceptionr exception)rrrMes r= list_budgetsrs@! <  #-..DE2,CCII  !"-MJ$         && mop  )**?@-0     s;B9AA*#A($A*'B9(A**B6<5B11B66B9N)Fr6collections.abcrr dataclassesrrtypesrtypingrr fastapir r r pydanticr rlitellm._loggingrlitellm.proxy._typesrrr$litellm.proxy.auth.user_api_key_authrlitellm.proxy.list_api.commonrr%litellm.proxy.list_api.list_frameworkrrrrrrrrrrr 7litellm.proxy.management_endpoints.management_v1.commonr!litellm.proxy.utilsr"6litellm.types.proxy.management_endpoints.management_v1r#r$r&r8r'r)r?rerArBjoin model_fieldsrHrLrhrmr7 frozensetr9rtrgetrr;r<r=rsD#-!"#//+1 C    Y, !566- e-Y& "%(;"<= e= y#~!67 U7))#XNcec3C%3I>O4PQ e;T1UV hIn4MN3gc:o./FN  Z [(  >@  F5.."@AB  +,./   ; ;  9J1K!KL; .!;  ; r<