fix: raise

2 years ago · 148747c3c6
parent ca5b29438b
commit 148747c3c6
2 changed files with 2 additions and 2 deletions
--- a/api/controllers/console/auth/forgot_password.py
+++ b/api/controllers/console/auth/forgot_password.py
@ -111,7 +111,7 @@ class ForgotPasswordResetApi(Resource):
            tenant = TenantService.get_join_tenants(account)
            if not tenant:
                if not dify_config.ALLOW_CREATE_WORKSPACE:
-                    return NotAllowedCreateWorkspace()
+                    raise NotAllowedCreateWorkspace()
                else:
                    tenant = TenantService.create_tenant(f"{account.name}'s Workspace")
                    TenantService.create_tenant_member(tenant, account, role="owner")
--- a/api/controllers/console/auth/login.py
+++ b/api/controllers/console/auth/login.py
@ -135,7 +135,7 @@ class EmailCodeLoginApi(Resource):
        tenant = TenantService.get_join_tenants(account)
        if not tenant:
            if not dify_config.ALLOW_CREATE_WORKSPACE:
-                return NotAllowedCreateWorkspace()
+                raise NotAllowedCreateWorkspace()
            else:
                tenant = TenantService.create_tenant(f"{account.name}'s Workspace")
                TenantService.create_tenant_member(tenant, account, role="owner")