h ddlmZmZmZddlmZmZmZmZm Z ddl m Z de de fdZ de dede fdZd eeefd ed edd fd Zdededeedeede f dZy ))AnyUnionOptional)GenerateKeyRequest1LiteLLM_ManagementEndpoint_MetadataFields_PremiumLiteLLM_TeamTableLitellmUserRolesUserAPIKeyAuth)_premium_user_checkuser_api_key_dictreturncz|jtjk(xs|jtjk(S)N) user_roler PROXY_ADMINPROXY_ADMIN_VIEW_ONLY)r s k/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/management_endpoints/common_utils.py_user_has_admin_viewr s7##'7'C'CC Q  & &*:*P*P Pteam_objc|jD]:}|j|j|jk(s*|jdk(s:yy)NadminTF)members_with_rolesuser_idrole)r rmembers r_is_user_team_adminrsD-- NN &6>>=N=V=V+VkkW$ . r object_data field_namevalueNcp|tvr t|jxsi|_||j|<y)z Helper function to set metadata fields that require premium user checks Args: object_data: The team data object to modify field_name: Name of the metadata field to set value: Value to set for the field N)rr metadata)rrrs r_set_object_metadata_fieldr" s6FF&//52K',K$rteam_idr max_budgetexisting_budget_idcK|/|jjd||didddiid{y|r*|jjd|id |id{y|jj||jxsd |jxsd d d di d{}|jj d||di||dd|j iidddd|j iiidd{y777X7 w)aC Helper function to Create/Update or Delete the budget within the team membership Args: tx: The transaction object team_id: The ID of the team user_id: The ID of the user max_budget: The maximum budget for the team existing_budget_id: The ID of the existing budget, if any user_api_key_dict: User API Key dictionary containing user information If max_budget is None, the user's budget is removed from the team membership. If max_budget exists, a budget is updated or created and linked to the team membership. Nuser_id_team_id)rr#litellm_budget_table disconnectT)wheredata budget_idr$)r$ created_by updated_byteam_membership)r+includeconnect)rr#r()createupdate)litellm_teammembershipr4litellm_budgettabler3rupsertr,)txr#rr$r%r new_budgets r_upsert_budget_and_membershipr:3sq,''..$'g&NO(<*>?/    $$++ 23 +,    --44$+339r+339r #D) 5J # # * * ""  #" Z-A-AB)' Z-A-AB)   + 1  sG,C?C7+C?C9AC?"C;#AC?1C=2C?9C?;C?=C?)typingrrrlitellm.proxy._typesrrrr r litellm.proxy.utilsr boolrrstrr"floatr:rrrBs''4Nt % 1B   -(*<<=-- - -&EE E  E ! E&Er