feat: register service WorkSpaceNotAllowedCreateError

langgenius · Sep 10, 2024 · 3f954e2 · 3f954e2
1 parent 4893631
commit 3f954e2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/services/account_service.py b/api/services/account_service.py
@@ -627,6 +627,9 @@ def register(
             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}")