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 84

Use immutable MCP settings type

Browse source
This commit is contained in:
Hashim Khan 2026-07-29 19:51:37 +05:00
parent a4f724316c
commit 5777efa284
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1,6 +1,7 @@
from __future__ import annotations
import json
from collections.abc import Mapping
from datetime import datetime
from typing import TYPE_CHECKING, Any, Optional
@ -17,7 +18,7 @@ DEFAULT_MCP_TOOL_SEARCH_TOP_K: int = 5
def get_mcp_tool_search_default_top_k(
user_api_key_dict: UserAPIKeyAuth | None = None,
litellm_settings: dict[str, object] | None = None,
litellm_settings: Mapping[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:
@ -37,7 +38,7 @@ def get_mcp_tool_search_default_top_k(
def resolve_mcp_tool_search_top_k(
explicit_top_k: object,
user_api_key_dict: UserAPIKeyAuth | None = None,
litellm_settings: dict[str, object] | None = None,
litellm_settings: Mapping[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: