uPj$;UdZddlZddlmZmZddlmZddlmZmZddl m Z ddl m Z m Z mZddlmZerdd lmZddlm Zdd lmZd Zeed <Gd de e Zy)zT VerificationToken repository for database operations on LiteLLM_VerificationToken. N)MappingSequence)datetime) TYPE_CHECKINGFinalLiteLLM_VerificationToken)BaseRepositoryDbRecordrecord_to_dict) TableActions) LiteLLM_DeletedVerificationToken) PrismaClient) aliasesconfig permissionsmetadata model_spendmodel_max_budgetrouter_settings budget_limitslitellm_budget_table_JSON_ENCODED_TOKEN_FIELDSc7eZdZdZed;fd ZededfdZededfdZede e fdZ d e d zde d zfd Z dd ed.ed zded zded zded zded zdeed zdeeefd zdeeefd zded zd eeefd zd!ed zd"ed zd#ed zd$eed zd%eed zd&eeefd zd/ed zd)ed zd*eed zde d zf*d0Z d?d ed1ed zd2ed zd3ed zde d zf d4Z d e deeeffd5Z!d ed6ede d zfd7Z"d ede d zfd8Z#d@d ed.ed zde d zfd9Z$d@d ed.ed zde d zfd:Z%xZ&S)AVerificationTokenRepositoryz@Repository for verification token (API key) database operations.returnct|}|SN)super prisma_client)selfr __class__s w/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/repositories/verification_token_repository.pyr z)VerificationTokenRepository.prisma_client.s-2W-B PrismaVerificationTokencB|jjjSr)r dblitellm_verificationtokenr!s r#tablez!VerificationTokenRepository.table3s!!$$>>>r$PrismaDeletedVerificationTokencB|jjjSr)r r' litellm_deletedverificationtokenr)s r# deleted_tablez)VerificationTokenRepository.deleted_table7s!!$$EEEr$ctSrrr)s r# model_classz'VerificationTokenRepository.model_class;s((r$recordNc L|yt|jDcic]5\}}||tvr%t|trt j |n|7}}}|jd}|jd||ni|d|i}tj|Scc}}w)z7Convert a database record to a VerificationToken model.Norganization_idorg_id) r itemsr isinstancestrjsonloadsgetr model_validate)r!r1fieldvaluedecodedr3datas r# _to_modelz%VerificationTokenRepository._to_model?s >!/v 6 < < > u 1K(KPZ[`bePf4::e$lq q  ")->!?{{8$0O4K 7G7X7 )77== s:B tokenid_fieldc@Kt|||d{S7wr)r find_by_id)r!rArBr"s r#rDz&VerificationTokenRepository.find_by_idQsW'x8888s  key_aliascK|jjd|id{}|r|j|dSy7w)zFind a token by key alias.rEwhereNr)r* find_manyr@)r!rErecordss r# find_by_aliasz)VerificationTokenRepository.find_by_aliasTsFBF**BVBV^ikt]uBVBv>'!*- -=ws"AAAuser_idc|K|jjd|id{}|j|S7w)z$Find all tokens belonging to a user.rLrGNr*rI_to_model_list)r!rLrJs r#find_by_user_idz+VerificationTokenRepository.find_by_user_id[>BF**BVBV^gip]qBVBrBF**BVBV^jlv]wBVBxC!?C? C)"C##%C)C% C) C?C'C?!C?#C)%C)'C?)C</C2 0C<7C?c|jd}dD]}|j|d|jdd}|||d<gd}|D]"}||vstj||||<$|S)a@Build archive data with only columns present in LiteLLM_DeletedVerificationToken. Serializes JSON columns to strings (the archive table stores them as JSON columns the same way the live table does) and maps ``org_id`` onto the ``organization_id`` column so the foreign key is preserved. T) exclude_none)object_permissionrrNr4r3)rrrrrrr) model_dumppopr8rq)r!rAr?r<r4rrs r#rz/VerificationTokenRepository._build_archive_dataNs*/)9)9t)9)LS "E HHUD ! "40  &,D" # ! 6E}"jje5U  6 r$spendcJK|j|d|idd{S7w)zUpdate token spend.rrAr}Nr~)r!rArs r# update_spendz(VerificationTokenRepository.update_spendks&[[%(87[KKKKs #!#cnK|j|dtjidd{S7w)z!Update the last_active timestamp. last_activerAr}N)r~rr_)r!rAs r#update_last_activez.VerificationTokenRepository.update_last_activeos.[[8I(JU\[]]]]s ,535c\Kddi}|||d<|j||dd{S7w)zBlock a token.r]TNrprAr}rr!rArpr?s r# block_tokenz'VerificationTokenRepository.block_tokenss;*3T):  !!+D [[w[???? #,*,c\Kddi}|||d<|j||dd{S7w)zUnblock a token.r]FNrprAr}rrs r# unblock_tokenz)VerificationTokenRepository.unblock_tokenzs;*3U);  !!+D [[w[????r)rr)rA)NNNNNNNNNNNNNNNNNNNNNNNN)NNNNNNNNNNNNNNNNNNN)NNNr)'__name__ __module__ __qualname____doc__propertyr r r*r.typer r0r r@r7rDrKlistrPrUrXr`floatrdictrobjectintrxr{boolrrrrrrr __classcell__)r"s@r#rr+sJ ?|$=>??F|,LMFF)T";<))>4>4MPT4T>$9c9S9G`cgGg9S5NQU5U,S,T:S5T, ,S,T:S5T, ,3,4@Y;Z, ,$/H*I, $ $#'#'#')-.2""#!%,004 $ $&*37+/37!!%+/-1 $5AA*A: A DL A D AS D Ac3h$&AV $t+AtAtA*A$JA #TzA#v+&-A :!A":#A$t%A&!%S D 0'A(S D()A*S&[)D0+A,d -A.$J/A0"Dj1A2s)d*3A4:5A6 c6k 7AL $ $#'#'#')-.2""#!%,004 $ $&*37+/37!!%+/-1 $58'8'*8': 8' DL 8' D 8'S D 8'c3h$&8'V $t+8't8't8'*8'$J8' #Tz8'#v+&-8' :!8'":#8'$t%8'&!%S D 0'8'(S D()8'*S&[)D0+8',d -8'.$J/8'0"Dj18'2s)d*38'4:58'6 #78'z"&# $#'#'#')-.2,004 $ $&*37+/37#+/-1+@@@@$J@@* @@ : @@ DL @@D@@S D @@c3h$&@@V $t+@@ #Tz@@#v+&-@@:@@:@@t@@ !%S D 0!@@"S D(#@@$S&[)D0%@@&'@@("Dj)@@*s)d*+@@, #T )-@@J"&)-)- $J $J   $J  #T ) 6)BtCQWKGX:LLEL>WZ^>^L^c^6ORV6V^@s@d @NgjnNn@@@#*@PilpPp@r$r)rr8collections.abcrrrtypingrr!litellm.models.verification_tokenr $litellm.repositories.base_repositoryr r r %litellm.repositories.prisma_protocolsr prisma.modelsrr+r%litellm.proxy.utilsrr__annotations__rr$r#rsc -' ?1 %E T@.1J"KT@r$