refactor(prompt): 重构提示词构建服务与数据模型

service/prompt/prompt_session_redis_service.go

@@ -9,15 +9,16 @@ import (
 	"github.com/gogf/gf/v2/frame/g"
 )
 
-// ==================== Redis 操作 ====================
+const (
+	redisKeyPrefix = "chat:session:%s"
+)
 
 // saveToRedis 保存会话数据到Redis
 func saveToRedis(ctx context.Context, sessionId string, requestMessages []map[string]any, responseMessages []map[string]any) error {
-	key := fmt.Sprintf("chat:session:%s", sessionId)
+	key := formatRedisKey(sessionId)
 
 	maxRounds := g.Cfg().MustGet(ctx, "session.maxRounds", 10).Int()
 	expireSeconds := g.Cfg().MustGet(ctx, "session.expireTime", 1800).Int64()
-	expireTime := time.Duration(expireSeconds) * time.Second
 
 	data := map[string]any{
 		"sessionId":       sessionId,
@@ -31,18 +32,29 @@ func saveToRedis(ctx context.Context, sessionId string, requestMessages []map[st
 		return fmt.Errorf("序列化会话数据失败: %w", err)
 	}
 
-	_, err = g.Redis().Do(ctx, "LPUSH", key, string(b))
-	if err != nil {
+	if err := executeRedisCommands(ctx, key, string(b), maxRounds, expireSeconds); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// formatRedisKey 格式化Redis键
+func formatRedisKey(sessionId string) string {
+	return fmt.Sprintf(redisKeyPrefix, sessionId)
+}
+
+// executeRedisCommands 执行Redis命令
+func executeRedisCommands(ctx context.Context, key string, value string, maxRounds int, expireSeconds int64) error {
+	if _, err := g.Redis().Do(ctx, "LPUSH", key, value); err != nil {
 		return fmt.Errorf("写入Redis失败: %w", err)
 	}
 
-	_, err = g.Redis().Do(ctx, "LTRIM", key, 0, maxRounds-1)
-	if err != nil {
+	if _, err := g.Redis().Do(ctx, "LTRIM", key, 0, maxRounds-1); err != nil {
 		return fmt.Errorf("裁剪Redis列表失败: %w", err)
 	}
 
-	_, err = g.Redis().Do(ctx, "EXPIRE", key, int64(expireTime.Seconds()))
-	if err != nil {
+	if _, err := g.Redis().Do(ctx, "EXPIRE", key, expireSeconds); err != nil {
 		return fmt.Errorf("设置过期时间失败: %w", err)
 	}
 
@@ -51,7 +63,7 @@ func saveToRedis(ctx context.Context, sessionId string, requestMessages []map[st
 
 // getFromRedis 从Redis获取会话历史
 func getFromRedis(ctx context.Context, sessionId string) ([]map[string]any, error) {
-	key := fmt.Sprintf("chat:session:%s", sessionId)
+	key := formatRedisKey(sessionId)
 
 	result, err := g.Redis().Do(ctx, "LRANGE", key, 0, -1)
 	if err != nil {
@@ -62,8 +74,17 @@ func getFromRedis(ctx context.Context, sessionId string) ([]map[string]any, erro
 		return []map[string]any{}, nil
 	}
 
+	sessions := parseRedisSessions(ctx, result.Strings())
+
+	reverseSlice(sessions)
+
+	return sessions, nil
+}
+
+// parseRedisSessions 解析Redis会话数据
+func parseRedisSessions(ctx context.Context, values []string) []map[string]any {
 	var sessions []map[string]any
-	values := result.Strings()
+
 	for _, str := range values {
 		var data map[string]any
 		if err := json.Unmarshal([]byte(str), &data); err != nil {
@@ -73,12 +94,14 @@ func getFromRedis(ctx context.Context, sessionId string) ([]map[string]any, erro
 		sessions = append(sessions, data)
 	}
 
-	// 反转（Redis 最新在前 → 时间正序）
-	for i, j := 0, len(sessions)-1; i < j; i, j = i+1, j-1 {
-		sessions[i], sessions[j] = sessions[j], sessions[i]
-	}
+	return sessions
+}
 
-	return sessions, nil
+// reverseSlice 反转切片
+func reverseSlice(s []map[string]any) {
+	for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
+		s[i], s[j] = s[j], s[i]
+	}
 }
 
 // GetSessionHistoryForInference 获取历史会话，返回扁平消息数组（给推理用）
@@ -92,23 +115,31 @@ func GetSessionHistoryForInference(ctx context.Context, sessionId string) ([]map
 		return []map[string]any{}, nil
 	}
 
+	return flattenHistoryMessages(historyData), nil
+}
+
+// flattenHistoryMessages 扁平化历史消息
+func flattenHistoryMessages(historyData []map[string]any) []map[string]any {
 	var messages []map[string]any
+
 	for _, round := range historyData {
-		if reqMsgs, ok := round["requestContent"].([]interface{}); ok {
-			for _, m := range reqMsgs {
-				if msg, ok := m.(map[string]interface{}); ok {
-					messages = append(messages, msg)
-				}
-			}
-		}
-		if respMsgs, ok := round["responseContent"].([]interface{}); ok {
-			for _, m := range respMsgs {
-				if msg, ok := m.(map[string]interface{}); ok {
-					messages = append(messages, msg)
-				}
-			}
-		}
+		appendMessagesFromField(round, "requestContent", &messages)
+		appendMessagesFromField(round, "responseContent", &messages)
 	}
 
-	return messages, nil
+	return messages
+}
+
+// appendMessagesFromField 从指定字段追加消息
+func appendMessagesFromField(data map[string]any, field string, messages *[]map[string]any) {
+	msgs, ok := data[field].([]interface{})
+	if !ok {
+		return
+	}
+
+	for _, m := range msgs {
+		if msg, ok := m.(map[string]interface{}); ok {
+			*messages = append(*messages, msg)
+		}
+	}
 }