From 9e4ee2beb10c6dadcb6e6fd6c192ac8c7955a7a2 Mon Sep 17 00:00:00 2001 From: Joe <1264204425@qq.com> Date: Thu, 26 Sep 2024 14:59:13 +0800 Subject: [PATCH] feat: add EmailPasswordLoginLimitError --- api/services/account_service.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/services/account_service.py b/api/services/account_service.py index 53e983d2e5..cf4d65a27d 100644 --- a/api/services/account_service.py +++ b/api/services/account_service.py @@ -12,6 +12,7 @@ from werkzeug.exceptions import Unauthorized from configs import dify_config from constants.languages import language_timezone_mapping, languages +from controllers.console.auth.error import EmailPasswordLoginLimitError from events.tenant_event import tenant_was_created from extensions.ext_redis import redis_client from libs.helper import RateLimiter, TokenManager @@ -289,7 +290,7 @@ class AccountService: cls, account: Optional[Account] = None, email: Optional[str] = None, language: Optional[str] = "en-US" ): if cls.email_code_login_rate_limiter.is_rate_limited(email): - raise RateLimitExceededError(f"Rate limit exceeded for email: {email}. Please try again later.") + raise EmailPasswordLoginLimitError() code = "".join([str(random.randint(0, 9)) for _ in range(6)]) token = TokenManager.generate_token(