scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

(test) proxy custom logger

Browse source
This commit is contained in:
ishaan-jaff 2023-12-11 14:22:55 -08:00
parent 3c8c5980d8
commit f3131f1254
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_aamazing_proxy_custom_logger.py
View file

@ -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