phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_key_generate_prisma.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-22 21:12:03 -08:00
parent bedb08bdef
commit 7079b951de
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_key_generate_prisma.py
View file

@ -737,6 +737,7 @@ def test_call_with_key_over_budget(prisma_client):
"user_api_key_user_id": user_id, "user_api_key_user_id": user_id,
} }
}, },
"response_cost": 0.00002,
}, },
completion_response=resp, completion_response=resp,
start_time=datetime.now(), start_time=datetime.now(),
@ -812,6 +813,7 @@ def test_call_with_key_over_budget_stream(prisma_client):
"user_api_key_user_id": user_id, "user_api_key_user_id": user_id,
} }
}, },
"response_cost": 0.00002,
}, },
completion_response=ModelResponse(), completion_response=ModelResponse(),
start_time=datetime.now(), start_time=datetime.now(),