mirror of
https://github.com/apache/superset.git
synced 2026-04-13 21:24:28 +00:00
108 lines
3.9 KiB
Python
108 lines
3.9 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
"""
|
|
System resources for providing instance configuration and stats
|
|
"""
|
|
|
|
import logging
|
|
|
|
from superset.mcp_service.app import mcp
|
|
from superset.mcp_service.auth import mcp_auth_hook
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@mcp.resource("instance://metadata")
|
|
@mcp_auth_hook
|
|
def get_instance_metadata_resource() -> str:
|
|
"""
|
|
Provide comprehensive metadata about the instance.
|
|
|
|
This resource gives LLMs context about:
|
|
- Available datasets and their popularity
|
|
- Dashboard and chart statistics
|
|
- Database connections
|
|
- Popular queries and usage patterns
|
|
- Available visualization types
|
|
- Feature flags and configuration
|
|
"""
|
|
try:
|
|
# Import the shared core and DAOs at runtime
|
|
# Create a shared core instance for the resource
|
|
from typing import Any, cast, Type
|
|
|
|
from superset.daos.base import BaseDAO
|
|
from superset.daos.chart import ChartDAO
|
|
from superset.daos.dashboard import DashboardDAO
|
|
from superset.daos.database import DatabaseDAO
|
|
from superset.daos.dataset import DatasetDAO
|
|
from superset.daos.tag import TagDAO
|
|
from superset.daos.user import UserDAO
|
|
from superset.mcp_service.mcp_core import InstanceInfoCore
|
|
from superset.mcp_service.system.schemas import InstanceInfo
|
|
from superset.mcp_service.system.system_utils import (
|
|
calculate_dashboard_breakdown,
|
|
calculate_database_breakdown,
|
|
calculate_instance_summary,
|
|
calculate_popular_content,
|
|
calculate_recent_activity,
|
|
)
|
|
|
|
instance_info_core = InstanceInfoCore(
|
|
dao_classes={
|
|
"dashboards": cast(Type[BaseDAO[Any]], DashboardDAO),
|
|
"charts": cast(Type[BaseDAO[Any]], ChartDAO),
|
|
"datasets": cast(Type[BaseDAO[Any]], DatasetDAO),
|
|
"databases": cast(Type[BaseDAO[Any]], DatabaseDAO),
|
|
"users": cast(Type[BaseDAO[Any]], UserDAO),
|
|
"tags": cast(Type[BaseDAO[Any]], TagDAO),
|
|
},
|
|
output_schema=InstanceInfo,
|
|
metric_calculators={
|
|
"instance_summary": calculate_instance_summary,
|
|
"recent_activity": calculate_recent_activity,
|
|
"dashboard_breakdown": calculate_dashboard_breakdown,
|
|
"database_breakdown": calculate_database_breakdown,
|
|
"popular_content": calculate_popular_content,
|
|
},
|
|
time_windows={
|
|
"recent": 7,
|
|
"monthly": 30,
|
|
"quarterly": 90,
|
|
},
|
|
logger=logger,
|
|
)
|
|
|
|
# Use the shared core's resource method
|
|
return instance_info_core.get_resource()
|
|
|
|
except Exception as e:
|
|
logger.error("Error generating instance metadata: %s", e)
|
|
# Return minimal metadata on error
|
|
from superset.utils import json
|
|
|
|
return json.dumps(
|
|
{
|
|
"error": "Unable to fetch complete metadata",
|
|
"tips": [
|
|
"Use list_datasets to explore available data",
|
|
"Use get_instance_info for basic stats",
|
|
],
|
|
}
|
|
)
|