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

new ai provider

Browse Source
This commit is contained in:
ScuroNeko
2026-01-19 16:47:11 +03:00
parent 4dd87ddecc
commit 96d40bffa4
9 changed files with 119 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
plugins/rp.go
View File

@@ -9,7 +9,6 @@ import (
"kurumibot/database/red"
"kurumibot/laniakea"
"kurumibot/utils/ai"
"os"
"strconv"
"strings"
@@ -146,7 +145,8 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
),
}
history, err := mdb.GetChatHistory(db, chatId)
index := red.RPGetIndex(db, ctx.FromID, waifuId)
history, err := mdb.GetChatHistory(db, chatId, index)
if err != nil {
ctx.Error(err)
return
@@ -159,24 +159,28 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
})
}
api := ai.NewOpenAIAPI(ai.CosmoRPUrl, os.Getenv("PAWAN_KEY"), "cosmorp-2.5")
//os.Getenv("PAWAN_KEY")
userMessage := strings.Join(ctx.Args, " ")
messages = append(messages, ai.Message{
Role: "system", Content: ai.FormatPrompt(preset.PostHistory, waifu.Name, ctx.Msg.From.FirstName),
}, ai.Message{
Role: "user", Content: userMessage,
})
err = mdb.UpdateChatHistory(db, chatId, "user", userMessage)
if index == 0 {
index += 1
}
err = mdb.UpdateChatHistory(db, chatId, "user", userMessage, index)
if err != nil {
ctx.Error(err)
return
}
m := ctx.Answer("Генерация запущена...")
api := ai.NewOpenAIAPI(ai.GPTBaseUrl, "", "deepseek-ai/deepseek-v3.1-terminus")
res, err := api.CreateCompletion(ai.CreateCompletionReq{
Messages: append([]ai.Message{systemPrompt}, messages...),
Verbosity: "low",
Temperature: 0.7,
Temperature: 1.0,
})
if err != nil {
ctx.Error(err)
@@ -185,8 +189,14 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
response := make([]string, 0)
for _, choice := range res.Choices {
m := choice.Message
messages = append(messages, m)
response = append(response, m.Content)
err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Content)
index += 1
err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Content, index)
}
err = red.RPSetIndex(db, ctx.FromID, waifuId, index)
if err != nil {
ctx.Error(err)
}
rpUser.UsedTokens = rpUser.UsedTokens + res.Usage.TotalTokens
err = rpRep.UpdateUser(rpUser)
@@ -199,12 +209,8 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
counter := red.RPGetCounter(db, ctx.FromID, waifuId) + 1
if counter == 5 {
res, err = api.CreateCompletion(ai.CreateCompletionReq{
Messages: append(messages, ai.Message{
Role: "user",
Content: "Сделай краткую выжимку нашего диалога",
}),
})
m := ctx.Answer("Запущено сжатие чата.")
res, err = api.CompressChat(messages)
if err != nil {
ctx.Error(err)
}
@@ -214,7 +220,9 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Error(err)
return
}
err = mdb.UpdateChatHistory(db, chatId, "assistant", res.Choices[0].Message.Content)
compressedHistory := res.Choices[0].Message.Content
m.Edit("\\[DEBUG] Compressed History:\n" + compressedHistory)
err = mdb.UpdateChatHistory(db, chatId, "assistant", compressedHistory, 0)
err = red.RPSetCounter(db, ctx.FromID, waifuId, 0)
if err != nil {
ctx.Error(err)