tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-13 20:23:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(mcp): Make MCP_TOOL_PREFIX_SEPARATOR configurable from env (#12603)

Browse Source 
The hardcoded hyphen (-) used as a tool name separator is now configurable via the MCP_TOOL_PREFIX_SEPARATOR environment variable.

- Replaced all hardcoded '-' instances with the MCP_TOOL_PREFIX_SEPARATOR constant.

- Updated error messages and docstrings to dynamically reference the new constant.

Tested and working fine.
Hope that helps!
This commit is contained in:
Juan Carlos Moreno
2025-07-15 13:05:38 -07:00
committed by GitHub
parent 45605f8362
commit a2f0bfc7e5
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/proxy/_experimental/mcp_server/utils.py
+6 -5
View File
@@ -3,13 +3,14 @@ MCP Server Utilities
"""
from typing import Tuple
import os
import importlib
# Constants
LITELLM_MCP_SERVER_NAME = "litellm-mcp-server"
LITELLM_MCP_SERVER_VERSION = "1.0.0"
LITELLM_MCP_SERVER_DESCRIPTION = "MCP Server for LiteLLM"
MCP_TOOL_PREFIX_SEPARATOR = "-"
MCP_TOOL_PREFIX_SEPARATOR = os.environ.get("MCP_TOOL_PREFIX_SEPARATOR", "-")
MCP_TOOL_PREFIX_FORMAT = "{server_name}{separator}{tool_name}"
def is_mcp_available() -> bool:
@@ -77,17 +78,17 @@ def is_tool_name_prefixed(tool_name: str) -> bool:
def validate_mcp_server_name(server_name: str, raise_http_exception: bool = False) -> None:
"""
Validate that MCP server name does not contain '-' (hyphen).
Validate that MCP server name does not contain 'MCP_TOOL_PREFIX_SEPARATOR'.
Args:
server_name: The server name to validate
raise_http_exception: If True, raises HTTPException instead of generic Exception
Raises:
Exception or HTTPException: If server name contains '-'
Exception or HTTPException: If server name contains 'MCP_TOOL_PREFIX_SEPARATOR'
"""
if server_name and '-' in server_name:
error_message = f"Server name cannot contain '-' (hyphen). Please use '_' (underscore) instead. Found: {server_name}"
if server_name and MCP_TOOL_PREFIX_SEPARATOR in server_name:
error_message = f"Server name cannot contain '{MCP_TOOL_PREFIX_SEPARATOR}'. Use an alternative character instead Found: {server_name}"
if raise_http_exception:
from fastapi import HTTPException
from starlette import status