^h~SSKJrJrJrJrJrJr SSKrSSKJ r SSK J r SSK J r SSKJr SSKJr "SS \ 5rg) )AnyDictListLiteralOptionalUnionN)verbose_proxy_logger) DualCache)CustomGuardrail)UserAPIKeyAuth) should_proceed_based_on_metadatac^\rSrSrU4SjrS\S\S\S\SS\ \ \ \ \44 S jr S\S\S\S 4S jrS\S\4S jrS rU=r$)myCustomGuardrail c 2>Xl[TU]"S0UD6 g)N)optional_paramssuper__init__)selfkwargs __class__s l/home/james-whalen/.local/lib/python3.13/site-packages/litellm/proxy/example_config_yaml/custom_guardrail.pyrmyCustomGuardrail.__init__ s & "6"user_api_key_dictcachedata call_type) completiontext_completion embeddingsimage_generation moderationaudio_transcriptionpass_through_endpointrerankreturnc&# URS5nU(a]UHWnURS5n[U[5(dM+SUR5;dMAUR SS5nXvS'MY [ R "SU5 U$7f)zZ Runs before the LLM API call Runs on only Input Use this if you want to MODIFY the input messagescontentlitellmz********z-async_pre_call_hook: Message after masking %s)get isinstancestrlowerreplacer debug)rrrrr _messagesmessage_contents rasync_pre_call_hook%myCustomGuardrail.async_pre_call_hooks.HHZ( $";;y1h,, HNN$44#+#3#3Iz#J-5 * % "" ;Y  sABB3B)r r"r#r$r%c# URS5nU(aQUHJnURS5n[U[5(dM+SUR5;dMA[ S5e gg7f)z Runs in parallel to LLM API call Runs on only Input This can NOT modify the input, only used to reject or accept a call before going to LLM API r*r+r,z+Guardrail failed words - `litellm` detectedN)r-r.r/r0 ValueError)rrrrr3r4r5s rasync_moderation_hook'myCustomGuardrail.async_moderation_hook;s]*HHZ( $";;y1h,, HNN$44()VWW % sAA-A-A-c# [R"SU5 [U[R5(aUR Hn[U[R 5(dM$[R"SU5 URR(dMX[URR[5(dMSURR;dM[S5e gg7f)z Runs on response from LLM API call It can be used to reject a response If a response contains the word "coffee" -> we will raise an exception z async_pre_call_hook response: %szasync_pre_call_hook choice: %scoffeez Guardrail failed Coffee DetectedN) r r2r.r, ModelResponsechoicesChoicesr4r+r/r9)rrrresponsechoices rasync_post_call_success_hook.myCustomGuardrail.async_post_call_success_hookXs ""#ExP h 5 5 6 6"**fgoo66(../OQWX...&v~~'='=sCC$(>(>>()KLL+ 7sA#C3)0C3'C3C3$C3)r)__name__ __module__ __qualname____firstlineno__rr r dictrrr Exceptionr/r6r:rC__static_attributes__ __classcell__)rs@rrr s#$)$$ $    $ % 3,- .$LXX*X # X:MM*MMrr)typingrrrrrrr,litellm._loggingr litellm.caching.cachingr %litellm.integrations.custom_guardrailr litellm.proxy._typesr *litellm.proxy.guardrails.guardrail_helpersr rrrrrSs-<<1-A/WdMdMr