55 lines
1.8 KiB
Go
55 lines
1.8 KiB
Go
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
|
||
}
|