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

some changes

Browse Source
This commit is contained in:
ScuroNeko
2026-01-21 15:58:30 +03:00
parent 02590a2a46
commit b8391e9c6b
4 changed files with 73 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
plugins/rp.go
View File

@@ -26,17 +26,26 @@ func RegisterRP(bot *laniakea.Bot) {
rp = rp.Command(rpUserPromptGet, "rpuserpget")
rp = rp.Command(rpUserPromptSet, "rpuserpset")
rp = rp.Command(generate, "g", "gen", "г")
rp = rp.Payload(debugTokens, "rp.tokens")
bot.AddPlugins(rp.Build())
}
func debugTokens(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rep := red.NewRPRepository(db)
waifuId := rep.GetSelectedWaifu(ctx.FromID)
tokens := rep.GetChatTokens(ctx.FromID, waifuId)
ctx.Answerf("%d", tokens)
}
func selectWaifu(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
waifuId, err := strconv.Atoi(ctx.Args[0])
if err != nil {
ctx.Error(err)
return
}
err = red.RPSetSelectedWaifu(db, ctx.FromID, waifuId)
rpRep := red.NewRPRepository(db)
err = rpRep.SetSelectedWaifu(ctx.FromID, waifuId)
if err != nil {
ctx.Error(err)
return
@@ -98,13 +107,14 @@ func rpScenarioList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Answer(strings.Join(out, "\n"))
}
func newChat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
waifuId := red.RPGetSelectedWaifu(db, ctx.FromID)
redisRpRep := red.NewRPRepository(db)
waifuId := redisRpRep.GetSelectedWaifu(ctx.FromID)
if waifuId == 0 {
ctx.Answer("Не выбрана вайфу")
return
}
chatId := uuid.New()
err := red.RPSetChatId(db, ctx.FromID, waifuId, chatId.String())
err := redisRpRep.SetChatId(ctx.FromID, waifuId, chatId.String())
if err != nil {
ctx.Error(err)
return
@@ -125,12 +135,12 @@ func newChat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
return
}
err = red.RPSetChatPrompt(db, ctx.FromID, waifuId, scenario.Prompt)
err = redisRpRep.SetChatPrompt(ctx.FromID, waifuId, scenario.Prompt)
if err != nil {
ctx.Error(err)
return
}
err = red.RPSetCounter(db, ctx.FromID, waifuId, 0)
err = redisRpRep.SetCounter(ctx.FromID, waifuId, 0)
if err != nil {
ctx.Error(err)
return
@@ -172,15 +182,16 @@ func rpUserPromptSet(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
waifuId := red.RPGetSelectedWaifu(db, ctx.FromID)
redisRpRep := red.NewRPRepository(db)
waifuId := redisRpRep.GetSelectedWaifu(ctx.FromID)
if waifuId == 0 {
ctx.Answer("Не выбрана вайфу")
return
}
chatId := red.RPGetChatId(db, ctx.FromID, waifuId)
chatId := redisRpRep.GetChatId(ctx.FromID, waifuId)
if chatId == "" {
chatId = uuid.New().String()
err := red.RPSetChatId(db, ctx.FromID, waifuId, chatId)
err := redisRpRep.SetChatId(ctx.FromID, waifuId, chatId)
if err != nil {
ctx.Error(err)
return
@@ -214,7 +225,7 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ai.FormatPrompt(preset.PreHistory, waifu.Name, ctx.Msg.From.FirstName),
fmt.Sprintf("Вот краткое описание твоего персонажа: %s", waifu.RpPrompt),
userPrompt,
red.RPGetChatPrompt(db, ctx.FromID, waifuId),
redisRpRep.GetChatPrompt(ctx.FromID, waifuId),
),
}
afterHistory := ai.Message{
@@ -267,17 +278,28 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
if err != nil {
ctx.Error(err)
}
rpUser.UsedTokens = rpUser.UsedTokens + res.Usage.TotalTokens
tokens := redisRpRep.GetChatTokens(ctx.FromID, waifuId)
tokens += int(res.Usage.CompletionTokens)
err = redisRpRep.SetChatTokens(ctx.FromID, waifuId, tokens)
if err != nil {
ctx.Error(err)
}
err = rpRep.UpdateUser(rpUser)
if err != nil {
ctx.Error(err)
return
}
m.Delete()
ctx.Answer(laniakea.EscapeMarkdown(agentAnswer.Content))
counter := red.RPGetCounter(db, ctx.FromID, waifuId) + 1
if counter == 10 {
m.Delete()
//kb := laniakea.NewInlineKeyboard()
//kb.AddCallbackButton("Перегенерировать", laniakea.CallbackData{Command: "rp.tokens"})
//ctx.Keyboard("Test", kb)
ctx.Answer(laniakea.EscapeMarkdown(agentAnswer.Content))
return
counter := redisRpRep.GetCounter(ctx.FromID, waifuId) + 1
if counter == 20 {
m := ctx.Answer("Запущено сжатие чата.")
history, err = mdb.GetChatHistory(db, chatId)
if err != nil {
@@ -304,7 +326,7 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
compressedHistory := res.Choices[0].Message.Content
chatId = uuid.New().String()
err := red.RPSetChatId(db, ctx.FromID, waifuId, chatId)
err := redisRpRep.SetChatId(ctx.FromID, waifuId, chatId)
if err != nil {
ctx.Error(err)
return
@@ -323,7 +345,7 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
counter = 0
m.Edit("Сжатие завершено")
}
err = red.RPSetCounter(db, ctx.FromID, waifuId, counter)
err = redisRpRep.SetCounter(ctx.FromID, waifuId, counter)
if err != nil {
ctx.Error(err)
}