g+ UddlZddlmZmZmZddlZddlmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZgaeeed<deeeefd ed ed ed eeeff d Z ddeed eefdZy)N)DictListOptional) get_secret)verbose_proxy_logger)initialize_callbacks_on_proxy) GuardrailGuardrailEventHooks GuardrailItemGuardrailItemSpecLakeraCategoryThresholds LitellmParamsSupportedGuardrailIntegrationsall_guardrailsguardrails_config premium_userconfig_file_pathlitellm_settingsreturncr tjd||D]R} |jD]<\}}td i|d|i}tj ||t j|<>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&$r4} tj(dj+t-| | d} ~ wwxYw) Nzvalidating guardrails passed guardrail_nameTpresidio logging_onlyr)valuerrrcallback_specific_paramsz error initializing guardrails {})rdebugitemsr rappendlitellmguardrail_name_config_mapsetr default_onupdate callback_args callbacksaddrlistlenr Exception exceptionformatstr)rrrritemkvguardrail_itemdefault_on_callbacksr guardrailcallbackdefault_on_callbacks_listes ]/var/www/openai/venv/lib/python3.12/site-packages/litellm/proxy/guardrails/init_guardrails.pyinitialize_guardrailsr8s 0""%CDUCV#WX%D   1!.!E!E1!E%%n57E11!4% & #u#% 'I & &y'?'? @ & &y';'; < $ + +I,C,C D##t+ ) 3 3Hw'8'88,00: --5#z1DRTXCY4Z@ !4( %))=$>! ( )A - )/)!1!1)A  000 && . 5 5c!f =  s,C+E9.AE9/E95AE99 F6/F11F6c  g}|D]E}|d}tjd|tjj Dcic]}|||vr||nd}}td+i|}d|vr|drt d+i|d}||d<|dr.|dj drtt|d|d<|dr.|dj drtt|d|d<|dtjjk(r?dd l m } | |d|d|d |d  } tjj!| n|dtj"jk(r?dd lm} | |d |d |d|d} tjj!| n|dtj(jk(rOddlm} | |d|d|d |d |j/d}tjj!|n|dtj0jk(rddlm}||d |d |d|d|j/dxsd}|ddurI|d|d t6j8j|d}tjj!|tjj!|nO|dtj:jk(rGddlm}||j/d|d |d }tjj!|n|dtj@jk(rwddl!m"}|j/d}| tGd||j/d|tj@j }tjj!|nQtI|dtr,d!|dvr$| tGd"ddl%}|d}|jMd!\}}tjd#||||jNjQ|}|jNjS||}|d$z }tTjVjY||}|t[d%|tTjVj]|}|j^ja|tc||}||d |d &} tjj!| nted'|dtg|d |(}!|j!|!|d Htid)|d*ycc}w),Nlitellm_paramszlitellm_params= %scategory_thresholdsapi_keyz os.environ/api_baser3r)AporiaGuardrailrmode)r=r<r event_hook)BedrockGuardrailguardrailIdentifierguardrailVersion)rr@rBrC)lakeraAI_Moderation)r=r<rr@r;)_OPTIONAL_PresidioPIIMaskingoutput_parse_piipresidio_ad_hoc_recognizersmock_redacted_text)rr@rFrGrHT)rFrr@rG)_ENTERPRISE_SecretDetectiondetect_secrets_config)rJr@r) GuardrailsAI guard_namezaGuardrailsAIException - Please pass the Guardrails AI guard name via 'litellm_params::guard_name')r=rLr.zTGuardrailsAIException - Please pass the config_file_path to initialize_guardrails_v2z@Initializing custom guardrail: %s, file_name: %s, class_name: %sz.pyz*Could not find a module specification for )rr@zUnsupported guardrail: )rr:z Guardrail List: r)5rrr__annotations__keysr startswithr-rrAPORIAr2litellm.proxy.guardrails.guardrail_hooks.aporia_air>r r&rBEDROCK;litellm.proxy.guardrails.guardrail_hooks.bedrock_guardrailsrALAKERA2litellm.proxy.guardrails.guardrail_hooks.lakera_airDgetPRESIDIO1litellm.proxy.guardrails.guardrail_hooks.presidiorEr post_call HIDE_SECRETS,enterprise.enterprise_hooks.secret_detectionrI GURDRAILS_AI6litellm.proxy.guardrails.guardrail_hooks.guardrails_airKr* isinstanceossplitpathdirnamejoin importlibutilspec_from_file_location ImportErrormodule_from_specloader exec_modulegetattr ValueErrorr print)"rrguardrail_listr3litellm_params_datar/_litellm_params_kwargsr:lakera_category_thresholdsr>_aporia_callbackrA_bedrock_callbackrD_lakera_callbackrE_presidio_callback_success_callbackrI_secret_detection_objectrK _guard_name_guardrails_ai_callbackra _guardrail _file_name _class_name directorymodule_file_pathspecmodule_guardrail_class_guardrail_callbackparsed_guardrails" r7init_guardrails_v2rVs N$ ((89""#79LM#22779" 9 .A)A"1%t K9 " '@)?@ "%8 8#$9:)A*%&;<* &5ON0 1 ) $i(33MB,/ >);T0U,Vy) * %j)44]C-0N:()9:)&1!/0B!C,:1-$2#5#56J#K#St" 01T9$@%)#,-=#>2<<BB0>51 %!!!(():;    $ $%7 8 ; '-::@@ A (C&4&8&89P&Q)&1()9:( $    $ $%= > ; '-::@@ A ),,\:K"w'3'++J7&=JJPP' #    $ $%< = ~k2C 8~k22'j ( 4J&0&6&6s&; #J & &R  (89I!ww||IzB   % >>99+GWXD|!@AQ@RS^^44T:F KK # #F +&v{; "2()9:)&1#     $ $%8 96~k7R6STU U$$%56)  ./"#o$r n-R 01g" sU)N)rftypingrrrr rlitellm._loggingr)litellm.proxy.common_utils.callback_utilsrlitellm.types.guardrailsr r r r r rrrrOr-booldictr8rrr7rs''1S')]#(6D&7!789666 6  #}  6r'+A2JA2smA2r