diff --git a/tests/local_testing/conftest.py b/tests/local_testing/conftest.py index 1421700c9..b3561d8a6 100644 --- a/tests/local_testing/conftest.py +++ b/tests/local_testing/conftest.py @@ -26,8 +26,11 @@ def setup_and_teardown(): from litellm import Router importlib.reload(litellm) + try: if hasattr(litellm, "proxy") and hasattr(litellm.proxy, "proxy_server"): + import litellm.proxy.proxy_server + importlib.reload(litellm.proxy.proxy_server) except Exception as e: print(f"Error reloading litellm.proxy.proxy_server: {e}")