update sth. at 2024-12-17 23:06:16

README.md

@@ -56,6 +56,7 @@ curl --location 'https://sanbo1200-duck2api.hf.space/hf/v1/chat/completions' \
      "messages": [{"role":"system","content":"你是一个辅助机器人"},{"role": "user", "content": "你的知识库最后什么日期"}],
      "stream": true
    }'
+
 ```
 
 ### 国内反代地址(maybe)

internal/duckgo/request.go

@@ -28,6 +28,32 @@ type XqdgToken struct {
 	ExpireAt time.Time  `json:"expire"`
 }
 
+// func InitXVQD(client httpclient.AuroraHttpClient, proxyUrl string) (string, error) {
+// 	if Token == nil {
+// 		Token = &XqdgToken{
+// 			Token: "",
+// 			M:     sync.Mutex{},
+// 		}
+// 	}
+// 	Token.M.Lock()
+// 	defer Token.M.Unlock()
+// 	if Token.Token == "" || Token.ExpireAt.Before(time.Now()) {
+// 		status, err := postStatus(client, proxyUrl)
+// 		if err != nil {
+// 			return "", err
+// 		}
+// 		defer status.Body.Close()
+// 		token := status.Header.Get("x-vqd-4")
+// 		if token == "" {
+// 			return "", errors.New("no x-vqd-4 token")
+// 		}
+// 		Token.Token = token
+// 		Token.ExpireAt = time.Now().Add(time.Minute * 3)
+// 	}
+//
+// 	return Token.Token, nil
+// }
+
 func InitXVQD(client httpclient.AuroraHttpClient, proxyUrl string) (string, error) {
 	if Token == nil {
 		Token = &XqdgToken{
@@ -37,20 +63,49 @@ func InitXVQD(client httpclient.AuroraHttpClient, proxyUrl string) (string, erro
 	}
 	Token.M.Lock()
 	defer Token.M.Unlock()
+
+	// 如果 token 已经失效或为空，尝试重新获取
 	if Token.Token == "" || Token.ExpireAt.Before(time.Now()) {
-		status, err := postStatus(client, proxyUrl)
-		if err != nil {
-			return "", err
-		}
-		defer status.Body.Close()
-		token := status.Header.Get("x-vqd-4")
-		if token == "" {
-			return "", errors.New("no x-vqd-4 token")
+		const maxRetries = 3                  // 设置最大重试次数
+		const retryInterval = 2 * time.Second // 设置每次重试间隔
+
+		for retries := 0; retries < maxRetries; retries++ {
+			// 发送请求
+			status, err := postStatus(client, proxyUrl)
+			if err != nil {
+				// 如果是非网络类错误，直接返回
+				if retries == maxRetries-1 {
+					return "", err
+				}
+				// 网络错误，等待重试
+				time.Sleep(retryInterval)
+				continue
+			}
+
+			// 获取 x-vqd-4 token
+			defer status.Body.Close()
+			token := status.Header.Get("x-vqd-4")
+			if token == "" {
+				// 如果没有获取到 token，判断是否是最后一次重试
+				if retries == maxRetries-1 {
+					return "", errors.New("no x-vqd-4 token after retries")
+				}
+				// 没有获取到 token，等待重试
+				time.Sleep(retryInterval)
+				continue
+			}
+
+			// 成功获取到 token
+			Token.Token = token
+			Token.ExpireAt = time.Now().Add(time.Minute * 3)
+			return Token.Token, nil
 		}
-		Token.Token = token
-		Token.ExpireAt = time.Now().Add(time.Minute * 3)
+
+		// 重试完仍未成功，返回错误
+		return "", errors.New("failed to get x-vqd-4 token after retries")
 	}
 
+	// 如果 token 已存在且有效，直接返回
 	return Token.Token, nil
 }