(test) cleanup proxy

This commit is contained in:
ishaan-jaff 2023-12-11 22:41:14 -08:00
parent d563e53948
commit ab90a66a8e

View file

@ -37,6 +37,8 @@ async def wrapper_startup_event():
# Make sure the fixture returns TestClient(app)
@pytest.fixture(autouse=True)
def client():
from litellm.proxy.proxy_server import cleanup_router_config_variables
cleanup_router_config_variables()
with TestClient(app) as client:
yield client