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

test: fix tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-22 20:20:17 -08:00
parent 3e8c8ef507
commit 11e3ee4411
4 changed files with 24 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_key_generate_prisma.py
View file

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