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/console/app/error.py
+++ b/api/controllers/console/app/error.py
@@ -119,3 +119,11 @@ class TracingConfigCheckError(BaseHTTPException):
    error_code = "trace_config_check_error"
    description = "Invalid Credentials."
    code = 400
+
+
+class InvokeRateLimitError(BaseHTTPException):
+    """Raised when the Invoke returns rate limit error."""
+
+    error_code = "rate_limit_error"
+    description = "Rate Limit Error"
+    code = 429