feat(mcp): add sql to search columns for query/saved-query list tools; add test

Author: aminghadersohi

superset/mcp_service/app.py

@@ -144,6 +144,10 @@ def get_default_instructions(
 - execute_sql: Execute SQL queries and get results (requires database_id)
 - save_sql_query: Save a SQL query to Saved Queries list
 - open_sql_lab_with_context: Generate SQL Lab URL with pre-filled sql
+- list_saved_queries: List saved SQL queries with filtering and search (1-based pagination)
+- get_saved_query_info: Get saved query details by ID or UUID
+- list_queries: List SQL query history with filtering and search (most recent first)
+- get_query_info: Get SQL query history details by ID
 
 Schema Discovery:
 - get_schema: Get schema metadata for chart/dataset/dashboard (columns, filters)

superset/mcp_service/query/tool/list_queries.py

@@ -104,7 +104,7 @@ def _serialize_query(obj: object, cols: list[str] | None) -> QueryInfo | None:
             item_serializer=_serialize_query,
             filter_type=QueryFilter,
             default_columns=DEFAULT_QUERY_COLUMNS,
-            search_columns=["tab_name"],
+            search_columns=["tab_name", "sql"],
             list_field_name="queries",
             output_list_schema=QueryList,
             all_columns=ALL_QUERY_COLUMNS,

superset/mcp_service/saved_query/tool/list_saved_queries.py

@@ -105,7 +105,7 @@ def _serialize_saved_query(
             item_serializer=_serialize_saved_query,
             filter_type=SavedQueryFilter,
             default_columns=DEFAULT_SAVED_QUERY_COLUMNS,
-            search_columns=["label", "description"],
+            search_columns=["label", "description", "sql"],
             list_field_name="saved_queries",
             output_list_schema=SavedQueryList,
             all_columns=ALL_SAVED_QUERY_COLUMNS,

tests/unit_tests/mcp_service/saved_query/tool/test_saved_query_tools.py

@@ -236,6 +236,23 @@ async def test_list_saved_queries_empty(mock_list, mcp_server):
         assert data["total_count"] == 0
 
 
+@patch("superset.daos.query.SavedQueryDAO.find_by_id")
+@pytest.mark.asyncio
+async def test_get_saved_query_info_by_uuid(mock_find, mcp_server):
+    """Test get saved query info by UUID string."""
+    saved_query = create_mock_saved_query(uuid="a1b2c3d4-5678-90ab-cdef-1234567890ab")
+    mock_find.return_value = saved_query
+    async with Client(mcp_server) as client:
+        result = await client.call_tool(
+            "get_saved_query_info",
+            {"request": {"identifier": "a1b2c3d4-5678-90ab-cdef-1234567890ab"}},
+        )
+        assert result.content is not None
+        data = json.loads(result.content[0].text)
+        assert data["id"] == 1
+        assert data["uuid"] == "a1b2c3d4-5678-90ab-cdef-1234567890ab"
+
+
 @patch("superset.daos.query.SavedQueryDAO.list")
 @pytest.mark.asyncio
 async def test_list_saved_queries_pagination_info(mock_list, mcp_server):