ScuroNeko/YaeMikoBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

small fix

Browse Source
This commit is contained in:
ScuroNeko
2026-02-02 17:33:03 +03:00
parent 7ab5ad91de
commit 3741017b02
2 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
plugins/rp.go
View File

@@ -605,14 +605,15 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return
}
err = mdb.UpdateChatHistory(db, chatId, "user", userMessage)
counter := redisRpRep.GetCounter(ctx.FromID, waifuId)
err = mdb.UpdateChatHistory(db, chatId, "user", userMessage, counter+1)
if err != nil {
ctx.Error(err)
return
}
agentAnswer := res.Choices[0].Message
answerContent := strings.TrimSpace(agentAnswer.Content)
err = mdb.UpdateChatHistory(db, chatId, agentAnswer.Role, answerContent)
err = mdb.UpdateChatHistory(db, chatId, agentAnswer.Role, answerContent, counter+2)
if err != nil {
ctx.Error(err)
}
@@ -629,7 +630,6 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
if err != nil {
ctx.Error(err)
}
counter := redisRpRep.GetCounter(ctx.FromID, waifuId)
err = redisRpRep.SetCounter(ctx.FromID, waifuId, counter+2)
if err != nil {
ctx.Error(err)
@@ -786,14 +786,14 @@ func _compress(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return
}
err = mdb.UpdateChatHistory(db, chatId, "assistant", compressedHistory)
err = mdb.UpdateChatHistory(db, chatId, "assistant", compressedHistory, 0)
if err != nil {
ctx.Error(err)
}
offset := utils.Min(len(history), 20)
for _, m := range history[len(history)-offset:] {
for i, m := range history[len(history)-offset:] {
tokens += len(m.Message)
err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Message)
err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Message, i+1)
if err != nil {
ctx.Error(err)
}