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

test(test_completion_cost.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-03 22:00:49 -08:00
parent 66565f96b1
commit 49b2dc4180
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_completion_cost.py
View file

@ -162,7 +162,11 @@ def test_cost_azure_embedding():
def test_cost_openai_image_gen(): def test_cost_openai_image_gen():
cost = litellm.completion_cost( cost = litellm.completion_cost(
model="dall-e-2", size="1024-x-1024", quality="standard", n=1 model="dall-e-2",
size="1024-x-1024",
quality="standard",
n=1,
call_type="image_generation",
) )
assert cost == 0.019922944 assert cost == 0.019922944