Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions marimo/_mcp/server/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@

from marimo._ai._tools.base import ToolContext
from marimo._ai._tools.tools_registry import SUPPORTED_BACKEND_AND_MCP_TOOLS
from marimo._dependencies.dependencies import DependencyManager
from marimo._loggers import marimo_logger

LOGGER = marimo_logger()
Expand All @@ -33,6 +34,12 @@ def setup_mcp_server(app: Starlette) -> None:
Returns:
StreamableHTTPSessionManager: MCP session manager
"""
if not DependencyManager.mcp.has():
from click import ClickException

msg = "MCP dependencies not available. Install with `pip install marimo[mcp]` or `uv add marimo[mcp]`"
raise ClickException(msg)

from mcp.server.fastmcp import FastMCP
from starlette.middleware.base import BaseHTTPMiddleware
from starlette.responses import JSONResponse
Expand Down
Loading