QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-05-12 19:35:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(mcp): remove @parse_request decorator for cleaner tool schemas (#38918)

Browse Source 
(cherry picked from commit d1903afc69)
This commit is contained in:
Amin Ghadersohi
2026-03-29 22:09:51 +02:00
committed by Michael S. Molina
parent df2a0416eb
commit e08c305396
22 changed files with 39 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
superset/mcp_service/sql_lab/tool/execute_sql.py
View File

@@ -22,8 +22,6 @@ Tool for executing SQL queries against databases using the unified
Database.execute() API with RLS, template rendering, and security validation.
"""
from __future__ import annotations
import logging
from decimal import Decimal
from typing import Any
@@ -48,7 +46,6 @@ from superset.mcp_service.sql_lab.schemas import (
StatementData,
StatementInfo,
)
from superset.mcp_service.utils.schema_utils import parse_request
logger = logging.getLogger(__name__)
@@ -63,7 +60,6 @@ logger = logging.getLogger(__name__)
destructiveHint=True,
),
)
@parse_request(ExecuteSqlRequest)
async def execute_sql(request: ExecuteSqlRequest, ctx: Context) -> ExecuteSqlResponse:
"""Execute SQL query against database using the unified Database.execute() API."""
await ctx.info(

2
superset/mcp_service/sql_lab/tool/open_sql_lab_with_context.py
View File

@@ -32,7 +32,6 @@ from superset.mcp_service.sql_lab.schemas import (
OpenSqlLabRequest,
SqlLabResponse,
)
from superset.mcp_service.utils.schema_utils import parse_request
from superset.mcp_service.utils.url_utils import get_superset_base_url
logger = logging.getLogger(__name__)
@@ -48,7 +47,6 @@ logger = logging.getLogger(__name__)
destructiveHint=False,
),
)
@parse_request(OpenSqlLabRequest)
def open_sql_lab_with_context(
request: OpenSqlLabRequest, ctx: Context
) -> SqlLabResponse:

4
superset/mcp_service/sql_lab/tool/save_sql_query.py
View File

@@ -23,8 +23,6 @@ so it appears in SQL Lab's "Saved Queries" list and can be
reloaded/shared via URL.
"""
from __future__ import annotations
import logging
from fastmcp import Context
@@ -38,7 +36,6 @@ from superset.mcp_service.sql_lab.schemas import (
SaveSqlQueryRequest,
SaveSqlQueryResponse,
)
from superset.mcp_service.utils.schema_utils import parse_request
logger = logging.getLogger(__name__)
@@ -53,7 +50,6 @@ logger = logging.getLogger(__name__)
destructiveHint=False,
),
)
@parse_request(SaveSqlQueryRequest)
async def save_sql_query(
request: SaveSqlQueryRequest, ctx: Context
) -> SaveSqlQueryResponse: