Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 2 additions & 19 deletions superset/mcp_service/system/tool/get_instance_info.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@
from superset.mcp_service.system.schemas import (
GetSupersetInstanceInfoRequest,
InstanceInfo,
UserInfo,
serialize_user_object,
)
from superset.mcp_service.system.system_utils import (
calculate_dashboard_breakdown,
Expand Down Expand Up @@ -110,24 +110,7 @@ def get_instance_info(
# Attach the authenticated user's identity to the response
user = getattr(g, "user", None)
if user is not None:
raw_roles = getattr(user, "roles", None)
user_roles = []
if raw_roles is not None:
try:
user_roles = [
role.name for role in raw_roles if hasattr(role, "name")
]
except TypeError:
logger.debug("Could not iterate user.roles: %s", type(raw_roles))
user_roles = []
result.current_user = UserInfo(
id=getattr(user, "id", None),
username=getattr(user, "username", None),
first_name=getattr(user, "first_name", None),
last_name=getattr(user, "last_name", None),
email=getattr(user, "email", None),
roles=user_roles,
)
result.current_user = serialize_user_object(user)

return result

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -221,6 +221,7 @@ async def test_get_instance_info_returns_current_user(self, mcp_server):
mock_g_user.first_name = "Sophie"
mock_g_user.last_name = "Beaumont"
mock_g_user.email = "sophie@preset.io"
mock_g_user.active = True
mock_g_user.roles = [mock_role]

with (
Expand Down
Loading