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

fix: get billing info in self-hosted edition from current workspace (#1774)

Browse Source
This commit is contained in:
zxhlyh 2023-12-18 17:54:16 +08:00 committed by GitHub
parent 7775f5785f
commit 1a5279a3ef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/services/workspace_service.py
View File

@ -1,3 +1,4 @@
from flask import current_app
from flask_login import current_user from flask_login import current_user
from extensions.ext_database import db from extensions.ext_database import db
from models.account import Tenant, TenantAccountJoin, TenantAccountJoinRole from models.account import Tenant, TenantAccountJoin, TenantAccountJoinRole
@ -31,10 +32,12 @@ class WorkspaceService:
).first() ).first()
tenant_info['role'] = tenant_account_join.role tenant_info['role'] = tenant_account_join.role
billing_info = BillingService.get_info(tenant_info['id']) edition = current_app.config['EDITION']
if edition == 'CLOUD':
billing_info = BillingService.get_info(tenant_info['id'])
if billing_info['can_replace_logo'] and TenantService.has_roles(tenant, [TenantAccountJoinRole.OWNER, TenantAccountJoinRole.ADMIN]): if billing_info['can_replace_logo'] and TenantService.has_roles(tenant, [TenantAccountJoinRole.OWNER, TenantAccountJoinRole.ADMIN]):
tenant_info['custom_config'] = tenant.custom_config_dict tenant_info['custom_config'] = tenant.custom_config_dict
# Get providers # Get providers
providers = db.session.query(Provider).filter( providers = db.session.query(Provider).filter(