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

test(router): cover single-deployment cost-map re-registration

Browse source
This commit is contained in:
Devin AI 2026-07-10 07:25:07 +00:00
parent d822457cfd
commit 832a53abc4
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/test_litellm/test_router_model_cost_isolation.py
View file

@ -447,6 +447,22 @@ def test_deployment_mode_persists_across_model_cost_map_reload():
assert litellm.model_cost[backend_key]["mode"] == "responses"
deployment = Deployment(
model_name="gpt-5.6",
litellm_params=LiteLLM_Params(
model="gpt-5.6",
custom_llm_provider="openai",
api_key="fake-key-reload",
),
model_info={"id": "gpt-56-responses-reload", "mode": "responses"},
)
router._register_deployment_in_model_cost(
deployment=deployment,
model_info=deployment.model_info.model_dump(exclude_none=True),
)
assert litellm.model_cost["gpt-56-responses-reload"]["mode"] == "responses"
assert litellm.model_cost[backend_key]["mode"] == "responses"
bridge_model_info, bridge_model = responses_api_bridge_check(
model="gpt-5.6",
custom_llm_provider="openai",