feat: add mcp abstractions to core (#36151)

This commit is contained in:
Ville Brofeldt
2025-11-25 08:23:59 -10:00
committed by GitHub
parent bae716fa83
commit a4860075d2
37 changed files with 1044 additions and 151 deletions

View File

@@ -25,9 +25,8 @@ and timeout protection.
import logging
from fastmcp import Context
from superset_core.mcp import tool
from superset.mcp_service.app import mcp
from superset.mcp_service.auth import mcp_auth_hook
from superset.mcp_service.sql_lab.execute_sql_core import ExecuteSqlCore
from superset.mcp_service.sql_lab.schemas import (
ExecuteSqlRequest,
@@ -38,8 +37,7 @@ from superset.mcp_service.utils.schema_utils import parse_request
logger = logging.getLogger(__name__)
@mcp.tool
@mcp_auth_hook
@tool
@parse_request(ExecuteSqlRequest)
async def execute_sql(request: ExecuteSqlRequest, ctx: Context) -> ExecuteSqlResponse:
"""Execute SQL query against database.