Skip to content

D
dify
Unverified Commit 1a5279a3 authored by zxhlyh's avatar zxhlyh Committed by GitHub
Browse files
Options

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

parent 7775f578
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
api/services/workspace_service.py
View file @ 1a5279a3
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: ...@@ -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(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment