youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 10:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: self host not create workspace

Browse Source
This commit is contained in:
Joe
2024-09-27 14:36:31 +08:00
parent d3d0d6451a
commit 9a6918069d
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
api/services/account_service.py
View File

@ -177,7 +177,10 @@ class AccountService:
account = AccountService.create_account( account = AccountService.create_account(
email=email, name=name, interface_language=interface_language, password=password email=email, name=name, interface_language=interface_language, password=password
) )
TenantService.create_owner_tenant_if_not_exist(account=account)
# SELF_HOSTED Just create account, not create tenant
if dify_config.EDITION != "SELF_HOSTED":
TenantService.create_owner_tenant_if_not_exist(account=account)
return account return account
@ -662,15 +665,16 @@ class RegisterService:
if open_id is not None or provider is not None: if open_id is not None or provider is not None:
AccountService.link_account_integrate(provider, open_id, account) AccountService.link_account_integrate(provider, open_id, account)
if not dify_config.ALLOW_CREATE_WORKSPACE: if dify_config.EDITION != "SELF_HOSTED":
raise WorkSpaceNotAllowedCreateError() if not dify_config.ALLOW_CREATE_WORKSPACE:
raise WorkSpaceNotAllowedCreateError()
tenant = TenantService.create_tenant(f"{account.name}'s Workspace") tenant = TenantService.create_tenant(f"{account.name}'s Workspace")
TenantService.create_tenant_member(tenant, account, role="owner") TenantService.create_tenant_member(tenant, account, role="owner")
account.current_tenant = tenant account.current_tenant = tenant
tenant_was_created.send(tenant) tenant_was_created.send(tenant)
db.session.commit() db.session.commit()
except WorkSpaceNotAllowedCreateError: except WorkSpaceNotAllowedCreateError:
db.session.rollback() db.session.rollback()
raise raise