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

memory leak

Browse Source
This commit is contained in:
ScuroNeko
2026-02-05 15:14:47 +03:00
parent b894cbd9c3
commit 0fded455b8
22 changed files with 420 additions and 170 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
plugins/admin.go
View File

@@ -12,31 +12,26 @@ func RegisterAdmin(b *laniakea.Bot) {
p.Command(uploadPhoto, "uploadPhoto")
p.Command(test, "test")
p.Middleware(laniakea.NewPluginMiddleware(func(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) bool {
p.Middleware(AdminMiddleware())
b.AddPlugins(p.Build())
}
func AdminMiddleware() *laniakea.PluginMiddleware {
return laniakea.NewPluginMiddleware(func(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) bool {
rep := psql.NewUserRepository(db)
u, err := rep.GetById(ctx.FromID)
if err != nil {
return false
}
return u.Group.IsAdmin
}))
b.AddPlugins(p.Build())
})
}
func test(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
ctx.Answer("Ok")
}
func uploadPhoto(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rep := psql.NewUserRepository(db)
user, err := rep.GetOrCreate(ctx.FromID, ctx.Msg.From.FirstName)
if err != nil {
ctx.Error(err)
return
}
if !user.Group.IsAdmin {
return
}
func uploadPhoto(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
ctx.SendAction(laniakea.ChatActionUploadPhoto)
photoId := ctx.Msg.Photo.Last().FileID
f, err := ctx.Bot.GetFile(&laniakea.GetFileP{FileId: photoId})
if err != nil {

60
plugins/ai.go Normal file
View File

@@ -0,0 +1,60 @@
package plugins
import (
"kurumibot/database/mdb"
"kurumibot/database/red"
"kurumibot/utils/ai"
"strings"
"git.nix13.pw/scuroneko/laniakea"
)
func RegisterAi(bot *laniakea.Bot) {
p := laniakea.NewPlugin("AI")
p.Command(gpt, "gpt")
bot.AddPlugins(p.Build())
}
func gpt(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
q := strings.Join(ctx.Args, " ")
api := ai.NewOpenAIAPI(ai.GPTBaseUrl, "", "anthropic/claude-sonnet-4")
defer api.Close()
aiRedisRep := red.NewAiRepository(db)
chatId, err := aiRedisRep.GetOrCreateChatId(ctx.FromID)
if err != nil {
ctx.Error(err)
return
}
history, err := mdb.GetGptChatHistory(db, chatId)
if err != nil {
ctx.Error(err)
return
}
aiHistory := make([]ai.Message, len(history))
for _, m := range history {
aiHistory = append(aiHistory, ai.Message{
Role: m.Role,
Content: m.Message,
})
}
m := ctx.Answer("Генерация запущена...")
res, err := api.CreateCompletion(aiHistory, q, 1.0)
if err != nil {
ctx.Error(err)
return
}
answer := res.Choices[0].Message.Content
m.Delete()
err = mdb.UpdateGptChatHistory(db, chatId, "user", q)
if err != nil {
ctx.Error(err)
}
err = mdb.UpdateGptChatHistory(db, chatId, "assistant", answer)
if err != nil {
ctx.Error(err)
}
ctx.Answer(answer)
}

21
plugins/economy.go
View File

@@ -17,20 +17,21 @@ import (
func RegisterEconomy(bot *laniakea.Bot) {
economy := laniakea.NewPlugin("Economy")
economy = economy.Command(profile, "profile", "профиль")
economy = economy.Command(work, "work", "работать")
economy = economy.Command(collect, "collect", "собрать")
economy = economy.Command(code, "code", "код")
economy.Command(profile, "profile", "профиль")
economy.Command(work, "work", "работать")
economy.Command(collect, "collect", "собрать")
economy.Command(code, "code", "код")
economy = economy.Command(vacancies, "vacancies", "вакансии")
economy = economy.Command(getAJob, "getajob", "устроиться")
economy.Command(vacancies, "vacancies", "вакансии")
economy.Command(getAJob, "getajob", "устроиться")
economy = economy.Command(aboutGroup, "group", "о группе")
economy.Command(aboutGroup, "group", "о группе")
economy = economy.Command(about, "about", "о боте")
economy.Command(about, "about", "о боте")
runner := laniakea.NewRunner("economy.PassiveIncome", passiveIncome).Timeout(time.Minute).Build()
bot.AddRunner(runner)
//bot.AddRunner(laniakea.NewRunner(
// "economy.PassiveIncome", passiveIncome,
//).Timeout(time.Minute).Build())
bot.AddPlugins(economy.Build())
}

47
plugins/logs.go
View File

@@ -1,20 +1,65 @@
package plugins
import (
"bytes"
"encoding/json"
"fmt"
"kurumibot/database/mdb"
"strings"
"time"
"git.nix13.pw/scuroneko/extypes"
"git.nix13.pw/scuroneko/laniakea"
"git.nix13.pw/scuroneko/slog"
)
func InitLogsPlugin() {}
func RegisterLogs(bot *laniakea.Bot) {
p := laniakea.NewPlugin("Logs")
p.Command(getLogs, "logs")
p.Command(getMsgLogs, "msglogs")
p.Middleware(AdminMiddleware())
bot.AddPlugins(p.Build())
}
func InitLogMiddleware() laniakea.Middleware {
return laniakea.NewMiddleware("LogMiddleware", logMiddleware).SetAsync(true).Build()
}
func getLogs(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
logs, err := mdb.GetConsoleLogs(db)
if err != nil {
ctx.Error(err)
return
}
out := encodeLogs(logs)
ctx.Answer(strings.Join(out, ""))
}
func getMsgLogs(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
logs, err := mdb.GetMessageLogs(db)
if err != nil {
ctx.Error(err)
return
}
out := encodeLogs(logs)
ctx.Answer(strings.Join(out, ""))
}
// Utils
func encodeLogs[T comparable](logs extypes.Slice[T]) extypes.Slice[string] {
out := make(extypes.Slice[string], len(logs))
buf := bytes.NewBuffer(nil)
enc := json.NewEncoder(buf)
enc.SetEscapeHTML(false)
enc.SetIndent("", " ")
for i, log := range logs {
_ = enc.Encode(log)
out[i] = fmt.Sprintf("`%s`", buf.String())
buf.Reset()
}
return out
}
func logMiddleware(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
if ctx.Msg == nil {
return

3
plugins/relations.go
View File

@@ -3,5 +3,6 @@ package plugins
import "git.nix13.pw/scuroneko/laniakea"
func RegisterRelations(b *laniakea.Bot) {
p := laniakea.NewPlugin("Relations")
b.AddPlugins(p.Build())
}

23
plugins/rp.go
View File

@@ -241,7 +241,7 @@ func rpPresetSet(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
func rpModelList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rep := psql.NewRPRepository(db)
rep := psql.NewAIRepository(db)
models, err := rep.GetAllModels()
if err != nil {
ctx.Error(err)
@@ -266,7 +266,8 @@ func rpModelSet(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Error(err)
return
}
model, err := rep.GetModel(ctx.Args[0])
aiRep := psql.NewAIRepository(db)
model, err := aiRep.GetModel(ctx.Args[0])
if err != nil {
ctx.Error(err)
return
@@ -328,7 +329,7 @@ func chatStat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Answer("Нет активного чата")
return
}
messageCount, err := mdb.GetChatHistorySize(db, chatId)
messageCount, err := mdb.GetRPChatHistorySize(db, chatId)
if err != nil {
ctx.Error(err)
return
@@ -604,7 +605,7 @@ func _getChatHistory(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) ([]
Content: ai.FormatPrompt(preset.PostHistory, waifu.Name, ctx.From.FirstName),
}
history, err := mdb.GetChatHistory(db, chatId)
history, err := mdb.GetRPChatHistory(db, chatId)
if err != nil {
return messages, err
}
@@ -661,14 +662,14 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
counter := redisRpRep.GetCounter(ctx.FromID, waifuId)
err = mdb.UpdateChatHistory(db, chatId, "user", userMessage, counter+1)
err = mdb.UpdateRPChatHistory(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, counter+2)
err = mdb.UpdateRPChatHistory(db, chatId, agentAnswer.Role, answerContent, counter+2)
if err != nil {
ctx.Error(err)
}
@@ -734,7 +735,7 @@ func regenerateResponse(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext)
ctx.Error(err)
return
}
history, err := mdb.GetChatHistory(db, chatId)
history, err := mdb.GetRPChatHistory(db, chatId)
if err != nil {
ctx.Error(err)
return
@@ -768,7 +769,7 @@ func regenerateResponse(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext)
return
}
err = mdb.DeleteChatEntry(db, answerToDelete)
err = mdb.DeleteRPChatEntry(db, answerToDelete)
if err != nil {
ctx.Error(err)
return
@@ -799,7 +800,7 @@ func _compress(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Error(err)
return
}
history, err := mdb.GetChatHistory(db, chatId)
history, err := mdb.GetRPChatHistory(db, chatId)
if err != nil {
ctx.Error(err)
return
@@ -839,14 +840,14 @@ func _compress(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return
}
err = mdb.UpdateChatHistory(db, chatId, "assistant", compressedHistory, 0)
err = mdb.UpdateRPChatHistory(db, chatId, "assistant", compressedHistory, 0)
if err != nil {
ctx.Error(err)
}
offset := utils.Min(len(history), 20)
for i, m := range history[len(history)-offset:] {
tokens += len(m.Message)
err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Message, i+1)
err = mdb.UpdateRPChatHistory(db, chatId, m.Role, m.Message, i+1)
if err != nil {
ctx.Error(err)
}