115 lines
3.4 KiB
Go
115 lines
3.4 KiB
Go
package prompt
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
sessionDao "prompts-core/dao"
|
|
"prompts-core/model/entity"
|
|
|
|
"prompts-core/common/util"
|
|
sessionDto "prompts-core/model/dto/prompt"
|
|
|
|
"gitea.com/red-future/common/beans"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
func SessionCallback(ctx context.Context, req *sessionDto.SessionCallbackReq) (res *sessionDto.SessionCallbackRes, err error) {
|
|
// 1. 解析AI返回的文本
|
|
result, err := util.ParseOutput(req.Text)
|
|
if err != nil {
|
|
g.Log().Errorf(ctx, "[会话回调] 解析模型输出失败 epicycleId=%d err=%v", req.EpicycleId, err)
|
|
return nil, err
|
|
}
|
|
|
|
// 2. 更新数据库
|
|
result["role"] = "assistant"
|
|
_, err = sessionDao.ComposeSession.Update(ctx, &entity.ComposeSession{
|
|
SQLBaseDO: beans.SQLBaseDO{Id: req.EpicycleId},
|
|
ResponseContent: result,
|
|
})
|
|
if err != nil {
|
|
g.Log().Errorf(ctx, "[会话回调] 更新数据库失败 epicycleId=%d err=%v", req.EpicycleId, err)
|
|
return nil, err
|
|
}
|
|
|
|
// 3. 获取当前轮次完整数据
|
|
session, err := sessionDao.ComposeSession.Get(ctx, &entity.ComposeSession{
|
|
SQLBaseDO: beans.SQLBaseDO{Id: req.EpicycleId},
|
|
})
|
|
if err != nil {
|
|
g.Log().Errorf(ctx, "[会话回调] 获取会话数据失败 epicycleId=%d err=%v", req.EpicycleId, err)
|
|
return nil, err
|
|
}
|
|
|
|
// 4. 转换 json 并存入 Redis
|
|
requestMessages := util.ConvertToMessages(session.RequestContent)
|
|
responseMessages := util.ConvertToMessages(session.ResponseContent)
|
|
|
|
if err = 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 &sessionDto.SessionCallbackRes{}, nil
|
|
}
|
|
|
|
// GetHistoryMessages 获取历史信息
|
|
func GetHistoryMessages(ctx context.Context, sessionId string) ([]map[string]any, error) {
|
|
maxRounds := g.Cfg().MustGet(ctx, "session.maxRounds", 10).Int()
|
|
|
|
// 1. 先从 Redis 拿
|
|
redisHistory, err := GetSessionHistoryForInference(ctx, sessionId)
|
|
if err == nil && len(redisHistory) > 0 {
|
|
return redisHistory, nil
|
|
}
|
|
|
|
// 2. Redis 没有 → fallback DB
|
|
sessions, _, err := sessionDao.ComposeSession.List(ctx, &entity.ComposeSession{
|
|
SessionId: sessionId,
|
|
}, 1, maxRounds)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("DB获取历史失败: %w", err)
|
|
}
|
|
|
|
var messages []map[string]any
|
|
|
|
for _, session := range sessions {
|
|
// request
|
|
reqMsgs := util.ConvertToMessages(session.RequestContent)
|
|
for _, m := range reqMsgs {
|
|
role := gconv.String(m["role"])
|
|
if role == "user" || role == "assistant" {
|
|
messages = append(messages, m)
|
|
}
|
|
}
|
|
|
|
// response
|
|
respMsgs := util.ConvertToMessages(session.ResponseContent)
|
|
for _, m := range respMsgs {
|
|
if m["role"] == nil {
|
|
m["role"] = "assistant"
|
|
}
|
|
messages = append(messages, m)
|
|
}
|
|
}
|
|
|
|
// 3. 回写 Redis
|
|
for _, session := range sessions {
|
|
reqMsgs := util.ConvertToMessages(session.RequestContent)
|
|
respMsgs := util.ConvertToMessages(session.ResponseContent)
|
|
for i := range respMsgs {
|
|
if respMsgs[i]["role"] == nil {
|
|
respMsgs[i]["role"] = "assistant"
|
|
}
|
|
}
|
|
if len(reqMsgs) > 0 || len(respMsgs) > 0 {
|
|
_ = saveToRedis(ctx, session.SessionId, reqMsgs, respMsgs)
|
|
}
|
|
}
|
|
return messages, nil
|
|
}
|