ai work

plugins/economy.go

@@ -37,7 +37,7 @@ func RegisterEconomy(bot *laniakea.Bot) {
 
 func about(msgCtx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
 	out := []string{
-		fmt.Sprintf("Go: %s", "1.25.0"),
+		fmt.Sprintf("Go: %s", "1.25.5"),
 		fmt.Sprintf("Версия laniakea: %s", laniakea.VersionString),
 	}
 	msgCtx.Answer(strings.Join(out, "\n"))
@@ -45,6 +45,8 @@ func about(msgCtx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
 
 func passiveIncome(b *laniakea.Bot) {
 	for {
+		time.Sleep(time.Minute * 5)
+
 		users, err := psql.GetAllUsers()
 		if err != nil {
 			b.Logger().Error(err)
@@ -97,8 +99,6 @@ func passiveIncome(b *laniakea.Bot) {
 
 			b.Logger().Debug(fmt.Sprintf("У %d было пассивно собрано. След. сбор через час\n", user.ID))
 		}
-
-		time.Sleep(time.Second * 5)
 	}
 }
 

plugins/testrp.go

@@ -3,6 +3,7 @@ package plugins
 import (
 	"fmt"
 	"kurumibot/database/mdb"
+	"kurumibot/database/psql"
 	"kurumibot/database/red"
 	"kurumibot/laniakea"
 	"kurumibot/utils/ai"
@@ -15,7 +16,7 @@ import (
 
 func RegisterTestRP(bot *laniakea.Bot) {
 	rp := laniakea.NewPlugin("RP")
-	rp = rp.Command(selectWaifu, "selwaifu")
+	rp = rp.Command(selectWaifu, "rpwaifu", "рпвайфу")
 	rp = rp.Payload(selectWaifu, "rp.selwaifu")
 	rp = rp.Command(newChat, "newchat")
 	rp = rp.Command(generate, "g", "gen")
@@ -47,11 +48,12 @@ func newChat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
 }
 
 func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
-	chatId := red.GetChatId(db, uint(ctx.FromID))
-	if chatId == "0" {
+	waifuId := red.RPGetSelectedWaifu(db, uint(ctx.FromID))
+	if waifuId == 0 {
 		ctx.Answer("Не выбрана вайфу")
 		return
 	}
+	chatId := red.GetChatId(db, uint(ctx.FromID))
 	if chatId == "" {
 		err := red.SaveChatId(db, uint(ctx.FromID), uuid.New().String())
 		if err != nil {
@@ -59,18 +61,23 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
 			return
 		}
 	}
+	waifu, err := psql.GetWaifuById(waifuId)
+	if err != nil {
+		ctx.Error(err)
+		return
+	}
 	messages := []ai.Message{
 		{
-			Role:    "system",
-			Content: ai.FormatPrompt(ai.PreHistoryPrompt, "Яэ Мико", ctx.Msg.From.FirstName),
-		},
-		{
-			Role:    "system",
-			Content: "Вот краткое описание твоего персонажа: у неё розовые волосы, лисьи уши и фиолетовые глаза. Её рост 166 см",
+			Role: "system",
+			Content: fmt.Sprintf(
+				"%s %s",
+				ai.FormatPrompt(ai.PreHistoryPrompt, waifu.Name, ctx.Msg.From.FirstName),
+				fmt.Sprintf("Вот краткое описание твоего персонажа: %s", waifu.RpPrompt),
+			),
 		},
 	}
 
-	api := ai.NewOpenAIAPI(ai.CosmoRPUrl, os.Getenv("PAWAN_KEY"), "cosmorp-2.5")
+	api := ai.NewOpenAIAPI(ai.CosmoRPUrl, os.Getenv("PAWAN_KEY"), "cosmorp-2.5-it")
 	history, err := mdb.GetChatHistory(db, chatId)
 	if err != nil {
 		ctx.Error(err)
@@ -84,7 +91,7 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
 	}
 	userMessage := strings.Join(ctx.Args, " ")
 	messages = append(messages, ai.Message{
-		Role: "system", Content: ai.FormatPrompt(ai.PostHistoryPrompt, "Яэ Мико", ctx.Msg.From.FirstName),
+		Role: "system", Content: ai.FormatPrompt(ai.PostHistoryPrompt, waifu.Name, ctx.Msg.From.FirstName),
 	}, ai.Message{
 		Role: "user", Content: userMessage,
 	})
@@ -95,8 +102,9 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
 	}
 	m := ctx.Answer("Генерация запущена...")
 	res, err := api.CreateCompletion(ai.CreateCompletionReq{
-		Model:    "cosmorp-2.5",
-		Messages: messages,
+		Messages:    messages,
+		Verbosity:   "low",
+		Temperature: 0.7,
 	})
 	if err != nil {
 		ctx.Error(err)

plugins/waifus.go

@@ -88,7 +88,7 @@ func waifuInfo(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
 		return
 	}
 
-	waifu, err := psql.GetWaifuById(waifuId)
+	waifu, err := psql.GetWaifuById(uint(waifuId))
 	if err != nil {
 		ctx.Error(err)
 		return