nat.plugins.mcp.utils

Functions

_get_or_create_enum(→ type[enum.Enum])	Get a cached enum class or create a new one.
truncate_session_id(→ str)	Truncate a session ID for logging purposes.
model_from_mcp_schema(→ type[pydantic.BaseModel])	Create a pydantic model from the input schema of the MCP tool

Module Contents

_get_or_create_enum(

name: str,

values: frozenset[str],

) → type[enum.Enum]

Get a cached enum class or create a new one.

This function ensures that enums with the same name and values always return the same class object. This is critical for Pydantic validation, which checks enum instances by class identity.

Uses cache to automatically cache enum classes by their name and values.

Args:

name: The name for the enum class values: Frozenset of enum values (frozenset is hashable for caching)

Returns:

An Enum class (cached or newly created)

truncate_session_id(session_id: str, max_length: int = 10) → str

Truncate a session ID for logging purposes.

Args:

session_id: The session ID to truncate max_length: Maximum length before truncation (default: 10)

Returns:

Truncated session ID with “…” if longer than max_length, otherwise full ID

model_from_mcp_schema(

name: str,

mcp_input_schema: dict,

) → type[pydantic.BaseModel]

Create a pydantic model from the input schema of the MCP tool