From f24d3ffdb6d95679b46bab0465b4e7ee334b10f3 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 20 Mar 2024 19:15:06 -0700 Subject: [PATCH] fix(proxy_server.py): fix import --- litellm/proxy/hooks/prompt_injection_detection.py | 4 ++-- litellm/proxy/proxy_server.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/litellm/proxy/hooks/prompt_injection_detection.py b/litellm/proxy/hooks/prompt_injection_detection.py index 56f1a2dbe..7692ca2b8 100644 --- a/litellm/proxy/hooks/prompt_injection_detection.py +++ b/litellm/proxy/hooks/prompt_injection_detection.py @@ -70,8 +70,8 @@ class _OPTIONAL_PromptInjectionDetection(CustomLogger): for prep in self.prepositions: phrase = " ".join(filter(None, [verb, adj, prep])).strip() if ( - len(phrase.split()) > 1 - ): # additional check to ensure more than 1 word + len(phrase.split()) > 2 + ): # additional check to ensure more than 2 words combinations.append(phrase.lower()) return combinations diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index dfaa99162..957bfc513 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1818,12 +1818,12 @@ class ProxyConfig: isinstance(callback, str) and callback == "detect_prompt_injection" ): - from enterprise.enterprise_hooks.prompt_injection_detection import ( - _ENTERPRISE_PromptInjectionDetection, + from litellm.proxy.hooks.prompt_injection_detection import ( + _OPTIONAL_PromptInjectionDetection, ) prompt_injection_detection_obj = ( - _ENTERPRISE_PromptInjectionDetection() + _OPTIONAL_PromptInjectionDetection() ) imported_list.append(prompt_injection_detection_obj) elif (