h&dZddlmZddlmZmZmZddlmZddl ddl m Z ddl m Z eZejdd gee g ee fd ed efd Zejdd gee g ee fd ed efdZejdd gee g defdZej-dd gee g ee fded efdZej-dd gee g ee fd efdZejdd gee g ee fded efdZy)z BUDGET MANAGEMENT All /budget management endpoints /budget/new /budget/info /budget/update /budget/delete /budget/settings /budget/list ) timedelta) APIRouterDepends HTTPException)duration_in_seconds)*)user_api_key_auth)jsonify_objectz /budget/newzbudget management)tags dependencies budget_objuser_api_key_dictcKddlm}m}|'tddtj j i|jF|j:tjtt|jz|_|jd }t|}|jj j#i||j$xs||j$xs|d  d{}|S7w) a 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}} - budget_reset_at: Optional[datetime] - Datetime when the initial budget is reset. Default is now. rlitellm_proxy_admin_name prisma_clientNerror status_codedetail)duration)secondsT exclude_none) created_by updated_by)data)litellm.proxy.proxy_serverrrrCommonProxyErrorsdb_not_connected_errorvaluebudget_reset_atbudget_durationdatetimeutcnowrr model_dumpr dblitellm_budgettablecreateuser_id)r rrrbudget_obj_jsonbudget_obj_jsonifiedresponses z/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/management_endpoints/budget_management_endpoints.py new_budgetr0s.S.EEKKL  !!)j.H.H.T%-__%6'1K1KL: & "!+++>O)/:"%%99@@ " +33O7O+33O7O AH OsC-C8/C60C8z/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}} - budget_reset_at: Optional[datetime] - Update the Datetime when the budget was last reset. rrNrrrzbudget_id is required budget_idTrr)wherer) rrrrr r!r"r3r(r)updater'r+)r rrrr.s r/ update_budgetr6Ms.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 connectedrr2z0Specify list of budget id's to query. Passed in=r3inr4)rrrlenbudgetsr(r) find_many)rrr.s r/ info_budgetr>ys9W>O4PQQ 4<<AKDLL>Z  #%%99CCT4<<01DH O sBBB Bz/budget/settingsr3c 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 rr8Nr2rr{}, your role={}r3r:TrtypeIntegerStringFloat)max_parallel_requests tpm_limit rpm_limitr$ max_budget soft_budget) field_name field_typefield_description field_value stored_in_dbfield_default_value)rrrr r!r" user_roleLitellmUserRoles PROXY_ADMINformatnot_allowed_accessr(r) find_firstr'BudgetNewRequest model_fieldsitems ConfigList descriptiongetdefaultappend) r3rr db_budget_rowdb_budget_row_dict allowed_args return_valrK field_info _stored_in_db _response_objs r/budget_settingsrfs"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 EsB9E/;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 rr8Nrrrr2r@r3r:)rrrr r!r"rQrRrSrTrUr(r)deleteid)rrrr.s r/ delete_budgetrls9.EEKKL  ""&6&B&BB+22%88>>%//  #%%99@@DGG$AH O sCCC CN)__doc__r%rfastapirrr*litellm.litellm_core_utils.duration_parserrlitellm.proxy._types$litellm.proxy.auth.user_api_key_authr litellm.proxy.utilsr routerpostrWUserAPIKeyAuthr0r6 BudgetRequestr>r\strrfrhBudgetDeleteRequestrlr/r{s 55J"B.   +,- )00A(B* *%*  *Z  +,- )00A(B$ $%$  $N  +,-  M  4  +,- )00A(BBB%B  BJ  +,-  )00A(B%  8  +,- )00A(B! !%!  !rz