diff --git a/litellm/tests/test_config.py b/litellm/tests/test_config.py index cd61101a3c..28d144e4dc 100644 --- a/litellm/tests/test_config.py +++ b/litellm/tests/test_config.py @@ -2,23 +2,30 @@ ## Unit tests for ProxyConfig class -import sys, os +import os +import sys import traceback + from dotenv import load_dotenv load_dotenv() -import os, io +import io +import os sys.path.insert( 0, os.path.abspath("../..") ) # Adds the parent directory to the system path -import pytest, litellm -from pydantic import BaseModel, ConfigDict -from litellm.proxy.proxy_server import ProxyConfig -from litellm.proxy.utils import encrypt_value, ProxyLogging, DualCache -from litellm.types.router import Deployment, LiteLLM_Params, ModelInfo from typing import Literal +import pytest +from pydantic import BaseModel, ConfigDict + +import litellm +from litellm.proxy.common_utils.encrypt_decrypt_utils import encrypt_value +from litellm.proxy.proxy_server import ProxyConfig +from litellm.proxy.utils import DualCache, ProxyLogging +from litellm.types.router import Deployment, LiteLLM_Params, ModelInfo + class DBModel(BaseModel): model_id: str @@ -28,6 +35,7 @@ class DBModel(BaseModel): model_config = ConfigDict(protected_namespaces=()) + @pytest.mark.asyncio async def test_delete_deployment(): """