compress wip
This commit is contained in:
@@ -7,6 +7,7 @@ import (
|
||||
"kurumibot/database/mdb"
|
||||
"kurumibot/database/psql"
|
||||
"kurumibot/database/red"
|
||||
"kurumibot/utils"
|
||||
"kurumibot/utils/ai"
|
||||
"strconv"
|
||||
"strings"
|
||||
@@ -617,30 +618,10 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
func compress(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
m := ctx.EditCallback("Запущено сжатие чата…", nil)
|
||||
_compress(ctx, db)
|
||||
m.Delete()
|
||||
return
|
||||
//chatId = uuid.New().String()
|
||||
//err = redisRpRep.SetChatId(ctx.FromID, waifuId, chatId)
|
||||
//if err != nil {
|
||||
// ctx.Error(err)
|
||||
// return
|
||||
//}
|
||||
//
|
||||
//err = mdb.UpdateChatHistory(db, chatId, "assistant", compressedHistory)
|
||||
//if err != nil {
|
||||
// ctx.Error(err)
|
||||
//}
|
||||
//offset := utils.Min(len(history), 20)
|
||||
//for _, m := range history[len(history)-offset:] {
|
||||
// err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Message)
|
||||
// if err != nil {
|
||||
// ctx.Error(err)
|
||||
// }
|
||||
//}
|
||||
//kb := laniakea.NewInlineKeyboard(1)
|
||||
//kb = kb.AddCallbackButton("Назад", "rp.chat_stat")
|
||||
//kb = kb.AddCallbackButton("На главную", "rp.info")
|
||||
//ctx.EditCallback("Сжатие завершено", kb)
|
||||
kb := laniakea.NewInlineKeyboard(1)
|
||||
kb = kb.AddCallbackButton("Назад", "rp.chat_stat")
|
||||
kb = kb.AddCallbackButton("На главную", "rp.info")
|
||||
m.Edit("Сжатие завершено")
|
||||
}
|
||||
|
||||
func _compress(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
@@ -686,6 +667,25 @@ func _compress(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
compressedHistory = strings.ReplaceAll(compressedHistory, "*", "")
|
||||
|
||||
ctx.Answer(compressedHistory)
|
||||
|
||||
chatId = uuid.New().String()
|
||||
err = redisRpRep.SetChatId(ctx.FromID, waifuId, chatId)
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
err = mdb.UpdateChatHistory(db, chatId, "assistant", compressedHistory)
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
}
|
||||
offset := utils.Min(len(history), 20)
|
||||
for _, m := range history[len(history)-offset:] {
|
||||
err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Message)
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var messagesMethodCount = []int{
|
||||
|
||||
Reference in New Issue
Block a user