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

fix(mcp): extract env var resolution to satisfy ruff PLR0915

Browse source
This commit is contained in:
mateo-berri 2026-06-04 04:20:01 +00:00
parent 3b1a90e6af
commit 21c7c38331
No known key found for this signature in database
1 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -868,6 +868,21 @@ class MCPServerManager:
f"Server ID {mcp_server.server_id} not found in registry"
)
def _resolve_env_vars_list(
self,
mcp_server: LiteLLM_MCPServerTable,
*,
credentials_are_encrypted: bool,
) -> Optional[List[Dict[str, Any]]]:
env_vars_list = _deserialize_json_list(getattr(mcp_server, "env_vars", None))
if credentials_are_encrypted:
from litellm.proxy._experimental.mcp_server.db import ( # noqa: PLC0415
decrypt_global_env_var_values,
)
decrypt_global_env_var_values(env_vars_list)
return env_vars_list
async def build_mcp_server_from_table(
self,
mcp_server: LiteLLM_MCPServerTable,
@ -879,13 +894,9 @@ class MCPServerManager:
static_headers_dict = _deserialize_json_dict(
getattr(mcp_server, "static_headers", None)
)
env_vars_list = _deserialize_json_list(getattr(mcp_server, "env_vars", None))
if credentials_are_encrypted:
from litellm.proxy._experimental.mcp_server.db import ( # noqa: PLC0415
decrypt_global_env_var_values,
)
decrypt_global_env_var_values(env_vars_list)
env_vars_list = self._resolve_env_vars_list(
mcp_server, credentials_are_encrypted=credentials_are_encrypted
)
credentials_dict = _deserialize_json_dict(
getattr(mcp_server, "credentials", None)
)