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

(test) model in SpendLogs table

Browse source
This commit is contained in:
ishaan-jaff 2024-01-24 11:10:13 -08:00
parent a842e6520c
commit 3e53742a4f
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

@ -825,6 +825,7 @@ def test_call_with_key_over_budget_stream(prisma_client):
) )
await track_cost_callback( await track_cost_callback(
kwargs={ kwargs={
"model": "sagemaker-chatgpt-v-2",
"stream": True, "stream": True,
"complete_streaming_response": resp, "complete_streaming_response": resp,
"litellm_params": { "litellm_params": {
@ -850,6 +851,7 @@ def test_call_with_key_over_budget_stream(prisma_client):
assert spend_log.request_id == request_id assert spend_log.request_id == request_id
assert spend_log.spend == float("5e-05") assert spend_log.spend == float("5e-05")
assert spend_log.model == "sagemaker-chatgpt-v-2"
# use generated key to auth in # use generated key to auth in
result = await user_api_key_auth(request=request, api_key=bearer_token) result = await user_api_key_auth(request=request, api_key=bearer_token)