diff --git a/api/controllers/console/workspace/workspace.py b/api/controllers/console/workspace/workspace.py index 0f99bf62e3..eb12ec77d8 100644 --- a/api/controllers/console/workspace/workspace.py +++ b/api/controllers/console/workspace/workspace.py @@ -26,6 +26,7 @@ from libs.helper import TimestampField from libs.login import login_required from models.account import Tenant, TenantStatus from services.account_service import TenantService +from services.feature_service import FeatureService from services.file_service import FileService from services.workspace_service import WorkspaceService @@ -68,6 +69,11 @@ class TenantListApi(Resource): tenants = TenantService.get_join_tenants(current_user) for tenant in tenants: + features = FeatureService.get_features(tenant.id) + if features.billing.enabled: + tenant.plan = features.billing.subscription.plan + else: + tenant.plan = "sandbox" if tenant.id == current_user.current_tenant_id: tenant.current = True # Set current=True for current tenant return {"workspaces": marshal(tenants, tenants_fields)}, 200