Coverage for .venv/lib/python3.13/site-packages/litellm/proxy/_experimental/mcp_server/guardrail_translation/__init__.py: 100%

1"""Guardrail translation mapping for MCP tool calls.""" 

2 

3from typing import Final 

4 

5from litellm.proxy._experimental.mcp_server.guardrail_translation.handler import ( 

6 MCPGuardrailTranslationHandler, 

7) 

8from litellm.types.utils import CallTypes 

9 

10# This mapping lives alongside the MCP server implementation because MCP 

11# integrations are managed by the proxy subsystem, not litellm.llms providers. 

12# Unified guardrails import this module explicitly to register the handler. 

13 

14guardrail_translation_mappings: Final = { 

15 CallTypes.call_mcp_tool: MCPGuardrailTranslationHandler, 

16} 

17 

18__all__ = ["MCPGuardrailTranslationHandler", "guardrail_translation_mappings"]