tPjk6 UdZddlmZmZddlmZmZmZmZm Z m Z ddl Z ddl m Z ddlmZddlmZddlmZmZdd lmZdd lmZdd lmZmZdd lmZdd lmZm Z eZ!ee"d<GddZ#e!jIdeegdgeefde de dedefdZ%e!jMdeegdgeefde de defdZ'e!jMdeegdge eddeefde de de(defd Z)e!jMd!eegdge edd"eefde de d#e(defd$Z*e!jWd%eegdgeddeefde de de(defd&Z, d-d'e d(e d)e(dzd*e fd+Z-e!j]d%eegdgeddeefde de de de(def d,Z/y).z2 CRUD endpoints for storing reusable credentials. )Finalcast) APIRouterDepends HTTPExceptionPathRequestResponseN)verbose_proxy_logger)CredentialAccessor)_get_masked_values)CommonProxyErrorsUserAPIKeyAuth)user_api_key_auth)encrypt_value_helper)handle_exception_on_proxyjsonify_object)CredentialsRepository)CreateCredentialItemCredentialItemrouterc2eZdZeddededzdefdZy)CredentialHelperUtilsN credentialnew_encryption_keyreturnci}|jxsijD]\}}t||||<t|j||j xsiS)z$>&8"44(),-' %/$>$>%!)( )@?R   +&&q)'**+s3B%"A4;A4"B%#A44 B"= BB""B%z /credentials/by_model/{model_id}z'The model ID to look up credentials forrIcrKddlm} | tdd|j|}| tdd|j |}| tddt |d d }t |jd ||i }|S#t$r%} tj| t| d} ~ wwxYww) rZr)r6Nr8zLLM router not foundr:r=r>rcrdz -credential-r) rGr6rrJrKr r model_namerPr rQr) r2r3rIr4r6rRr masked_credential_valuesrrWs r&get_credential_by_modelrls6+  C8NO O!00: =C8IJ J#-#H#H#R  $C8IJ J*<  !+  +$//0 XJG6  +&&q)'**+s)B7A;BB7 B4 B//B44B7z#/credentials/{credential_name:path}cvKddlm} |'tddtjj it |j|d{tjDcgc]}|j|k7s|c}t_ ddd S7@cc}w#t$r}t|cYd}~Sd}~wwxYww) rZrr7Nr8r9r:TzCredential deleted successfullyrD) rGr7rrrHr%rdelete_by_namer\r]rrPr)r2r3rr4r7credrWs r&delete_credentialrqs9 ,  !2!I!I!O!OP $M2AA/RRR5<4K4K"wDtOcOcgvOv4"w,MNN S#x ,(++,s_B9ABBB'B<BBB9BB B6! B1+B6,B91B66B9 db_credential updated_patchrrct|j|j|j}tj ||}|jr|j|_|jrH|jj Dcic]\}}|| }}}|jj||jr; d|jvri|_|jj|j|Scc}}w)z( Update a credential in the DB. )rr!r r!)rrr!r rr'r"update)rrrsrmerged_credentialrTkvencrypted_paramss r&update_db_credentialrzs .%55%55'99  #8"Q"Q# ++,@,P,P)--4H4Z4Z4`4`4b"cDAq1a4"c"c++223CD++$ $5$E$E E02  -))001E1U1UV #ds C=c Kddlm} |'tddtjj it |}|j|d{}| tddt||}td t|j} |j|i| d |ji d{|j} d} tj D]} | j|k(s| } n| t#| j$xsi} |j$r| j'|j$t#| j(xsi}|j(r|j'|j(t+| | | }| |k7r7tj Dcgc]}|j|k7s|c}t_t-j.|gd ddS77%cc}w#t0$r}t3|d}~wwxYww)rZrrnNr8r9r:r=zCredential not found in DB.r?rArBrTzCredential updated successfullyrD)rGr7rrrHr%r find_by_namerzrrrLupdate_by_namerNrr\r]dictr rur!rr rOrPr)r2r3rrr4r7credentials_repositoryrrrvcredential_object_jsonifiednew_nameexisting_in_memoryrpin_memory_valuesin_memory_infoupdated_in_memorycrWs r&update_credentialr"s91+  !2!I!I!O!OP )>m(L%;%H%H%YY  C8UV V#7 z#R-1 0A0L0L0N!O. #%33 -/774   ,;;48++ D##6%)"   )&*+=+O+O+USU&V ++ '' (D(DE$();)K)K)Qr$RN))%%j&@&@A'5 ("2 .(  ?*6=6M6M*vQRQbQbfuQu1*v'  1 13D2E F,MNNQ Z <+w +'**+smHAG(GA(G(;G <4G(1B,G(G#2G#6&G(HG( G(#G(( H1 G<<HHr))0__doc__typingrrfastapirrrrr r r\litellm._loggingr .litellm.litellm_core_utils.credential_accessorr *litellm.litellm_core_utils.litellm_loggingr litellm.proxy._typesrr$litellm.proxy.auth.user_api_key_authr0litellm.proxy.common_utils.encrypt_decrypt_utilsrlitellm.proxy.utilsrr+litellm.repositories.credentials_repositoryrlitellm.types.utilsrrr__annotations__rpostrXgetr_r.rhrldeleterqrzpatchrr/r(r&rs ON1MIBBQIMD   "+,- ! " )00A(B ?+ ?+?+%?+& ?+  ?+D+,- ! " )00A(B, ,,&,  ,,1+,- ! "!   1lm(/0A(B + +++& +  +>&+,- ! "!  *ST(/0A(B !+ !+!+!+& !+  !+H)+,- ! " 1lm(/0A(B , ,,,& ,  ,:&*$!$!$d $ $N)+,- ! " 1lm(/0A(B =+ =+=+=+ =+ & =+  =+r(