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 36

(test) proxy custom logger

Browse source
This commit is contained in:
ishaan-jaff 2023-12-11 11:56:47 -08:00
parent ea89a8a938
commit fa7bcd5825
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_amazing_proxy_custom_logger.py
View file

@ -31,7 +31,7 @@ config_fp = f"{filepath}/test_configs/test_custom_logger.yaml"
python_file_path = f"{filepath}/test_configs/custom_callbacks.py"
save_worker_config(config=config_fp, model=None, alias=None, api_base=None, api_version=None, debug=False, temperature=None, max_tokens=None, request_timeout=600, max_budget=None, telemetry=False, drop_params=True, add_function_to_prompt=False, headers=None, save=False, use_queue=False)
@pytest.fixture(scope="session")
@pytest.fixture(scope="session", autouse=True)
def app():
app = FastAPI()
app.include_router(router)
@ -54,8 +54,6 @@ headers = {
"Authorization": f"Bearer {token}"
}
@pytest.mark.no_parallel
def test_embedding(client):
try:
# Your test data
@ -92,7 +90,7 @@ def test_embedding(client):
except Exception as e:
pytest.fail("LiteLLM Proxy test failed. Exception", e)
@pytest.mark.no_parallel
def test_chat_completion(client):
try:
# Your test data
@ -138,7 +136,7 @@ def test_chat_completion(client):
except Exception as e:
pytest.fail("LiteLLM Proxy test failed. Exception", e)
@pytest.mark.no_parallel
def test_chat_completion_stream(client):
try:
# Your test data