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

fix(ci): resolve remaining mypy, MCP test, and model prices failures

Browse source 
- cost_tracking_settings.py: fix mypy return type by using temp variables with str() cast inside if-blocks
- test_mcp_server.py: add tool_name_to_display_name/description dicts to test_add_update_server_without_alias
- test_utils.py: add /vertex_ai/live to supported_endpoints enum

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Harshit28j 2026-03-05 09:43:27 +05:30
parent 2ebaa32d1c
commit 05a8834069
3 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/proxy/management_endpoints/cost_tracking_settings.py
View file

@ -59,24 +59,24 @@ def _resolve_model_for_cost_lookup(model: str) -> Tuple[str, Optional[str]]:
model_info = first_deployment.get("model_info", {})
# Check base_model first (needed for Azure custom deployment names)
base_model = model_info.get("base_model") or litellm_params.get( # type: ignore[assignment]
_base_model = model_info.get("base_model") or litellm_params.get(
"base_model"
)
if base_model:
if _base_model:
verbose_proxy_logger.debug(
f"Resolved model '{model}' to base_model '{base_model}' from router"
f"Resolved model '{model}' to base_model '{_base_model}' from router"
)
custom_llm_provider = litellm_params.get("custom_llm_provider")
return base_model, custom_llm_provider
return str(_base_model), custom_llm_provider
resolved_model: Optional[str] = litellm_params.get("model")
_resolved_model = litellm_params.get("model")
if resolved_model:
if _resolved_model:
verbose_proxy_logger.debug(
f"Resolved model '{model}' to '{resolved_model}' from router"
f"Resolved model '{model}' to '{_resolved_model}' from router"
)
custom_llm_provider = litellm_params.get("custom_llm_provider")
return resolved_model, custom_llm_provider
return str(_resolved_model), custom_llm_provider
except Exception as e:
verbose_proxy_logger.debug(
f"Could not resolve model '{model}' from router: {e}"

2
tests/mcp_tests/test_mcp_server.py
View file

@ -1508,6 +1508,8 @@ async def test_add_update_server_without_alias():
mock_mcp_server.args = []
mock_mcp_server.env = None
mock_mcp_server.spec_path = None
mock_mcp_server.tool_name_to_display_name = {}
mock_mcp_server.tool_name_to_description = {}
# OAuth fields - set explicitly to None to avoid MagicMock objects
mock_mcp_server.client_id = None
mock_mcp_server.client_secret = None

1
tests/test_litellm/test_utils.py
View file

@ -754,6 +754,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
"/v1/audio/transcriptions",
"/v1/audio/speech",
"/v1/ocr",
"/vertex_ai/live",
],
},
},