tPj2UdZddlZddlmZddlmZddlmZmZm Z ddl ddl m Z ddl mZdd lmZmZdd lmZdd lmZeZeed <ej1d dgee gee fdedefdZej1ddgee gee fdedefdZej1ddgee gdefdZej?ddgee gee fde defdZ!ej?ddgee gee fdefdZ"ej1ddgee gee fde#defdZ$y)z BUDGET MANAGEMENT All /budget management endpoints /budget/new /budget/info /budget/update /budget/delete /budget/settings /budget/list N)Mapping)Final) APIRouterDepends HTTPException)*)user_api_key_auth)get_budget_reset_time)_user_has_admin_viewvalidate_budget_duration)jsonify_object)BudgetRepositoryrouterz /budget/newzbudget management)tags dependencies budget_objuser_api_key_dictcKddlm}ddlm}m}|'t ddt jji|jJtj|jr|jdkrt ddd |ji|jJtj|jr|jdkrt ddd |jit|j|j1t!|jdkDrdd lm} ||j|j*'|jt-|j |_|j/d }t1|} t3|j4j7i||j8xs||j8xs|dd{} | S#t&$r}t ddt)|id}~wwxYw7/#t:$r/}t=||st ddd|j>did}~wwxYww)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. r)UniqueViolationErrorlitellm_proxy_admin_name prisma_clientNerror status_codedetail;max_budget must be a non-negative finite number. Received: *H99H>>Iz/budget/updatec6Kddlm}m}|'tddtj j i|jtdddi|jJtj|jr|jdkrtddd |ji|jJtj|jr|jdkrtddd |jit|j|j1t|jdkDrdd lm} ||j|j&d |j(vrd t+|j ini}t-i|j/d|d|j0xs|i}t3|j4j7d|ji|d{}|S#t$$r}tddt'|id}~wwxYw7/w)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. rrNrrrrzbudget_id is requiredrr r!r8r#T) exclude_unsetr(r?)wherer))r+rrrr,r-r.r?r/r0r1r2r r$r3r4r5r"r6r7model_fields_setr r r9r<rr:update) rrrrr"r@recomputed_reset_atrBrCs rD update_budgetrLusA.S.EEKKL  #W>U4VWW($-- @U@U2VZdZoZorsZsZ[e[p[pZqrs  )4==AWAW3X\f\r\ruv\v[\f\r\r[stu  Z778"".3z7R7R3SVW3W  K %j&A&A B  % % 16GzOjOj6j 1*B\B\]^  9G ##$#7 !  +33O7O 9-];AAHHJ001 !IH O/ KC#a&8IJ J K$s7D8H;G. BH'H(H. H7HHHz /budget/infor)cKddlm}|tdddit|jdk(rtddd |jit |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 connectedrrz0Specify list of budget id's to query. Passed in=r?inrH)r+rrr4budgetsrr: find_many)r)rrCs rD info_budgetrSs9W>O4PQQ 4<<AOPTP\P\~^_  -];AAKKT4<<01LH O sBB B B z/budget/settingsr?c Kddlm}|'tddtjj it |s6tddtjj d|jit|jjd|i d{}||jd }ni}d d id d id d id did did did did}g}tjjD]\\}}||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 rrNNrrr , your role=r?rPTr%typeIntegerStringFloatObject)max_parallel_requests tpm_limit rpm_limitr$r/r2r3) field_name field_typefield_description field_value stored_in_dbfield_default_value)r+rrr,r-r.r not_allowed_access user_rolerr: find_firstr9BudgetNewRequest model_fieldsitems ConfigList descriptiongetdefaultappend) r?rr db_budget_rowdb_budget_row_dict allowed_args return_valr_ field_info _stored_in_db _response_objs rDbudget_settingsrws"9.EEKKL  1 2 1 D D J JKB.22:tD*$.$6$6 M   m , - CrsBEEAErs  #55"BK/C   +,- )00A(BK K%K  K\  +,- )00A(BI I%I  IX  +,-  M  0  +,- )00A(B<<%<  <~  +,-  )00A(B%  .  +,- )00A(B %  r