phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix tests on viewing spend logs

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-05 13:17:03 -07:00
parent f9a3e343bb
commit 48cf7ac52f
1 changed files with 12 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/tests/test_key_generate_prisma.py
View file

@ -1610,7 +1610,10 @@ def test_call_with_key_over_budget_no_cache(prisma_client):
proxy_logging_obj=proxy_logging_obj, proxy_logging_obj=proxy_logging_obj,
) )
# test spend_log was written and we can read it # test spend_log was written and we can read it
spend_logs = await view_spend_logs(request_id=request_id) spend_logs = await view_spend_logs(
request_id=request_id,
user_api_key_dict=UserAPIKeyAuth(api_key=generated_key),
)
print("read spend logs", spend_logs) print("read spend logs", spend_logs)
assert len(spend_logs) == 1 assert len(spend_logs) == 1
@ -1730,7 +1733,10 @@ def test_call_with_key_over_model_budget(prisma_client):
proxy_logging_obj=proxy_logging_obj, proxy_logging_obj=proxy_logging_obj,
) )
# test spend_log was written and we can read it # test spend_log was written and we can read it
spend_logs = await view_spend_logs(request_id=request_id) spend_logs = await view_spend_logs(
request_id=request_id,
user_api_key_dict=UserAPIKeyAuth(api_key=generated_key),
)
print("read spend logs", spend_logs) print("read spend logs", spend_logs)
assert len(spend_logs) == 1 assert len(spend_logs) == 1
@ -2299,7 +2305,10 @@ async def test_proxy_load_test_db(prisma_client):
await asyncio.sleep(120) await asyncio.sleep(120)
try: try:
# call spend logs # call spend logs
spend_logs = await view_spend_logs(api_key=generated_key) spend_logs = await view_spend_logs(
api_key=generated_key,
user_api_key_dict=UserAPIKeyAuth(api_key=generated_key),
)
print(f"len responses: {len(spend_logs)}") print(f"len responses: {len(spend_logs)}")
assert len(spend_logs) == n assert len(spend_logs) == n