添加对上下文的改动（测试）

api/main.go

@@ -2,17 +2,37 @@ package handler
 
 import (
 	"bufio"
+	"bytes"
 	"encoding/json"
 	"fmt"
+	"io"
+	"mime/multipart"
 	"net/http"
+	"os"
+	"path/filepath"
 	"strings"
 	"time"
+
+	"github.com/google/uuid"
+)
+
+// TokenCount 定义了 token 计数的结构
+type TokenCount struct {
+	PromptTokens     int `json:"prompt_tokens"`
+	CompletionTokens int `json:"completion_tokens"`
+	TotalTokens      int `json:"total_tokens"`
+}
+
+const (
+	MaxContextTokens = 4000 // 最大上下文 token 数
 )
 
+// YouChatResponse 定义了从 You.com API 接收的单个 token 的结构。
 type YouChatResponse struct {
 	YouChatToken string `json:"youChatToken"`
 }
 
+// OpenAIStreamResponse 定义了 OpenAI API 流式响应的结构。
 type OpenAIStreamResponse struct {
 	ID      string   `json:"id"`
 	Object  string   `json:"object"`
@@ -21,27 +41,32 @@ type OpenAIStreamResponse struct {
 	Choices []Choice `json:"choices"`
 }
 
+// Choice 定义了 OpenAI 流式响应中 choices 数组的单个元素的结构。
 type Choice struct {
 	Delta        Delta  `json:"delta"`
 	Index        int    `json:"index"`
 	FinishReason string `json:"finish_reason"`
 }
 
+// Delta 定义了流式响应中表示增量内容的结构。
 type Delta struct {
 	Content string `json:"content"`
 }
 
+// OpenAIRequest 定义了 OpenAI API 请求体的结构。
 type OpenAIRequest struct {
 	Messages []Message `json:"messages"`
 	Stream   bool      `json:"stream"`
 	Model    string    `json:"model"`
 }
 
+// Message 定义了 OpenAI 聊天消息的结构。
 type Message struct {
 	Role    string `json:"role"`
 	Content string `json:"content"`
 }
 
+// OpenAIResponse 定义了 OpenAI API 非流式响应的结构。
 type OpenAIResponse struct {
 	ID      string         `json:"id"`
 	Object  string         `json:"object"`
@@ -50,17 +75,20 @@ type OpenAIResponse struct {
 	Choices []OpenAIChoice `json:"choices"`
 }
 
+// OpenAIChoice 定义了 OpenAI 非流式响应中 choices 数组的单个元素的结构。
 type OpenAIChoice struct {
 	Message      Message `json:"message"`
 	Index        int     `json:"index"`
 	FinishReason string  `json:"finish_reason"`
 }
 
+// ModelResponse 定义了 /v1/models 响应的结构。
 type ModelResponse struct {
 	Object string        `json:"object"`
 	Data   []ModelDetail `json:"data"`
 }
 
+// ModelDetail 定义了模型列表中单个模型的详细信息。
 type ModelDetail struct {
 	ID      string `json:"id"`
 	Object  string `json:"object"`
@@ -68,6 +96,7 @@ type ModelDetail struct {
 	OwnedBy string `json:"owned_by"`
 }
 
+// modelMap 存储 OpenAI 模型名称到 You.com 模型名称的映射。
 var modelMap = map[string]string{
 	"deepseek-reasoner":  "deepseek_r1",
 	"deepseek-chat":      "deepseek_v3",
@@ -94,6 +123,7 @@ var modelMap = map[string]string{
 	"command-r-plus":     "command_r_plus",
 }
 
+// getReverseModelMap 创建并返回 modelMap 的反向映射（You.com 模型名称 -> OpenAI 模型名称）。
 func getReverseModelMap() map[string]string {
 	reverse := make(map[string]string, len(modelMap))
 	for k, v := range modelMap {
@@ -102,24 +132,43 @@ func getReverseModelMap() map[string]string {
 	return reverse
 }
 
+// mapModelName 将 OpenAI 模型名称映射到 You.com 模型名称。
 func mapModelName(openAIModel string) string {
 	if mappedModel, exists := modelMap[openAIModel]; exists {
 		return mappedModel
 	}
-	return "deepseek_v3"
+	return "deepseek_v3" // 默认模型
 }
 
+// reverseMapModelName 将 You.com 模型名称映射回 OpenAI 模型名称。
 func reverseMapModelName(youModel string) string {
 	reverseMap := getReverseModelMap()
 	if mappedModel, exists := reverseMap[youModel]; exists {
 		return mappedModel
 	}
-	return "deepseek-chat"
+	return "deepseek-chat" // 默认模型
 }
 
+// originalModel 存储原始的 OpenAI 模型名称。
 var originalModel string
 
+// NonceResponse 定义了获取 nonce 的响应结构
+type NonceResponse struct {
+	Uuid string `json:"uuid"`
+}
+
+// UploadResponse 定义了文件上传的响应结构
+type UploadResponse struct {
+	Filename     string `json:"filename"`
+	UserFilename string `json:"user_filename"`
+}
+
+// 定义最大查询长度
+const MaxQueryLength = 2000
+
+// Handler 是处理所有传入 HTTP 请求的主处理函数。
 func Handler(w http.ResponseWriter, r *http.Request) {
+	// 处理 /v1/models 请求（列出可用模型）
 	if r.URL.Path == "/v1/models" || r.URL.Path == "/api/v1/models" {
 		w.Header().Set("Content-Type", "application/json")
 		w.Header().Set("Access-Control-Allow-Origin", "*")
@@ -151,7 +200,8 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	if r.URL.Path != "/v1/chat/completions" && r.URL.Path != "/api/v1/chat/completions" {
+	// 处理非 /v1/chat/completions 请求（服务状态检查）
+	if r.URL.Path != "/v1/chat/completions" && r.URL.Path != "/none/v1/chat/completions" && r.URL.Path != "/such/chat/completions" {
 		w.Header().Set("Content-Type", "application/json")
 		json.NewEncoder(w).Encode(map[string]string{
 			"status":  "You2Api Service Running...",
@@ -160,6 +210,7 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	// 设置 CORS 头部
 	w.Header().Set("Access-Control-Allow-Origin", "*")
 	w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
 	w.Header().Set("Access-Control-Allow-Headers", "*")
@@ -169,13 +220,15 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	// 验证 Authorization 头部
 	authHeader := r.Header.Get("Authorization")
 	if !strings.HasPrefix(authHeader, "Bearer ") {
 		http.Error(w, "Missing or invalid authorization header", http.StatusUnauthorized)
 		return
 	}
-	dsToken := strings.TrimPrefix(authHeader, "Bearer ")
+	dsToken := strings.TrimPrefix(authHeader, "Bearer ") // 提取 DS token
 
+	// 解析 OpenAI 请求体
 	var openAIReq OpenAIRequest
 	if err := json.NewDecoder(r.Body).Decode(&openAIReq); err != nil {
 		http.Error(w, "Invalid request body", http.StatusBadRequest)
@@ -183,13 +236,25 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 	}
 
 	originalModel = openAIReq.Model
-	lastMessage := openAIReq.Messages[len(openAIReq.Messages)-1].Content
+
+	// 转换 system 消息为 user 消息
+	openAIReq.Messages = convertSystemToUser(openAIReq.Messages)
+
+	// 计算 token 数（使用字符估算方法）
+	totalTokens, err := countTokens(openAIReq.Messages)
+	if err != nil {
+		http.Error(w, "Failed to count tokens", http.StatusInternalServerError)
+		return
+	}
+
+	// 构建 You.com 聊天历史
 	var chatHistory []map[string]interface{}
 	for _, msg := range openAIReq.Messages {
 		chatMsg := map[string]interface{}{
 			"question": msg.Content,
 			"answer":   "",
 		}
+		// 如果是 assistant 的消息, 则交换 question 和 answer
 		if msg.Role == "assistant" {
 			chatMsg["question"] = ""
 			chatMsg["answer"] = msg.Content
@@ -197,27 +262,114 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 		chatHistory = append(chatHistory, chatMsg)
 	}
 
-	chatHistoryJSON, _ := json.Marshal(chatHistory)
+	chatHistoryJSON, _ := json.Marshal(chatHistory) // 将聊天历史序列化为 JSON
 
+	// 创建 You.com API 请求
 	youReq, _ := http.NewRequest("GET", "https://you.com/api/streamingSearch", nil)
 
-	q := youReq.URL.Query()
-	q.Add("q", lastMessage)
-	q.Add("page", "1")
-	q.Add("count", "10")
-	q.Add("safeSearch", "Moderate")
-	q.Add("mkt", "zh-HK")
-	q.Add("enable_worklow_generation_ux", "true")
-	q.Add("domain", "youchat")
-	q.Add("use_personalization_extraction", "true")
-	q.Add("pastChatLength", fmt.Sprintf("%d", len(chatHistory)-1))
-	q.Add("selectedChatMode", "custom")
-	q.Add("selectedAiModel", mapModelName(openAIReq.Model))
-	q.Add("enable_agent_clarification_questions", "true")
-	q.Add("use_nested_youchat_updates", "true")
-	q.Add("chat", string(chatHistoryJSON))
-	youReq.URL.RawQuery = q.Encode()
+	// 生成必要的 ID
+	chatId := uuid.New().String()
+	conversationTurnId := uuid.New().String()
+	traceId := fmt.Sprintf("%s|%s|%s", chatId, conversationTurnId, time.Now().Format(time.RFC3339))
+
+	// 如果超过限制，使用文件上传
+	if totalTokens > MaxContextTokens {
+		// 1. 获取 nonce
+		nonceResp, err := getNonce(dsToken)
+		if err != nil {
+			http.Error(w, "Failed to get nonce", http.StatusInternalServerError)
+			return
+		}
+
+		// 2. 创建临时文件，包含所有消息内容
+		var fileContent strings.Builder
+		for i, msg := range openAIReq.Messages {
+			if i > 0 {
+				fileContent.WriteString("\n\n")
+			}
+			fileContent.WriteString(fmt.Sprintf("%s: %s", msg.Role, msg.Content))
+		}
+
+		tempFile := fmt.Sprintf("temp_%s.txt", nonceResp.Uuid)
+		if err := os.WriteFile(tempFile, []byte(fileContent.String()), 0644); err != nil {
+			http.Error(w, "Failed to create temp file", http.StatusInternalServerError)
+			return
+		}
+		defer os.Remove(tempFile)
+
+		// 3. 上传文件
+		uploadResp, err := uploadFile(dsToken, tempFile)
+		if err != nil {
+			http.Error(w, "Failed to upload file", http.StatusInternalServerError)
+			return
+		}
+
+		// 4. 修改消息列表，只保留文件引用
+		openAIReq.Messages = []Message{
+			{
+				Role:    "user",
+				Content: fmt.Sprintf("Please review the attached file: %s", uploadResp.UserFilename),
+			},
+		}
+
+		// 5. 添加文件源信息
+		sources := []map[string]interface{}{
+			{
+				"source_type":   "user_file",
+				"filename":      uploadResp.Filename,
+				"user_filename": uploadResp.UserFilename,
+				"size_bytes":    len(fileContent.String()),
+			},
+		}
+		sourcesJSON, _ := json.Marshal(sources)
+
+		// 更新查询参数
+		q := youReq.URL.Query()
+		q.Add("sources", string(sourcesJSON))
+		q.Add("chatId", chatId)
+		q.Add("queryTraceId", chatId)
+		q.Add("conversationTurnId", conversationTurnId)
+		q.Add("traceId", traceId)
+		q.Add("q", openAIReq.Messages[0].Content)
+		q.Add("page", "1")
+		q.Add("count", "10")
+		q.Add("safeSearch", "Moderate")
+		q.Add("mkt", "zh-HK")
+		q.Add("enable_worklow_generation_ux", "true")
+		q.Add("domain", "youchat")
+		q.Add("use_personalization_extraction", "true")
+		q.Add("pastChatLength", fmt.Sprintf("%d", len(chatHistory)-1))
+		q.Add("selectedChatMode", "custom")
+		q.Add("selectedAiModel", mapModelName(openAIReq.Model))
+		q.Add("enable_agent_clarification_questions", "true")
+		q.Add("use_nested_youchat_updates", "true")
+		q.Add("chat", string(chatHistoryJSON))
+		youReq.URL.RawQuery = q.Encode()
+	} else {
+		// 构建常规查询参数
+		q := youReq.URL.Query()
+		q.Add("q", openAIReq.Messages[len(openAIReq.Messages)-1].Content)
+		q.Add("chatId", chatId)
+		q.Add("queryTraceId", chatId)
+		q.Add("conversationTurnId", conversationTurnId)
+		q.Add("traceId", traceId)
+		q.Add("page", "1")
+		q.Add("count", "10")
+		q.Add("safeSearch", "Moderate")
+		q.Add("mkt", "zh-HK")
+		q.Add("enable_worklow_generation_ux", "true")
+		q.Add("domain", "youchat")
+		q.Add("use_personalization_extraction", "true")
+		q.Add("pastChatLength", fmt.Sprintf("%d", len(chatHistory)-1))
+		q.Add("selectedChatMode", "custom")
+		q.Add("selectedAiModel", mapModelName(openAIReq.Model))
+		q.Add("enable_agent_clarification_questions", "true")
+		q.Add("use_nested_youchat_updates", "true")
+		q.Add("chat", string(chatHistoryJSON))
+		youReq.URL.RawQuery = q.Encode()
+	}
 
+	// 设置 You.com API 请求头
 	youReq.Header = http.Header{
 		"sec-ch-ua-platform":         {"Windows"},
 		"Cache-Control":              {"no-cache"},
@@ -227,7 +379,7 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 		"sec-ch-ua-mobile":           {"?0"},
 		"sec-ch-ua-arch":             {"x86"},
 		"sec-ch-ua-full-version":     {"133.0.3065.39"},
-		"Accept":                     {"text/event-stream"},
+		"Accept":                     {"text/event-stream"}, // 重要：接受 SSE 流
 		"User-Agent":                 {"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36 Edg/133.0.0.0"},
 		"sec-ch-ua-platform-version": {"19.0.0"},
 		"Sec-Fetch-Site":             {"same-origin"},
@@ -236,6 +388,7 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 		"Host":                       {"you.com"},
 	}
 
+	// 设置 You.com API 请求的 Cookie
 	cookies := getCookies(dsToken)
 	var cookieStrings []string
 	for name, value := range cookies {
@@ -243,29 +396,32 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 	}
 	youReq.Header.Add("Cookie", strings.Join(cookieStrings, ";"))
 
+	// 根据 OpenAI 请求的 stream 参数选择处理函数
 	if !openAIReq.Stream {
-		handleNonStreamingResponse(w, youReq)
+		handleNonStreamingResponse(w, youReq) // 处理非流式响应
 		return
 	}
 
-	handleStreamingResponse(w, youReq)
+	handleStreamingResponse(w, youReq) // 处理流式响应
 }
 
+// getCookies 根据提供的 DS token 生成所需的 Cookie。
 func getCookies(dsToken string) map[string]string {
 	return map[string]string{
 		"guest_has_seen_legal_disclaimer": "true",
 		"youchat_personalization":         "true",
-		"DS":                              dsToken,
-		"you_subscription":                "youpro_standard_year",
+		"DS":                              dsToken,                // 关键的 DS token
+		"you_subscription":                "youpro_standard_year", // 示例订阅信息
 		"youpro_subscription":             "true",
-		"ai_model":                        "deepseek_r1",
+		"ai_model":                        "deepseek_r1", // 示例 AI 模型
 		"youchat_smart_learn":             "true",
 	}
 }
 
+// handleNonStreamingResponse 处理非流式请求。
 func handleNonStreamingResponse(w http.ResponseWriter, youReq *http.Request) {
 	client := &http.Client{
-		Timeout: 60 * time.Second,
+		Timeout: 60 * time.Second, // 设置超时时间
 	}
 	resp, err := client.Do(youReq)
 	if err != nil {
@@ -277,22 +433,24 @@ func handleNonStreamingResponse(w http.ResponseWriter, youReq *http.Request) {
 	var fullResponse strings.Builder
 	scanner := bufio.NewScanner(resp.Body)
 
+	// 设置 scanner 的缓冲区大小（可选，但对于大型响应很重要）
 	buf := make([]byte, 0, 64*1024)
 	scanner.Buffer(buf, 1024*1024)
 
+	// 逐行扫描响应，寻找 youChatToken 事件
 	for scanner.Scan() {
 		line := scanner.Text()
 		if strings.HasPrefix(line, "event: youChatToken") {
-			scanner.Scan()
+			scanner.Scan() // 读取下一行 (data 行)
 			data := scanner.Text()
 			if !strings.HasPrefix(data, "data: ") {
-				continue
+				continue // 如果不是 data 行，则跳过
 			}
 			var token YouChatResponse
 			if err := json.Unmarshal([]byte(strings.TrimPrefix(data, "data: ")), &token); err != nil {
-				continue
+				continue // 如果解析失败，则跳过
 			}
-			fullResponse.WriteString(token.YouChatToken)
+			fullResponse.WriteString(token.YouChatToken) // 将 token 添加到完整响应中
 		}
 	}
 
@@ -301,19 +459,20 @@ func handleNonStreamingResponse(w http.ResponseWriter, youReq *http.Request) {
 		return
 	}
 
+	// 构建 OpenAI 格式的非流式响应
 	openAIResp := OpenAIResponse{
 		ID:      "chatcmpl-" + fmt.Sprintf("%d", time.Now().Unix()),
 		Object:  "chat.completion",
 		Created: time.Now().Unix(),
-		Model:   reverseMapModelName(mapModelName(originalModel)),
+		Model:   reverseMapModelName(mapModelName(originalModel)), // 映射回 OpenAI 模型名称
 		Choices: []OpenAIChoice{
 			{
 				Message: Message{
 					Role:    "assistant",
-					Content: fullResponse.String(),
+					Content: fullResponse.String(), // 完整的响应内容
 				},
 				Index:        0,
-				FinishReason: "stop",
+				FinishReason: "stop", // 停止原因
 			},
 		},
 	}
@@ -325,8 +484,9 @@ func handleNonStreamingResponse(w http.ResponseWriter, youReq *http.Request) {
 	}
 }
 
+// handleStreamingResponse 处理流式请求。
 func handleStreamingResponse(w http.ResponseWriter, youReq *http.Request) {
-	client := &http.Client{}
+	client := &http.Client{} // 流式请求不需要设置超时，因为它会持续接收数据
 	resp, err := client.Do(youReq)
 	if err != nil {
 		http.Error(w, err.Error(), http.StatusInternalServerError)
@@ -334,40 +494,159 @@ func handleStreamingResponse(w http.ResponseWriter, youReq *http.Request) {
 	}
 	defer resp.Body.Close()
 
+	// 设置流式响应的头部
 	w.Header().Set("Content-Type", "text/event-stream")
 	w.Header().Set("Cache-Control", "no-cache")
 	w.Header().Set("Connection", "keep-alive")
 
 	scanner := bufio.NewScanner(resp.Body)
+	// 逐行扫描响应，寻找 youChatToken 事件
 	for scanner.Scan() {
 		line := scanner.Text()
 
 		if strings.HasPrefix(line, "event: youChatToken") {
-			scanner.Scan()
-			data := scanner.Text()
+			scanner.Scan()         // 读取下一行 (data 行)
+			data := scanner.Text() // 获取数据行
 
 			var token YouChatResponse
-			json.Unmarshal([]byte(strings.TrimPrefix(data, "data: ")), &token)
+			json.Unmarshal([]byte(strings.TrimPrefix(data, "data: ")), &token) // 解析 JSON
 
+			// 构建 OpenAI 格式的流式响应块
 			openAIResp := OpenAIStreamResponse{
 				ID:      "chatcmpl-" + fmt.Sprintf("%d", time.Now().Unix()),
 				Object:  "chat.completion.chunk",
 				Created: time.Now().Unix(),
-				Model:   reverseMapModelName(mapModelName(originalModel)),
+				Model:   reverseMapModelName(mapModelName(originalModel)), // 映射回 OpenAI 模型名称
 				Choices: []Choice{
 					{
 						Delta: Delta{
-							Content: token.YouChatToken,
+							Content: token.YouChatToken, // 增量内容
 						},
 						Index:        0,
-						FinishReason: "",
+						FinishReason: "", // 流式响应中通常为空
 					},
 				},
 			}
 
-			respBytes, _ := json.Marshal(openAIResp)
-			fmt.Fprintf(w, "data: %s\n\n", string(respBytes))
-			w.(http.Flusher).Flush()
+			respBytes, _ := json.Marshal(openAIResp)          // 将响应块序列化为 JSON
+			fmt.Fprintf(w, "data: %s\n\n", string(respBytes)) // 写入响应数据
+			w.(http.Flusher).Flush()                          // 立即刷新输出
+		}
+	}
+
+}
+
+// 获取上传文件所需的 nonce
+func getNonce(dsToken string) (*NonceResponse, error) {
+	req, _ := http.NewRequest("GET", "https://you.com/api/get_nonce", nil)
+	req.Header.Set("Cookie", fmt.Sprintf("ds_token=%s", dsToken))
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var nonceResp NonceResponse
+	if err := json.NewDecoder(resp.Body).Decode(&nonceResp); err != nil {
+		return nil, err
+	}
+	return &nonceResp, nil
+}
+
+// 上传文件
+func uploadFile(dsToken, filePath string) (*UploadResponse, error) {
+	file, err := os.Open(filePath)
+	if err != nil {
+		return nil, err
+	}
+	defer file.Close()
+
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+	part, err := writer.CreateFormFile("file", filepath.Base(filePath))
+	if err != nil {
+		return nil, err
+	}
+
+	if _, err := io.Copy(part, file); err != nil {
+		return nil, err
+	}
+	writer.Close()
+
+	req, _ := http.NewRequest("POST", "https://you.com/api/upload", body)
+	req.Header.Set("Content-Type", writer.FormDataContentType())
+	req.Header.Set("Cookie", fmt.Sprintf("ds_token=%s", dsToken))
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var uploadResp UploadResponse
+	if err := json.NewDecoder(resp.Body).Decode(&uploadResp); err != nil {
+		return nil, err
+	}
+	return &uploadResp, nil
+}
+
+// 计算消息的 token 数（使用字符估算方法）
+func countTokens(messages []Message) (int, error) {
+	totalTokens := 0
+	for _, msg := range messages {
+		content := msg.Content
+		englishCount := 0
+		chineseCount := 0
+
+		// 遍历每个字符
+		for _, r := range content {
+			if r <= 127 { // ASCII 字符（英文和符号）
+				englishCount++
+			} else { // 非 ASCII 字符（中文等）
+				chineseCount++
+			}
+		}
+
+		// 计算 tokens：英文字符 * 0.3 + 中文字符 * 0.6
+		tokens := int(float64(englishCount)*0.3 + float64(chineseCount)*0.6)
+
+		// 加上角色名的 token（约 2 个）
+		totalTokens += tokens + 2
+	}
+	return totalTokens, nil
+}
+
+// 将 system 消息转换为第一条 user 消息
+func convertSystemToUser(messages []Message) []Message {
+	if len(messages) == 0 {
+		return messages
+	}
+
+	var systemContent strings.Builder
+	var newMessages []Message
+	var systemFound bool
+
+	// 收��所有 system 消息
+	for _, msg := range messages {
+		if msg.Role == "system" {
+			if systemContent.Len() > 0 {
+				systemContent.WriteString("\n")
+			}
+			systemContent.WriteString(msg.Content)
+			systemFound = true
+		} else {
+			newMessages = append(newMessages, msg)
 		}
 	}
+
+	// 如果有 system 消息，将其作为第一条 user 消息
+	if systemFound {
+		newMessages = append([]Message{{
+			Role:    "user",
+			Content: systemContent.String(),
+		}}, newMessages...)
+	}
+
+	return newMessages
 }

go.mod

@@ -3,6 +3,7 @@ module you2api
 go 1.22.2
 
 require (
+	github.com/google/uuid v1.6.0
 	github.com/prometheus/client_golang v1.18.0
 	go.uber.org/zap v1.26.0
 )

go.sum

@@ -8,6 +8,8 @@ github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaS
 github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
 github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
+github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
+github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/matttproud/golang_protobuf_extensions/v2 v2.0.0 h1:jWpvCLoY8Z/e3VKvlsiIGKtc+UG6U5vzxaoagmhXfyg=
 github.com/matttproud/golang_protobuf_extensions/v2 v2.0.0/go.mod h1:QUyp042oQthUoa9bqDv0ER0wrtXnBruoNd7aNjkbP+k=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
@@ -20,6 +22,8 @@ github.com/prometheus/common v0.45.0 h1:2BGz0eBc2hdMDLnO/8n0jeB3oPrt2D08CekT0lne
 github.com/prometheus/common v0.45.0/go.mod h1:YJmSTw9BoKxJplESWWxlbyttQR4uaEcGyv9MZjVOJsY=
 github.com/prometheus/procfs v0.12.0 h1:jluTpSng7V9hY0O2R9DzzJHYb2xULk9VTR1V1R/k6Bo=
 github.com/prometheus/procfs v0.12.0/go.mod h1:pcuDEFsWDnvcgNzo4EEweacyhjeA9Zk3cnaOZAZEfOo=
+github.com/sashabaranov/go-openai v1.20.2 h1:nilzF2EKzaHyK4Rk2Dbu/aJEZbtIvskDIXvfS4yx+6M=
+github.com/sashabaranov/go-openai v1.20.2/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
 github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk=
 github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
 go.uber.org/goleak v1.2.0 h1:xqgm/S+aQvhWFTtR0XK3Jvg7z8kGV8P4X14IzwN3Eqk=