scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

Resolve strict MCP search lint violations

Browse source
This commit is contained in:
Hashim Khan 2026-07-29 19:35:09 +05:00
parent f97c1441a3
commit a4f724316c
3 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/models/object_permission.py
View file

@ -25,4 +25,4 @@ class LiteLLM_ObjectPermissionTable(LiteLLMPydanticObjectBase):
blocked_tools: Optional[List[str]] = []
search_tools: Optional[List[str]] = []
mcp_tool_search_enabled: Optional[bool] = None
mcp_tool_search_top_k: Optional[int] = None
mcp_tool_search_top_k: int | None = None

14
litellm/proxy/_experimental/mcp_server/tool_search.py
View file

@ -16,8 +16,8 @@ DEFAULT_MCP_TOOL_SEARCH_TOP_K: int = 5
def get_mcp_tool_search_default_top_k(
user_api_key_dict: Optional["UserAPIKeyAuth"] = None,
litellm_settings: Optional[dict[str, Any]] = None,
user_api_key_dict: UserAPIKeyAuth | None = None,
litellm_settings: dict[str, object] | None = None,
) -> int:
"""Resolve the default top_k for mcp_tool_search (per-key, then global, then 5)."""
if user_api_key_dict is not None:
@ -35,9 +35,9 @@ def get_mcp_tool_search_default_top_k(
def resolve_mcp_tool_search_top_k(
explicit_top_k: Any,
user_api_key_dict: Optional["UserAPIKeyAuth"] = None,
litellm_settings: Optional[dict[str, Any]] = None,
explicit_top_k: object,
user_api_key_dict: UserAPIKeyAuth | None = None,
litellm_settings: dict[str, object] | None = None,
) -> int:
default_top_k = get_mcp_tool_search_default_top_k(user_api_key_dict, litellm_settings)
if explicit_top_k is None:
@ -45,7 +45,9 @@ def resolve_mcp_tool_search_top_k(
return coerce_top_k(explicit_top_k, default=default_top_k)
def coerce_top_k(value: Any, default: int = DEFAULT_MCP_TOOL_SEARCH_TOP_K) -> int:
def coerce_top_k(value: object, default: int = DEFAULT_MCP_TOOL_SEARCH_TOP_K) -> int:
if not isinstance(value, (int, float, str, bytes, bytearray)):
return default
try:
result = int(value)
except (TypeError, ValueError):

2
litellm/proxy/_types.py
View file

@ -1018,7 +1018,7 @@ class LiteLLM_ObjectPermissionBase(LiteLLMPydanticObjectBase):
models: Optional[List[str]] = None
search_tools: Optional[List[str]] = None
mcp_tool_search_enabled: Optional[bool] = None
mcp_tool_search_top_k: Optional[int] = None
mcp_tool_search_top_k: int | None = None
from litellm.models.team import BudgetLimitEntry as BudgetLimitEntry # noqa: E402