From c8fbadeec2b65f9946015bb0076bca2f0ebddc9d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 13 Mar 2024 21:24:37 -0700 Subject: [PATCH] (fix) importing PromptInjectionDetection --- litellm/proxy/proxy_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 79a8b58508..5a141fa034 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1773,7 +1773,7 @@ class ProxyConfig: isinstance(callback, str) and callback == "detect_prompt_injection" ): - from litellm.proxy.enterprise.enterprise_hooks.prompt_injection_detection import ( + from enterprise.enterprise_hooks.prompt_injection_detection import ( _ENTERPRISE_PromptInjectionDetection, )