From 5cfd8a7118906d10df27d927483b35751ef5fcc3 Mon Sep 17 00:00:00 2001
From: Zhang Minghan <zmh13054618081@dingtalk.com>
Date: Thu, 23 Nov 2023 21:01:01 +0800
Subject: [PATCH] fix retries error

---
 adapter/request.go           | 5 ++---
 addition/article/generate.go | 4 ++--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/adapter/request.go b/adapter/request.go
index 9f097e0..850b6c1 100644
--- a/adapter/request.go
+++ b/adapter/request.go
@@ -33,19 +33,18 @@ func NewChatRequest(props *ChatProps, hook globals.Hook) error {
 
 	retries := getRetries(props.MaxRetries)
 	props.Current++
-	if props.Current > 1 {
-		fmt.Println(fmt.Sprintf("retrying chat request for %s (attempt %d/%d, error: %s)", props.Model, props.Current, retries, err.Error()))
-	}
 
 	if IsAvailableError(err) {
 		if isQPSOverLimit(props.Model, err) {
 			// sleep for 0.5s to avoid qps limit
 
+			fmt.Println(fmt.Sprintf("qps limit for %s, sleep and retry (times: %d)", props.Model, props.Current))
 			time.Sleep(500 * time.Millisecond)
 			return NewChatRequest(props, hook)
 		}
 
 		if props.Current < retries {
+			fmt.Println(fmt.Sprintf("retrying chat request for %s (attempt %d/%d, error: %s)", props.Model, props.Current+1, retries, err.Error()))
 			return NewChatRequest(props, hook)
 		}
 	}
diff --git a/addition/article/generate.go b/addition/article/generate.go
index ec27fd1..b35f854 100644
--- a/addition/article/generate.go
+++ b/addition/article/generate.go
@@ -48,10 +48,10 @@ func CreateGenerationWorker(c *gin.Context, user *auth.User, model string, promp
 	titles := ParseTitle(title)
 	result := make(chan Response, len(titles))
 
-	for _, title := range titles {
+	for _, name := range titles {
 		go func(title string) {
 			result <- GenerateArticle(c, user, model, hash, title, prompt, enableWeb)
-		}(title)
+		}(name)
 	}
 
 	return len(titles), result