scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-01 02:02:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 92

fix(cost): read deepseek override_dates on the Asia/Shanghai calendar

Browse source 
The shipped DeepSeek rows model PRC public holidays and make-up workdays, which the State Council notice declares on China Standard Time, so the rows now set weekday_timezone: Asia/Shanghai and their override_dates follow the Shanghai calendar instead of UTC.
This commit is contained in:
shrey kharbanda 2026-09-24 06:12:35 +00:00
parent 09746bff48
commit 33374a9207
3 changed files with 33 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/model_prices_and_context_window_backup.json
View file

@ -58298,7 +58298,8 @@
"2026-10-10"
]
}
]
],
"weekday_timezone": "Asia/Shanghai"
},
"output_cost_per_token": 1.2e-06,
"source": "https://api-docs.deepseek.com/quick_start/pricing",
@ -58513,7 +58514,8 @@
"2026-10-10"
]
}
]
],
"weekday_timezone": "Asia/Shanghai"
},
"output_cost_per_token": 3.96e-06,
"source": "https://api-docs.deepseek.com/quick_start/pricing",

6
model_prices_and_context_window.json
View file

@ -58298,7 +58298,8 @@
"2026-10-10"
]
}
]
],
"weekday_timezone": "Asia/Shanghai"
},
"output_cost_per_token": 1.2e-06,
"source": "https://api-docs.deepseek.com/quick_start/pricing",
@ -58513,7 +58514,8 @@
"2026-10-10"
]
}
]
],
"weekday_timezone": "Asia/Shanghai"
},
"output_cost_per_token": 3.96e-06,
"source": "https://api-docs.deepseek.com/quick_start/pricing",

25
tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py
View file

@ -738,6 +738,31 @@ def test_is_off_peak_without_override_dates_is_unchanged():
assert _is_off_peak(deepseek, datetime(2026, 1, 1, 2, 0, tzinfo=timezone.utc)) is False
def test_override_dates_use_weekday_timezone_calendar_on_shipped_deepseek_rows():
"""The shipped deepseek/deepseek-flash row reads its override_dates on the
Asia/Shanghai calendar: the make-up Sunday bills weekday hours and the holiday is
off-peak all day."""
from datetime import datetime, timezone
import litellm
block: Final = litellm.model_cost["deepseek/deepseek-flash"]["off_peak_pricing"]
assert block.get("weekday_timezone") == "Asia/Shanghai"
shanghai_make_up_sunday = datetime(2026, 1, 3, 17, 0, tzinfo=timezone.utc)
make_up_sunday_peak = datetime(2026, 1, 4, 2, 0, tzinfo=timezone.utc)
plain_sunday = datetime(2026, 1, 11, 2, 0, tzinfo=timezone.utc)
holiday = datetime(2026, 1, 1, 2, 0, tzinfo=timezone.utc)
assert _is_off_peak(block, shanghai_make_up_sunday) is True, (
f"{shanghai_make_up_sunday.isoformat()} is 2026-01-04 in Shanghai, a make-up workday"
)
assert _is_off_peak(block, make_up_sunday_peak) is False, (
f"{make_up_sunday_peak.isoformat()} is a make-up workday peak hour"
)
assert _is_off_peak(block, plain_sunday) is True
assert _is_off_peak(block, holiday) is True
def test_get_token_base_cost_override_dates_bill_holiday_rates():
"""End to end through _get_token_base_cost: a date listed on an override rule bills the
off-peak input rate during what would be a weekday peak hour."""