From 93ae0474ce2c7fb842825fcf964665bdaea24e3d Mon Sep 17 00:00:00 2001 From: Rhys Date: Mon, 23 Feb 2026 11:05:03 +0700 Subject: [PATCH] fix: use local model cost map in Azure cost calculation test --- tests/test_litellm/test_gpt_image_cost_calculator.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_litellm/test_gpt_image_cost_calculator.py b/tests/test_litellm/test_gpt_image_cost_calculator.py index fdcee806862..37e8647ca5b 100644 --- a/tests/test_litellm/test_gpt_image_cost_calculator.py +++ b/tests/test_litellm/test_gpt_image_cost_calculator.py @@ -273,6 +273,8 @@ class TestGPTImage15OutputImageTokens: """ Test that output tokens are correctly calculated in cost calculation for Azure. """ + # Set environment variable to use local model cost map + os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True" # Simulate gpt-image-1.5 response with output_tokens_details usage = Usage( prompt_tokens=200,