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

fix(mcp): add get_secret_str import and guard None-resolved env vars in static_headers

Browse source
This commit is contained in:
Shriniwas Ramesh Suram 2026-06-23 14:13:16 -04:00
parent 5999eb2cb6
commit 152b2290b3
1 changed files with 15 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
litellm/proxy/_experimental/mcp_server/mcp_server_manager.py
View file

@ -33,6 +33,7 @@ from pydantic import AnyUrl
import litellm
from litellm._logging import verbose_logger
from litellm.secret_managers.main import get_secret_str
from litellm.constants import (
MCP_CLIENT_TIMEOUT,
MCP_HEALTH_CHECK_TIMEOUT,
@ -1064,10 +1065,20 @@ class MCPServerManager:
getattr(mcp_server, "static_headers", None)
)
if static_headers_dict:
static_headers_dict = {
k: get_secret(v) if isinstance(v, str) and v.startswith("os.environ/") else v
for k, v in static_headers_dict.items()
}
resolved: dict[str, str] = {}
for k, v in static_headers_dict.items():
if isinstance(v, str) and v.startswith("os.environ/"):
secret = get_secret_str(v)
if secret is None:
verbose_logger.warning(
f"MCP static_header {k!r}: env var {v!r} not set; keeping literal value"
)
resolved[k] = v
else:
resolved[k] = secret
else:
resolved[k] = v
static_headers_dict = resolved
env_vars_list = self._resolve_env_vars_list(
mcp_server,
env_vars_are_encrypted=(