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/core/errors/error.py
+++ b/api/core/errors/error.py
@@ -43,3 +43,8 @@ class ModelCurrentlyNotSupportError(Exception):
    Custom exception raised when the model not support
    """
    description = "Model Currently Not Support"
+
+
+class InvokeRateLimitError(Exception):
+    """Raised when the Invoke returns rate limit error."""
+    description = "Rate Limit Error"