"""

MCP (Model Context Protocol) integration for Feast Feature Server.

This module provides MCP support for Feast by integrating with fastapi_mcp

to expose Feast functionality through the Model Context Protocol.

"""

import logging

from typing import Optional

from feast.feature_store import FeatureStore

logger = logging.getLogger(__name__)

try:

from fastapi_mcp import FastApiMCP

MCP_AVAILABLE = True

except ImportError:

logger.warning(

"fastapi_mcp is not installed. MCP support will be disabled. "

"Install it with: pip install fastapi_mcp"

)

MCP_AVAILABLE = False

# Create placeholder classes for testing

FastApiMCP = None

def add_mcp_support_to_app(app, store: FeatureStore, config) -> Optional["FastApiMCP"]:

"""Add MCP support to the FastAPI app if enabled in configuration."""

if not MCP_AVAILABLE:

logger.warning("MCP support requested but fastapi_mcp is not available")

return None

try:

# Create MCP server from the FastAPI app

mcp = FastApiMCP(

app,

name=getattr(config, "mcp_server_name", "feast-feature-store"),

description="Feast Feature Store MCP Server - Access feature store data and operations through MCP",

)

# Mount the MCP server to the FastAPI app

mcp.mount()

logger.info(

"MCP support has been enabled for the Feast feature server at /mcp endpoint"

)

logger.info(

f"MCP integration initialized for {getattr(config, 'mcp_server_name', 'feast-feature-store')} "

f"v{getattr(config, 'mcp_server_version', '1.0.0')}"

)

return mcp

except Exception as e:

logger.error(f"Failed to initialize MCP integration: {e}")

return None