phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(llm_guard.py): enable key-specific llm guard check

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-26 17:21:51 -07:00
parent bec093675c
commit e10eb8f6fe
8 changed files with 26 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/proxy/hooks/prompt_injection_detection.py
View file

@ -199,6 +199,7 @@ class _OPTIONAL_PromptInjectionDetection(CustomLogger):
async def async_moderation_hook(
self,
data: dict,
user_api_key_dict: UserAPIKeyAuth,
call_type: Literal["completion", "embeddings", "image_generation"],
):
self.print_verbose(