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

fix(proxy_server.py): fix import

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-20 19:15:06 -07:00
parent 3bb0e24cb7
commit f24d3ffdb6
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/proxy_server.py
View file

@ -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 (