From 49b2dc41801f413f4e2cbe5bf65be8ccc3b981dd Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 3 Feb 2024 22:00:49 -0800 Subject: [PATCH] test(test_completion_cost.py): fix test --- litellm/tests/test_completion_cost.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_completion_cost.py b/litellm/tests/test_completion_cost.py index b55f9c9d6e..bb460b76bd 100644 --- a/litellm/tests/test_completion_cost.py +++ b/litellm/tests/test_completion_cost.py @@ -162,7 +162,11 @@ def test_cost_azure_embedding(): def test_cost_openai_image_gen(): 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