From 5165894b1ad2209fe681841173447d1808226065 Mon Sep 17 00:00:00 2001 From: Joe <1264204425@qq.com> Date: Sun, 29 Sep 2024 21:54:05 +0800 Subject: [PATCH] feat: add invite menmber workspace all create --- api/services/account_service.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/api/services/account_service.py b/api/services/account_service.py index caebeb9401..4649462760 100644 --- a/api/services/account_service.py +++ b/api/services/account_service.py @@ -684,17 +684,15 @@ class RegisterService: if open_id is not None or provider is not None: AccountService.link_account_integrate(provider, open_id, account) - if not dify_config.ALLOW_CREATE_WORKSPACE: - raise WorkSpaceNotAllowedCreateError() - tenant = TenantService.create_tenant(f"{account.name}'s Workspace") - TenantService.create_tenant_member(tenant, account, role="owner") - account.current_tenant = tenant - tenant_was_created.send(tenant) + if dify_config.ALLOW_CREATE_WORKSPACE: + tenant = TenantService.create_tenant(f"{account.name}'s Workspace") + TenantService.create_tenant_member(tenant, account, role="owner") + account.current_tenant = tenant + tenant_was_created.send(tenant) db.session.commit() except WorkSpaceNotAllowedCreateError: db.session.rollback() - raise except Exception as e: db.session.rollback() logging.error(f"Register failed: {e}")