prompts-core

service/session_redis_service.go

@@ -0,0 +1,181 @@
+package service
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"strings"
+	"time"
+
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+// Message 消息结构（content 支持 string 或 []string）
+type Message struct {
+	Role    string `json:"role"`           // user / assistant / system
+	Content any    `json:"content"`        // 内容：string 或 []string
+	Type    string `json:"type,omitempty"` // text / file（可选扩展）
+}
+
+// GetContentString 获取 Content 的字符串形式
+func (m Message) GetContentString() string {
+	switch v := m.Content.(type) {
+	case string:
+		return v
+	case []interface{}:
+		var parts []string
+		for _, item := range v {
+			if s, ok := item.(string); ok {
+				parts = append(parts, s)
+			}
+		}
+		return strings.Join(parts, "\n")
+	default:
+		b, _ := json.Marshal(m.Content)
+		return string(b)
+	}
+}
+
+// SessionRoundData Redis存储的单轮会话数据
+type SessionRoundData struct {
+	SessionId       string    `json:"sessionId"`       // 会话ID
+	RequestContent  []Message `json:"requestContent"`  // 用户请求会话
+	ResponseContent []Message `json:"responseContent"` // AI回调会话
+	Timestamp       int64     `json:"timestamp"`       // 存入时间戳
+}
+
+// GetSessionHistory 获取多轮会话历史（供推理时使用）
+func (s *sessionService) GetSessionHistory(ctx context.Context, sessionId string) ([]SessionRoundData, error) {
+	return s.getFromRedis(ctx, sessionId)
+}
+
+// BuildMessages 根据Redis历史构建完整的Messages数组
+func (s *sessionService) BuildMessages(ctx context.Context, sessionId string, currentMessages []Message) ([]Message, error) {
+	// 获取历史会话
+	history, err := s.getFromRedis(ctx, sessionId)
+	if err != nil {
+		return nil, fmt.Errorf("获取历史会话失败: %w", err)
+	}
+
+	var allMessages []Message
+
+	// 按时间顺序拼接历史消息
+	for _, round := range history {
+		allMessages = append(allMessages, round.RequestContent...)
+		allMessages = append(allMessages, round.ResponseContent...)
+	}
+
+	// 添加当前轮次的请求消息
+	allMessages = append(allMessages, currentMessages...)
+
+	return allMessages, nil
+}
+
+// ==================== Redis 操作 ====================
+
+// saveToRedis 保存会话数据到Redis
+// sessionId: 会话ID作为key
+// 最大10轮，超出替换最早的，过期时间30分钟
+func (s *sessionService) saveToRedis(ctx context.Context, sessionId string, requestMessages []Message, responseMessages []Message) error {
+	key := fmt.Sprintf("chat:session:%s", 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 := SessionRoundData{
+		SessionId:       sessionId,
+		RequestContent:  requestMessages,
+		ResponseContent: responseMessages,
+		Timestamp:       time.Now().Unix(),
+	}
+
+	// 序列化
+	b, err := json.Marshal(data)
+	if err != nil {
+		return fmt.Errorf("序列化会话数据失败: %w", err)
+	}
+
+	// 写入 Redis（LPUSH 添加到最前面，新的在前）
+	_, err = g.Redis().Do(ctx, "LPUSH", key, string(b))
+	if err != nil {
+		return fmt.Errorf("写入Redis失败: %w", err)
+	}
+
+	// 裁剪到最新10轮（保留前10条）
+	_, err = g.Redis().Do(ctx, "LTRIM", key, 0, maxRounds-1)
+	if err != nil {
+		return fmt.Errorf("裁剪Redis列表失败: %w", err)
+	}
+
+	// 重置过期时间
+	_, err = g.Redis().Do(ctx, "EXPIRE", key, int64(expireTime.Seconds()))
+	if err != nil {
+		return fmt.Errorf("设置过期时间失败: %w", err)
+	}
+
+	return nil
+}
+
+// getFromRedis 从Redis获取会话历史
+func (s *sessionService) getFromRedis(ctx context.Context, sessionId string) ([]SessionRoundData, error) {
+	key := fmt.Sprintf("chat:session:%s", sessionId)
+
+	// 获取列表中所有数据（最多10条）
+	result, err := g.Redis().Do(ctx, "LRANGE", key, 0, -1)
+	if err != nil {
+		return nil, fmt.Errorf("从Redis获取数据失败: %w", err)
+	}
+
+	if result == nil || result.IsNil() {
+		return []SessionRoundData{}, nil
+	}
+
+	// 解析数据
+	var sessions []SessionRoundData
+
+	// 将结果转换为字符串数组
+	values := result.Strings()
+	for _, str := range values {
+		var data SessionRoundData
+		if err := json.Unmarshal([]byte(str), &data); err != nil {
+			g.Log().Warningf(ctx, "[会话] 解析Redis数据失败 err=%v", err)
+			continue
+		}
+		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, nil
+}
+
+// GetSessionHistoryForInference 获取历史会话，直接返回Message数组（给推理用）
+func (s *sessionService) GetSessionHistoryForInference(ctx context.Context, sessionId string) ([]Message, error) {
+	// 从Redis获取历史会话数据
+	historyData, err := s.getFromRedis(ctx, sessionId)
+	if err != nil {
+		return nil, fmt.Errorf("获取历史会话失败: %w", err)
+	}
+
+	// 如果没有任何历史数据，返回空
+	if len(historyData) == 0 {
+		return []Message{}, nil
+	}
+
+	// 把SessionRoundData转换成扁平的Message数组
+	var messages []Message
+	for _, round := range historyData {
+		// 先加用户的请求
+		messages = append(messages, round.RequestContent...)
+		// 再加AI的回答
+		messages = append(messages, round.ResponseContent...)
+	}
+
+	return messages, nil
+}