fix import error guardrails

This commit is contained in:
Ishaan Jaff 2024-08-19 20:59:18 -07:00
parent 0cba845382
commit 6b7a88d350
2 changed files with 9 additions and 1 deletions

View file

@ -1,3 +1,5 @@
import os
import sys
from typing import Dict
import litellm
@ -5,6 +7,10 @@ from litellm._logging import verbose_proxy_logger
from litellm.proxy.proxy_server import LiteLLM_TeamTable, UserAPIKeyAuth
from litellm.types.guardrails import *
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
def can_modify_guardrails(team_obj: Optional[LiteLLM_TeamTable]) -> bool:
if team_obj is None: