scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Merge bac9a36ca1 into eb0e3f8c18

Browse source
This commit is contained in:
cursor[bot] 2026-08-27 19:28:10 -05:00 • committed by GitHub
commit 8aabb96c4a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_litellm/proxy/test_proxy_server.py
View file

@ -4067,6 +4067,7 @@ class TestPriceDataReloadIntegration:
A revision this pod has not applied takes effect here even one minute into a 6h
interval; a missing row is a no-op
"""
from litellm import utils as litellm_utils
from litellm.proxy.proxy_server import ProxyConfig
proxy_config = ProxyConfig()
@ -4095,6 +4096,12 @@ class TestPriceDataReloadIntegration:
from litellm.litellm_core_utils.get_model_cost_map import ModelCostMapReloaded
original_model_cost = litellm.model_cost.copy()
# A reload replays runtime registrations, so any gpt-3.5-turbo entry a
# prior test in this worker persisted would be re-registered on top of
# the mocked catalog and defeat the exact-equality check below. Isolate
# the registry for the duration of this test.
original_registry = dict(litellm_utils._runtime_registered_model_cost)
litellm_utils._runtime_registered_model_cost.clear()
try:
with (
patch(
@ -4118,6 +4125,8 @@ class TestPriceDataReloadIntegration:
assert proxy_config.model_cost_map_applied_revision == 4
finally:
litellm.model_cost = original_model_cost
litellm_utils._runtime_registered_model_cost.clear()
litellm_utils._runtime_registered_model_cost.update(original_registry)
_invalidate_model_cost_lowercase_map()
def test_distributed_reload_ignores_already_applied_request(self):