hg5"ddlZddlmZddlmZmZmZmZmZddlZddl Z ddl Z ddl m Z ddl mZejdZ e j!ej"ded ej"d ed ej"d ed ej"ded ej"ded ej"ded ej"ded ej"ddddej"dddej"ddej(d d!gd d"#ej*d$ej,d%eed&eed'eed(eed)eed*eed+eed,eded-d.efd/Ze j!ej"d0ed1 ej"d2ed3 ej"d4ed5 ej"d6ed7 ej"ded8 ej"ded9 ej"d ed: ej"d;ed< ej"d=ed> ej*d$ej,d?eed@eedAeedBeed+eed(eed'eedCeedDeefdEZe j!ej"dFedG ej"dHedI ej*d$ej,dJeedKeefdLZdMeedNeefdOZdPedQedNeeeeffdRZdSeeeefdTeedMedNeeeeffdUZdSeeeefdNdfdVZdWeeefdNeeeffdXZ dSeeeefdYedNe!eeffdZZ"e j!d[\ej"d]dd^_ej"d`dabej"dcdddej"dedfbej*d$ej,dQedgeedhedMeef diZ#y)jN)datetime)LiteralOptionalListDictAny)Table)KeysManagementClientcy)z,Manage API keys for the LiteLLM proxy serverNr b/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/client/cli/commands/keys.pykeysr s rz--pagezPage number for pagination)typehelpz--sizezNumber of items per pagez --user-idzFilter keys by user IDz --team-idzFilter keys by team IDz--organization-idzFilter keys by organization IDz --key-hashzFilter by specific key hashz --key-aliaszFilter by key aliasz--return-full-objectTzReturn the full key object)is_flagdefaultrz--include-team-keysz!Include team keys in the response)rrz--format output_formattablejsonzOutput format (table or json))rrrctxpagesizeuser_idteam_idorganization_idkey_hash key_aliasinclude_team_keys)rrreturn_full_objectc t|jd|jd} | j|||||||| | } t| tsJ| dk(rt j | yt jdt| jdgd| jd d td } | jd d| jdd| jdd| jdd| jdd| jdgD]}| jt|jddt|jddt|jddt|jddt|jddt j| y)zList all API keysbase_urlapi_key) rrrrrrrr!r rdatazShowing rz keys out of total_countrzAPI KeystitlezKey HashcyanstyleAliasgreenUser IDmagentazTeam IDyellowSpendredtokenrrrspendN)r objlist isinstancedictrich print_jsonprintlengetr add_columnadd_rowstr)rrrrrrrrr rr!clientresponserkeys rr8r8s@"#''*"5swwy7I JF{{  '-+ H h %% % X& Xc(,,vr":;#a&!:;;<    ( ( \!**"8"8!9:E 2*J OO ,kkm## 2 JJqzzD 1kkm 2 sdCB=7C 0C>=CC;!C66C;>G*.G% 0F:G%>G" G%!G""G%%G*z--keysz*Comma-separated list of API keys to deletez --key-aliasesz-Comma-separated list of key aliases to deleter key_aliasescZt|jd|jd}|r-|jdDcgc]}|jc}nd}|r-|jdDcgc]}|jc}nd} |j ||}t j |ycc}wcc}w#tjj$r} tjd| jjd  | jj} t j | tj"#tj$rAtj| jj d Ytj"wxYwd} ~ wwxYw) zDelete API keys by key or aliasr#r$rLN)rrdr%rMTrN)r r7rPrQdeleter;r<rWrXrYrTrZrDr[rrSr\r]) rrrdrCk keys_lista aliases_listrDrbrcs rrfrfs: "#''*"5swwy7I JF8!AGGI>SWL ==i\=J X& 5>    ( ( \!**"8"8!9:E 2*J OO ,kkm## 2 JJqzzD 1kkm 2 sGB4-B9 )B>>F*.F% 0E:F%>F" F%!F""F%%F* created_sincereturnc|sy d|vrtj|dStj|dS#t$r0tjd|ddtj wxYw) z1Parse and validate the created_since date filter.N_z%Y-%m-%d_%H:%Mz%Y-%m-%dzError: Invalid date format 'z%'. Use YYYY-MM-DD_HH:MM or YYYY-MM-DDTrN)rstrptime ValueErrorrTrZr])rks r_parse_created_since_filterrqsq  - $$]4DE E$$]J? ?  1-@eflpqkkms 559A. source_clientsource_base_urlc^tjd|g}d}d} |jd||}t|tsJd|j dg}|s |S|j |tjd|d t|d t||kr |S|dz }) z5Fetch all keys from source instance using pagination.z"Fetching keys from source server: dT)r!rrz$Expected dict response from list APIrz Fetched page : z keys)rTrZr8r9r:r?extendr>)rrrs source_keysr page_sizesource_response page_keyss r_fetch_all_keys_with_paginationr}s JJ3O3DEFK DI ',,4V_,`/40X2XX0#''3    9% ]4&3y>*:%@A y>I %    ! rrycreated_since_dtc|s|Sg}|D]}|jd}|st|ts(d|vr&tj|j dd}ntj|}|j r|j d}||k\s|j|tjdt|dt|d ||S) z/Filter keys by created_since date if specified. created_atTZ+00:00N)tzinfoz Filtered z keys to z keys created since ) r?r9rBr fromisoformatreplacerappendrTrZr>)ryr~rk filtered_keysrEkey_created_atkey_dts r_filter_keys_by_created_sincers M. .#..(%33N4J4J3PX4YZF%33NCF==#^^4^8F--!((-" JJ3{+,Ic-6H5II]^k]lmn rc Htjdtd}|jdd|jdd|jd d |D]}|j d d }|rJt |t r:d |vr6tj|jdd}|jd}|jt |j dd t |j dd t |tj|y)z?Display a table of keys that would be imported in dry-run mode.z --- DRY RUN MODE ---zKeys that would be importedr(z Key Aliasr.r+r/r0Createdr*rr5rrrz%Y-%m-%d %H:%MrrN)rTrZr r@r?r9rBrrrstrftimerAr;r=)ryrrErdts r_display_dry_run_tablers JJ'( 5 6E [0 Yi0 Yf-WW\2. *c**$!// 0B0B30QRB!#-=!>J c#''+r23SB9O5PRUV`Rab JJurrEcLi}dD]}|j|s||||<|S)z:Prepare key data for import by extracting relevant fields.)rFrGr6rrrrIrJ)r?)rE import_datafields r_prepare_key_import_datars6Kj 775>!$UK j r dest_clientc d}d}|D]g} t|}|jd i|}tjd||dz }|j dd}tjd|i||fS#t $rG}|dz }|j dd}tjd|dt |d Yd }~d }~wwxYw) z>Import each key to the destination instance and return counts.rzGenerated key: rurzN/Au✓ Imported key: u✗ Failed to import key rwTrNNr )rrVrTrZr? ExceptionrB) ryrimported_count failed_countrErrDrrbs r_import_keys_to_destinationrsNL T237K,{++:k:H JJ 3 4 a N U3I JJ+I;7 8& < ''  T A L U3I JJ29+RAxHd S S TsA$A55 C>=CCimport)namez--source-base-urlz?Base URL of the source LiteLLM proxy server to import keys from)requiredrz--source-api-keyz/API key for authentication to the source server)rz --dry-runz6Show what would be imported without actually importingz--created-sincezVOnly import keys created after this date/time (format: YYYY-MM-DD_HH:MM or YYYY-MM-DD)source_api_keydry_runct|}t||}t|jd|jd} t||}|r t |||}|st j dyt j dt|d|r t|yt||\} } t j dt j d| t j d | t j d t|y#tjj$r} t j d | jjd  | jj} t!j"| t j(#tj$$rAt j | jj&d Yt j(wxYwd} ~ wt*$r<} t j dt-| d t j(d} ~ wwxYw)z-Import API keys from another LiteLLM instancer#r$z!No keys found in source instance.NzFound z keys in source instance.z Import completed:z Successfully imported: z Failed to import: z Total keys processed: rMTrNr%zError: )rqr r7r}rrTrZr>rrrWrXrYrDr[rr;r<rSr\r]rrB) rrsrrrkr~rrrryrrrbrcs r import_keysr+s3=A).IM&swwz':CGGI?@ ),89 -c+.>-?@A    ( ( \!**"8"8!9:E 2*J OO ,kkm## 2 JJqzzD 1kkm 2  WSVH%40kkmsT2D2/D"A5DI5.G?$0F(G?(>G<&G?;G<<G?? I 7II)$rrtypingrrrrrrTr;rW rich.tabler rr groupcommandoptionintrBChoice pass_contextContextboolr8floatrVrfrqr}rrrtuplerrr rrrs 55 (    hS'CD hS'AB k*BC k*BC !2RS l+HI m#,AB $dDGcd #T8[\  w' (  ( . . 3-. 3-.c] . c] . c] .sm.}..?+..]eCJTDDCE$.b js)QR k*OP ie*LM l+cd m#,DE k*MN k*MN m#,QR js)]^( ( SM(c]( E? ( sm ( } (c](c](}( SM(_SOOFeNQS(V hS'ST oC.]^ Xc]#_U$ x} (AS  3GZ]bfgkloqtltgubv8d38n%9A(9K\_ $sCx.:T#s(^(<. $sCx. T#s(^ (d38n%(4H( 38_(:8 $-n 'XY k4.fg t0 0),0>Fsm0VZ0kstwkx0hZ0r