hdZddlmZmZddlZddlZdZdZdZejjddZ d Z d e fd Zd ed efd Zded dfdZded ed efdZded efdZded eeeffdZded e fdZdd ede d dfdZy)z MCP Server Utilities )TupleAnyNzlitellm-mcp-serverz1.0.0zMCP Server for LiteLLMMCP_TOOL_PREFIX_SEPARATOR-z#{server_name}{separator}{tool_name}returncN tjdy#t$rYywxYw)zF Returns True if the MCP module is available, False otherwise mcpTF) importlib import_module ImportErrorh/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/_experimental/mcp_server/utils.pyis_mcp_availablers*& s  $$ server_namec&|jddS)zD Normalize server name by replacing spaces with underscores  _)replace)rs rnormalize_server_namers   sC ((rpayloadcVt|dr#|jrt|jdt|dr#|jrt|jdt |dd}t |dd}|s|r t |}n |r t |}t|dr||_yy)a Validate and normalize MCP server payload fields (server_name and alias). This function: 1. Validates that server_name and alias don't contain the MCP_TOOL_PREFIX_SEPARATOR 2. Normalizes alias by replacing spaces with underscores 3. Sets default alias if not provided (using server_name as base) Args: payload: The payload object containing server_name and alias fields Raises: HTTPException: If validation fails rT)raise_http_exceptionaliasN)hasattrrvalidate_mcp_server_namergetattrr)rrrs r)validate_and_normalize_mcp_server_payloadr s w &7+>+> !4!44Pw W]] TJ GWd +E'=$7K [%k2 %e,w  !r tool_namecPt|}tj|t|S)z Add server name prefix to tool name Args: tool_name: Original tool name server_name: MCP server name Returns: Prefixed tool name in format: server_name::tool_name )r separatorr)rMCP_TOOL_PREFIX_FORMATformatr)rrformatted_server_names radd_server_prefix_to_tool_namer%Ds02+> ! ( ()+ ) rserverct|dr|jr |jSt|dr|jr |jSt|dr |jSy)zRReturn the prefix for a server: alias if present, else server_name, else server_idrr server_id)rrrr()r&s rget_server_prefixr*WsSvwFLL||v}%&*<*<!!!v{# rprefixed_tool_namecvt|vr.|jtd}t|dk(r |d|dfS|dfS)z Remove server name prefix from tool name Args: prefixed_tool_name: Tool name with server prefix Returns: Tuple of (original_tool_name, server_name) rr))rsplitlen)r+partss rget_server_name_prefix_tool_mcpr2asI!$66"(()BAF u:?8U1X% % r !!rct|vS)z Check if tool name has server prefix Args: tool_name: Tool name to check Returns: True if tool name is prefixed, False otherwise )r)rs ris_tool_name_prefixedr4qs % 11rrc|rDt|vr;dtd|}|r"ddlm}ddlm}||j d|it |yy) a[ 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 'MCP_TOOL_PREFIX_SEPARATOR' zServer name cannot contain 'z/'. Use an alternative character instead Found: r) HTTPException)statuserror) status_codedetailN)rfastapir6 starletter7HTTP_400_BAD_REQUEST Exception)rr error_messager6r7s rrr}sq0K?67P6QRABMANO  - ("77/  M* *@{r)F)__doc__typingrrosr LITELLM_MCP_SERVER_NAMELITELLM_MCP_SERVER_VERSIONLITELLM_MCP_SERVER_DESCRIPTIONenvirongetrr"boolrstrrrr%r*r2r4rr rrrJs /$!9JJNN+FL>$)s)s) "s"t"Hc&cc""c3h" 2S 2T 2+#+T+VZ+r