From ced89398a5abb74d7a53544ae3b9daa00c71da12 Mon Sep 17 00:00:00 2001
From: JustSong <songquanpeng@foxmail.com>
Date: Thu, 18 May 2023 15:27:15 +0800
Subject: [PATCH] chore: rewrite 429 prompt text (close #96)

---
 controller/relay.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/controller/relay.go b/controller/relay.go
index 4803794b..b10c27f2 100644
--- a/controller/relay.go
+++ b/controller/relay.go
@@ -75,6 +75,9 @@ func countToken(text string) int {
 func Relay(c *gin.Context) {
 	err := relayHelper(c)
 	if err != nil {
+		if err.StatusCode == http.StatusTooManyRequests {
+			err.OpenAIError.Message = "负载已满，请稍后再试，或升级账户以提升服务质量。"
+		}
 		c.JSON(err.StatusCode, gin.H{
 			"error": err.OpenAIError,
 		})