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

some fixes and improvements

Browse Source
This commit is contained in:
ScuroNeko
2026-01-23 11:09:45 +03:00
parent 43e2f6be9d
commit 336abd5517
4 changed files with 29 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
plugins/rp.go
View File

@@ -407,17 +407,12 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
Role: "user",
Content: userMessage,
})
err = mdb.UpdateChatHistory(db, chatId, "user", userMessage)
if err != nil {
ctx.Error(err)
return
}
m := ctx.Answer("Генерация запущена...")
kb := laniakea.NewInlineKeyboard(1).AddCallbackButton("Отменить", "rp.cancel")
m := ctx.Keyboard("Генерация запущена...", kb)
api := ai.NewOpenAIAPI(ai.GPTBaseUrl, "", rpUser.Model.Key)
res, err := api.CreateCompletion(ai.CreateCompletionReq{
Messages: messages,
Verbosity: "low",
Temperature: 1.0,
})
if err != nil {
@@ -429,6 +424,11 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return
}
err = mdb.UpdateChatHistory(db, chatId, "user", userMessage)
if err != nil {
ctx.Error(err)
return
}
agentAnswer := res.Choices[0].Message
err = mdb.UpdateChatHistory(db, chatId, agentAnswer.Role, agentAnswer.Content)
if err != nil {
@@ -448,7 +448,7 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
m.Delete()
kb := laniakea.NewInlineKeyboard(1).AddCallbackButton("Сжать чать", "rp.compress_chat")
kb = laniakea.NewInlineKeyboard(1).AddCallbackButton("Сжать чать", "rp.compress_chat")
ctx.Keyboard(laniakea.EscapeMarkdown(agentAnswer.Content), kb)
}