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 17:20:50 +03:00
parent 9bd5354250
commit cec3a35d83
7 changed files with 82 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
database/psql/waifus.go
View File

@@ -50,7 +50,7 @@ func GetFreeWaifusWithRarity(rarity int) ([]*Waifu, error) {
return waifus, tx.Error
}
func GetWaifuById(id uint) (*Waifu, error) {
func GetWaifuById(id int) (*Waifu, error) {
waifu := new(Waifu)
tx := database.PostgresDatabase.Joins("Owner").Find(waifu, id)
return waifu, tx.Error

39
database/red/rp_chats.go
View File

@@ -2,43 +2,48 @@ package red
import (
"context"
"errors"
"fmt"
"kurumibot/laniakea"
)
var ctx = context.Background()
func RPSetSelectedWaifu(db *laniakea.DatabaseContext, userId, waifuId uint) error {
func RPSetSelectedWaifu(db *laniakea.DatabaseContext, userId, waifuId int) error {
key := fmt.Sprintf("ai.chats.rp.%d", userId)
return db.Redis.Set(ctx, key, waifuId, 0).Err()
}
func RPGetSelectedWaifu(db *laniakea.DatabaseContext, userId uint) uint {
func RPGetSelectedWaifu(db *laniakea.DatabaseContext, userId int) int {
key := fmt.Sprintf("ai.chats.rp.%d", userId)
res := db.Redis.Get(ctx, key)
if res.Err() != nil {
return 0
}
i, _ := res.Int()
return uint(i)
return i
}
func GetChatId(db *laniakea.DatabaseContext, userId uint) string {
waifuId := RPGetSelectedWaifu(db, userId)
if waifuId == 0 {
return "0"
}
key := fmt.Sprintf("ai.chats.rp.%d.%d", userId, waifuId)
func RPSetChatId(db *laniakea.DatabaseContext, userId, waifuId int, chatId string) error {
key := fmt.Sprintf("ai.chats.rp.%d.%d.chat", userId, waifuId)
return db.Redis.Set(context.Background(), key, chatId, 0).Err()
}
func RPGetChatId(db *laniakea.DatabaseContext, userId, waifuId int) string {
key := fmt.Sprintf("ai.chats.rp.%d.%d.chat", userId, waifuId)
res := db.Redis.Get(ctx, key)
if res.Err() != nil {
return ""
}
return res.Val()
}
func SaveChatId(db *laniakea.DatabaseContext, userId uint, chatId string) error {
waifuId := RPGetSelectedWaifu(db, userId)
if waifuId == 0 {
return errors.New("no chat found")
}
key := fmt.Sprintf("ai.chats.rp.%d.%d", userId, waifuId)
return db.Redis.Set(context.Background(), key, chatId, 0).Err()
func RPSetChatPrompt(db *laniakea.DatabaseContext, userId, waifuId int, prompt string) error {
key := fmt.Sprintf("ai.chats.rp.%d.%d.prompt", userId, waifuId)
return db.Redis.Set(context.Background(), key, prompt, 0).Err()
}
func RPGetChatPrompt(db *laniakea.DatabaseContext, userId, waifuId int) string {
key := fmt.Sprintf("ai.chats.rp.%d.%d.prompt", userId, waifuId)
res := db.Redis.Get(ctx, key)
if res.Err() != nil {
return ""
}
return res.Val()
}