g($hdZddlmZmZmZddlddlmZddlm Z eZ e jddgeegeefd e d e fd Ze jd dgeegeefd e d e fd Ze jddgeegdefdZe j%ddgeegeefded e fdZe j%ddgeegeefd e fdZe jddgeegeefded e fdZy)z BUDGET MANAGEMENT All /budget management endpoints /budget/new /budget/info /budget/update /budget/delete /budget/settings /budget/list ) APIRouterDepends HTTPException)*)user_api_key_auth)jsonify_objectz /budget/newzbudget management)tags dependencies budget_objuser_api_key_dictcPKddlm}m}|'tddtj j i|jd}t|}|jjji||jxs||jxs|d  d{}|S7w) ag Create a new budget object. Can apply this to teams, orgs, end-users, keys. Parameters: - budget_duration: Optional[str] - Budget reset period ("30d", "1h", etc.) - budget_id: Optional[str] - The id of the budget. If not provided, a new id will be generated. - max_budget: Optional[float] - The max budget for the budget. - soft_budget: Optional[float] - The soft budget for the budget. - max_parallel_requests: Optional[int] - The max number of parallel requests for the budget. - tpm_limit: Optional[int] - The tokens per minute limit for the budget. - rpm_limit: Optional[int] - The requests per minute limit for the budget. - model_max_budget: Optional[dict] - Specify max budget for a given model. Example: {"openai/gpt-4o-mini": {"max_budget": 100.0, "budget_duration": "1d", "tpm_limit": 100000, "rpm_limit": 100000}} rlitellm_proxy_admin_name prisma_clientNerror status_codedetailT exclude_none) created_by updated_by)data) litellm.proxy.proxy_serverrrrCommonProxyErrorsdb_not_connected_errorvalue model_dumprdblitellm_budgettablecreateuser_id)r r rrbudget_obj_jsonbudget_obj_jsonifiedresponses s/var/www/openai/venv/lib/python3.12/site-packages/litellm/proxy/management_endpoints/budget_management_endpoints.py new_budgetr(s,S.EEKKL  !+++>O)/:"%%99@@ " +33O7O+33O7O AH OsBB&B$B&z/budget/updatechKddlm}m}|'tddtj j i|jtdddi|jjjd |jii|jd d |jxs|i d{}|S7w)a= Update an existing budget object. Parameters: - budget_duration: Optional[str] - Budget reset period ("30d", "1h", etc.) - budget_id: Optional[str] - The id of the budget. If not provided, a new id will be generated. - max_budget: Optional[float] - The max budget for the budget. - soft_budget: Optional[float] - The soft budget for the budget. - max_parallel_requests: Optional[int] - The max number of parallel requests for the budget. - tpm_limit: Optional[int] - The tokens per minute limit for the budget. - rpm_limit: Optional[int] - The requests per minute limit for the budget. - model_max_budget: Optional[dict] - Specify max budget for a given model. Example: {"openai/gpt-4o-mini": {"max_budget": 100.0, "budget_duration": "1d", "tpm_limit": 100000, "rpm_limit": 100000}} rrNrrrzbudget_id is required budget_idTrr)wherer) rrrrrrrr+r r!updaterr#)r r rrr&s r' update_budgetr.Cs,S.EEKKL  #W>U4VWW"%%99@@J001 ###6 +33O7O AH OsB'B2)B0*B2z /budget/inforc Kddlm}|tdddit|jdk(rtddd |ji|j j jd d |jii d{}|S7w) z Get the budget id specific information Parameters: - budgets: List[str] - The list of budget ids to get information for rrNrrzNo db connectedrr*z0Specify list of budget id's to query. Passed in=r+inr,)rrrlenbudgetsr r! find_many)rrr&s r' info_budgetr6ns9W>O4PQQ 4<<AKDLL>Z  #%%99CCT4<<01DH O sBBB Bz/budget/settingsr+c Kddlm}|'tddtjj i|j tjk7rAtdddjtjj |j i|jjjd|i d{}||jd }ni}d d id d id d id did did did}g}tj j#D]\\}}||vs d } t%|||d |j&xsd|j)|d| |j*} |j-| ^|S7w)z Get list of configurable params + current value for a budget item + description of each field Used on Admin UI. Query Parameters: - budget_id: str - The budget id to get information for rr0Nr*rr{}, your role={}r+r2TrtypeIntegerStringFloat)max_parallel_requests tpm_limit rpm_limitbudget_duration max_budget soft_budget) field_name field_typefield_description field_value stored_in_dbfield_default_value)rrrrrr user_roleLitellmUserRoles PROXY_ADMINformatnot_allowed_accessr r! find_firstrBudgetNewRequest model_fieldsitems ConfigList descriptiongetdefaultappend) r+r r db_budget_rowdb_budget_row_dict allowed_args return_valrD field_info _stored_in_db _response_objs r'budget_settingsr_s"9.EEKKL  ""&6&B&BB+22%88>>%//  (**>>III&JM *5545H#))!4i(i("H-w'( LJ"2"?"?"E"E"G J  % M&%' 3F;","8"8">B.22:tD*$.$6$6 M   m ,#H GsB9E/;E->%//  #%%99CCEEH OFsB5C7B>8Cz/budget/deletecKddlm}|'tddtjj i|j tjk7rAtdddjtjj |j i|jjjd |ji d{}|S7w) zP Delete budget Parameters: - id: str - The budget id to delete rr0Nrrrr*r8r+r2)rrrrrrrJrKrLrMrNr r!deleteid)rr rr&s r' delete_budgetres9.EEKKL  ""&6&B&BB+22%88>>%//  #%%99@@DGG$AH O sCCC CN)__doc__fastapirrrlitellm.proxy._types$litellm.proxy.auth.user_api_key_authrlitellm.proxy.utilsrrouterpostrPUserAPIKeyAuthr(r. BudgetRequestr6rUstrr_raBudgetDeleteRequestrer'rss 65"B.   +,- )00A(B# #%#  #L  +,- )00A(B# #%#  #L  +,-  M  4  +,- )00A(BCC%C  CL  +,-  )00A(B%  8  +,- )00A(B! !%!  !rr