feat 临时

backend/ai/ai_client.go

@@ -0,0 +1,59 @@
+package ai
+
+import (
+	"fmt"
+)
+
+type Client struct {
+	enableSearch   bool    // 是否开启搜索
+	searchStrategy string  // 搜索策略（开了搜索才生效）
+	temperature    float32 // 采样温度
+}
+
+func NewAIClient(enableSearch bool, searchStrategy string, temperature float32) *Client {
+	return &Client{
+		enableSearch:   enableSearch,
+		searchStrategy: searchStrategy,
+		temperature:    temperature,
+	}
+}
+
+func (c *Client) RequestAI(pre string, prompt []string, cb func(content string, i int) error) error {
+	inToken, outToken := int32(0), int32(0)
+	messages := []Message{{
+		Role:    "user",
+		Content: pre,
+	}}
+	for i, step := range prompt {
+		messages = append(messages, Message{
+			Role:    "user",
+			Content: step,
+		})
+		r, err := request("qwen3-max", Parameters{
+			//r, err := request("deepseek-v3.2", Parameters{
+			EnableSearch: c.enableSearch,
+			SearchOptions: SearchOptions{
+				SearchStrategy: c.searchStrategy,
+			},
+			ResultFormat: "message",
+			Temperature:  c.temperature,
+		}, messages)
+		if err != nil {
+			return err
+		}
+		inToken += r.Usage.InputTokens
+		outToken += r.Usage.OutputTokens
+		if len(r.Output.Choices) == 0 {
+			fmt.Printf("AI response err, r: %#+v\n", r)
+			return err
+		} else {
+			messages = append(messages, r.Output.Choices[0].Message)
+		}
+		fmt.Printf("AI 回答 %v: %v\n", i, r.Output.Choices[0].Message.Content)
+		if err = cb(r.Output.Choices[0].Message.Content, i); err != nil {
+			return err
+		}
+	}
+	fmt.Printf("AI finished, use token: in-%v out-%v\n", inToken, outToken)
+	return nil
+}