prompts-core

service/session_service.go

@@ -0,0 +1,54 @@
+package service
+
+import (
+	"context"
+	"prompts-core/dao"
+	"prompts-core/model/dto"
+	"prompts-core/model/entity"
+
+	"gitea.com/red-future/common/beans"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+var Session = &sessionService{}
+
+type sessionService struct{}
+
+// SessionCallback 会话回调处理
+func (s *sessionService) SessionCallback(ctx context.Context, req *dto.SessionCallbackReq) (res *beans.ResponseEmpty, err error) {
+	// 1. 解析AI返回的文本
+	result, err := sessionParseModelOutput(req.Text)
+	if err != nil {
+		g.Log().Errorf(ctx, "[会话回调] 解析模型输出失败 epicycleId=%d err=%v", req.EpicycleId, err)
+		return nil, err
+	}
+
+	// 2. 更新数据库（AI回调内容）
+	_, err = dao.ComposeSession.UpdateById(ctx, req.EpicycleId, map[string]any{
+		entity.ComposeSessionCol.ResponseContent: result,
+	})
+	if err != nil {
+		g.Log().Errorf(ctx, "[会话回调] 更新数据库失败 epicycleId=%d err=%v", req.EpicycleId, err)
+		return nil, err
+	}
+
+	// 3. 获取当前轮次的完整数据
+	session, err := dao.ComposeSession.GetById(ctx, req.EpicycleId)
+	if err != nil {
+		g.Log().Errorf(ctx, "[会话回调] 获取会话数据失败 epicycleId=%d err=%v", req.EpicycleId, err)
+		return nil, err
+	}
+
+	// 4. 写入 Redis（多轮记忆）
+	requestMessages := s.convertToMessages(session.RequestContent)
+	responseMessages := s.convertToMessages(session.ResponseContent)
+	if err = s.saveToRedis(ctx, session.SessionId, requestMessages, responseMessages); err != nil {
+		g.Log().Errorf(ctx, "[会话回调] Redis存储失败 sessionId=%s id=%d err=%v",
+			session.SessionId, session.Id, err)
+		return nil, err
+	}
+
+	g.Log().Infof(ctx, "[会话回调] 存储成功 sessionId=%s id=%d requestLen=%d responseLen=%d",
+		session.SessionId, session.Id, len(requestMessages), len(responseMessages))
+	return &beans.ResponseEmpty{}, nil
+}