From f3131f12546dc06e48de35b2bb13b0e76cb1b427 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 11 Dec 2023 14:22:55 -0800 Subject: [PATCH] (test) proxy custom logger --- litellm/tests/test_aamazing_proxy_custom_logger.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_aamazing_proxy_custom_logger.py b/litellm/tests/test_aamazing_proxy_custom_logger.py index 9ed68d78b67..7ae3baec69e 100644 --- a/litellm/tests/test_aamazing_proxy_custom_logger.py +++ b/litellm/tests/test_aamazing_proxy_custom_logger.py @@ -20,6 +20,8 @@ import importlib, inspect from fastapi.testclient import TestClient from fastapi import FastAPI from litellm.proxy.proxy_server import router, save_worker_config, initialize # Replace with the actual module where your FastAPI router is defined + + filepath = os.path.dirname(os.path.abspath(__file__)) config_fp = f"{filepath}/test_configs/test_custom_logger.yaml" python_file_path = f"{filepath}/test_configs/custom_callbacks.py" @@ -31,7 +33,7 @@ async def wrapper_startup_event(): initialize(config=config_fp) # Use the app fixture in your client fixture -@pytest.fixture(autouse=True) +@pytest.fixture() def client(): with TestClient(app) as client: yield client