catch openai rate limit error (#7658)

2024-08-26 19:36:44 +08:00
parent 7cda73f192
commit 1473083a41
7 changed files with 53 additions and 0 deletions
--- a/api/controllers/web/error.py
+++ b/api/controllers/web/error.py
@@ -125,3 +125,11 @@ class WebSSOAuthRequiredError(BaseHTTPException):
    error_code = "web_sso_auth_required"
    description = "Web SSO authentication required."
    code = 401
+
+
+class InvokeRateLimitError(BaseHTTPException):
+    """Raised when the Invoke returns rate limit error."""
+
+    error_code = "rate_limit_error"
+    description = "Rate Limit Error"
+    code = 429