fix: session title 补全

This commit is contained in:
fuzhongyun 2026-01-09 18:31:51 +08:00
parent 44155c058d
commit 03652cb588
1 changed files with 15 additions and 4 deletions

View File

@ -32,16 +32,18 @@ type Do struct {
} }
func NewDo( func NewDo(
sessionImpl *impl.SessionImpl,
sysImpl *impl.SysImpl, sysImpl *impl.SysImpl,
taskImpl *impl.TaskImpl, taskImpl *impl.TaskImpl,
hisImpl *impl.ChatHisImpl, hisImpl *impl.ChatHisImpl,
conf *config.Config, conf *config.Config,
) *Do { ) *Do {
return &Do{ return &Do{
conf: conf, conf: conf,
sysImpl: sysImpl, sessionImpl: sessionImpl,
hisImpl: hisImpl, sysImpl: sysImpl,
taskImpl: taskImpl, hisImpl: hisImpl,
taskImpl: taskImpl,
} }
} }
@ -275,6 +277,15 @@ func (d *Do) startMessageHandler(
} }
d.hisImpl.AddWithData(AiRes) d.hisImpl.AddWithData(AiRes)
hisLog.HisId = AiRes.HisID hisLog.HisId = AiRes.HisID
// 查询当前session
cond := builder.NewCond().And(builder.Eq{"session_id": requireData.Session})
sessionMap, _ := d.sessionImpl.GetOneBySearch(&cond)
requireData.SessionInfo.Title = sessionMap["title"].(string)
// 当前 session title为空 ,更新为用户输入
if requireData.SessionInfo.Title == "" {
d.sessionImpl.UpdateByCond(&cond, &model.AiSession{Title: requireData.Req.Text})
}
} }
_ = entitys.MsgSend(client, entitys.Response{ _ = entitys.MsgSend(client, entitys.Response{