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

ai work

Browse Source
This commit is contained in:
ScuroNeko
2026-01-13 21:01:10 +03:00
parent cec3a35d83
commit 7d540962a1
6 changed files with 79 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
plugins/testrp.go
View File

@@ -58,6 +58,11 @@ func newChat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Error(err)
return
}
err = red.RPSetCounter(db, ctx.FromID, waifuId, 0)
if err != nil {
ctx.Error(err)
return
}
ctx.Answer("Был создан новый чат.")
}
@@ -69,7 +74,8 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
chatId := red.RPGetChatId(db, ctx.FromID, waifuId)
if chatId == "" {
err := red.RPSetChatId(db, ctx.FromID, waifuId, uuid.New().String())
chatId = uuid.New().String()
err := red.RPSetChatId(db, ctx.FromID, waifuId, chatId)
if err != nil {
ctx.Error(err)
return
@@ -80,30 +86,30 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Error(err)
return
}
messages := []ai.Message{
{
Role: "system",
Content: fmt.Sprintf(
"%s %s %s",
ai.FormatPrompt(ai.PreHistoryPrompt, waifu.Name, ctx.Msg.From.FirstName),
fmt.Sprintf("Вот краткое описание твоего персонажа: %s", waifu.RpPrompt),
red.RPGetChatPrompt(db, ctx.FromID, waifuId),
),
},
systemPrompt := ai.Message{
Role: "system",
Content: fmt.Sprintf(
"%s %s %s",
ai.FormatPrompt(ai.PreHistoryPrompt, waifu.Name, ctx.Msg.From.FirstName),
fmt.Sprintf("Вот краткое описание твоего персонажа: %s", waifu.RpPrompt),
red.RPGetChatPrompt(db, ctx.FromID, waifuId),
),
}
api := ai.NewOpenAIAPI(ai.CosmoRPUrl, os.Getenv("PAWAN_KEY"), "cosmorp-2.5")
history, err := mdb.GetChatHistory(db, chatId)
if err != nil {
ctx.Error(err)
return
}
messages := make([]ai.Message, 0)
for _, m := range history {
messages = append(messages, ai.Message{
Role: m.Role,
Content: m.Message,
})
}
api := ai.NewOpenAIAPI(ai.CosmoRPUrl, os.Getenv("PAWAN_KEY"), "cosmorp-2.5")
userMessage := strings.Join(ctx.Args, " ")
messages = append(messages, ai.Message{
Role: "system", Content: ai.FormatPrompt(ai.PostHistoryPrompt, waifu.Name, ctx.Msg.From.FirstName),
@@ -117,7 +123,7 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
m := ctx.Answer("Генерация запущена...")
res, err := api.CreateCompletion(ai.CreateCompletionReq{
Messages: messages,
Messages: append([]ai.Message{systemPrompt}, messages...),
Verbosity: "low",
Temperature: 0.7,
})
@@ -131,5 +137,36 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
response = append(response, m.Content)
err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Content)
}
m.Edit(laniakea.EscapeMarkdown(strings.Join(response, "\n")))
m.Delete()
ctx.Answer(laniakea.EscapeMarkdown(strings.Join(response, "\n")))
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: "Сделай краткую выжимку нашего диалога",
}),
})
if err != nil {
ctx.Error(err)
}
chatId = uuid.New().String()
err := red.RPSetChatId(db, ctx.FromID, waifuId, chatId)
if err != nil {
ctx.Error(err)
return
}
err = mdb.UpdateChatHistory(db, chatId, "assistant", res.Choices[0].Message.Content)
err = red.RPSetCounter(db, ctx.FromID, waifuId, 0)
if err != nil {
ctx.Error(err)
return
}
} else {
err = red.RPSetCounter(db, ctx.FromID, waifuId, counter)
if err != nil {
ctx.Error(err)
}
}
}