tPj" UddlZ ddlmZmZddlmZmZmZmZm Z m Z ddl Z ddl m Z m Z mZmZmZddlmZddlmZddlmZddldd lmZmZdd lmZdd lmZdd lm Z m!Z!dd l"m#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)ddl*m+Z+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:ddl;mZ>ddl?m@Z@er$ddlAmBZBddlCmDZEddlCmFZGddlCmHZIddlCmJZKdd lCmLZMe ZNeeOd!<Gd"d#eZPGd$d%eZQGd&d'eZRGd(d)eZSGd*d+eZTGd,d-eZUGd.d/eZVGd0d1eZWGd2d3eZXGd4d5eZYGd6d7eZZGd8d9eZ[e d:e0d;d'fd<Z\e d:e2d;d)fd=Z\e d:e4d;d+fd>Z\e d:e6d;d-fd?Z\e d:e8d;d/fd@Z\e d:e:d;d%fdAZ\e d:edTgYddddddZd[de ef d\e^dzd]e^dzd^e^dzd_e^dzd`e^dzdaeodbeodce^dzdEe_fddZpdVeldFe.dzd;e^dzfdeZqeNjdfdTge egesUe efdgedEe_fdhZtdieae^e]fdjePd;eae^e]ffdkZueNjdldTge egesdmndDe^dVevdEee_e effdoZweNjdpdTge egeyesUe efdVezdEe_fdqZ{eNjdrdTge egeyesUe jddste jddute efdve^dzdwe^dzdEe_fdxZ}eNjdydTge egesUe efdDe^dEe_fdzZ~eNjdydTge eg{e efdVedEe_fd|ZeNjd}dTge egeUe,e efdVed~edEe_d;efdZde^dFe.d;eLfdZeNjddTge egeUe,e efdVedEe_fdZeNjddTge eg{e efdVedEe_fdZdedDe^dFe.d;eeLeffdZy)N)MappingSequence) TYPE_CHECKING AnnotatedFinalProtocolcastoverload) APIRouterDepends HTTPExceptionRequeststatus) TypeAdapter)verbose_proxy_logger)uuid)*)can_user_call_modelget_user_object)user_api_key_auth)get_budget_reset_time) new_budget update_budget)get_daily_activity)_set_object_metadata_field_user_has_admin_view)&handle_update_object_permission_common prepare_object_permission_upsert)get_new_internal_user_defaultsmanagement_endpoint_wrapper) PrismaClient)BudgetRepository)ObjectPermissionRepository)OrganizationRepository) OrganizationMembershipRepository)TeamRepository)UserRepository)VerificationTokenRepository)SpendAnalyticsPaginatedResponse)_update_dictionary) TracebackType)LiteLLM_BudgetTable)LiteLLM_ObjectPermissionTable)LiteLLM_OrganizationMembership)LiteLLM_OrganizationTable)LiteLLM_UserTablerouterc(eZdZededzfdZy)_ObjectPermissionRowreturnNcyNselfs ~/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/management_endpoints/organization_endpoints.pyobject_permission_idz)_ObjectPermissionRow.object_permission_idUs25)__name__ __module__ __qualname__propertystrr;r7r<r:r3r3Ts 5cDj55r<r3c&eZdZdeeefddfdZy)_UserTableClientwherer4zPrismaUserTable | Nonec Kywr6r7r9rDs r: find_uniquez_UserTableClient.find_uniqueZN)r=r>r?rrAobjectrGr7r<r:rCrCYs]wsF{';]@X]r<rCc&eZdZdeeefddfdZy)_BudgetTableClientdatar4PrismaBudgetTablec Kywr6r7r9rMs r:createz_BudgetTableClient.create^rHrINr=r>r?rrArJrQr7r<r:rLrL]sRf!5R:MRr<rLc&eZdZdeeefddfdZy)_ObjectPermissionTableClientrMr4PrismaObjectPermissionTablec Kywr6r7rPs r:rQz#_ObjectPermissionTableClient.createbrHrINrRr7r<r:rTrTas\f!5\:W\r<rTc eZdZ ddeeefdeeefdzddfdZ ddeeefdeeefdzddfd Z ddeeefdzdeeefdzdd fd Z ddeeefdeeefdeeefdzddfd Z ddeeefdeeefdzddfd Z y)_OrganizationTableClientNrMincluder4PrismaOrganizationTablec Kywr6r7)r9rMrYs r:rQz_OrganizationTableClient.createfs $'rIrDzPrismaOrganizationTable | Nonec Kywr6r7r9rDrYs r:rGz$_OrganizationTableClient.find_uniquej +.rIz!Sequence[PrismaOrganizationTable]c Kywr6r7r]s r: find_manyz"_OrganizationTableClient.find_manyns /2rIc Kywr6r7)r9rDrMrYs r:updatez_OrganizationTableClient.updatets %(rIc Kywr6r7r]s r:deletez_OrganizationTableClient.delete{r^rIr6)NN) r=r>r?rrArJrQrGr`rbrdr7r<r:rXrXesDQU(CK((3:3;3G$3N( "( SW/S&[)/4;CK4H44O/ )/ .2/32sF{#d*2f%,2 - 204 (sF{#(c6k"(f%, ( # (SW/S&[)/4;CK4H44O/ )/r<rXceZdZdeeefddfdZ ddeeefdeeefdzddfd Z ddeeefdzdd fd Zdeeefdeeefddfd Z deeefddfd Z deeefde fdZ y)"_OrganizationMembershipTableClientrMr4PrismaOrganizationMembershipc Kywr6r7rPs r:rQz)_OrganizationMembershipTableClient.createrHrINrDrYz#PrismaOrganizationMembership | Nonec Kywr6r7r]s r:rGz._OrganizationMembershipTableClient.find_uniques 03rIz&Sequence[PrismaOrganizationMembership]c Kywr6r7rFs r:r`z,_OrganizationMembershipTableClient.find_manys 36rIc Kywr6r7r9rDrMs r:rbz)_OrganizationMembershipTableClient.updates ),rIc Kywr6r7rFs r:rdz)_OrganizationMembershipTableClient.deleterHrIc Kywr6r7rFs r: delete_manyz._OrganizationMembershipTableClient.delete_manyrHrIr6) r=r>r?rrArJrQrGr`rbrdintror7r<r:rfrfs]f!5]:X]SW4S&[)44;CK4H44O4 .4 487S&[)D07 17-S&[)-18f1E- '-f'#v+"6e;`eHwsF{';HHr<rfc&eZdZdeeefdefdZy)_TeamTableClientrDr4c Kywr6r7rFs r:roz_TeamTableClient.delete_manyrHrINr=r>r?rrArJrpror7r<r:rrrrHwsF{';HHr<rrc&eZdZdeeefdefdZy)_VerificationTokenTableClientrDr4c Kywr6r7rFs r:roz)_VerificationTokenTableClient.delete_manyrHrINrtr7r<r:rwrwrur<rwc4eZdZdeeefdeeefddfdZy)_ObjectPermissionTxClientrDrMr4rUc Kywr6r7rls r:upsertz _ObjectPermissionTxClient.upserts (+rIN)r=r>r?rrArJr|r7r<r:rzrzs1,S&[),18f1E, &,r<rzc4eZdZdeeefdeeefddfdZy)_BudgetTxClientrDrMr4zPrismaBudgetTable | Nonec Kywr6r7rls r:rbz_BudgetTxClient.updaterHrIN)r=r>r?rrArJrbr7r<r:r~r~s'v'#v+"6vgc6k>RvWqvr<r~cBeZdZeddZeddZeddZy)_TransactionTablescyr6r7r8s r:litellm_objectpermissiontablez0_TransactionTables.litellm_objectpermissiontablesLOr<cyr6r7r8s r:litellm_budgettablez&_TransactionTables.litellm_budgettables8;r<cyr6r7r8s r:litellm_organizationtablez,_TransactionTables.litellm_organizationtablesGJr<N)r4rz)r4r~)r4rX)r=r>r?r@rrrr7r<r:rrs, OO ;; JJr<rc DeZdZd dZdeedzdedzdddedzfdZy) _TransactionManagerr4c Kywr6r7r8s r: __aenter__z_TransactionManager.__aenter__rHrIexc_typeN exc_value tracebackzTracebackType | Nonec Kywr6r7)r9rrrs r: __aexit__z_TransactionManager.__aexit__s rI)r4r)r=r>r?rtype BaseExceptionboolrr7r<r:rrsC;}%,!4'*   r<r repositoryr4cyr6r7rs r:_tablersBEr<cyr6r7rs r:rrsVYr<cyr6r7rs r:rrsNQr<cyr6r7rs r:rrsber<cyr6r7rs r:rr>Ar<cyr6r7rs r:rrrr<cyr6r7rs r:rrsX[r<c|j}|Sr6)table)r prisma_tables r:rrs#-"2"2L r<organization_iduser_api_key_dict prisma_clientcKt|ry|jsttjdddlm}m}t|j||d|d{}|ttjd|jxsgD];}|j|k(s|jtjjk(s;yttjd7w)z Verify the caller is either a proxy admin or an org admin of the given organization. Raises HTTPException(403) if the caller does not have access. Nz+You do not have access to this organization status_codedetailr)proxy_logging_objuser_api_key_cacheF)user_idrruser_id_upsertr)rrr rHTTP_403_FORBIDDENlitellm.proxy.proxy_serverrrrorganization_membershipsr user_roleLitellmUserRoles ORG_ADMINvalue)rrrrr caller_userms r:_verify_org_accessrs-.  $ $11@  Q.!))#-+  K11@   1 1 7R   /AKKCSC]C]CcCc4c  --< #sAC(C&AC( 'C(C(_STR_OBJECT_DICT_ADAPTER budget_id_BUDGET_SETTABLE_FIELDSorganization_aliasmodels_ORG_COLUMN_FIELDSbudget_updates updated_byc|jd}d|vr dt|tr t|ndini}i||d|iS)a Budget-row columns to write. ``budget_reset_at`` tracks any sent ``budget_duration``: recomputed for a new duration, cleared alongside a ``None`` duration so no stale reset timestamp survives. Other sent fields (including a ``None`` clear) are written as-is. budget_durationbudget_reset_at)rNr)get isinstancerAr)rrrrecomputed_reset_ats r:build_budget_write_datarse ,//0ABO  . JTUdfiJj%oFpt  On N 3 N\: NNr<raw_datac|j}d|vrW|jdi}|rCtjj }|j D]\}}||vs ||||<|S)z Transform organization update request to handle UI payload format. The UI sends nested budget data in 'litellm_budget_table', but our model expects flat budget fields at the top level. litellm_budget_table)copypopr, model_fieldskeysitems)rtransformed_data budget_data budget_fieldskeyrs r:=handle_nested_budget_structure_in_organization_update_requestr)s'mmo!11-112H"M #6#C#C#H#H#JM)//1 2 U-'E,=,1$S) 2 r<z/organization/newzorganization management)tags dependenciesresponse_modelrMc  Kddlm}m}m}|'t ddt j ji|j|jtjk7rt ddd|ji|'t ddt jji|jJtj|jr|jdkrt d dd |ji|jJtj|jr|jdkrt d dd |jid}|j ^ t#t%|j'd |j i d{}|#t(j+|j-}|j0 t2j4j7}t8j;|j=d}|j?D cic] \} } | |vs | | } } } t2j+| } t8j;|jA| j=d} t#tC|jEi| |j xs||j xs|dd{}|j0|_tG||d{} tI|jJdk(rnNtI|jJdk(rt d ddi|jJD]}tM|||d{tOdi|j=d||j xs||j xs|d}tPD](}tS||dtU||tS||*t8j;|jA|j=d}tWjXdt<j[|dt#t]|jEi|ddid{}|S7#t.$rYwxYwcc} } w777&7(w)a Allow orgs to own teams Set org level budgets + model access. Only admins can create orgs. # Parameters - organization_alias: *str* - The name of the organization. - models: *List* - The models the organization has access to. - budget_id: *Optional[str]* - The id for a budget (tpm/rpm/max budget) for the organization. ### IF NO BUDGET ID - CREATE ONE WITH THESE PARAMS ### - max_budget: *Optional[float]* - Max budget for org - tpm_limit: *Optional[int]* - Max tpm limit for org - rpm_limit: *Optional[int]* - Max rpm limit for org - model_rpm_limit: *Optional[Dict[str, int]]* - The RPM (Requests Per Minute) limit per model for this organization. - model_tpm_limit: *Optional[Dict[str, int]]* - The TPM (Tokens Per Minute) limit per model for this organization. - max_parallel_requests: *Optional[int]* - [Not Implemented Yet] Max parallel requests for org - soft_budget: *Optional[float]* - [Not Implemented Yet] Get a slack alert when this soft budget is reached. Don't block requests. - model_max_budget: *Optional[dict]* - Max budget for a specific model - budget_duration: *Optional[str]* - Frequency of reseting org budget - metadata: *Optional[dict]* - Metadata for organization, store information for organization. Example metadata - {"extra_info": "some info"} - blocked: *bool* - Flag indicating if the org is blocked or not - will stop all calls from keys with this org_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) - object_permission: Optional[LiteLLM_ObjectPermissionBase] - organization-specific object permission. Example - {"vector_stores": ["vector_store_1", "vector_store_2"]}. IF null or {} then no object permission. - allowed_models: Optional[List[str]] - List of models the organization is allowed to access. If not set, defaults to the models field. Case 1: Create new org **without** a budget_id ```bash curl --location 'http://0.0.0.0:4000/organization/new' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data '{ "organization_alias": "my-secret-org", "models": ["model1", "model2"], "max_budget": 100 }' ``` Case 2: Create new org **with** a budget_id ```bash curl --location 'http://0.0.0.0:4000/organization/new' --header 'Authorization: Bearer sk-1234' --header 'Content-Type: application/json' --data '{ "organization_alias": "my-secret-org", "models": ["model1", "model2"], "budget_id": "428eeaa8-f3ac-4e85-a8fb-7dc8d7aa8689" }' ``` r)litellm_proxy_admin_name llm_routerrNerrorrz,Only admins can create orgs. Your role is = ;max_budget must be a non-negative finite number. Received: O>]>]>c>c4dee "DMM$//,Jdoo`aNaZ[_[j[jZklm   #T]]4;K;K-LPTP`P`cdPd[\`\l\l[mno  :>  , '-n].K'L'X'X "3";";<(Y("K&+<+K+KKLbLbLd+e( ~~ 3??DDF 5DDTYY\`YEab 0:0@0@0BY1a=FXq!tY Y/>>|L 4DD  ( (d)K L  &&6}&EFMM/77S;S/77S;S N  !**)? #)#   # #$) t{{ q {  fA%aJLde e e f8 ))) &1$,,H0H$,,H0H ; 4 % 1 &, dE* #;"J"J$$%5%:%:%:%MN#8$**EYbc*:de"#9-#HIPP " (. QH Oi"    Z #( f*sEQ?3QQ)Q|d{nd} | d| j>in| rddini}t2j5|jAi| |d|j i}|jBjE}|4d{}| H|jFjId| j>i| jJ| jJdd{| rrt2j5|jAtMtO| |j }|jPjSd |jTi|d{|jVjSd|i|dddd!"d{}dddd{|S#t$r}td dt|id}~wwxYw77cc} wcc} w77=777Y7K#1d{7swYSxYww)#aX Partial update of an organization (RESTful PATCH, RFC 7396 merge-patch semantics). A sent field is written and an omitted one is left untouched (presence is read from ``model_fields_set``). Clear tokens are per field: budget limits and ``metadata`` clear with ``null``, ``models`` with ``[]``, and ``object_permission`` with ``null`` (it merges when sent, so an empty ``{}`` is rejected). ``organization_alias`` is required and cannot be cleared. Validation failures return 422; the object-permission upsert, budget-row write, and org-row write are one transaction. rrNrrrrr0irr)validate_model_max_budgetrz4organization_alias cannot be cleared; it is requiredrz1models cannot be set to null; send [] to clear itTrzgobject_permission cannot be an empty object; send null to clear it, or a non-empty object to set grantsr1rrr2r3r,)new_object_permissionrJrr;r)rQrbrDrMrr:r=),rrr rrrrrrrrmodel_max_budget;litellm.proxy.management_endpoints.key_management_endpointsrMr?rAmodel_fields_setrrr,rrrr$rGrrrrr3rr;rdbtxrr|recorddictrrrbrr)rrMrrrMer7 field_valuespresent_fieldsrrorg_column_updatesobject_permission_clearedobject_permission_upsertobject_permission_writeorganization_write_data tx_managerrUbudget_write_datars r:update_organization_v2rbsg,9.EEKKL    (x   "DMM$//,Jdoo`aNaZ[_[j[jZklm   #T]]4;K;K-LPTP`P`cdPd[\`\l\l[mno     K %d&;&; <t4449P9P9XST  4(((T[[-@PQ   )$2H2H2S2Sae2S2fC   '+#  .44J=4Y-Z-f-f /2.g.(!(J?J[\]  3BB4??CTUL 11NESxEW\`wWwUL$77xNx73A a%UN`E`5,u% % a70:n0LJ + ,RT7 (;n'L'oQUQgQgkoQo  ! ! - /"&"8"8"C"CQU"C"V*C*X*X'    $ / !9!N!NO0I%t,r &>%M%M$$ $ ) /77  &.;-=-=-@-@-BJ  R # /2299-/G/\/\]6==6==:   '?'O'O,,T2I.ZkZsZs2t-uv( ((//"$=$G$GH&0  !# < < C C$o6( $tTR!D!  #  . O} KC#a&8IJ J K* (y a  2    #    . OsD R) Q BR)8Q59-R)&Q8'A R)2 Q;Q;R) RR%A1R)RA4R) R R)ARR A3R R  +R5R6R: R)RR) Q2Q--Q22R)8R); R)R) R RRR)R&R R&!R)z/organization/deletecKddlm}|'tddtjj i|j tjk7rtdddig}|jD]}tt|jd |i d{tt|jd |i d{tt|jd |i d{tt|j!d |id d d d  d{}|tddd|di|j#||S777e74w)zx Delete an organization # Parameters: - organization_ids: List[str] - The organization ids to delete. rrNrrrrz*Only proxy admins can delete organizationsrrTr:rDrYrNz Organization=z not found)rrr rrrrrrr rr&ror%r(r$rdappend)rMrr deleted_orgsr deleted_orgs r:delete_organizationrhs"9.EEKKL  ""&6&B&BBIJ  L00)^M23??GXZiFj?kkk5mDEQQ$o6R   0?@LLTegvSwLxxx"#9-#HIPP$o6 $tTRQ    =0A!LM  K(')* ' l  y sHBEE-EE-E4E52E'E(/EEEEz/organization/listz3Filter organizations by exact organization_id match)default descriptionz[Filter organizations by partial organization_alias match. Supports case-insensitive search.r& org_aliascKddlm}|tdddi|'tddtjj ii}|r||d <|r|d d |d <t |r7tt|j|r|ndd d d dd{}|Stt|jd|jid{}|Dcgc]}|j}}|r||r@||vrg}|S||d <tt|j|d d d dd{}|Sd|i|d <tt|j|d d d dd{}|Sg}|S77cc}w7M7w)a Get a list of organizations with optional filtering. Parameters: org_id: Optional[str] Filter organizations by exact organization_id match org_alias: Optional[str] Filter organizations by partial organization_alias match (case-insensitive) Example: ``` curl --location --request GET 'http://0.0.0.0:4000/organization/list?org_alias=my-org' --header 'Authorization: Bearer sk-1234' ``` Example with org_id: ``` curl --location --request GET 'http://0.0.0.0:4000/organization/list?org_id=123e4567-e89b-12d3-a456-426614174000' --header 'Authorization: Bearer sk-1234' ``` rrNrrNo db connectedrrr insensitive)containsmoderT)rr;r<rdrrr) rrr rrrrrr$r`r%rr) r&rkrrwhere_conditionsrorg_memberships membershipmembership_org_idss r:list_organizationrusF9W>O4PQQ.EEKKL  24 .4*+!!2 -. -. 6} EFPP&6"D-1dTRQ  V OK(..N}.]'^'h'h/778(i( " Sb$bJZ%?%?$b$b !33!H4 O1;A$%67%+,B=,Q%R%\%\.48'+%)!&]& H. O8<=O6P !23!'(>}(M!N!X!X*04#'!%"Y" OH OW " %c sTBE8E+9E8 E- E8E/'?E8&E4'9E8 E6! E8-E8/E86E8z/organization/infocHKddlm}|tdddit|||d{t t |j d |id d d d iid d d d{}|tddditj|j}|S7r7<w)z* Get the org specific information rrNrrrmrr1rTrYuser)rr;r<r,rdrNzOrganization not found) rrr rrr$rG$LiteLLM_OrganizationTableWithMembersrr)rrrrresponse_pydantic_objs r:info_organizationrz5s9W>O4PQQ '+#  ##9-#HIUU /2$(D !%  V  HW>V4WXX#G#V#VW_WjWjWl#m  3 s!)B"B7B"#B $;B" B")rrctKddlm}|tdddit|jdk(rtddd |ji|jD]}t ||| d{t t|jd d |jiid did{}|S7E7w)z8 DEPRECATED: Use GET /organization/info instead rrNrrrmrrz6Specify list of organization id's to query. Passed in=r1rrrTrd) rrr r organizationsrrr$r`)rMrrr&rs r:deprecated_info_organizationr}ds9W>O4PQQ 4  !#UVZVhVhUijk  $$  "/'    ##9-#HISS 4););"<='.TH O s$A,B8.B4/>B8-B6.B86B8z/organization/member_add http_requestc "K ddlm}|tdddit|j||d{t t |jd |ji d{}|td dd t|d dit|jtr |j}n |jg}g}g}|D]G}t||j| d{\} } |j| |j| It|j||S777G#t$r} t!j"d| t| trPt%t| dd| dt&j(t| ddt| dt*j,t| t$r| t%dt/| zt&j(t| ddt*j,d} ~ wwxYww)a [BETA] Add new members (either via user_email or user_id) to an organization If user doesn't exist, new user row will also be added to User Table Only proxy_admin or org_admin of organization, allowed to access this endpoint. # Parameters: - organization_id: str (required) - member: Union[List[Member], Member] (required) - role: Literal[LitellmUserRoles] (required) - user_id: Optional[str] - user_email: Optional[str] Note: Either user_id or user_email must be provided for each member. Example: ``` curl -X POST 'http://0.0.0.0:4000/organization/member_add' -H 'Authorization: Bearer sk-1234' -H 'Content-Type: application/json' -d '{ "organization_id": "45e3e396-ee08-4a61-a88e-16b3ce7e0849", "member": { "role": "internal_user", "user_id": "krrish247652@berri.ai" }, "max_budget_in_organization": 100.0 }' ``` The following is executed in this function: 1. Check if organization exists 2. Creates a new Internal User if the user_id or user_email is not found in LiteLLM_UserTable 3. Add Internal User to the `LiteLLM_OrganizationMembership` table rrNrrrmrr1rrrNr2)memberrr)r updated_users updated_organization_membershipsz'Error adding member to organization: %srzAuthentication Error()paramNoner)messagerrcodezAuthentication Error, )rrr rrrr$rGrrrr!add_member_to_organizationreOrganizationAddMemberResponserr exceptionProxyExceptionProxyErrorTypes auth_errorrHTTP_500_INTERNAL_SERVER_ERRORrA) rMr~rrr7r;rrr updated_userupdated_organization_membershiprXs r:organization_member_addrs1hG <  CBS8TU U ! 00/'   288N}8]1^1j1j$d&:&:;2k2 , ! % ,J7SWYjlpKqJrs  dkk4 (kkG{{mG8: ]_( UF1 $ 4 4+ /   . , 3 34S T U- 00'-M  Q , 0  &&'PRST a ' 8/DQCq-IJ$//a&1Q v/T/TU   > *G,s1v5 ++!Wf-66    s_H2D4D.7D4.D0/A;D4*D2+AD4-H.D40D42D44 H =C HH  H user_emailcKtddd|di} t|jjd|id{}||t j |j}|S7-#t$r|wxYww) zA Find a member if the user_email is in LiteLLM_UserTable rrz%Unique user not found for user_email=z]. Potential duplicate OR non-existent user_email in LiteLLM_UserTable. Use 'user_id' instead.rrrN)r r'rrGrr0rr)rrnot_unique_user_email_errorexisting_user_email_row existing_user_email_row_pydantics r:find_member_if_emailr s *7 .J.J.P.PP# #v 99 9-HIPP-#'<<+/+?+?0 "499-Q    * * 6?II^SQUQ[Q[Q]M^I/99A!1IRVRqRq!r NN_```$/)PTPoPop&7 9-HIPP-#'<<+/+?+?0 "9-Q  /55UVc5d.e.q.q)#||'+';';, ,T2/r/ ) % ) 0#DTEYEYDZZghlhthtgu!vw  9\8j8j ) 4 4$ 4 G9 .65o , .h 7 QRVR^R^Q__pqurFrFqGGIJKILM  "s" a  ) ( &&'RTUVsQ"A P9O 7P9O#AP9O&P92AO,2O)3O,7AP9P) CP9P, A.P9P/*P99P2:AP9P5AP9 P7 AP9Q" P9#P9&P9)O,, P&5,P!!P&&P9,P9/P92P95P97P99 QQQQ"z/organization/member_deletec$K ddlm}|'tddtjj it |j||d{|j;|j/t|j|d{}|j|_ tt|jd|j|jd i d{}|S77^7 #t$r}tj d ||d}~wwxYww) z. Delete a member from an organization rrNrrrr1rrrz+Error deleting member from organization: %s)rrr rrrrrrrrrr%rdrrr)rMrrrmember_to_deleterXs r:organization_member_deleters"<  !2!I!I!O!OP ! 00/'   ?? &4<<+?3GYf3g-g #2::DL(./OP]/^(_(f(f)#||'+';';,)g) #  % .h#  &&'TVWXs`DA C'C!5C'C#AC'C%C' D!C'#C'%C'' D 0DD  Drc BK d}d}d}|j8tt|jd|jid{}|F|j: t|j jd|jid{}|||jxsttj}t||j}|j|d d{} | tj| j}n|t!|d kDrt#d d d i|$tj|j}nO|$tj|j}n)t#dd d|jd|ji|%td|jd|jtt%|j'||j|j(dd{} t*j| j} || fS77#t $r }td|jd|d}~wwxYw77`#t $r}td|d|d|d}~wwxYww)z Add a member to an organization - Checks if member.user_id or member.user_email is in LiteLLM_UserTable - If not found, create a new user in LiteLLM_UserTable - Add user to organization in LiteLLM_OrganizationMembership NrrrzdPotential NON-Existent or Duplicate user email in DB: Error finding a unique instance of user_email=z in LiteLLM_UserTable.: )rrrw)rMrr rrzIMultiple users with this email found in db. Please use 'user_id' instead.rrNzUser not found for user_id=z and user_email=z2User does not exist in LiteLLM_UserTable. user_id=)rrrrzError adding member=z to organization=r)rrr'rGrrrr?rArrr insert_datar0rrrr r%rQrr) rrrrexisting_user_id_rowrrXrnew_user_defaults_returned_user_organization_membershiporganization_memberships r:rrsEa04 #"& >> %)/}0M)N)Z)Z &..1*[*$  'F,=,=,I 0>}0M0S0S0_0_'):):;1`1+' ',C,K"(.."EC 4EG'E!,,(  $1#<#>~?X?X?Z[ $ 0 ' (  !lm % 0+::;R;];];_`K ! -+::;O;Z;Z;\]K#>v~~>NN^_e_p_p^q!rs   DV^^DTTdekevevdwx  177WXe7f0g0n0n#2&..#[[1o1 +  *M)[)[ $ / / 1* 333y$ +  z|B|M|M{NNfghfij h6+  a/x7HHYY[\][^_``asJAI: I I:4I II AI:2I53D$I:I8*I:JI:I I2I--I22I:8I:: JJJJ)rcollections.abcrrtypingrrrrr r fastapir r r rrpydanticrlitellm._loggingr litellm._uuidrlitellm.proxy._typeslitellm.proxy.auth.auth_checksrr$litellm.proxy.auth.user_api_key_authr)litellm.proxy.common_utils.timezone_utilsr>litellm.proxy.management_endpoints.budget_management_endpointsrr8litellm.proxy.management_endpoints.common_daily_activityr/litellm.proxy.management_endpoints.common_utilsrr8litellm.proxy.management_helpers.object_permission_utilsrr&litellm.proxy.management_helpers.utilsrr litellm.proxy.utilsr!&litellm.repositories.budget_repositoryr"1litellm.repositories.object_permission_repositoryr#,litellm.repositories.organization_repositoryr$'litellm.repositories.table_repositoriesr%$litellm.repositories.team_repositoryr&$litellm.repositories.user_repositoryr'2litellm.repositories.verification_token_repositoryr(>litellm.types.proxy.management_endpoints.common_daily_activityr) litellm.utilsr*typesr+ prisma.modelsr,rNr-rUr.rgr/rZr0PrismaUserTabler1__annotations__r3rCrLrTrXrfrrrwrzr~rrrrJrAUserAPIKeyAuthrrWr frozensetrrrrrrpostNewOrganizationResponseNewOrganizationRequestr rrpr*rpatchrxrHr@OrganizationUpdateRequestV2rbrdr!DeleteOrganizationRequestrhQueryrurzOrganizationRequestr}rOrganizationMemberAddRequestrrrOrganizationMemberUpdateRequestrOrganizationMemberDeleteRequestr OrgMembertuplerr7r<r:rs  .FF 1"OBKX-CXOT??-#FSB 686 ^x^SS]8]/x/6II(IxIIHI,, whwKK( E'E,@E E Y1Y6TY Y Q-Q2LQ Q e7e<`e e A~A*<A A A~A*<A A [2[7V[ [   !''    " "   ))%) ) )X#.d3;.?"@%@!*+>+K+K+P+P+R!SWbVc!cc%';X&FGEGOGCK,@OcOV]^aci^iVjO& 0 # $+,-*  )00A(B} }%}  }@"2 # $  $(!+/(/0A(BDDjDd DDjD : D 4Z D  DD"DjD&D  DN $& 4Z, # $+,-7 )00A(Bn n%n  nbCK 3 #v+6( # $+,-7 NN %N!9J1K!KLNNb # $+,-<= )00A(B/ #/%/  /d # $+,-<=  'tAvw)GMMq)00A(B a $JaTza & a  aH # $+,-7  )00A(B&!&!%&!  &!R # $+,- )00A(B %  D # $+,-0   )00A(Bt &t t &t # t  t n,3,|,Pa,.! # $+,-6  )00A(BN )N%N Nb! # $+,-)00A(B) ))%)  )XRa RaRa Ra A AB Rar<