feat: 修改AI接口请求，增加重试机制和状态码打印；优化AUR包查询逻辑，移除不必要的转换和循环

workers/ai.go

@@ -3,6 +3,7 @@ package workers
 import (
 	"fmt"
 	"log"
+	"net/http"
 	"strings"
 	"time"
 
@@ -131,10 +132,10 @@ func handleChatRequest(OPENAI_API_KEY, OPENAI_BaseURL, MODEL, rawMsg, UID string
 		"frequency_penalty": 0,
 		"top_p":             1,
 	}
-	const maxRetry = 2
-	for retries := 0; retries <= maxRetry; retries++ {
+
 	request := gorequest.New()
 	resp, body, errs := request.Post(OPENAI_BaseURL).
+		Retry(3, 5*time.Second, http.StatusServiceUnavailable, http.StatusBadGateway).
 		Set("Content-Type", "application/json").
 		Set("Authorization", "Bearer "+OPENAI_API_KEY).
 		Send(requestBody).
@@ -160,11 +161,7 @@ func handleChatRequest(OPENAI_API_KEY, OPENAI_BaseURL, MODEL, rawMsg, UID string
 			log.Println("choices为空")
 			return "api解析失败"
 		}
-		} else {
-			log.Printf("请求失败，状态码：%d，重试中...（%d/%d）\n", resp.StatusCode, retries+1, maxRetry)
-			time.Sleep(time.Second * 1)
-			// return "请求失败: " + fmt.Sprintf("%d", resp.StatusCode)
-		}
+
 	}
 	return "请求失败!"
 }

workers/pkg.go

@@ -93,7 +93,11 @@ func (a *Pkg) GetMsg() string {
 			searchResult = searchMap["results"].([]interface{})[0].(map[string]interface{})
 
 		} else {
-			searchUrl := "https://aur.archlinux.org/rpc/v5/search/" + suggestions[0] + "?by=name"
+
+			maxVotes := 0.0
+			var searchResults []map[string]interface{}
+			for _, suggestion := range suggestions {
+				searchUrl := "https://aur.archlinux.org/rpc/v5/info/" + suggestion
 				_, body, errs = request.Get(searchUrl).End()
 				if len(errs) > 0 {
 					fmt.Println("searchUrl err:", errs)
@@ -103,23 +107,23 @@ func (a *Pkg) GetMsg() string {
 				if err != nil {
 					fmt.Println(err)
 				}
+				results := searchMap["results"].([]interface{})
 
-			searchResults := searchMap["results"].([]interface{})
+				searchResults = append(searchResults, results[0].(map[string]interface{}))
+				// searchResults = append(searchResults, searchMap["results"].([]interface{})[0].(map[string]interface{}))
 				// println("searchResults:", len(searchResults))
-			maxVotes := 0.0
+			}
 
 			// 增加根据投票人数排序
-			if len(searchResults) > 1 {
 			for _, result := range searchResults {
-					resultMap := result.(map[string]interface{})
-					numVotes := resultMap["NumVotes"].(float64)
+				// resultMap := result.(map[string]interface{})
+				numVotes := result["NumVotes"].(float64)
 				if numVotes > maxVotes {
 					maxVotes = numVotes
-						searchResult = resultMap
+					searchResult = result
 				}
 			}
-				// fmt.Println(len(searchResult))
-			}
+
 		}
 		maintainer, ok := searchResult["Maintainer"].(string)
 		if !ok {
@@ -150,9 +154,7 @@ func (a *Pkg) GetMsg() string {
 
 	}
 	result := resultSlipe[0].(map[string]interface{})
-	// 输出结果字符串
-	// fmt.Println("result:", result)
-	// if result != nil {
+
 	var msg string
 	last_update := result["last_update"].(string)
 	last_update = strings.Replace(last_update, "T", " ", 1)