uPj<UdZddlZddlmZddlmZddlmZmZddlm Z ddl m Z m Z ddl mZmZmZdd lmZer dd lmZdd lmZe ee Zeed <d Zeed<Gddee Zy)z? Team repository for database operations on LiteLLM_TeamTable. N)Mapping)datetime) TYPE_CHECKINGFinal) TypeAdapter)LiteLLM_TeamTableMember)BaseRepositoryDbRecordrecord_to_dict) TableActions)Prisma)models_MEMBERS_WITH_ROLES_ADAPTER)metadata model_spendmodel_max_budgetrouter_settings budget_limitsmembers_with_roles_JSON_ENCODED_TEAM_FIELDSc$ eZdZdZededfdZededfdZedee fdZ de d zde d zfd Z d d d e deed zfdZd5d e de de d zffd Zde de d zfdZde dee fdZde dee fdZde dee fdZ d6d e de d zde d zdee d zdee d zdee efd zdee efd zded zded zdee d zded zd ed zd!ed zd"e d zd#e d zde f d$Z d7d e de d zde d zdee d zdee d zdee efd zdee efd zded zded zdee d zded zd ed zd!ed zd"e d zd%ed zd#e d zde d zf"d&Z d8d e d'e d zd(e d zd)e d zde d zf d*Zd+e dee effd,Zd e d-ede d zfd.Z d e de de d zfd/Z!d e de de d zfd0Z"d e de de d zfd1Z#d e de de d zfd2Z$d e dee de d zfd3Z%d e dee de d zfd4Z&xZ'S)9TeamRepositoryz(Repository for team database operations.returnzprisma_models.LiteLLM_TeamTablecB|jjjSN) prisma_clientdblitellm_teamtableselfs i/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/repositories/team_repository.pytablezTeamRepository.table&s!!$$666z&prisma_models.LiteLLM_DeletedTeamTablecB|jjjSr)rrlitellm_deletedteamtabler s r" deleted_tablezTeamRepository.deleted_table*s!!$$===r$ctSr)rr s r" model_classzTeamRepository.model_class.s  r$recordNc |yt|jDcic]5\}}||tvr%t|trt j |n|7}}}tj|Scc}}w)z*Convert a database record to a Team model.N) r itemsr isinstancestrjsonloadsrmodel_validate)r!r*fieldvaluedatas r" _to_modelzTeamRepository._to_model2sw >!/v 6 < < > u 1J(JzZ_adOe4::e$kp p  !//55  s:A2txrteam_idcK|jd|d{}|sy|dd}t|trtj|n|}|sgSt j |S7Ow)a(Return the team's members_with_roles. The caller must already hold ``TEAM_ADVISORY_LOCK_SQL`` for this team_id on ``tx`` before calling this. ``None`` when the team row is gone, which is only possible under that lock if a delete committed before this read, as opposed to ``[]`` for a team that simply has no members. A plain read is enough here because the advisory lock, not a row lock, is what serializes this against a concurrent writer: ``SELECT ... FOR UPDATE`` would additionally take a row lock on ``LiteLLM_TeamTable``, and the access-group endpoints lock an access group and then a team row, so a team-row-first lock here can deadlock with them. The advisory lock cannot, since those endpoints never take it. zESELECT members_with_roles FROM "LiteLLM_TeamTable" WHERE team_id = $1Nrr) query_rawr-r.r/r0rvalidate_python)r!r6r7rows raw_valueparseds r"get_members_with_roles_lockedz,TeamRepository.get_members_with_roles_locked>spLL S   7#78 1;Is1K 9-QZI*::6BB sA*A(AA*id_fieldc@Kt|||d{S7wr)super find_by_id)r!r7r? __class__s r"rBzTeamRepository.find_by_idYsW'::::s  team_aliascK|jjd|id{}|r|j|dSy7w)zFind a team by alias.rDwhereNr)r# find_manyr5)r!rDrecordss r" find_by_aliaszTeamRepository.find_by_alias\sD#zz33<:T3UU >>'!*- -Vs"AAAorganization_idc|K|jjd|id{}|j|S7w)z,Find all teams belonging to an organization.rKrFNr#rH_to_model_list)r!rKrIs r"find_by_organization_idz&TeamRepository.find_by_organization_idcs=#zz33;Lo:^3__""7++`s "<:<user_idcK|jjdd|iid{}|j|S7w)z&Find all teams where user is a member.membershasrFNrMr!rPrIs r"find_by_memberzTeamRepository.find_by_memberhsA#zz339ugFV:W3XX""7++Y $><>cK|jjdd|iid{}|j|S7w)z&Find all teams where user is an admin.adminsrSrFNrMrTs r" find_by_adminzTeamRepository.find_by_adminmsA#zz338eWEU:V3WW""7++XrVrXrRrr max_budget soft_budgetrmax_parallel_requests tpm_limit rpm_limitbudget_durationobject_permission_idcXKd|i}|||d<|||d<|||d<|||d<|tj||d<|tj||d<|||d <| | |d <| | |d <| | |d <| | |d <| | |d<|||d<|||d<|j|d{S7w)zCreate a new team.r7NrDrKrXrRrrrZr[rr\r]r^r_r`)r/dumpscreate)r!r7rDrKrXrRrrrZr[rr\r]r^r_r`r4s r" create_teamzTeamRepository.create_teamrs&+4W)=  !!+D   &&5D" #  #DN  %DO  ))-4F)GD% &  #zz(3D   !!+D   ""-D   #DN ,,AD( )  )D   )D   &&5D" #  ++?D' ([[&&&&sB!B*#B($B*blockedchKi}|||d<|||d<|||d<|||d<|tj||d<|tj||d<|||d<| | |d <| | |d <| | |d <| | |d <| | |d <|||d<|||d<|||d<|j||dd{S7w)zUpdate a team.NrDrKrXrRrrrZr[rr\r]r^r_rer`r7r?)r/rbupdate)r!r7rDrKrXrRrrrZr[rr\r]r^r_rer`r4s r" update_teamzTeamRepository.update_teams,(*,  !!+D   &&5D" #  #DN  %DO  ))-4F)GD% &  #zz(3D   !!+D   ""-D   #DN ,,AD( )  )D   )D   &&5D" #  %DO  ++?D' ([[$[CCCCsB)B2+B0,B2 deleted_bydeleted_by_api_keylitellm_changed_bycK|j|d{}|y|j|}||d<||d<||d<tj|d<|jj j 4d{}|jj|d{|jjd|id{dddd{|S77b7@77#1d{7swY|SxYww) zDelete a team and archive it to the deleted teams table. Uses a transaction to ensure atomicity of the archive-then-delete operation. Nrjrkrl deleted_at)r4r7rF) rB_build_archive_datarutcnowrrr6r&rcrdelete)r!r7rjrkrlteam archive_datar6s r" delete_teamzTeamRepository.delete_teams!OOG44 <"66t< %/ \"-? )*-? )*%-__%6 \"%%((++- J J--44,4G G G&&--Y4H-I I I J J 5 J G I J J J J s|C?CA&C?>C!?C? C)"C##%C)C% C) C?C'C?!C?#C)%C)'C?)C</C2 0C<7C?rrcd|ji}|j|j|d<|j|j|d<|j|j|d<|j|d<|j |d<|j r>tj|j Dcgc]}|jc}|d<|jr"tj|j|d<|j|j|d <|j|j|d <|j |jnd |d <|j|d <|j|j|d<|j |j |d<|j"|j"|d<|j$|j$|d<|j&|j&|d<|j(|d<|j*r"tj|j*|d<|j,r"tj|j,|d<|j."tj|j.|d<|j0xsg|d<|j2xsg|d<|j4xsg|d<|j6|j6|d<|j8|d<|Scc}w)zQBuild archive data dict with only columns that exist in LiteLLM_DeletedTeamTable.r7rDrKr`rXrRrrrZr[gspendrr\r]r^r_budget_reset_atrerrrteam_member_permissionsaccess_group_idspoliciesmodel_idallow_team_guardrail_config)r7rDrKr`rXrRrr/rb model_dumprrZr[rvrr\r]r^r_rwrerrrrxryrzr{r|)r!rrr4ms r"roz"TeamRepository._build_archive_datas*3T\\)B ?? &!%D     +&*&:&:D" #  $ $ 0+/+D+DD' (X,,Y  " ")-TMdMd4eQ\\^4e)fD% & ==#zz$--8D  ?? &!%D     '"&"2"2D &*jj&< #W X  % % 1,0,F,FD( ) >> % $D  >> % $D     +&*&:&:D" #    +&*&:&:D" #,,Y   "&**T-=-=">D   '+zz$2G2G'HD# $    +&*jj1E1E&FD" #*.*F*F*L" &'#'#8#8#>B  ==.BZ == $#}}D .2.N.N *+ A5fs'KrvcJK|j|d|idd{S7w)zUpdate team spend.rvr7rgN)rh)r!r7rvs r" update_spendzTeamRepository.update_spend$s&[[7E*:Y[OOOOs #!#cK|j|dd{sy|jjd|idd|iid{}|j|S7A7w)z9Add a member to a team using atomic array push operation.r7rgNrRpushrGr4existsr#rhr5r!r7rPr*s r" add_memberzTeamRepository.add_member(sn[[9[==="jj//g&fg./0  ~~f%%> !AA+AAAAcK|j|d{}|y|jDcgc] }||k7s | }}|j|d|idd{S7Dcc}w7 w)zRemove a member from a team. Note: Prisma doesn't support atomic array removal, so we use a read-modify-write pattern here. For high-concurrency scenarios, consider using raw SQL with array_remove(). NrRr7rg)rBrRrh)r!r7rPrrr~rRs r" remove_memberzTeamRepository.remove_member3si!OOG44 <%)\\BQ'\!BB[[9g*>[SSS 5CS0A%AA% AAA%A#A%A%cK|j|dd{sy|jjd|idd|iid{}|j|S7A7w)z9Add an admin to a team using atomic array push operation.r7rgNrXrrrrs r" add_adminzTeamRepository.add_adminAsn[[9[==="jj//g&VW-.0  ~~f%%> rcK|j|d{}|y|jDcgc] }||k7s | }}|j|d|idd{S7Dcc}w7 w)zRemove an admin from a team. Note: Prisma doesn't support atomic array removal, so we use a read-modify-write pattern here. For high-concurrency scenarios, consider using raw SQL with array_remove(). NrXr7rg)rBrXrh)r!r7rPrrarXs r" remove_adminzTeamRepository.remove_adminLsi!OOG44 <$(KK@q1<@@[[8V* rcK|j|d{}|y|jDcgc] }||vs| }}|j|d|idd{S7Ccc}w7 w)zRemove models from a team's allowed models list. Note: Prisma doesn't support atomic array removal, so we use a read-modify-write pattern here. For high-concurrency scenarios, consider using raw SQL with array_remove(). Nrr7rg)rBrrh)r!r7rrrr~current_modelss r" remove_modelszTeamRepository.remove_modelsesi!OOG44 <,0KK Kq1F? K K[[8^*Dy[YYY 5!LYs0A$AA$ AAA$A"A$A$)r7)NNNNNNNNNNNNNN)NNNNNNNNNNNNNNN)NNN)(__name__ __module__ __qualname____doc__propertyr r#r'typerr)r r5r.listr r>rBrJrOrUrYrobjectfloatintrdboolrirtdictrorrrrrrr __classcell__)rCs@r"rr#s 2 7|$EF77>|,TU>>!T"34!! 64 64E4L 6ChCCQUV\Q]`dQdC6;;s;K\_cKc;c6G$6N,S,TJ[E\, ,C,D9J4K, ,3,48I3J,"&&*#'$(:>04#'$(#',0 $ $&*+/!1'1'$J1't 1' S D 1' cT! 1'$CK0471'#v+&-1'DL1'T\1'S D 1' #Tz1':1':1't1' "Dj!1'" #1'l"&&*#'$(:>04#'$(#',0 $ $&*#+/#4D4D$J4Dt 4D S D 4D cT! 4D$CK0474D#v+&-4DDL4DT\4DS D 4D #Tz4D:4D:4Dt4D !4D""Dj#4D$ T !%4Dr"&)-)- $J $J   $J  T ! 6,(9,d3;>O,\P#PeP@QTX@XP & &c &>ORV>V & T3 T TARUYAY T &s &S &=NQU=U & R# R R@QTX@X R & &T#Y &CTW[C[ & Z3 ZS ZFWZ^F^ Zr$r)rr/collections.abcrrtypingrrpydanticrlitellm.models.teamrr $litellm.repositories.base_repositoryr r r %litellm.repositories.prisma_protocolsr prismarr prisma_modelsrr__annotations__rrr$r"rso #' 9 ?.%0f%>U>$5NZ^$56NZr$