AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-15 23:05:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

support workspace billing info (#15510)

Browse Source
This commit is contained in:
Jyong 2025-03-11 18:38:23 +08:00 committed by GitHub
parent 90a1508b87
commit da01b460fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/controllers/console/workspace/workspace.py
View File

@ -26,6 +26,7 @@ from libs.helper import TimestampField
from libs.login import login_required from libs.login import login_required
from models.account import Tenant, TenantStatus from models.account import Tenant, TenantStatus
from services.account_service import TenantService from services.account_service import TenantService
from services.feature_service import FeatureService
from services.file_service import FileService from services.file_service import FileService
from services.workspace_service import WorkspaceService from services.workspace_service import WorkspaceService
@ -68,6 +69,11 @@ class TenantListApi(Resource):
tenants = TenantService.get_join_tenants(current_user) tenants = TenantService.get_join_tenants(current_user)
for tenant in tenants: 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: if tenant.id == current_user.current_tenant_id:
tenant.current = True # Set current=True for current tenant tenant.current = True # Set current=True for current tenant
return {"workspaces": marshal(tenants, tenants_fields)}, 200 return {"workspaces": marshal(tenants, tenants_fields)}, 200