scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 74

test(proxy): use local_model_cost_map fixture for alias listing test

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
jesus 2026-09-18 19:59:24 +00:00
parent 0a81c6d3a8
commit 8549189cbb
1 changed files with 16 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
tests/test_litellm/proxy/test_proxy_utils.py
View file

@ -2243,32 +2243,27 @@ def test_create_model_info_response_resolves_mode_through_deployment_model():
{"team-embeddings": {"model": "my-embeddings", "hidden": False}},
],
)
def test_create_model_info_response_resolves_model_group_alias_to_target(model_group_alias):
def test_create_model_info_response_resolves_model_group_alias_to_target(model_group_alias, local_model_cost_map):
"""A `model_group_alias` row must report the metadata of the group it points at,
not the cost-map generalization or nothing that the alias name resolves to."""
from litellm import Router
saved_model_cost = dict(litellm.model_cost)
try:
router = Router(
model_list=[
{
"model_name": "my-embeddings",
"litellm_params": {"model": "openai/text-embedding-3-small"},
}
],
model_group_alias=model_group_alias,
)
router = Router(
model_list=[
{
"model_name": "my-embeddings",
"litellm_params": {"model": "openai/text-embedding-3-small"},
}
],
model_group_alias=model_group_alias,
)
alias_response = create_model_info_response(
model_id="team-embeddings", provider="openai", llm_router=router
)
target_response = create_model_info_response(
model_id="my-embeddings", provider="openai", llm_router=router
)
finally:
litellm.model_cost.clear()
litellm.model_cost.update(saved_model_cost)
alias_response = create_model_info_response(
model_id="team-embeddings", provider="openai", llm_router=router
)
target_response = create_model_info_response(
model_id="my-embeddings", provider="openai", llm_router=router
)
assert alias_response["id"] == "team-embeddings"
for field in ("mode", "max_input_tokens", "max_output_tokens"):