scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 178

fix: use local model cost map in Azure cost calculation test

Browse source
This commit is contained in:
Rhys 2026-02-23 11:05:03 +07:00
parent 51b229156e
commit 93ae0474ce
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_litellm/test_gpt_image_cost_calculator.py
View file

@ -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,