QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-08-05 13:32:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(mcp): add event_logger instrumentation to MCP tools (#37859)

Browse Source 
(cherry picked from commit 4dfece9ee5)
(cherry picked from commit adfe91c9167c2d5cbd98d0a28f6e945d3fb81244)
This commit is contained in:
Amin Ghadersohi
2026-02-12 10:50:20 -05:00
committed by Joe Li
parent 82d8293f86
commit 94e4cb68e1
22 changed files with 975 additions and 491 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

@@ -26,6 +26,7 @@ import logging
from fastmcp import Context
from superset.extensions import event_logger
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
@@ -67,7 +68,8 @@ async def execute_sql(request: ExecuteSqlRequest, ctx: Context) -> ExecuteSqlRes
try:
# Use the ExecuteSqlCore to handle all the logic
sql_tool = ExecuteSqlCore(use_command_mode=False, logger=logger)
result = sql_tool.run_tool(request)
with event_logger.log_context(action="mcp.execute_sql.query_execution"):
result = sql_tool.run_tool(request)
# Log successful execution
if hasattr(result, "data") and result.data: