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

test: address review comments - add global state cleanup and parallel_function_calling assertion

Browse source 
- Wrap test body in try/finally to restore os.environ and litellm.model_cost
  after the test completes, preventing state leakage to subsequent tests
- Add assertion for supports_parallel_function_calling (the fourth field
  added to _get_model_info_helper, now fully covered by the test)
This commit is contained in:
Abhishek Deshmukh 2026-04-04 01:23:13 +05:30
parent 5ca2041ba1
commit 27264323eb
1 changed files with 39 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

62
tests/test_litellm/test_utils.py
View file

@ -1174,31 +1174,47 @@ def test_get_model_info_shows_supports_xhigh_reasoning_effort():
where get_model_info omitted reasoning-effort capability fields even
though they exist in model_cost.
"""
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
original_env_var = os.environ.get("LITELLM_LOCAL_MODEL_COST_MAP")
original_model_cost = getattr(litellm, "model_cost", None)
try:
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
model = "gpt-5.4"
info = litellm.get_model_info(model)
model_cost = litellm.model_cost[model]
model = "gpt-5.4"
info = litellm.get_model_info(model)
model_cost = litellm.model_cost[model]
assert info.get("supports_xhigh_reasoning_effort") == model_cost.get(
"supports_xhigh_reasoning_effort"
), (
f"get_model_info should return supports_xhigh_reasoning_effort={model_cost.get('supports_xhigh_reasoning_effort')!r} "
f"for {model!r}, got {info.get('supports_xhigh_reasoning_effort')!r}"
)
assert info.get("supports_none_reasoning_effort") == model_cost.get(
"supports_none_reasoning_effort"
), (
f"get_model_info should return supports_none_reasoning_effort={model_cost.get('supports_none_reasoning_effort')!r} "
f"for {model!r}, got {info.get('supports_none_reasoning_effort')!r}"
)
assert info.get("supports_minimal_reasoning_effort") == model_cost.get(
"supports_minimal_reasoning_effort"
), (
f"get_model_info should return supports_minimal_reasoning_effort={model_cost.get('supports_minimal_reasoning_effort')!r} "
f"for {model!r}, got {info.get('supports_minimal_reasoning_effort')!r}"
)
assert info.get("supports_xhigh_reasoning_effort") == model_cost.get(
"supports_xhigh_reasoning_effort"
), (
f"get_model_info should return supports_xhigh_reasoning_effort={model_cost.get('supports_xhigh_reasoning_effort')!r} "
f"for {model!r}, got {info.get('supports_xhigh_reasoning_effort')!r}"
)
assert info.get("supports_none_reasoning_effort") == model_cost.get(
"supports_none_reasoning_effort"
), (
f"get_model_info should return supports_none_reasoning_effort={model_cost.get('supports_none_reasoning_effort')!r} "
f"for {model!r}, got {info.get('supports_none_reasoning_effort')!r}"
)
assert info.get("supports_minimal_reasoning_effort") == model_cost.get(
"supports_minimal_reasoning_effort"
), (
f"get_model_info should return supports_minimal_reasoning_effort={model_cost.get('supports_minimal_reasoning_effort')!r} "
f"for {model!r}, got {info.get('supports_minimal_reasoning_effort')!r}"
)
assert info.get("supports_parallel_function_calling") == model_cost.get(
"supports_parallel_function_calling"
), (
f"get_model_info should return supports_parallel_function_calling={model_cost.get('supports_parallel_function_calling')!r} "
f"for {model!r}, got {info.get('supports_parallel_function_calling')!r}"
)
finally:
if original_env_var is None:
os.environ.pop("LITELLM_LOCAL_MODEL_COST_MAP", None)
else:
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = original_env_var
if original_model_cost is not None:
litellm.model_cost = original_model_cost
@pytest.mark.parametrize(