o ưi4 @sUddlmZmZmZmZmZddlZddlmZddlm Z ddl m Z ddl m Z mZmZgaeeed<  ddeedeed eefd d Zd ee d dfddZdeeeefdededed eeeff ddZdS))AnyDictListOptionalcastN)Router)verbose_proxy_logger)initialize_callbacks_on_proxy) Guardrail GuardrailItemGuardrailItemSpecall_guardrailsconfig_file_path llm_routercCsLddlm}g}|D]}|jtt|||d}|r||q t|ddS)NrIN_MEMORY_GUARDRAIL_HANDLER) guardrailrr)guardrail_list)+litellm.proxy.guardrails.guardrail_registryrZinitialize_guardrailrr append_populate_router_guardrail_list)r rrrrrZinitialized_guardrailr_/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/guardrails/init_guardrails.pyinit_guardrails_v2s  rrreturnc Csddlm}ddlm}ddlm}|durtddSg}|D]K}|d}|d}|d i}d} |r=|j |} t |d rF| nt |} ||pNd | d d | d d | d| dd| |d} | | q!||_tdt|ddS)z Populate the router's guardrail_list from initialized guardrails. This enables load balancing across multiple guardrail deployments with the same guardrail_name. rr)r)GuardrailTypedDictNz>Router not initialized yet, skipping guardrail_list population guardrail_idguardrail_namelitellm_params model_dumprmodeapi_keyapi_base)rr!r"r#)rrcallbackidz%Populated router guardrail_list with z guardrails)rrZlitellm.proxy.proxy_serverrZlitellm.types.routerrrdebuggetZ guardrail_id_to_custom_guardrailhasattrrdictrrlen) rrrrZrouter_guardrail_listrrrrr$ params_dictZrouter_guardrailrrrr+sJ          rguardrails_config premium_userlitellm_settingsc Cs8ztd||D]!} |D]\}}td i|d|i}t||tj|<qq t}i} tD]8} t| j t| j | | j | j durl| j D]} | tj vr\|| | jdurk| dkrkddi| d<qPq4t|} t| dkrt| |||| dtjWSty} z tdt| | d} ~ ww) Nzvalidating guardrails passed rTZpresidio logging_onlyr)valuer-rr.callback_specific_paramsz error initializing guardrails {}r)rr&itemsr r rlitellmZguardrail_name_config_mapsetrZ default_onupdate callback_args callbacksaddr/listr*r Exception exceptionformatstr)r,r-rr.itemkvZguardrail_itemZdefault_on_callbacksr1rr$Zdefault_on_callbacks_listerrrinitialize_guardrailsfsP             rB)NN)typingrrrrrr3rZlitellm._loggingrZ)litellm.proxy.common_utils.callback_utilsr Zlitellm.types.guardrailsr r r r __annotations__r=rrboolr)rBrrrrs:    ;