test(test_completion_cost.py): fix test

2026-06-27 09:10:03 +00:00 · 2024-02-03 22:00:49 -08:00
parent 66565f96b1
commit 49b2dc4180
1 changed files with 5 additions and 1 deletions
@@ -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