61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
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.AddCommand(p.NewCommand(gpt, "gpt").SkipCommandAutoGen())
|
|
bot.AddPlugins(p)
|
|
}
|
|
|
|
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)
|
|
}
|