tPj  UddlmZmZmZddlZddlmZddlmZddlm Z ddl m Z m Z m Z gaee ed< ddeededzd edzfd Zd ee d dfd Zdeeee fdededed eee ff dZy))AnyFinalcastN)Router)verbose_proxy_logger)initialize_callbacks_on_proxy) Guardrail GuardrailItemGuardrailItemSpecall_guardrailsconfig_file_path llm_routerc.ddlm}g}|D]:} |jtt|||d}|s*|j|<t|y#t t f$r0}tjd|jd|Yd}~d}~wwxYw)NrIN_MEMORY_GUARDRAIL_HANDLERconfig) guardrailr rsourcez\Skipping guardrail '%s': invalid configuration, proxy is starting WITHOUT this guardrail: %sguardrail_name)guardrail_list) +litellm.proxy.guardrails.guardrail_registryrinitialize_guardrailrr ValueError TypeErrorrerrorgetappend_populate_router_guardrail_list)r r rrrrinitialized_guardrail init_errors m/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/guardrails/init_guardrails.pyinit_guardrails_v2r"s X-/N#9  $?$T$Ty)4!1% %U% ! !  ! !"7 8!9*$>BI&  & &n ./    s#AB$&BBrreturnc rddlm}ddlm}ddlm}|t jdyg}|D]}|jd}|jd}|jd i}d} |r|jj|} t|d r|jn t|} ||xsd | jd d | jd d | jd| jdd| |} |j| ||_t jdt|y)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_idrlitellm_params model_dumprmodeapi_keyapi_base)rr*r+r,)rr'callbackidz2Populated router guardrail_list with %s guardrails)rrlitellm.proxy.proxy_serverrlitellm.types.routerr%rdebugr guardrail_id_to_custom_guardrailhasattrr(dictrrlen) rrrr%router_guardrail_listrr&rr'r- params_dictrouter_guardrails r!rr4s3X57""#cd=?#7  }}^4 "'78'mm,Tt}i} tD]} tj| jtj| j| j| j| jdusk| jD]A} | t jvr|j| | jdus5| dk(s;ddi| d<Ct!|} t#| dkDrt%| |||| t jS#t&$r} tj(d| | d} ~ wwxYw) Nz validating guardrails passed %srTpresidio logging_onlyr)valuer;r r<callback_specific_paramsz error initializing guardrails %s)rr1itemsr r rlitellmguardrail_name_config_mapsetr default_onupdate callback_args callbacksaddr?listr5r Exception exception)r:r;r r<itemkvguardrail_itemdefault_on_callbacksrArr-default_on_callbacks_listes r!initialize_guardrailsrVes .""#EGXY% FD    F1!.!E!E1!E%%n57E11!4 F F'*e*, ' ZI & &y'?'? @ & &y';'; < $ + +I,C,C D##t+ ) 3 3ZHw'8'88,00: --5#z1DRTXCY4Z@ Z Z ,00D+E! ( )A - )/)!1!1)A  000 &&'I1Ms,C)E7,AE7-E73AE77 FFF)NN)typingrrrrDrlitellm._loggingr)litellm.proxy.common_utils.callback_utilsrlitellm.types.guardrailsr r r r rL__annotations__r4strr"rboolrVrBr9r!r^s##1SQP&(]#($( $CJCDjC CB-qDO-q-qb4D&7!789444 4  #}  4r9