tPj_UddlZddlZddlmZmZddlmZmZddlZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZmZddlmZdZeed <Gd d Zy) N)datetimetimezone)AnyFinal)verbose_proxy_logger)uuid) GenerateKeyRequestGenerateKeyResponse KeyRequestLiteLLM_AuditLogsLitellm_EntityTypeLiteLLM_VerificationTokenLitellmTableNamesRegenerateKeyRequestUpdateKeyRequestUserAPIKeyAuth)_hash_token_if_neededzlitellm/"LITELLM_PREFIX_STORED_VIRTUAL_KEYSc eZdZe ddededededzfdZe dde de de dededzf dZ e dde dzde dedededzf d Ze dded ee dedededzf d Zedd ed ededzfdZe ddededededzfdZed edefdZed ee fdZededzdedzfdZedefdZedefdZedededzfdZy)KeyManagementEventHooksNdataresponseuser_api_key_dictlitellm_changed_bycdKddlm}m}m}ddlm}|j dur. tj|jdd{|r|jd} tj|t!t#t%j&t)j*t,j.|||||j0t2j4|j6xsd d | d  tj9|j:xsd |j6|j<|j>d{y7#t$r!}tjd|Yd}~+d}~wwxYw75#t$r }tjd|Yd}~yd}~wwxYww)z Hook that runs after a successful /key/generate request Handles the following: - Sending Email with Key Details - Storing Audit Logs for key generation - Storing Generated Key in DB rcreate_audit_log_for_updateget_audit_log_changed_byis_audit_logging_enabledlitellm_proxy_admin_nameT exclude_noneNz$Failed to send key created email: %srrr!created id updated_at changed_bychanged_by_api_key table_name object_idactionupdated_values before_value request_data virtual-key-) secret_name secret_tokenteam_idz1Failed to store virtual key in secret manager: %s) +litellm.proxy.management_helpers.audit_logsrrrlitellm.proxy.proxy_serverr!send_invite_emailr_send_key_created_email model_dump Exceptionrwarningmodel_dump_jsonasyncio create_taskr strruuid4rnowrutcapi_keyrKEY_TABLE_NAMEtoken_id$_store_virtual_key_in_secret_manager key_aliaskeyr6) rrrrrrrr!e_updated_valuess s/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/hooks/key_management_event_hooks.pyasync_key_generated_hookz0KeyManagementEventHooks.async_key_generated_hooks  H  ! !T ) X-EEhFYFYgkFYFlmmm $ %%-%=%=4%=%PO   +!2tzz|,#+<< #=#;/A.?5M$ ,=+D+D#4#C#C"*"3"3"9r('6%)" * a)NN NNP X=N=N) rrOrrrrrrr!initial_secret_namerYr6rKs rMasync_key_rotated_hookz.KeyManagementEventHooks.async_key_rotated_hooks  H   1 1 = f-=-G-G-r\ZjZpZpYqKr#)1););)dt~~)dQd$))G#"))99d!K-SS(;$3%-\\# T %))F# T)AA!,,$,?#3#=#=B   $ %*:*@*@*L   +!2tzz|,#+<< #=#;/A.?5M$ ,=+B+B#4#C#C"2"8"8('/'?'?T'?'R%5%E%ESW%E%X" +M %/ f$,,-acdee f   T ( ()OQR S S TsxH%BG &G'G 4G86G67G8;C H%G G3G.(H%.G33H%6G88 H"HH%H""H%keys_being_deletedcKddlm}m}m}ddlm}|r|j |D]} | j| jd} tj|tttjtj t"j$|||||jt&j(| jdd | t*j-| d{y7w) z Post /key/delete processing hook Handles the following: - Storing Audit Logs for key deletion rrr NTr"r$deletedz{}r'r1)rd)r7rrrr8r!keysr]r>r?r@r rArrBrrCrrDrrFr(_delete_virtual_keys_from_secret_manager) rrdrrrrrrr!rJ_key_rows rMasync_key_deleted_hookz.KeyManagementEventHooks.async_key_deleted_hooks  H $ %$))*?) 99$..D.A##/%6"4::<0'/||HLL'A'?3E2C9Q( 0A/F/F'8'G'G&)ii#,+/)1&  2&NNbtNuuusC8D:D;Dr4r5r6cKtjtjjdurddlm}t tj |rttjdd}ttjdd}tj|d{}tjd|||tj jtj|||||d{yyyy7_7 w) z Store a virtual key in the secret manager Args: secret_name: Name of the virtual key secret_token: Value of the virtual key (example: sk-1234) NTrBaseSecretManagertags descriptionz6Creating secret with %s and tags=%s and description=%s)r4ro secret_valuernoptional_params)litellm_key_management_settingsstore_virtual_keys+litellm.secret_managers.base_secret_managerrm isinstancesecret_manager_clientr_r#_get_secret_manager_optional_paramsrdebugasync_write_secret_get_secret_name)r4r5r6rmrnrorqs rMrHz  F$3+#5%H8Z6   8\ ***,H(D)'||,EF mT *} & &G     & &'XZacf g scC0B2C0CCCA,C02 B>;C0=B>>C0C C- C(#C0(C--C0c ddlm}tjj |}t |dkDry ddlm}d|jdgvryy#t $rY'wxYw) z Check if email sending is enabled via v2 enterprise loggers or v0 alerting config. Returns True only if email is actually configured, preventing any email processing when the user has not opted in. rBaseEmailLogger callback_typeT)general_settingsemailalertingF) >litellm_enterprise.enterprise_callbacks.send_emails.base_emailrrrlogging_callback_managerget_custom_loggers_for_typelenrr8rr)rinitialized_email_loggersrs rM_is_email_sending_enabledz1KeyManagementEventHooks._is_email_sending_enableds|  07/O/O/k/k-0l0 %,-12 @ &**:r: :   s4A A A cKtjstjdyddlm}m} ddlm}ddl m }tjj|}t|dkDr||jdd d t j"d |jd d |jd d|jdd|jdd|jdd|jdd }|D])}t%||s|j'|d{+y d|jdgvrddlm}|d t j"d |jd d |jd d|jdd|jdd|jdd|jdd }t/j0|j2j5|yy7#t($rYwxYww)z Send key created email if email sending is enabled. This method is non-blocking - it will return silently if email is not configured, and will log warnings instead of raising exceptions on failure. z5Email sending not enabled, skipping key created emailNr)rproxy_logging_objr)SendKeyCreatedEmailEventrrJr% key_createdzAPI Key Createdr]spend max_budgetuser_idr6 Default TeamrI virtual_keyevent event_group event_messager]rrrr6rI)send_key_created_email_eventrr) WebhookEvent) rrrr]rrrr6rI) webhook_event)rrrryr8rrrr9litellm_enterprise.types.enterprise_callbacks.send_emailsrrrrrrrr KEYrvsend_key_created_emailrlitellm.proxy._typesrr?r@slack_alerting_instance&send_key_created_or_user_invited_email) rrrrrrr email_loggerrs rMr:z/KeyManagementEventHooks._send_key_created_emails'@@B & &'^ _ R    07/O/O/k/k-0l0 %,-10 ( UB 7' 2 6 6"3",,w3",,w4'||L#>$LLD9$LLNC≪= %>L!,@*AA9>B %22 &**:r: : 9 #.22/ll7B/ll7C0#<< c: Y5 Y?",,{D9 E   !99``"'a  ;   sB3H CG<G<'G:(G<.C H :G<< HH HH r[cKtjstjdy ddlm} ddlm}||jdd d tjd |jd d |jd d|jdd|jdd|jdd|jd| }tjj|}t|dkDr/|D])}t!||s|j#|d{+yy#t $rtjdYywxYw#t $rtjdYywxYw7Pw)z Send key rotated email if email sending is enabled. This method is non-blocking - it will return silently if email is not configured, and will log warnings instead of raising exceptions on failure. z5Email sending not enabled, skipping key rotated emailNrrz  .A  & &'e f    & &'c d  4sW+E3D)E CE3 E3 E1!E3)E E3 E  E3 E.+E3-E..E3)N)__name__ __module__ __qualname__ staticmethodr r rrArNrrrVrrrcr listrrjrHr`r{rhrxboolrr:rarrMrrs *. :a :a%:a*:a $J :a:ax *. ,,,,* ,  $J ,,\ *. B"T)B3B&B* B  $J BBH *. 0v0v !:;0v0v* 0v  $J 0v0vdSVadgkak> #  t >fcfcff ! !:;!!F+t+ ++Zt:HHHT55#PT*55rr)r?rSrrtypingrrrrlitellm._loggingr litellm._uuidrrr r r r r rrrrrlitellm.proxy.utilsrr__annotations__rrrrMrsG '1   6-7"E6]]r