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 (