hdZddlZddlZddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZddlZddlmZmZmZmZmZmZddlmZddlZddlmZddlmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;dd lZ>m?Z?m@Z@dd lAmBZBdd lCmDZDmEZEmFZFmGZGdd lHmIZIdd lJmKZKddlLmMZMddlNmOZOmPZPddlQmRZRmSZSmTZTddlUmVZVddlWmXZXddlYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_eZ`deade;debfdZc ddeRde eade eade e#fdZdde e*e$fdeede;defdeef dZgd e$de;defd!eedeef d"Zhe`jd#d$geeBge$%ePeeBedd&'fde*d(ede;d)e eafd*Zjde:d+e eadeRde;d,eade eaf d-Zkde*de eRde eafd.Zld/e$d0e"d1eVdebfd2Zme`jd3d$geeBg4ePeeBedd&'fde:d(ede;d)e eafd5Znd6eed7e$deefd8Zod9e e)e e)fd:ebfd;Zpde eRde4d:ebfd<Zqde4d7e$fd=Zrde;d>e$ddfd?Zsde4d>e$deRde;d,eade e e&e e#ff d@Ztde4d>e$dAe e&ddfdBZude4d>e$deRde;d,eade e$e e&e e#ff dCZve`jdDd$geeBge0%ePeeBfde4de;fdEZwd7e$de5de ebe e)ffdFZxe`jdGd$geeBg4ePeeBfde5de;fdHZye`jdId$geeBge7%ePeeBfde6d(ede;fdJZzdKe e)dLe0de e^fdMZ{e`jdNd$geeBge[%ePeeBfdeZde;fdOZ|e`jdPd$geeBg4ePeeBedd&'fded(ede;d)e eafdQZ}de;d e$fdRZ~dSe$d1eVde$fdTZdUeadeRdVe2de2fdWZe`jdXd$geeBg4ePejddYZeeBfd(edeade;fd[Ze`jd\d$geeBg4ePeeBfded(ede;fd]Ze`jd^d$geeBg4ePeeBfded(ede;fd_Ze`jd`eeBe e$fd(ede;fdaZe`jdbd$ge]eeBgcejdddZejddeZejddfZejddgZejdhdidhjejdkdldhdmnejddoZejdpdqZeeBf d(ede eadre eade eadse eadteduedve eadweade;fdxZe`jdyd$geeBg4ePejdddZdeeBfd(ede eadre eade;fdzZ ddeRduedtede e e$effd{Zd!eed|eaddfd}Ze`jd~d$geeBgddde e$iiejddYZejddZejdhdidhjejdddhdmneeBfde eadse eadteduede;f dZde$de eade eafdZe`jdd$geeBg4ePeeBfde8d(ede;fdZe`jdd$geeBg4ePeeBfde9d(ede;fdZe`jdd$geeBg4ePejddYZeeBfdeade;de\fdZe`jdd$geeBg4eeBfde_d(ede;de$fdZe`jdeXd$gddddddhdkdeeBf de eade eade eade eade eadteduede eade;fdZy)za TEAM MANAGEMENT All /team management endpoints /team/new /team/info /team/update /team/delete N)datetimetimezone)AnyDictListOptionalTupleUnioncast) APIRouterDependsHeader HTTPExceptionRequeststatus) BaseModel)verbose_proxy_logger) BlockTeamRequestCommonProxyErrorsDeleteTeamRequestLiteLLM_AuditLogs1LiteLLM_ManagementEndpoint_MetadataFields_PremiumLiteLLM_ModelTableLiteLLM_OrganizationTableLiteLLM_TeamMembershipLiteLLM_TeamTableLiteLLM_TeamTableCachedObjLiteLLM_UserTableLitellmTableNamesLitellmUserRolesMemberNewTeamRequestProxyErrorTypesProxyExceptionSpecialManagementEndpointEnumsSpecialModelNamesSpecialProxyStringsTeamAddMemberResponseTeamInfoResponseObjectTeamInfoResponseObjectTeamTableTeamListResponseObjectTeamMemberAddRequestTeamMemberDeleteRequestTeamMemberUpdateRequestTeamMemberUpdateResponseTeamModelAddRequestTeamModelDeleteRequestUpdateTeamRequestUserAPIKeyAuth) allowed_route_check_inside_routecan_org_access_modelget_team_objectget_user_object)user_api_key_auth)_is_user_team_admin_set_object_metadata_field_upsert_budget_and_membership_user_has_admin_view)get_daily_activity)&handle_update_object_permission_common)TeamMemberPermissionChecks)add_new_membermanagement_endpoint_wrapper) PrismaClient_premium_user_checkhandle_exception_on_proxy)Router)SpendAnalyticsPaginatedResponse)BulkTeamMemberAddRequestBulkTeamMemberAddResponse GetTeamMemberPermissionsResponseTeamListResponseTeamMemberAddResult"UpdateTeamMemberPermissionsRequestteam_iduser_api_key_dictreturncttjydtjvr|tjdvSy)NFavailable_teams)litellmdefault_internal_user_paramsrMrNs m/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/management_endpoints/team_endpoints.py_is_available_teamrVjs9++3G@@@'>>?PQQQ  prisma_clientteam_idsuser_idc Kdd|ii}|d|gi|d<|jjj|ddid{}g}|D]*}|jt di|j ,|S77w) z)Get all team memberships for a given userrMinNrZlitellm_budget_tableTwhereinclude)dblitellm_teammembership find_manyappendr model_dump)rXrYrZ where_objteam_memberships returned_tmtms rUget_all_team_membershipsrkrs 3litellm.proxy.management_endpoints.budget_management_endpointsrr team_aliasreplaceuuiduuid4hexrvgetrwpop)rlrmrNrnrqrrrwteam_member_budget_tables rU _create_team_member_budget_tablers6 "DOO++C56htzz|?O?O>P Q #4::<#3#3"45 %/#) 00 , & j)1)+:& !**z"d ' sBCCAC team_table updated_kvcKddlm}ddlm}|ji|_|jj d}|xt |trh|||||d{}tjd|jd ||j d i|d <|j|d d<nt|||| d{}|jd d|S7}7w) zD Add budget if none exists If budget exists, update it rrp) update_budgetNr{)rwrurxz"Updated team member budget table: z, with team_member_budget=rzrlrmrNrnrn) r|rqr}rrzr isinstancestrrinforwrr)rrNrnrrqrr{ budget_rows rU _upsert_team_member_budget_tablers 6" &//334KL(Z8Ms-S('/-0    !!01E1E0FF`as`t u  >>* % -%'Jz ":D:N:N :67<)/1   NN'. -  s%A"C&$C"%A$C& C$ C&$C&z /team/newzteam management)tags dependenciesresponse_modelzThe litellm-changed-by header enables tracking of actions performed by authorized users on behalf of other users, providing an audit trail for accountability) description http_requestlitellm_changed_byc K ddlm}m}m}m}|t dddi|j jjd{}|r|j|r t d d |j#ttj|_ nE|j|jd d d{} | t ddd|jdi|j|jt j"k7rz|j$N|j$B|j$|j$kDr)t ddd|j$d|ji|j&N|j&B|j&|j&kDr)t ddd|j&d|ji|j(N|j(B|j(|j(kDr)t ddd|j(d|ji|j*`t-|j*dkDrH|j*D]9} | |j*vst ddd|j*d|j.i|j.cd} |j0D]} | j.|j.k(sd} | dur0|j0j3t5d|j.d} |j6t9|j6t:rt=t?j@|j6|j.xs||j.xs|}|j jBjEi|j?dd{}|jF} tI||d{}|j?}|jJ!tM||||jJd{}tOd,i|| |d}tPD]'}tS||tU||tS|| )|jVdd!l,m-}||jV"|_.g}|j0|j0}g|_|j_d}|ja|#}|j jjE|d$di%d{}tc|j|&}te|||||'d{tfjhdur|j?d}t?j@|t(}tkjl|tottjtqjrttjv|xs|j.xs||jxtzj||jd)|d* + |j_S77C7f7H7717#t~$r|j;cYSwxYw#t~$r}t|d}~wwxYww)-u Allow users to create a new team. Apply user permissions to their team. 👉 [Detailed Doc on setting team budgets](https://docs.litellm.ai/docs/proxy/team_budgets) Parameters: - team_alias: Optional[str] - User defined team alias - team_id: Optional[str] - The team id of the user. If none passed, we'll generate it. - members_with_roles: List[{"role": "admin" or "user", "user_id": ""}] - A list of users and their roles in the team. Get user_id when making a new user via `/user/new`. - team_member_permissions: Optional[List[str]] - A list of routes that non-admin team members can access. example: ["/key/generate", "/key/update", "/key/delete"] - metadata: Optional[dict] - Metadata for team, store information for team. Example metadata = {"extra_info": "some info"} - tpm_limit: Optional[int] - The TPM (Tokens Per Minute) limit for this team - all keys with this team_id will have at max this TPM limit - rpm_limit: Optional[int] - The RPM (Requests Per Minute) limit for this team - all keys associated with this team_id will have at max this RPM limit - max_budget: Optional[float] - The maximum budget allocated to the team - all keys for this team_id will have at max this max_budget - budget_duration: Optional[str] - The duration of the budget for the team. Doc [here](https://docs.litellm.ai/docs/proxy/team_budgets) - models: Optional[list] - A list of models associated with the team - all keys for this team_id will have at most, these models. If empty, assumes all models are allowed. - blocked: bool - Flag indicating if the team is blocked or not - will stop all calls from keys with this team_id. - members: Optional[List] - Control team members via `/team/member/add` and `/team/member/delete`. - tags: Optional[List[str]] - Tags for [tracking spend](https://litellm.vercel.app/docs/proxy/enterprise#tracking-spend-for-custom-tags) and/or doing [tag-based routing](https://litellm.vercel.app/docs/proxy/tag_routing). - organization_id: Optional[str] - The organization id of the team. Default is None. Create via `/organization/new`. - model_aliases: Optional[dict] - Model aliases for the team. [Docs](https://docs.litellm.ai/docs/proxy/team_based_routing#create-team-with-model-alias) - guardrails: Optional[List[str]] - Guardrails for the team. [Docs](https://docs.litellm.ai/docs/proxy/guardrails) - object_permission: Optional[LiteLLM_ObjectPermissionBase] - team-specific object permission. Example - {"vector_stores": ["vector_store_1", "vector_store_2"]}. IF null or {} then no object permission. - team_member_budget: Optional[float] - The maximum budget allocated to an individual team member. - team_member_key_duration: Optional[str] - The duration for a team member's key. e.g. "1d", "1w", "1mo" Returns: - team_id: (str) Unique team id - used for tracking spend across multiple keys for same team id. _deprecated_params: - admins: list - A list of user_id's for the admin role - users: list - A list of user_id's for the user role Example Request: ``` curl --location 'http://0.0.0.0:4000/team/new' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data '{ "team_alias": "my-new-team_2", "members_with_roles": [{"role": "admin", "user_id": "user-1234"}, {"role": "user", "user_id": "user-2434"}] }' ``` ``` curl --location 'http://0.0.0.0:4000/team/new' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data '{ "team_alias": "QA Prod Bot", "max_budget": 0.000000001, "budget_duration": "1d" }' ``` r)_license_checkcreate_audit_log_for_updatelitellm_proxy_admin_namerXNerrorNo db connected status_codedetail) team_countzOLicense is over limit. Please contact support@berri.ai to upgrade your license.team find_uniquerM table_name query_typez Team id = z0 already exists. Please use a different team id.z/tpm limit higher than user max. User tpm limit=z . User role=z/rpm limit higher than user max. User rpm limit=z1max budget higher than user max. User max budget=z6Model not in allowed user models. User allowed models=z . User id=FTadmin)rolerZ model_aliases created_by updated_by exclude_none)rlrXr)model_idobject_permission_id) object_data field_namevalueget_budget_reset_timervdb_datalitellm_model_table)rlr`)rMmemberrlcomplete_team_datarXrNrdefaultcreated id updated_at changed_bychanged_by_api_keyr object_idactionupdated_values before_value request_datara)Alitellm.proxy.proxy_serverrrrrXrrblitellm_teamtablecountis_team_count_over_limitrMrrrget_data user_roler PROXY_ADMIN tpm_limit rpm_limitrumodelslenrZmembers_with_rolesrer!rrdictrjsondumpslitellm_modeltablecreater_set_object_permissionrnrrrgetattrr:rv)litellm.proxy.common_utils.timezone_utilsrbudget_reset_atrfjsonify_team_objectr,_add_team_members_to_teamrRstore_audit_logsasyncio create_taskrrnowrutcapi_keyrTEAM_TABLE_NAME ExceptionrD)rlrrNrrrrrX total_teams_existing_team_idmcreating_user_in_listr _model_idr model_dictr data_jsonrfieldrrcomplete_team_data_dictteam_rowteam_member_add_request_updated_valueses rUnew_teamrsTQ+    CBS8TU U*,,>>DDFF >BB"C  h  << tzz|,DL'4&<&< M'='! !,# #:dll^;k!l  ' ' / **.>.J.JJ*%//;NN%6%@%@@# ##RSdSnSnRoo{}N}X}X|Y"Z*%//;NN%6%@%@@# ##RSdSnSnRoo{}N}X}X|Y"Z+%00<OO&7&B&BB# ##TUfUqUqTrr~@Q@[@[\"]{{&3/@/G/G+H1+LA 1 8 88+(+ '+absbzbza{|FGXG`G`Fa*b$%  $ $ 0$) !11>>%6%>%>>,0)2%-''..1B1J1JK     )j9K9KT.R!3"jj););<,44P8P,44P8P"  -//BBII>%***=> J# I&<'&   IIK  " " .>#,"3#'#:#: I/  !5 GEtU#/* 2$!$.G  - - 9 W1F 2 B B2  . ,.  0 0  A5X2>X?FX2A X2=C!X2X X2?XX2X2X2X2 X2 X2X2X/,X2-Y.X//X22 Y ; YY  YrrcDK|}|jt|jtrtt j |j|j xs||j xs|}|A|jjji|j dd{}nX|jjjd|ii|j di|j ddd{}|j}|S7k7w) z4 Upsert model table and return the model id NrTrrlr)updaterr_rl) rrrrrrrZrbrrupsertr)rlrrXrNrrrrs rU_update_model_tablers)I %*T5G5G*N/**T%7%78(00L4L(00L4L   ,//BBIIC*//T/BC J J -//BBIIX&L!3!8!8d!8!KLL!3!8!8d!8!KL J JMM s%B.D 0D1AD  D D D cK|y|jV|jjj|jj dd{}|`|j Sy7w)z Creates the LiteLLM_ObjectPermissionTable record for the team. - Handles permissions for vector stores and mcp servers. Returns the object_permission_id if created, otherwise None. NTrr)object_permissionrblitellm_objectpermissiontablerrfr)rlrXcreated_object_permissions rUrr&sz )""@@GG++66D6IH  "  "(===   sAA+A)A+r organization llm_routerc |j|jk(ryt|jdkDrytjj |jvrnR|jt|jdk(rt dddi|jD]}t||||jry|jrm|jjrW|j|jjkDr4t ddd|jd |jjd i|jDcgc]}|j}}|jr$|jDcgc]}|jc}ng}|Dcgc]%}||vr|tjj k7r|'}}t|dkDrt ddd |d i|jry|jrm|jjrW|j|jjkDr4t ddd |jd|jjd i|j ry|jrm|jj rW|j |jj kDr4t ddd|j d|jj d iycc}wcc}wcc}w)a2 Validate that a team can be moved to an organization. - The org must have access to the team's models - The team budget cannot be greater than the org max_budget - The team's user_id must be a member of the org - The team's tpm/rpm limit must be less than the org's tpm/rpm limit TrrrzeCannot move team to organization. Team has access to all proxy models, but the organization does not.r)model org_objectrz6Cannot move team to organization. Team has max_budget z4 that is greater than the organization's max_budget .z3Cannot move team to organization. Team has user_id z* that is not a member of the organization.z5Cannot move team to organization. Team has tpm_limit z3 that is greater than the organization's tpm_limit z5Cannot move team to organization. Team has rpm_limit z3 that is greater than the organization's rpm_limit )organization_idrrr&all_proxy_modelsrrr5rur]rrZusersr'default_user_idrr)rrrrr team_members org_members not_in_orgs rUvalidate_team_org_changer>sM |;;; <  !#  - - 3 3|7J7J J  [[ C $4$9E $+)%   - -  - - 8 8 OOl??JJ JQRVRaRaQbcWXdXyXyXDXDWEEFG  (,'>'>?!AII?L?=I=O=Ol&8&891999UWK K A)<)L)L)R)R$R J  :Nzl[EF     - -  - - 7 7 NN\>>HH HPQUQ_Q_P`aTUaUvUvU@U@TAABC     - -  - - 7 7 NN\>>HH HPQUQ_Q_P`aTUaUvUvU@U@TAABC  Q@9s?K".K' *K,z /team/update)rrc Kddlm}ddlm}m}m}m}m} m} |'tddtjji|jtddd itjd ||j j"j%d |ji d{} | td dd|ji|j&t)|j&dkDr|'tddtj*ji|j j,j%d|j&idddd{} | td dd|j&it/t1d,i| j3t5d,i| j3|n+|j&t)|j&dk(rd|_|j7d} |j8ddlm}||j8}|| d<|j>"tA| | |j>|d{} n| jCdd|jDtG| | d{} tH}|D]}|| vs| |tK| |d| vr:| jCdtM|| jN|||d{}||| d<|jQ| } |j j"jSd |ji| d!di"d{}| |jtddd#jU|itjVd$|j||jtYd,i|j3| | %d{tZj\dur| j7d&}t7j^|t`'}t7j^| t`'}tcjd|tgtatijjtmjntpjr|xs|jtxs||jvtxjz|jd(||) *|j|d+S7 7f7z7I777w)-aZ Use `/team/member_add` AND `/team/member/delete` to add/remove new team members You can now update team budget / rate limits via /team/update Parameters: - team_id: str - The team id of the user. Required param. - team_alias: Optional[str] - User defined team alias - team_member_permissions: Optional[List[str]] - A list of routes that non-admin team members can access. example: ["/key/generate", "/key/update", "/key/delete"] - metadata: Optional[dict] - Metadata for team, store information for team. Example metadata = {"team": "core-infra", "app": "app2", "email": "ishaan@berri.ai" } - tpm_limit: Optional[int] - The TPM (Tokens Per Minute) limit for this team - all keys with this team_id will have at max this TPM limit - rpm_limit: Optional[int] - The RPM (Requests Per Minute) limit for this team - all keys associated with this team_id will have at max this RPM limit - max_budget: Optional[float] - The maximum budget allocated to the team - all keys for this team_id will have at max this max_budget - budget_duration: Optional[str] - The duration of the budget for the team. Doc [here](https://docs.litellm.ai/docs/proxy/team_budgets) - models: Optional[list] - A list of models associated with the team - all keys for this team_id will have at most, these models. If empty, assumes all models are allowed. - blocked: bool - Flag indicating if the team is blocked or not - will stop all calls from keys with this team_id. - tags: Optional[List[str]] - Tags for [tracking spend](https://litellm.vercel.app/docs/proxy/enterprise#tracking-spend-for-custom-tags) and/or doing [tag-based routing](https://litellm.vercel.app/docs/proxy/tag_routing). - organization_id: Optional[str] - The organization id of the team. Default is None. Create via `/organization/new`. - model_aliases: Optional[dict] - Model aliases for the team. [Docs](https://docs.litellm.ai/docs/proxy/team_based_routing#create-team-with-model-alias) - guardrails: Optional[List[str]] - Guardrails for the team. [Docs](https://docs.litellm.ai/docs/proxy/guardrails) - object_permission: Optional[LiteLLM_ObjectPermissionBase] - team-specific object permission. Example - {"vector_stores": ["vector_store_1", "vector_store_2"]}. IF null or {} then no object permission. - team_member_budget: Optional[float] - The maximum budget allocated to an individual team member. - team_member_key_duration: Optional[str] - The duration for a team member's key. e.g. "1d", "1w", "1mo" Example - update team TPM Limit ``` curl --location 'http://0.0.0.0:4000/team/update' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data-raw '{ "team_id": "8d916b1c-510d-4894-a334-1c16a93344f5", "tpm_limit": 100 }' ``` Example - Update Team `max_budget` budget ``` curl --location 'http://0.0.0.0:4000/team/update' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data-raw '{ "team_id": "8d916b1c-510d-4894-a334-1c16a93344f5", "max_budget": 10 }' ``` r)_cache_team_object)rrrrXproxy_logging_objuser_api_key_cacheNrrrrNo team id passed inz/team/update - %srMr_Team not found, passed team_id=r T)r]r r^z/Organization not found, passed organization_id=)rrr) exclude_unsetrrr)rrrnrNrn)rexisting_team_row)rrr)rlrrXrNrrrr)r_rlr`zTeam doesn't exist. Got={}z$Successfully updated team - %s, info)rMrrrrrupdatedrr)rMrlra)>litellm.proxy.auth.auth_checksrrrrrrXrrrrdb_not_connected_errorrrMrdebugrbrrr r no_llm_routerlitellm_organizationtablerrrfrrrvrrrnrrrhandle_update_object_permissionr_update_team_metadata_fieldrrrrformatrrrRrrrrrrrrrrrrrZrrr)rlrrNrrrrrrXrrrorganization_rowrrreset_at_team_metadata_fieldsrrr _before_value _after_values rU update_teamr*svB.EEKKL   ||W>T4UVV2D9+..@@LL$,,'M >t||nMN  (S1E1E-F-J  2C2Q2Q2W2W(X "/!1!1!K!K!W!W$d&:&:;-1DA"X"    #NtOcOcNde  !"D%6%A%A%CD2S5E5P5P5RS!    )c$2F2F.G1.L#.J 'S(9M9MN)1 $% *;(!#66/    +T2 ): /  N& J :e#4#@ '%  '*$'-&//'/%=     %.Jz "22:2FJ0077dll+*D18   8++39@@JK  DhFVFVW   -F0C0C0EF-+  4')..D.A  =#>  JJz3?  '.4::<('||HLL91 0(00 0/'8'@'@0@@"ll$#/!.   $ '' ::K" @    sB&S (R4)B(S R7C.S R:3S 4R=5S  S AS SAS 'S(BS (S)D S 7S :S =S S S S rrcKddlm}t||j|d{}|||d<t j d||S7%w)a Handle the update of object permission for a team. - IF there's no object_permission_id, then create a new entry in LiteLLM_ObjectPermissionTable - IF there's an object_permission_id, then update the entry in LiteLLM_ObjectPermissionTable rrX)rexisting_object_permission_idrXNrzupdated object_permission_id: )rrXr>rrr)rrrXrs rUr"r"ssg9"H&7&L&L#"',@ ()"",-A,B C  s"A A &A r premium_usercDt|tr:|jdk(r+|dur&tdtj j yt|trF|D]@}|jdk(s|dustd|dtj j dyy)NrTz,Assigning team admins is a premium feature. z0Assigning team admins is a premium feature. Got=z. )rr!r ValueErrorrnot_premium_userrr)rr.rs rU_check_team_member_admin_addr2s&&!fkkW&< t #>?P?a?a?g?g>hi  $ FD !Avv t+$J1#RPaPrPrPxPxOyy{| "rWc|tdddi|jtdddi|jtdddi t|j|y#t$r}tddt |id}~wwxYw) NrrrrrrzNo member/members passed in)rr.)rrMrr2rr)rXrlr.rs rUteam_call_validation_checksr4s W>O4PQQ ||W>T4UVV {{W.K$L  G$;;%  GWc!f4EFFGs A!! B*BBcgdtffd }t|jtr||jn3t|jtr|jD] }|| t|jtrKt t |jk(r*t djtjddt|jtrft dk(rXt d|jjd|jjd jtjddt d kDr&tjd jd y y )z Check if a member already exists in the team. This check is done BEFORE we create/fetch the user, so it only prevents obvious duplicates where both user_id and user_email match exactly. rc(|j<jD]-}|j|jk(sj|/|j=jD]-}|j|jk(sj|/yyN)rZrre user_email)rexisting_memberrinvalid_team_memberss rU_check_member_duplicationzDteam_member_add_duplication_check.._check_member_duplications >> %#4#G#G"**fnn<(//7$H    (#4#G#G"--1B1BB(//7$H )rWz0All users are already in team. Existing members=400messagetypeparamcodez&User already in team. Member: user_id=z , user_email=z. Existing members=rz1Some users are already in team. Existing members=z. Duplicate members=N)r!rrrlistrr$rr#team_member_already_in_teamrZr8rr)rlrr;rr:s ` @rU!team_member_add_duplication_checkrEs} 8& 8$++v&!$++. DKK &A %a ($++t$-A)Bc$++FV)VFGXGkGkFlm <<   DKK (S1E-F!-K'>"/&7-E LL+A2,( j$   .%(// ;-0 2 22]( U\\ T\\3q6  , # #!Y!`!`t||SV" syAG  3D6>D4?D6AG )FFF G G 4D66 F?=E<<FG F G3GGG rRc*Kt|jtr|jj}|jK|j ?|D]:}|j |j |j k(s*|j|_<d}|j D]Q}|j|j|jk(s'|j 5|j |j k(sOd}n|s|j j|yyt|jtr|jD]}|jK|j ?|D]:}|j |j |j k(s*|j|_<d}|j D]Q}|j|j|jk(s'|j 5|j |j k(sOd}n|r|j j|yyw)z*Update the team's members_with_roles list.NFT) rrr! model_copyrZr8rrer)rlrrRrOusermember_already_existsr9nms rU_update_team_members_listr\Ms $++v&[[++-     %**?*?*K%OO/:+@+@@)-J& &!&1DDO"".#++z/A/AA%%1#..*2G2GG(,% E%  1 1 8 8 D% DKK &++Bzz!bmm&?)D2t"--7W%)\\ * %* !#5#H#HJJ*/F/F"**/TMM-'22bmmC,0)$I)"55<B;B0B;B2B;B4A B;&B9' B;2B;4B;rJc DKddlm}m}m}m}m} t |||tt|}t|j||d|ddd{}|tdd d t|d di tdi|j} t|| t!|| d{t#|| |||d{\} } } | tdd d|jdi t%di| j| | dS#t$r}|d}~wwxYw77r7\w)a Add new members (either via user_email or user_id) to a team If user doesn't exist, new user row will also be added to User Table Only proxy_admin or admin of team, allowed to access this endpoint. ``` curl -X POST 'http://0.0.0.0:4000/team/member_add' -H 'Authorization: Bearer sk-1234' -H 'Content-Type: application/json' -d '{"team_id": "45e3e396-ee08-4a61-a88e-16b3ce7e0849", "member": {"role": "user", "user_id": "krrish247652@berri.ai"}}' ``` r)rr.rXrr)rXrlr.NFTrMrXrparent_otel_spanrcheck_cache_only check_db_onlyrrTeam not found for team_id=rMr)rlrrLrz Team with id z not found)rRrSra)rrr.rXrrr4rr rBr6rMrrrfrErMrr() rlrNrr.rXrrrrrr_rRrSs rUteam_member_addrfs~4#'% }5M- #-+ 6wtYPT7U6VW  +L->-I-I-KL% , 0+-  (1'/%=   :L-!9W dll^:.V$W   !   ! ! # #!9 e  2  sYD D/D DAD )D*D DAD  DDDD D D c&d}g}|jD]{}|j(|j|j|jk(rd}7|j(|j|j|jk(rd}k|j|}||fS)z+Cleanup members_with_roles list for a team.FT)rrZr8re)rrlis_member_in_teamnew_team_membersrs rU_cleanup_members_with_rolesrj s %'  1 1 LL $ %  ) $   OO ' (1<</ $  "2 . ..rW/team/member_deletecKddlm}|tdddi|jtddd i|j|j tddd i|j jjd |ji d{}|(tddd j|jitdi|j}|jtjjk7r6t!||s)tdddjd|jit#||\}}|stdddi||_|Dcgc]}|j}}|j jj'd |jidt)j*|id{} i} |j|j| d<n|j |j | d<|j j,j/| d{} | t1| t2rt5| dkDr| D]} g} |j| j6vs| j6} | j9|j|j j,j'd| jidd| iid{|S7Kcc}w777w)a [BETA] delete members (either via user_email or user_id) from a team If user doesn't exist, an exception will be raised ``` curl -X POST 'http://0.0.0.0:8000/team/member_delete' -H 'Authorization: Bearer sk-1234' -H 'Content-Type: application/json' -d '{ "team_id": "45e3e396-ee08-4a61-a88e-16b3ce7e0849", "user_id": "krrish247652@berri.ai" }' ``` rr,Nrrrrrr2Either user_id or user_email needs to be passed inrMrTeam id={} does not exist in dbrGrrIrk)rrlzUser not found in teamrrrZr8teamssetra)rrXrrMrZr8rbrrr$rrfrr rrr9rjrrrrlitellm_usertablerdrrCrroremove)rlrNrX_existing_team_rowrrhrir_db_new_team_members_key_valexisting_user_rows existing_user team_lists rUteam_member_deleterz#s:9W>O4PQQ ||W>T4UVV || 7QR  -//AAMM$,,' N !>EEdllST  *L,>,I,I,KL ##'7'C'C'I'II#/:K ell)+<+D+D  +F+ +'' W>V4WXX+;(@P'Q1 'Q'Q0077 t|| #DJJ/C$D E 8 AG ||!\\   $ $ ,//AAKK L %%t,5G1H11L/MI||}222)//   .#&&88??!=#8#8"E9#56 @ 0 UN(R s_BK+KCK+"K9A K+K$A&K+-K'.AK+2A!K+K) K+K+'K+)K+z/team/member_updatec Kddlm}m}|tdddi|jtddd i|j d k(r|s tdd |j |jtddd i|jjjd |jid{}|(tdddj|jitdi|j}|jtj j"k7r6t%||s)tdddjd|jit'||j|d{}|d}d} |j |j } nT|jH|dj(D]6} | j| j|jk(s*| j } n| tdddj|id} |dD]} | j | k(s| j*} n|jj-4d{} t/| |j| |j0| |d{dddd{|j g}|j(D]l} | j | k(rJ|j3t5| j |j |jxs | j\|j3| n||_|Dcgc]}|j}}|jjj7d |jidt9j:|id{t=|j| |j|j0S7 7O7r7H7;#1d{7swYLxYwcc}w7[w)zE [BETA] Update team member budgets and team member role r)r.rXNrrrrrrra Assigning team admins is a premium feature. You must be a LiteLLM Enterprise user to use this feature. If you have a license please set `LITELLM_LICENSE` in your env. Get a 7 day trial key here: https://www.litellm.ai/#trial. Pricing: https://www.litellm.ai/#pricingrmrMrrnrGrrIrk)rrMrN team_infoz,User id doesn't exist in team table. Data={}rh)txrMrZruexisting_budget_idrN)rZrr8rr)rMrZr8rPra)rr.rXrrMrrZr8rbrrr$rrfrr rrr9r|rrwr}r;rPrer!rrrr/)rlrrNr.rXrsrreturned_team_inforreceived_user_idridentified_budget_idrjr}rrr^s rUteam_member_updaters"GW>O4PQQ ||W>T4UVV yyGL`   || 7QR  -//AAMM$,,' N !>EEdllST  *L,>,I,I,KL ##'7'C'C'I'II#/:K ell)+<+D+D  8A! +82 $K0J'+ ||<<  $(5HHF  ,1B1Bdoo1U#)>> I GNNtT  +/ !34 ::) )#%<< 5 ""$  +LL$..3/       yy%'  33F~~!11## &!YY#'??#Gf6G6G##F+4)5 %@L'M1 'M'M0077dll+& 3C(DE8   $  ??22  }82B      6(N sB/O)1N=2B?O)1O2AO)O)AO)$-O)OO))O >O?O  O)O BO)'O">A O) O' 2O)O)O)O  O) OO OO)membersresponsec g}|D]}d}|jD]_}|jr|j|jk(s'|js5|j|jk(sO|j}nd}|jD]:}|js|j|jk(s*|j}n|j t |j|jd|||S)zS Convert TeamAddMemberResponse into individual TeamMemberAddResult objects NT)rZr8successrTupdated_team_membership)rRrZr8rfrSrerK)rrresultsrrTrYrrjs rU_create_results_from_responser&s*,G **D4<<6>>#A!!doo9J9J&J#0  +#'33B~~"**">*,--/'4  !,,)(?   #6 NrWz/team/bulk_member_addc Kddlm}ddlm}|#t dd|j j i|jre|jjjdd i d{}|Dcgc]$}t|j|jd &c}|_|jst d ddid}t|j|kDrt d dd|di t!t#|j$|j|j&|d{}t)|j|}t+|j$|t|jt|d|j-S7cc}w7g#t.$r} t1j2| t5| } |jD cgc]%} t7| j| jd| 'ncc} w}} t+|j$|t|jdt|jdcYd} ~ Sd} ~ wwxYww)a" Bulk add multiple members to a team at once. This endpoint reuses the same logic as /team/member_add but provides a bulk-friendly response format. Parameters: - team_id: str - The ID of the team to add members to - members: List[Member] - List of members to add to the team - all_users: Optional[bool] - Flag to add all users on Proxy to the team - max_budget_in_team: Optional[float] - Maximum budget allocated to each user within the team Returns: - results: List of individual member addition results - total_requested: Total number of members requested for addition - successful_additions: Number of successful additions - failed_additions: Number of failed additions - updated_team: The updated team object Example request: ```bash curl --location 'http://0.0.0.0:4000/team/bulk_member_add' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data '{ "team_id": "team-1234", "members": [ { "user_id": "user1", "role": "user" }, { "user_email": "user2@example.com", "role": "admin" } ], "max_budget_in_team": 100.0 }' ``` r)rr,Nrrr created_atdesc)orderrY)rZr8rrzAt least one member is requiredzMaximum z members can be added at once)rMrrPrlrN)rMrtotal_requestedsuccessful_additionsfailed_additionsr_F)rZr8rr)r|rrrXrrr all_usersrbrqrdr!rZr8rrrfr,rMrPrrHrfrr exceptionrrK) rlrNrrXall_users_in_dbrYMAX_BATCH_SIZErrr error_messagers rUbulk_team_member_addrMs:d78.EEKKL   ~~ - 0 0 B B L L(!M!  (    ??    <<>?  N 4<<>)x'77TUV  , (% ||#'#:#: 0   0 hG(LL -!$W!,,.   Q  2 *  &&q)A ,,  !,,#     )LL -!" .    suA)I +F,I 4)F AI .:F(F)AFI  I F I.I *G43AI;I<I II z /team/deletecKddlm}m}m}|t dddi|j t ddd ig}|j D]f} |j jjd |i d{} | t tdi| j} |j| htjdur|j D]}|j|ddd{} | "| j!d} t#j$|t't)t+j,t/j0t2j4|xs|j6xs||j8t:j<|dd|  |j?|j dd{|D]w} | j@} g}| D]H}|jtCtE| jF|j6|jH|Jt#jJ|d{y|j?|j dd{}|S7#t$rt d dd |iwxYw777T7/w)a delete team and associated team keys Parameters: - team_ids: List[str] - Required. List of team IDs to delete. Example: ["team-1234", "team-5678"] ``` curl --location 'http://0.0.0.0:4000/team/delete' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data-raw '{ "team_ids": ["8d916b1c-510d-4894-a334-1c16a93344f5"] }' ``` r)rrrXNrrrrrrrMrrrTrrrrdeletedz{}rrkey) team_id_listr)rMrZr8rra)&rrrrXrrYrbrrrrrfrerRrrrrrrrrrrrrrrZrrr delete_datarrzr-rMr8gather)rlrrNrrrrX team_rowsrM team_row_baseteam_row_pydanticr _team_rowrtasks team_member deleted_teamss rU delete_teamrs8 W>O4PQQ }}W>T4UVV*,I== #&&88DD$g.E  $%.K 0H0H0JK*+!&4'}}G:G:P:PF};Q;5H 4 8I   +!2tzz|,#+<< #=#5$4,44$43+<+D+D#4#D#D")('+%. " %>  # #5 # QQQ22 'K LL"0 ( 0 0 + 3 3#.#9#9 '8  (nne$$$!&(33]]v4M U  #B7)!LM  5<R& %suAJ  +I#6I 7 I#A%J (J)CJ 8J9A9J 2J3&J J J  I##I??J J J  J c|jtjjk(s'|jtjjk(ry|j |j k(ry|j |jDcgc]}|j c}vr3tdddj|j |j iycc}w)Nrrz-User={} not authorized to access this team={}r) rr rrPROXY_ADMIN_VIEW_ONLYrMrZrrr$)rNrrs rUvalidate_membershiprMs ##'7'C'C'I'II  & &*:*P*P*V*V V !!Z%7%77  %88) )HOO%--z/A/A  )sCr|cVtjj|jvr|t }|jD]1}|tjjk7s!|j |3|j D]}|j |t||_|Sr7)r&r rrrpaddget_model_namesrC)r|r team_modelsrs rU_unfurl_all_proxy_modelsrhs **00I4D4DD  " # %%E)::@@@&& //1E OOE "2 ,  rWr{team_info_response_objectcK |jjjd|id{}||_|S7 #t$rt j d|Y|SwxYww)NrwrzATeam member budget table not found, passed team_member_budget_id=)rblitellm_budgettablerrrrr)r{rXr team_budgets rU_add_team_member_budget_tablerys  ),,@@LL 56M  >I!: %$   !!OPeOf g  %$  s1A'+?= ?A'?!A$ A'#A$$A'z /team/infoz!Team ID in the request parameters)rrc tKddlm}ddlm} |t t j ddi|t t jdd i |jjjd |id d i d{}|t t|td#i|j|j!|ddt#j$d{}|g}|d}|D]}|t'|ddz }d|i}|D]%} | j} | j+dd't-||gdd{} t/|t(r |d#i|} n.t/|t0r|d#i|j} n|} | j2| j2j5dnd} | t7| || d{} t9|| || } | S7^#t$r"t t jdd|diwxYw73#t$r| j)} YwxYw77k#t$r}t;j<dj?|tAjBt/|trYtEt'|ddtG|dtHjJt'|ddt'|d t jL!t/|tDr|tEd"tG|ztHjJt'|ddt jL!d}~wwxYww)$a# get info on team + related keys Parameters: - team_id: str - Required. The unique identifier of the team to get info on. ``` curl --location 'http://localhost:4000/team/info?team_id=your_team_id_here' --header 'Authorization: Bearer your_api_key_here' ``` r)r*r,NrzDatabase not connected. Connect a database to your proxy - https://docs.litellm.ai/docs/simple_proxy#managing-auth---virtual-keysrr>z(Malformed request. No team id passed in.rMrTr^z Team not found, passed team id: r )rNrrfind_all)rMrrexpiresspendtokenrZr{)r{rXr)rMr|keysrhz\litellm.proxy.management_endpoints.team_endpoints.py::team_info - Exception occurred - {} {}rzAuthentication Error()r@Nonerr=zAuthentication Error, ra)'r|r*rrXrrHTTP_500_INTERNAL_SERVER_ERRORHTTP_422_UNPROCESSABLE_ENTITYrbrrrHTTP_404_NOT_FOUNDrrrfrrrrrrrkrrrzrrr)rrr$ traceback format_excr$rr# auth_errorHTTP_400_BAD_REQUEST)rrMrNr*rXr|rrkrri _team_infor{response_objectrs rUr|r|s,E8t  "AAa  ?"@@!#MN  #&&88DD$g.0$7E   ! /(B9+?+?+AB #++!LLN ,   <D  EGQ// %(IC !nn& GGGT " 5 G9d   i &8E9EJ  9 -8R9;O;O;QRJ8:J "".    # #$; <  ! ,<&;+*4 J1 (   ] "55!%EgYa#PQ   * !hhj !  $"  "" k r r9'')  a ' 8/DSVHA-NO$//a&1Q v/J/JK   > *G,s1v5 ++!Wf-,,    s L8>H3.G=G> G AH3H ,H3H&H3=H/>BH3H1H3L8G+H  H3H,(H3+H,,H31H33 L59*++ ##55<<$,,'y$.?=F~>t||nMN  MA A5A3%A5z /team/unblockcKddlm}| td|jjj d|j iddid{}|td d d |j i |S7$w) ay Blocks all calls from keys with this team id. Parameters: - team_id: str - Required. The unique identifier of the team to unblock. Example: ``` curl --location 'http://0.0.0.0:4000/team/unblock' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data '{ "team_id": "team-1234" }' ``` rr,NrrMrFrrrrrrrs rU unblock_teamrLs29*++ ##55<<$,,'y%.@=F~>t||nMN  Mrz/team/availablec Kddlm}|'tddtjj it ttttjtjjdnd}|tdddi|jjjd|j i d{}|td dd it#di|j%}|Dcgc]}||j&vs|}}|jj(j+d d |ii d{}|Dcgc]}t-di|j%} }| S7cc}w73cc}ww)Nrr,rrrrQzNo available teams for user to join. See how to set available teams here: https://docs.litellm.ai/docs/proxy/self_serve#all-settings-for-self-serve--sso-flowrZrrzUser not foundrMr\ra)rrXrrrrr rrrrRrSrrbrqrrZrrfrorrdr) rrNrrXrQ user_infouser_info_correct_typeravailable_teams_dbavailable_teams_correct_types rUlist_available_teamsrws 9.EEKKL  c33?  0 0 4 45F G Oy  $&&88DD+334EI-.  /H1E1E1GH)D8N8T8T,TO -//AAKK412 L  Column to sort by (e.g. 'team_id', 'team_alias', 'created_at')asczSort order ('asc' or 'desc')r r~page page_sizesort_by sort_orderc Kddlm} | tddd| i|)| jtj k7r | j }|dz |z} i} |r|| d <|r|d d | d <|r|| d <|r | jjjd|id{} | tddd|itdi| j}|d|ji| d <n&||jvr|| d <ntddd|igd}d}|r*||vr&|jdvrd}||ji}| jjj!| | ||r|nddid{}| jjj#| d{}| |z }|r|Dcgc]}|jc}ng||||dS72#t$rtddd|iwxYw77Wcc}ww)a Get a paginated list of teams with filtering and sorting options. Parameters: user_id: Optional[str] Only return teams which this user belongs to organization_id: Optional[str] Only return teams which belong to this organization team_id: Optional[str] Filter teams by exact team_id match team_alias: Optional[str] Filter teams by partial team_alias match page: int The page number to return page_size: int The number of items per page sort_by: Optional[str] Column to sort by (e.g. 'team_id', 'team_alias', 'created_at') sort_order: str Sort order ('asc' or 'desc') rr,NrrzNo db connected. prisma client=rrBrM insensitivecontainsmoder~r rZrrzUser not found, passed user_id=r\z User is not a member of team_id=)rMr~r)rrrrrr_skiptaker)rototalrr total_pagesra)rrXrrr rrZrbrqrrrrfrolowerrrdr)rrZr rMr~rrrrrNrXrwhere_conditions user_objectuser_object_correct_typevalid_sort_columnsorder_byro total_countrrs rU list_team_v2rst9>}oNO  ,66:J:V:VV#++ 1H !D(*&-#"!* & .=*+  - 0 0 B B N N '*!O!K  #B7)!LM $5#P{7M7M7O#P ?+/1I1O1O*P Y ' 066 6*1 Y '#CG9!MN  AH700    _ 4JZ--/0 ""44>>  "hv(> ? E&((::@@GW@XXK!LI-.K;@6$//#6R"  ] #B7)!LM  > Y 7saA*G5-+G G G C G5(G,)-G5G.G5(G0? G5 G G))G5.G50G5z /team/listc JKddlm}t||s(tdddj |j i|'td dt jji|jjjd d i d{}g}|rK|D]E}|js|jD]'}d |vs|d |d |k(s|j|)Gn|}|Dcgc]}|j} }t|| |d{} g} |D]}g} | D]-} | j|jk(s| j| /|jj jd|jid{} | jt#di|j%| |d| j/d|Y|t0j2jk(r| Dcgc]}|j4|} }| S| Dcgc]}|j4|k(r|} }| S7cc}w7"7#t&$rT}dj |j|j%t)|}t+j,|Yd}~vd}~wwxYwcc}wcc}ww)a ``` curl --location --request GET 'http://0.0.0.0:4000/team/list' --header 'Authorization: Bearer sk-1234' ``` Parameters: - user_id: str - Optional. If passed will only return teams that the user_id is a member of. - organization_id: str - Optional. If passed will only return teams that belong to the organization_id. Pass 'default_organization' to get all teams without organization_id. rr,)rNrequested_user_idirzCOnly admin users can query all teams/other teams. Your user role={}rNrrT)r`rZrrMr)rhrzWInvalid team object for team_id: {}. team_object={}. Error: {} c$t|ddxsdS)Nr~)r)xs rUzlist_team.. s71lB+G+M2+MrW)rra)rrXr4rr$rrrrrbrrdrrerMrklitellm_verificationtokenr+rfrrrrsortr%DEFAULT_ORGANIZATIONr )rrZr rNrXrfiltered_responserr _team_idsrireturned_responses_team_membershipsrjrrteam_exceptions rU list_teamr? s,9 ++w ^ee%//  .EEKKL  #%%77AA !4 BH D&&"55F!V+"9-9"9-8)006 6%*;<$ :  ""sBJ#H,J#-J#J#J#J#,H/?J#H4+J#?AJ#H7J# -H9:8J#2JJJ#J(J#/J#7J#9 JA J J#J J#cLK |dz |z}|jjjd{}|jjj||ddid{}||fS7:7 #t$r&}t j d|gdfcYd}~Sd}~wwxYww)a5 Get paginated list of teams from team table Parameters: prisma_client: PrismaClient - The database client page_size: int - Number of teams per page page: int - Page number (1-based) Returns: Tuple[List[LiteLLM_TeamTable], int] - (list of teams, total count) rBNr~r)rrrz.[Non-Blocking] Error getting paginated teams: r)rbrrrdrrr)rXrrrrrors rUget_paginated_teamsr s qI%),,>>DDFF $&&88BBIlE-BC  k!! G  &&(J17Jz ": .&0&%9Jz " %GrWz/team/filter/uiFr)rrinclude_in_schema responsesz$Team alias in the request parameters2zNumber of items per pagec@Kddlm}|tdddi |dz |z}i}|r|d d |d <|r|d d |d <|jjj |||d did{}|sgS|S7 #t $r} tddt| d} ~ wwxYww)a [PROXY-ADMIN ONLY] Filter teams based on partial match of team_id or team_alias with pagination. Args: user_id (Optional[str]): Partial user ID to search for user_email (Optional[str]): Partial email to search for page (int): Page number for pagination (starts at 1) page_size (int): Number of items per page (max 100) user_api_key_dict (UserAPIKeyAuth): User authentication information Returns: List[LiteLLM_SpendLogs]: Paginated list of matching user records rr,NrrrrrBrrrMr~rrrzError searching teams: )rrXrrbrrdrr) rMr~rrrNrXrrrors rU ui_view_teamsr  sJ9W>O4PQQ!XqI% #%+ Y ' &%. \ * $&&88BB"( C  I   X6McRSfX4VWWXsGBA A4'A2(A4/B0A41B2A44 B=BBBrH new_modelsc|j}|*t|dk(rtjjg}n |j}t t ||z}|S)z8 Add new models to a team's allowed model list. r)rrr&r rrCrp)rHr current_modelsupdated_modelss rUadd_new_models_to_teamr5 sV __N"s>':a'?+<<BBC!#nz9:;N rWz/team/model/addcTKddlm}|tdddi|jjj d|j i d{}|td dd |j itdi|j}|jtjjk7rt|| std ddit||j}|jjj!d|j id|id{}|S77w)a  Add models to a team's allowed model list. Only proxy admin or team admin can add models. Parameters: - team_id: str - Required. The team to add models to - models: List[str] - Required. List of models to add to the team Example Request: ``` curl --location 'http://0.0.0.0:4000/team/model/add' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data '{ "team_id": "team-1234", "models": ["gpt-4", "claude-2"] }' ``` rr,NrrrrrMrrrrGr5Only proxy admin or team admin can modify team models)rHr rrra)rrXrrbrrrMrrfrr rrr9rrr)rlrrNrXrrHr r_s rUteam_model_addrF sA:9W>O4PQQ#%%77CC$,,'DH>t||nMN  !98#6#6#89H ##'7'C'C'I'II#/( TU  ,X$++VN&));;BB$,,'x.HCL =4s%A D(D$C D(D&D(&D(z/team/model/deletecKddlm}|tdddi|jjj d|j i d{}|td dd |j itdi|j}|jtjjk7rt|| std ddi|jxsg}|Dcgc]}||jvs|}}|jjjd|j id|id{} | S7cc}w7 w)a Remove models from a team's allowed model list. Only proxy admin or team admin can remove models. Parameters: - team_id: str - Required. The team to remove models from - models: List[str] - Required. List of models to remove from the team Example Request: ``` curl --location 'http://0.0.0.0:4000/team/model/delete' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data '{ "team_id": "team-1234", "models": ["gpt-4"] }' ``` rr,NrrrrrMrrrrGrrrrra)rrXrrbrrrMrrfrr rrr9rr) rlrrNrXrrHr rr r_s rUteam_model_deleter sg:9W>O4PQQ#%%77CC$,,'DH>t||nMN  !98#6#6#89H ##'7'C'C'I'II#/( TU  __*N"0HA1DKK3GaHNH'));;BB$,,'x.HCL G8Is7A ED;BE"D=6D=::E4E5E=Ez/team/permissions_listc rKddlm}m}m}|t dddit |||d|dd d{}|t d dd |it di|j}t|d rt|jtjjk7rMt||s@t|j|s)t dddj!d|ji|j"t%j&|_t)||j"t%j*S7w)z4 Get the team member permissions for a team rrXrrNrrrrFTrarrerrGrTrrIrJ)rMteam_member_permissionsall_available_permissionsra)rrXrrrr6rrfrKrr rrr9rVrMr$rr?default_team_member_permissionsrI)get_all_available_team_member_permissions)rMrNrXrrrrs rUrr sc W>O4PQQ.#-+ :7)DE  +L->-I-I-KL !;/  ' '+;+G+G+M+M M#/:L #&../ ell&&..  008 & F F H 1 , 1 I I"<"f"f"h Ws1D7D4DD7z/team/permissions_updatec Kddlm}m}m}|t dddit |j ||d|dd d{}|t d dd |j itdi|j}t|d rt|jtjjk7rMt||s@t|j |s)t dddj!d|j i|j"j$j'd|j id|j(id{}|S77 w)z7 Update the team member permissions for a team rrNrrrrFTrarrerrGrTrrIrJrMrrra)rrXrrrr6rMrrfrKrr rrr9rVr$rbrrr) rlrrNrXrrrrr_s rUupdate_team_member_permissionsr st W>O4PQQ. #-+ :4<<.IJ  +L->-I-I-KL !;/  ' '+;+G+G+M+M M#/:L #&../ ell&&..  '));;BB$,,'')E)E FCL WLs";EEC=E;E<EEz/team/daily/activity)rr start_dateend_daterexclude_team_idsc :Kddlm} m} m} | 't ddt j ji|r|jdnd} d} |r|r|jdnd} t|st|j| d| |j| d d{}|(t d dd j|ji| |j} n3| D].}||jvst d dd j|ii}| rdt| i|d<| j j"j%|d{}|Dcic]}|j&d|j(i}}t+| dd| || |||||| d{S77Kcc}w7 w)a Get daily activity for specific teams or all teams. Args: team_ids (Optional[str]): Comma-separated list of team IDs to filter by. If not provided, returns data for all teams. start_date (Optional[str]): Start date for the activity period (YYYY-MM-DD). end_date (Optional[str]): End date for the activity period (YYYY-MM-DD). model (Optional[str]): Filter by model name. api_key (Optional[str]): Filter by API key. page (int): Page number for pagination. page_size (int): Number of items per page. exclude_team_ids (Optional[str]): Comma-separated list of team IDs to exclude. Returns: SpendAnalyticsPaginatedResponse: Paginated response containing daily activity data. rrNrrr,FT)rZrXuser_id_upsertrrbrrdrzUser= {} not foundzGUser does not belong to Team= {}. Call `/user/info` to see user's teamsr\rMrr~litellm_dailyteamspend) rXrentity_id_field entity_identity_metadata_fieldexclude_entity_idsrrrrrr)rrXrrrrrrsplitr<r7rZrbr$rorCrbrrdrMr~r=)rYrrrrrrrrNrXrr team_ids_listexclude_team_ids_listrrMwhere_condition team_aliasestteam_alias_metadatas rUget_team_daily_activityr.c s> .EEKKL  ,4HNN3'M15+;  " "3 '  1 2)%--' 1.??/    1889J9R9RS   %OOM))//1'$'#%n%u%u '& )O&*D,?%@ "&));;EEFL:F45 L!,,//$#+!10    W H sJBFFAF*AFFF F/F F FFFr7)rrB)__doc__rrrrrrtypingrrrrr r r fastapir r rrrrpydanticrrRlitellm._loggingrr|rrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3rr4r5r6r7$litellm.proxy.auth.user_api_key_authr8/litellm.proxy.management_endpoints.common_utilsr9r:r;r<;litellm.proxy.management_endpoints.tag_management_endpointsr=8litellm.proxy.management_helpers.object_permission_utilsr>>litellm.proxy.management_helpers.team_member_permission_checksr?&litellm.proxy.management_helpers.utilsr@rAlitellm.proxy.utilsrBrCrDlitellm.routerrE>litellm.types.proxy.management_endpoints.common_daily_activityrF7litellm.types.proxy.management_endpoints.team_endpointsrGrHrIrJrKrLrouterrboolrVrkrfloatrrpostrrrrr*r"r2r4rErMrVr\rrfrjrzrrrrrrrrQueryr|rrrintrrrr#r rrrrrr.rarWrUrDs<   '@@@NN1!!!!!!!!!D C  " 4PT+/9?G} !2&  11 2&&&& &  &R,!,%,, ,  ,`  +,-$   )00A(B(. t) T+ T+T+&T+! T+ T+n! !sm! !& ! " ! c] !H L)c]0[ [+D[RX[ [|+,GDU