many new
This commit is contained in:
3
.env
3
.env
@@ -6,4 +6,5 @@ PSQL_NAME=kurumi_bot
|
||||
MONGO_HOST=127.0.0.1
|
||||
MONGO_USER=kurumi_bot
|
||||
MONGO_PASS=kurumi_bot
|
||||
MONGO_NAME=kurumi
|
||||
MONGO_NAME=kurumi
|
||||
REDIS_HOST=127.0.0.1
|
||||
@@ -1,6 +0,0 @@
|
||||
# User ONLY for bot. Rename to .env.production
|
||||
TG_TOKEN=
|
||||
PREFIXES=/;!
|
||||
DEBUG=true
|
||||
USE_REQ_LOG=true
|
||||
WRITE_TO_FILE=true
|
||||
@@ -1,4 +1,4 @@
|
||||
FROM golang:1.25.1-alpine3.22 AS builder
|
||||
FROM golang:1.25.5-alpine3.23 AS builder
|
||||
WORKDIR /usr/src/kurumi
|
||||
COPY go.mod go.sum ./
|
||||
RUN go mod download
|
||||
|
||||
2
Makefile
2
Makefile
@@ -1,3 +1,3 @@
|
||||
build:
|
||||
docker build -t git.nix13.pw/scuroneko/kurumibotgo:latest -t git.nix13.pw/scuroneko/kurumibotgo:0.1.4 -f ./Dockerfile .
|
||||
docker build -t git.nix13.pw/scuroneko/kurumibotgo:latest -t git.nix13.pw/scuroneko/kurumibotgo:0.2.0 -f ./Dockerfile .
|
||||
docker push git.nix13.pw/scuroneko/kurumibotgo --all-tags
|
||||
40
database/mdb/rp_chats.go
Normal file
40
database/mdb/rp_chats.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package mdb
|
||||
|
||||
import (
|
||||
"context"
|
||||
"kurumibot/database"
|
||||
"kurumibot/laniakea"
|
||||
"time"
|
||||
|
||||
"go.mongodb.org/mongo-driver/v2/bson"
|
||||
)
|
||||
|
||||
type RPChatMessage struct {
|
||||
ChatID string `bson:"chat_id"`
|
||||
Role string `bson:"role"`
|
||||
Message string `bson:"message"`
|
||||
}
|
||||
|
||||
func GetChatHistory(db *laniakea.DatabaseContext, chatId string) ([]*RPChatMessage, error) {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
defer cancel()
|
||||
col := database.GetMongoCollection(db, "rp_chat_history")
|
||||
cursor, err := col.Find(ctx, bson.M{"chat_id": chatId})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
result := make([]*RPChatMessage, 0)
|
||||
err = cursor.All(ctx, &result)
|
||||
return result, err
|
||||
}
|
||||
func UpdateChatHistory(db *laniakea.DatabaseContext, chatId, role, message string) error {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
defer cancel()
|
||||
col := database.GetMongoCollection(db, "rp_chat_history")
|
||||
_, err := col.InsertOne(ctx, RPChatMessage{
|
||||
chatId,
|
||||
role,
|
||||
message,
|
||||
})
|
||||
return err
|
||||
}
|
||||
@@ -13,17 +13,16 @@ import (
|
||||
)
|
||||
|
||||
type User struct {
|
||||
ID int
|
||||
TelegramID int `gorm:"column:tg_id"`
|
||||
Balance decimal.Decimal
|
||||
Name string
|
||||
GroupID int `gorm:"default:1"`
|
||||
Group *Group
|
||||
Level int `gorm:"default:1"`
|
||||
Exp int `gorm:"default:0"`
|
||||
WorkID int `gorm:"default:1"`
|
||||
Work *Work
|
||||
WorkTime time.Time
|
||||
ID int
|
||||
Balance decimal.Decimal
|
||||
Name string
|
||||
GroupID int `gorm:"default:1"`
|
||||
Group *Group
|
||||
Level int `gorm:"default:1"`
|
||||
Exp int `gorm:"default:0"`
|
||||
WorkID int `gorm:"default:1"`
|
||||
Work *Work
|
||||
WorkTime time.Time
|
||||
|
||||
AutoID sql.NullInt64
|
||||
Auto *ShopAuto
|
||||
@@ -64,10 +63,10 @@ func GetOrCreateUser(tgId int, name string) (*User, error) {
|
||||
return user, err
|
||||
}
|
||||
|
||||
func CreateUser(tgId int, name string) (*User, error) {
|
||||
func CreateUser(id int, name string) (*User, error) {
|
||||
user := &User{
|
||||
TelegramID: tgId,
|
||||
Name: name,
|
||||
ID: id,
|
||||
Name: name,
|
||||
}
|
||||
tx := database.PostgresDatabase.Create(user)
|
||||
return user, tx.Error
|
||||
@@ -75,7 +74,7 @@ func CreateUser(tgId int, name string) (*User, error) {
|
||||
|
||||
func GetUser(telegramId int) (*User, error) {
|
||||
user := new(User)
|
||||
tx := database.PostgresDatabase.Joins("Group").Joins("Work").Joins("Auto").Joins("Business").Joins("Maid").Joins("Miner").Joins("Fraction").Preload("Pair").Take(user, "tg_id=?", telegramId)
|
||||
tx := database.PostgresDatabase.Joins("Group").Joins("Work").Joins("Auto").Joins("Business").Joins("Maid").Joins("Miner").Joins("Fraction").Preload("Pair").Take(user, "users.id=?", telegramId)
|
||||
return user, tx.Error
|
||||
}
|
||||
|
||||
|
||||
44
database/red/rp_chats.go
Normal file
44
database/red/rp_chats.go
Normal file
@@ -0,0 +1,44 @@
|
||||
package red
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"kurumibot/laniakea"
|
||||
)
|
||||
|
||||
var ctx = context.Background()
|
||||
|
||||
func RPSetSelectedWaifu(db *laniakea.DatabaseContext, userId, waifuId uint) 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 {
|
||||
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)
|
||||
}
|
||||
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)
|
||||
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()
|
||||
}
|
||||
@@ -1,12 +1,17 @@
|
||||
package database
|
||||
|
||||
import "github.com/redis/go-redis/v9"
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"github.com/redis/go-redis/v9"
|
||||
)
|
||||
|
||||
var RedisClient *redis.Client
|
||||
|
||||
func ConnectRedis() {
|
||||
RedisClient = redis.NewClient(&redis.Options{
|
||||
Addr: "redis:6379",
|
||||
Addr: fmt.Sprintf("%s:6379", os.Getenv("REDIS_HOST")),
|
||||
Password: "",
|
||||
DB: 0,
|
||||
})
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
services:
|
||||
postgres:
|
||||
image: postgres:17.6-alpine3.22
|
||||
image: postgres:18.1-alpine
|
||||
networks:
|
||||
- bot
|
||||
restart: always
|
||||
@@ -33,7 +33,7 @@ services:
|
||||
- ./scripts/mongo:/docker-entrypoint-initdb.d
|
||||
- mongo_data:/data/db
|
||||
redis:
|
||||
image: redis:8.2.1-alpine
|
||||
image: redis:8.4-alpine
|
||||
networks:
|
||||
- bot
|
||||
restart: always
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
name: "kurumibot"
|
||||
include:
|
||||
- db.docker-compose.yml
|
||||
|
||||
services:
|
||||
bot:
|
||||
env_file:
|
||||
|
||||
23
go.mod
23
go.mod
@@ -1,15 +1,16 @@
|
||||
module kurumibot
|
||||
|
||||
go 1.25.0
|
||||
go 1.25.5
|
||||
|
||||
require (
|
||||
github.com/fatih/color v1.18.0
|
||||
github.com/google/uuid v1.6.0
|
||||
github.com/joho/godotenv v1.5.1
|
||||
github.com/redis/go-redis/v9 v9.14.0
|
||||
github.com/redis/go-redis/v9 v9.17.2
|
||||
github.com/shopspring/decimal v1.4.0
|
||||
go.mongodb.org/mongo-driver/v2 v2.3.0
|
||||
go.mongodb.org/mongo-driver/v2 v2.4.1
|
||||
gorm.io/driver/postgres v1.6.0
|
||||
gorm.io/gorm v1.31.0
|
||||
gorm.io/gorm v1.31.1
|
||||
)
|
||||
|
||||
require (
|
||||
@@ -18,19 +19,19 @@ require (
|
||||
github.com/golang/snappy v1.0.0 // indirect
|
||||
github.com/jackc/pgpassfile v1.0.0 // indirect
|
||||
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
|
||||
github.com/jackc/pgx/v5 v5.7.6 // indirect
|
||||
github.com/jackc/pgx/v5 v5.8.0 // indirect
|
||||
github.com/jackc/puddle/v2 v2.2.2 // indirect
|
||||
github.com/jinzhu/inflection v1.0.0 // indirect
|
||||
github.com/jinzhu/now v1.1.5 // indirect
|
||||
github.com/klauspost/compress v1.18.0 // indirect
|
||||
github.com/klauspost/compress v1.18.2 // indirect
|
||||
github.com/mattn/go-colorable v0.1.14 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/xdg-go/pbkdf2 v1.0.0 // indirect
|
||||
github.com/xdg-go/scram v1.1.2 // indirect
|
||||
github.com/xdg-go/scram v1.2.0 // indirect
|
||||
github.com/xdg-go/stringprep v1.0.4 // indirect
|
||||
github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 // indirect
|
||||
golang.org/x/crypto v0.42.0 // indirect
|
||||
golang.org/x/sync v0.17.0 // indirect
|
||||
golang.org/x/sys v0.36.0 // indirect
|
||||
golang.org/x/text v0.29.0 // indirect
|
||||
golang.org/x/crypto v0.47.0 // indirect
|
||||
golang.org/x/sync v0.19.0 // indirect
|
||||
golang.org/x/sys v0.40.0 // indirect
|
||||
golang.org/x/text v0.33.0 // indirect
|
||||
)
|
||||
|
||||
23
go.sum
23
go.sum
@@ -9,6 +9,8 @@ github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
|
||||
github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
|
||||
github.com/golang/snappy v1.0.0 h1:Oy607GVXHs7RtbggtPBnr2RmDArIsAefDwvrdWvRhGs=
|
||||
github.com/golang/snappy v1.0.0/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
|
||||
github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
|
||||
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo=
|
||||
@@ -19,6 +21,8 @@ github.com/jackc/pgx/v5 v5.7.5 h1:JHGfMnQY+IEtGM63d+NGMjoRpysB2JBwDr5fsngwmJs=
|
||||
github.com/jackc/pgx/v5 v5.7.5/go.mod h1:aruU7o91Tc2q2cFp5h4uP3f6ztExVpyVv88Xl/8Vl8M=
|
||||
github.com/jackc/pgx/v5 v5.7.6 h1:rWQc5FwZSPX58r1OQmkuaNicxdmExaEz5A2DO2hUuTk=
|
||||
github.com/jackc/pgx/v5 v5.7.6/go.mod h1:aruU7o91Tc2q2cFp5h4uP3f6ztExVpyVv88Xl/8Vl8M=
|
||||
github.com/jackc/pgx/v5 v5.8.0 h1:TYPDoleBBme0xGSAX3/+NujXXtpZn9HBONkQC7IEZSo=
|
||||
github.com/jackc/pgx/v5 v5.8.0/go.mod h1:QVeDInX2m9VyzvNeiCJVjCkNFqzsNb43204HshNSZKw=
|
||||
github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo=
|
||||
github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4=
|
||||
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
|
||||
@@ -31,6 +35,8 @@ github.com/klauspost/compress v1.16.7 h1:2mk3MPGNzKyxErAw8YaohYh69+pa4sIQSC0fPGC
|
||||
github.com/klauspost/compress v1.16.7/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
|
||||
github.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo=
|
||||
github.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ=
|
||||
github.com/klauspost/compress v1.18.2 h1:iiPHWW0YrcFgpBYhsA6D1+fqHssJscY/Tm/y2Uqnapk=
|
||||
github.com/klauspost/compress v1.18.2/go.mod h1:R0h/fSBs8DE4ENlcrlib3PsXS61voFxhIs2DeRhCvJ4=
|
||||
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
|
||||
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
||||
github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
|
||||
@@ -44,6 +50,7 @@ github.com/redis/go-redis/v9 v9.12.1 h1:k5iquqv27aBtnTm2tIkROUDp8JBXhXZIVu1InSgv
|
||||
github.com/redis/go-redis/v9 v9.12.1/go.mod h1:huWgSWd8mW6+m0VPhJjSSQ+d6Nh1VICQ6Q5lHuCH/Iw=
|
||||
github.com/redis/go-redis/v9 v9.14.0 h1:u4tNCjXOyzfgeLN+vAZaW1xUooqWDqVEsZN0U01jfAE=
|
||||
github.com/redis/go-redis/v9 v9.14.0/go.mod h1:huWgSWd8mW6+m0VPhJjSSQ+d6Nh1VICQ6Q5lHuCH/Iw=
|
||||
github.com/redis/go-redis/v9 v9.17.2 h1:P2EGsA4qVIM3Pp+aPocCJ7DguDHhqrXNhVcEp4ViluI=
|
||||
github.com/redis/go-redis/v9 v9.17.2/go.mod h1:u410H11HMLoB+TP67dz8rL9s6QW2j76l0//kSOd3370=
|
||||
github.com/shopspring/decimal v1.4.0 h1:bxl37RwXBklmTi0C79JfXCEBD1cqqHt0bbgBAGFp81k=
|
||||
github.com/shopspring/decimal v1.4.0/go.mod h1:gawqmDU56v4yIKSwfBSFip1HdCCXN8/+DMd9qYNcwME=
|
||||
@@ -56,6 +63,8 @@ github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c=
|
||||
github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
|
||||
github.com/xdg-go/scram v1.1.2 h1:FHX5I5B4i4hKRVRBCFRxq1iQRej7WO3hhBuJf+UUySY=
|
||||
github.com/xdg-go/scram v1.1.2/go.mod h1:RT/sEzTbU5y00aCK8UOx6R7YryM0iF1N2MOmC3kKLN4=
|
||||
github.com/xdg-go/scram v1.2.0 h1:bYKF2AEwG5rqd1BumT4gAnvwU/M9nBp2pTSxeZw7Wvs=
|
||||
github.com/xdg-go/scram v1.2.0/go.mod h1:3dlrS0iBaWKYVt2ZfA4cj48umJZ+cAEbR6/SjLA88I8=
|
||||
github.com/xdg-go/stringprep v1.0.4 h1:XLI/Ng3O1Atzq0oBs3TWm+5ZVgkq2aqdlvP9JtoZ6c8=
|
||||
github.com/xdg-go/stringprep v1.0.4/go.mod h1:mPGuuIYwz7CmR2bT9j4GbQqutWS1zV24gijq1dTyGkM=
|
||||
github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 h1:ilQV1hzziu+LLM3zUTJ0trRztfwgjqKnBWNtSRkbmwM=
|
||||
@@ -63,6 +72,10 @@ github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78/go.mod h1:aL8wCCfTfS
|
||||
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
|
||||
go.mongodb.org/mongo-driver/v2 v2.3.0 h1:sh55yOXA2vUjW1QYw/2tRlHSQViwDyPnW61AwpZ4rtU=
|
||||
go.mongodb.org/mongo-driver/v2 v2.3.0/go.mod h1:jHeEDJHJq7tm6ZF45Issun9dbogjfnPySb1vXA7EeAI=
|
||||
go.mongodb.org/mongo-driver/v2 v2.3.1 h1:WrCgSzO7dh1/FrePud9dK5fKNZOE97q5EQimGkos7Wo=
|
||||
go.mongodb.org/mongo-driver/v2 v2.3.1/go.mod h1:jHeEDJHJq7tm6ZF45Issun9dbogjfnPySb1vXA7EeAI=
|
||||
go.mongodb.org/mongo-driver/v2 v2.4.1 h1:hGDMngUao03OVQ6sgV5csk+RWOIkF+CuLsTPobNMGNI=
|
||||
go.mongodb.org/mongo-driver/v2 v2.4.1/go.mod h1:jHeEDJHJq7tm6ZF45Issun9dbogjfnPySb1vXA7EeAI=
|
||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||
golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
|
||||
@@ -73,6 +86,8 @@ golang.org/x/crypto v0.41.0 h1:WKYxWedPGCTVVl5+WHSSrOBT0O8lx32+zxmHxijgXp4=
|
||||
golang.org/x/crypto v0.41.0/go.mod h1:pO5AFd7FA68rFak7rOAGVuygIISepHftHnr8dr6+sUc=
|
||||
golang.org/x/crypto v0.42.0 h1:chiH31gIWm57EkTXpwnqf8qeuMUi0yekh6mT2AvFlqI=
|
||||
golang.org/x/crypto v0.42.0/go.mod h1:4+rDnOTJhQCx2q7/j6rAN5XDw8kPjeaXEUR2eL94ix8=
|
||||
golang.org/x/crypto v0.47.0 h1:V6e3FRj+n4dbpw86FJ8Fv7XVOql7TEwpHapKoMJ/GO8=
|
||||
golang.org/x/crypto v0.47.0/go.mod h1:ff3Y9VzzKbwSSEzWqJsJVBnWmRwRSHt/6Op5n9bQc4A=
|
||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||
@@ -87,6 +102,8 @@ golang.org/x/sync v0.16.0 h1:ycBJEhp9p4vXvUZNszeOq0kGTPghopOL8q0fq3vstxw=
|
||||
golang.org/x/sync v0.16.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
|
||||
golang.org/x/sync v0.17.0 h1:l60nONMj9l5drqw6jlhIELNv9I0A4OFgRsG9k2oT9Ug=
|
||||
golang.org/x/sync v0.17.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
||||
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
@@ -102,6 +119,8 @@ golang.org/x/sys v0.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI=
|
||||
golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
|
||||
golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k=
|
||||
golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||
golang.org/x/sys v0.40.0 h1:DBZZqJ2Rkml6QMQsZywtnjnnGvHza6BTfYFWY9kjEWQ=
|
||||
golang.org/x/sys v0.40.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||
@@ -116,6 +135,8 @@ golang.org/x/text v0.28.0 h1:rhazDwis8INMIwQ4tpjLDzUhx6RlXqZNPEM0huQojng=
|
||||
golang.org/x/text v0.28.0/go.mod h1:U8nCwOR8jO/marOQ0QbDiOngZVEBB7MAiitBuMjXiNU=
|
||||
golang.org/x/text v0.29.0 h1:1neNs90w9YzJ9BocxfsQNHKuAT4pkghyXc4nhZ6sJvk=
|
||||
golang.org/x/text v0.29.0/go.mod h1:7MhJOA9CD2qZyOKYazxdYMF85OwPdEr9jTtBpO7ydH4=
|
||||
golang.org/x/text v0.33.0 h1:B3njUFyqtHDUI5jMn1YIr5B0IE2U0qck04r6d4KPAxE=
|
||||
golang.org/x/text v0.33.0/go.mod h1:LuMebE6+rBincTi9+xWTY8TztLzKHc/9C1uBCG27+q8=
|
||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||
@@ -130,3 +151,5 @@ gorm.io/gorm v1.30.2 h1:f7bevlVoVe4Byu3pmbWPVHnPsLoWaMjEb7/clyr9Ivs=
|
||||
gorm.io/gorm v1.30.2/go.mod h1:8Z33v652h4//uMA76KjeDH8mJXPm1QNCYrMeatR0DOE=
|
||||
gorm.io/gorm v1.31.0 h1:0VlycGreVhK7RF/Bwt51Fk8v0xLiiiFdbGDPIZQ7mJY=
|
||||
gorm.io/gorm v1.31.0/go.mod h1:XyQVbO2k6YkOis7C2437jSit3SsDK72s7n7rsSHd+Gs=
|
||||
gorm.io/gorm v1.31.1 h1:7CA8FTFz/gRfgqgpeKIBcervUn3xSyPUmr6B2WXJ7kg=
|
||||
gorm.io/gorm v1.31.1/go.mod h1:XyQVbO2k6YkOis7C2437jSit3SsDK72s7n7rsSHd+Gs=
|
||||
|
||||
@@ -95,7 +95,7 @@ func passiveIncome(b *laniakea.Bot) {
|
||||
user.IncomeTime = time.Now().Add(-time.Hour * 2)
|
||||
database.PostgresDatabase.Save(user)
|
||||
|
||||
b.Logger().Debug(fmt.Sprintf("У %d было пассивно собрано. След. сбор через час\n", user.TelegramID))
|
||||
b.Logger().Debug(fmt.Sprintf("У %d было пассивно собрано. След. сбор через час\n", user.ID))
|
||||
}
|
||||
|
||||
time.Sleep(time.Second * 5)
|
||||
|
||||
@@ -1,34 +1,112 @@
|
||||
package plugins
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"kurumibot/database/mdb"
|
||||
"kurumibot/database/red"
|
||||
"kurumibot/laniakea"
|
||||
"kurumibot/utils/ai"
|
||||
"log"
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
func RegisterTestRP(bot *laniakea.Bot) {
|
||||
rp := laniakea.NewPlugin("RP")
|
||||
rp = rp.Command(selectWaifu, "selwaifu")
|
||||
rp = rp.Payload(selectWaifu, "rp.selwaifu")
|
||||
rp = rp.Command(newChat, "newchat")
|
||||
rp = rp.Command(generate, "g", "gen")
|
||||
|
||||
bot.AddPlugins(rp.Build())
|
||||
}
|
||||
|
||||
func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
api := ai.NewOpenAIAPI(ai.CosmoRPUrl, os.Getenv("PAWAN_KEY"), "cosmorp-2.5")
|
||||
res, err := api.CreateCompletion(ai.CreateCompletionReq{
|
||||
Model: "cosmorp-2.5",
|
||||
Messages: []ai.Message{
|
||||
{
|
||||
Role: "developer",
|
||||
Content: "123",
|
||||
},
|
||||
{
|
||||
Role: "user",
|
||||
Content: strings.Join(ctx.Args, " "),
|
||||
},
|
||||
},
|
||||
})
|
||||
log.Println(res, err)
|
||||
func selectWaifu(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
waifuId, err := strconv.Atoi(ctx.Args[0])
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
return
|
||||
}
|
||||
err = red.RPSetSelectedWaifu(db, uint(ctx.FromID), uint(waifuId))
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
return
|
||||
}
|
||||
ctx.Answer(fmt.Sprintf("Была выбрана вайфу %d", waifuId))
|
||||
}
|
||||
|
||||
func newChat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
err := red.SaveChatId(db, uint(ctx.FromID), uuid.New().String())
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
return
|
||||
}
|
||||
ctx.Answer("Был создан новый чат.")
|
||||
}
|
||||
|
||||
func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
chatId := red.GetChatId(db, uint(ctx.FromID))
|
||||
if chatId == "0" {
|
||||
ctx.Answer("Не выбрана вайфу")
|
||||
return
|
||||
}
|
||||
if chatId == "" {
|
||||
err := red.SaveChatId(db, uint(ctx.FromID), uuid.New().String())
|
||||
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 см",
|
||||
},
|
||||
}
|
||||
|
||||
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
|
||||
}
|
||||
for _, m := range history {
|
||||
messages = append(messages, ai.Message{
|
||||
Role: m.Role,
|
||||
Content: m.Message,
|
||||
})
|
||||
}
|
||||
userMessage := strings.Join(ctx.Args, " ")
|
||||
messages = append(messages, ai.Message{
|
||||
Role: "system", Content: ai.FormatPrompt(ai.PostHistoryPrompt, "Яэ Мико", ctx.Msg.From.FirstName),
|
||||
}, ai.Message{
|
||||
Role: "user", Content: userMessage,
|
||||
})
|
||||
err = mdb.UpdateChatHistory(db, chatId, "user", userMessage)
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
return
|
||||
}
|
||||
m := ctx.Answer("Генерация запущена...")
|
||||
res, err := api.CreateCompletion(ai.CreateCompletionReq{
|
||||
Model: "cosmorp-2.5",
|
||||
Messages: messages,
|
||||
})
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
return
|
||||
}
|
||||
response := make([]string, 0)
|
||||
for _, choice := range res.Choices {
|
||||
m := choice.Message
|
||||
response = append(response, m.Content)
|
||||
err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Content)
|
||||
}
|
||||
m.Edit(laniakea.EscapeMarkdown(strings.Join(response, "\n")))
|
||||
}
|
||||
|
||||
2025
requests.log
Normal file
2025
requests.log
Normal file
File diff suppressed because it is too large
Load Diff
3
scripts/mongo/00-init.js
Normal file
3
scripts/mongo/00-init.js
Normal file
@@ -0,0 +1,3 @@
|
||||
db.createCollection("logs");
|
||||
db.createCollection("msg_logs");
|
||||
db.createCollection("codes");
|
||||
@@ -1,2 +0,0 @@
|
||||
db.createCollection("logs");
|
||||
db.createCollection("mds_logs");
|
||||
12
scripts/postgres/00-init.sql
Normal file
12
scripts/postgres/00-init.sql
Normal file
@@ -0,0 +1,12 @@
|
||||
SET statement_timeout = 0;
|
||||
SET lock_timeout = 0;
|
||||
SET idle_in_transaction_session_timeout = 0;
|
||||
SET transaction_timeout = 0;
|
||||
SET client_encoding = 'UTF8';
|
||||
SET standard_conforming_strings = on;
|
||||
SET check_function_bodies = false;
|
||||
SET xmloption = content;
|
||||
SET client_min_messages = warning;
|
||||
SET row_security = off;
|
||||
SET default_tablespace = '';
|
||||
SET default_table_access_method = heap;
|
||||
21
scripts/postgres/01-works.sql
Normal file
21
scripts/postgres/01-works.sql
Normal file
@@ -0,0 +1,21 @@
|
||||
BEGIN TRANSACTION;
|
||||
CREATE TABLE works (
|
||||
id serial NOT NULL PRIMARY KEY,
|
||||
name text NOT NULL,
|
||||
required_level integer DEFAULT 1 NOT NULL,
|
||||
money_income numeric(10,0) DEFAULT 1 NOT NULL,
|
||||
min_exp integer DEFAULT 1 NOT NULL,
|
||||
max_exp integer DEFAULT 1 NOT NULL
|
||||
);
|
||||
|
||||
INSERT INTO works VALUES (1, 'Уборщик в маке', 1, 13000, 20, 28);
|
||||
INSERT INTO works VALUES (2, 'Рабочий на заводе', 3, 20000, 48, 65);
|
||||
INSERT INTO works VALUES (3, 'Продавец в пятёрочке', 5, 25000, 76, 103);
|
||||
INSERT INTO works VALUES (4, 'Учитель', 7, 40000, 104, 141);
|
||||
INSERT INTO works VALUES (5, 'Повар', 9, 50000, 175, 237);
|
||||
INSERT INTO works VALUES (6, 'Полицейский', 11, 60000, 260, 351);
|
||||
INSERT INTO works VALUES (7, 'Художник', 13, 80000, 305, 412);
|
||||
INSERT INTO works VALUES (8, 'Программист', 15, 100000, 350, 474);
|
||||
|
||||
ALTER SEQUENCE works_id_seq RESTART WITH 9;
|
||||
COMMIT TRANSACTION;
|
||||
55
scripts/postgres/02-users.sql
Normal file
55
scripts/postgres/02-users.sql
Normal file
@@ -0,0 +1,55 @@
|
||||
CREATE TABLE groups (
|
||||
id serial NOT NULL PRIMARY KEY,
|
||||
name text,
|
||||
is_admin boolean DEFAULT false NOT NULL,
|
||||
is_vip boolean DEFAULT false NOT NULL,
|
||||
multiplier numeric DEFAULT 1 NOT NULL,
|
||||
sale double precision DEFAULT 1 NOT NULL,
|
||||
max_multigen integer DEFAULT 2 NOT NULL,
|
||||
is_tester boolean DEFAULT false NOT NULL,
|
||||
max_waifus integer DEFAULT 3 NOT NULL
|
||||
);
|
||||
|
||||
CREATE TABLE users (
|
||||
id int NOT NULL PRIMARY KEY,
|
||||
balance decimal(20,0) NOT NULL DEFAULT 0,
|
||||
name text,
|
||||
group_id int REFERENCES groups(id),
|
||||
level int DEFAULT 1,
|
||||
exp int DEFAULT 0,
|
||||
work_id int REFERENCES works(id),
|
||||
work_time timestamp DEFAULT now(),
|
||||
|
||||
auto_id int DEFAULT NULL,
|
||||
business_id int DEFAULT NULL,
|
||||
maid_id int DEFAULT NULL,
|
||||
miner_id int DEFAULT NULL,
|
||||
|
||||
income_time timestamp DEFAULT now(),
|
||||
btc decimal(16,6) DEFAULT 0,
|
||||
invested decimal(20, 0) DEFAULT 0,
|
||||
pair_id int DEFAULT NULL,
|
||||
greeting text DEFAULT 'Привет',
|
||||
donat int DEFAULT 0,
|
||||
fraction_id int DEFAULT NULL,
|
||||
|
||||
money_income decimal(20, 0) DEFAULT 0,
|
||||
exp_income int DEFAULT 0,
|
||||
btc_income decimal(16, 6) DEFAULT 0,
|
||||
|
||||
waifu_search_time timestamp DEFAULT now()
|
||||
);
|
||||
CREATE UNIQUE INDEX users_uindex ON users(id);
|
||||
CREATE UNIQUE INDEX groups_uindex ON groups(id);
|
||||
|
||||
BEGIN TRANSACTION;
|
||||
INSERT INTO groups VALUES (1, '✨Пользователь✨', false, false, 1.0, 1, 3, false, 3);
|
||||
INSERT INTO groups VALUES (3, '👾Премиум🌟', false, true, 5, 0.75, 7, false, 3);
|
||||
INSERT INTO groups VALUES (2, '🔮VIP🔮', false, true, 2.5, 0.9, 5, false, 3);
|
||||
INSERT INTO groups VALUES (4, '💎Админ💎', true, true, 10, 0.6, 10, true, 3);
|
||||
INSERT INTO groups VALUES (5, '💰🔱СОЗДАТЕЛЬ🔱💰', true, true, 20, 0.5, 10, true, 3);
|
||||
INSERT INTO groups VALUES (6, '⚠Тостер⚠', false, true, 3, 0.85, 3, true, 3);
|
||||
INSERT INTO groups VALUES (1488, '📚Экономист📈', false, true, 14.88, 0.6, 3, false, 3);
|
||||
INSERT INTO groups VALUES (1337, '🏳🌈ToP GaY In ThE WorlD🏳🌈🌈', false, true, 13.37, 0.7, 3, false, 3);
|
||||
ALTER SEQUENCE groups_id_seq RESTART WITH 8;
|
||||
COMMIT TRANSACTION;
|
||||
11
scripts/postgres/03-fractions.sql
Normal file
11
scripts/postgres/03-fractions.sql
Normal file
@@ -0,0 +1,11 @@
|
||||
BEGIN TRANSACTION;
|
||||
CREATE TABLE fractions(
|
||||
id serial PRIMARY KEY,
|
||||
name text,
|
||||
owner_id int REFERENCES users(id),
|
||||
money decimal(20, 0),
|
||||
exp int DEFAULT 0,
|
||||
level int DEFAULT 0
|
||||
);
|
||||
CREATE UNIQUE INDEX fractions_uindex ON fractions(id);
|
||||
COMMIT TRANSACTION;
|
||||
15
scripts/postgres/04-waifus.sql
Normal file
15
scripts/postgres/04-waifus.sql
Normal file
@@ -0,0 +1,15 @@
|
||||
BEGIN TRANSACTION;
|
||||
CREATE TABLE waifus(
|
||||
id serial PRIMARY KEY,
|
||||
name text DEFAULT '' NOT NULL,
|
||||
rarity int DEFAULT 3 NOT NULL,
|
||||
exp_bonus decimal(4,2) DEFAULT 1,
|
||||
money_bonus decimal(4, 2) DEFAULT 1,
|
||||
market_price decimal(20, 0) DEFAULT 1000000,
|
||||
fandom text DEFAULT '' NOT NULL,
|
||||
image text DEFAULT '' NOT NULL,
|
||||
owner_id int REFERENCES users(id)
|
||||
);
|
||||
CREATE UNIQUE INDEX waifus_uindex ON waifus(id);
|
||||
CREATE INDEX waifus_index ON waifus(fandom, owner_id);
|
||||
COMMIT TRANSACTION;
|
||||
84
scripts/postgres/05-shops.sql
Normal file
84
scripts/postgres/05-shops.sql
Normal file
@@ -0,0 +1,84 @@
|
||||
BEGIN TRANSACTION;
|
||||
CREATE TABLE shop_auto (
|
||||
id serial NOT NULL PRIMARY KEY,
|
||||
name text DEFAULT ''::text NOT NULL,
|
||||
price numeric(20,0) DEFAULT '999999999999'::bigint NOT NULL
|
||||
);
|
||||
CREATE UNIQUE INDEX shop_auto_uindex ON shop_auto(id);
|
||||
|
||||
CREATE TABLE shop_business (
|
||||
id serial NOT NULL PRIMARY KEY ,
|
||||
name text DEFAULT ''::text NOT NULL,
|
||||
price numeric(20,0) DEFAULT 999999999 NOT NULL,
|
||||
income numeric(20,0) DEFAULT 0 NOT NULL
|
||||
);
|
||||
CREATE UNIQUE INDEX shop_business_uindex ON shop_business(id);
|
||||
|
||||
CREATE TABLE shop_maid (
|
||||
id serial NOT NULL PRIMARY KEY,
|
||||
name text DEFAULT ''::text NOT NULL,
|
||||
price numeric(20,0) DEFAULT 0 NOT NULL,
|
||||
income numeric(20,0) DEFAULT 0 NOT NULL
|
||||
);
|
||||
CREATE UNIQUE INDEX shop_maid_uindex ON shop_maid(id);
|
||||
|
||||
CREATE TABLE shop_miner (
|
||||
id serial NOT NULL,
|
||||
name text DEFAULT ''::text NOT NULL,
|
||||
price numeric(20,0) DEFAULT 1 NOT NULL,
|
||||
income numeric(20,6) DEFAULT 1 NOT NULL
|
||||
);
|
||||
CREATE UNIQUE INDEX shop_miner_uindex ON shop_miner(id);
|
||||
|
||||
INSERT INTO shop_auto VALUES (1, 'Москвич 412', 40000);
|
||||
INSERT INTO shop_auto VALUES (2, 'ВАЗ 2105', 100000);
|
||||
INSERT INTO shop_auto VALUES (3, 'Nissan Skyline GT-R R32', 225000);
|
||||
INSERT INTO shop_auto VALUES (4, 'Toyota AE86 Sprinter Trueno', 450000);
|
||||
INSERT INTO shop_auto VALUES (5, 'Toyota Mark II X90 Tourer V', 800000);
|
||||
INSERT INTO shop_auto VALUES (6, 'Mazda RX-7 FD3S', 1200000);
|
||||
INSERT INTO shop_auto VALUES (7, 'Mitsubishi Lancer Evolution X', 1650000);
|
||||
INSERT INTO shop_auto VALUES (8, 'Subaru Impreza WRX STI II', 1875000);
|
||||
|
||||
INSERT INTO shop_business VALUES (1, 'Стенд с лимонадом', 200000, 1500);
|
||||
INSERT INTO shop_business VALUES (2, 'Союзпечать', 1000000, 7000);
|
||||
INSERT INTO shop_business VALUES (3, 'Маленький магазин вещей из Китая', 2500000, 17500);
|
||||
INSERT INTO shop_business VALUES (4, 'Ларёк с шаурмой', 3000000, 21000);
|
||||
INSERT INTO shop_business VALUES (5, 'Магазин тофу', 5000000, 35000);
|
||||
INSERT INTO shop_business VALUES (6, 'Соц. сеть', 10000000, 70000);
|
||||
INSERT INTO shop_business VALUES (7, 'Продажа машин из Японии', 20000000, 140000);
|
||||
INSERT INTO shop_business VALUES (8, 'Автосервис', 40000000, 280000);
|
||||
INSERT INTO shop_business VALUES (9, 'Компания по разработке игр', 100000000, 700000);
|
||||
INSERT INTO shop_business VALUES (10, 'Аниме студия', 250000000, 1725000);
|
||||
INSERT INTO shop_business VALUES (11, 'Сеть магазинов', 500000000, 3500000);
|
||||
INSERT INTO shop_business VALUES (12, 'Сеть заводов', 1000000000, 7000000);
|
||||
INSERT INTO shop_business VALUES (13, 'Автосалон', 10000000000, 70000000);
|
||||
INSERT INTO shop_business VALUES (14, 'Бренд электроники', 50000000000, 350000000);
|
||||
INSERT INTO shop_business VALUES (15, 'Автомобильная марка', 100000000000, 700000000);
|
||||
|
||||
INSERT INTO shop_maid VALUES (1, 'Кавайная горничная', 50, 7500);
|
||||
INSERT INTO shop_maid VALUES (2, 'Кавайная в кружеве', 100, 15000);
|
||||
INSERT INTO shop_maid VALUES (3, 'Тору', 250, 30000);
|
||||
INSERT INTO shop_maid VALUES (4, 'Лоли горничные', 500, 52500);
|
||||
INSERT INTO shop_maid VALUES (5, 'Рам и Рем', 2000, 150000);
|
||||
INSERT INTO shop_maid VALUES (6, 'Неко-горничная', 4500, 225000);
|
||||
|
||||
INSERT INTO shop_miner VALUES (1, 'Старый ПК', 20000, 0.000001);
|
||||
INSERT INTO shop_miner VALUES (2, 'Новый ПК', 120000, 0.000005);
|
||||
INSERT INTO shop_miner VALUES (3, 'Маленький майнер', 250000, 0.000010);
|
||||
INSERT INTO shop_miner VALUES (4, 'Средний майнер', 600000, 0.000025);
|
||||
INSERT INTO shop_miner VALUES (5, 'Большой майнер', 1000000, 0.000050);
|
||||
INSERT INTO shop_miner VALUES (6, 'Огромный майнер', 2500000, 0.000100);
|
||||
INSERT INTO shop_miner VALUES (7, 'Маленькая ферма майнеров', 10000000, 0.000500);
|
||||
INSERT INTO shop_miner VALUES (8, 'Средняя ферма майнкеров', 100000000, 0.005000);
|
||||
INSERT INTO shop_miner VALUES (9, 'Большая ферма майнеров', 500000000, 0.025000);
|
||||
INSERT INTO shop_miner VALUES (10, 'Огромная ферма майнеров', 5000000000, 0.250000);
|
||||
INSERT INTO shop_miner VALUES (11, 'Гигантская ферма майнеров', 10000000000, 0.500000);
|
||||
INSERT INTO shop_miner VALUES (12, 'Комната майнеров', 25000000000, 1.000000);
|
||||
INSERT INTO shop_miner VALUES (13, 'Этаж с майнерми', 100000000000, 5.000000);
|
||||
INSERT INTO shop_miner VALUES (14, 'Небоскреб майнеров', 250000000000, 10.000000);
|
||||
|
||||
ALTER SEQUENCE shop_auto_id_seq RESTART WITH 9;
|
||||
ALTER SEQUENCE shop_business_id_seq RESTART WITH 16;
|
||||
ALTER SEQUENCE shop_maid_id_seq RESTART WITH 7;
|
||||
ALTER SEQUENCE shop_miner_id_seq RESTART WITH 15;
|
||||
COMMIT TRANSACTION;
|
||||
@@ -1,171 +0,0 @@
|
||||
SET statement_timeout = 0;
|
||||
SET lock_timeout = 0;
|
||||
SET idle_in_transaction_session_timeout = 0;
|
||||
SET transaction_timeout = 0;
|
||||
SET client_encoding = 'UTF8';
|
||||
SET standard_conforming_strings = on;
|
||||
SELECT pg_catalog.set_config('search_path', '', false);
|
||||
SET check_function_bodies = false;
|
||||
SET xmloption = content;
|
||||
SET client_min_messages = warning;
|
||||
SET row_security = off;
|
||||
SET default_tablespace = '';
|
||||
SET default_table_access_method = heap;
|
||||
|
||||
CREATE TABLE public.groups (
|
||||
id integer NOT NULL,
|
||||
name text,
|
||||
is_admin boolean DEFAULT false NOT NULL,
|
||||
is_vip boolean DEFAULT false NOT NULL,
|
||||
multiplier numeric DEFAULT 1 NOT NULL,
|
||||
sale double precision DEFAULT 1 NOT NULL,
|
||||
max_multigen integer DEFAULT 2 NOT NULL,
|
||||
is_tester boolean DEFAULT false NOT NULL,
|
||||
max_waifus integer DEFAULT 3 NOT NULL
|
||||
);
|
||||
ALTER TABLE public.groups OWNER TO kurumi_bot;
|
||||
CREATE SEQUENCE public.groups_id_seq START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
|
||||
ALTER SEQUENCE public.groups_id_seq OWNER TO kurumi_bot;
|
||||
ALTER SEQUENCE public.groups_id_seq OWNED BY public.groups.id;
|
||||
|
||||
CREATE TABLE public.shop_autos (
|
||||
id integer NOT NULL,
|
||||
name text DEFAULT ''::text NOT NULL,
|
||||
price numeric(20,0) DEFAULT '999999999999'::bigint NOT NULL
|
||||
);
|
||||
ALTER TABLE public.shop_autos OWNER TO kurumi_bot;
|
||||
CREATE SEQUENCE public.shop_autos_id_seq START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
|
||||
ALTER SEQUENCE public.shop_autos_id_seq OWNER TO kurumi_bot;
|
||||
ALTER SEQUENCE public.shop_autos_id_seq OWNED BY public.shop_autos.id;
|
||||
|
||||
CREATE TABLE public.shop_businesses (
|
||||
id integer NOT NULL,
|
||||
name text DEFAULT ''::text NOT NULL,
|
||||
price numeric(20,0) DEFAULT 999999999 NOT NULL,
|
||||
income numeric(20,0) DEFAULT 0 NOT NULL
|
||||
);
|
||||
ALTER TABLE public.shop_businesses OWNER TO kurumi_bot;
|
||||
CREATE SEQUENCE public.shop_businesses_id_seq START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
|
||||
ALTER SEQUENCE public.shop_businesses_id_seq OWNER TO kurumi_bot;
|
||||
ALTER SEQUENCE public.shop_businesses_id_seq OWNED BY public.shop_businesses.id;
|
||||
|
||||
CREATE TABLE public.shop_maids (
|
||||
id integer NOT NULL,
|
||||
name text DEFAULT ''::text NOT NULL,
|
||||
price numeric(20,0) DEFAULT 0 NOT NULL,
|
||||
income numeric(20,0) DEFAULT 0 NOT NULL
|
||||
);
|
||||
ALTER TABLE public.shop_maids OWNER TO kurumi_bot;
|
||||
CREATE SEQUENCE public.shop_maids_id_seq AS integer START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
|
||||
ALTER SEQUENCE public.shop_maids_id_seq OWNER TO kurumi_bot;
|
||||
ALTER SEQUENCE public.shop_maids_id_seq OWNED BY public.shop_maids.id;
|
||||
|
||||
CREATE TABLE public.shop_miners (
|
||||
id integer NOT NULL,
|
||||
name text DEFAULT ''::text NOT NULL,
|
||||
price numeric(20,0) DEFAULT 1 NOT NULL,
|
||||
income numeric(20,6) DEFAULT 1 NOT NULL
|
||||
);
|
||||
ALTER TABLE public.shop_miners OWNER TO kurumi_bot;
|
||||
CREATE SEQUENCE public.shop_miners_id_seq START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
|
||||
ALTER SEQUENCE public.shop_miners_id_seq OWNER TO kurumi_bot;
|
||||
ALTER SEQUENCE public.shop_miners_id_seq OWNED BY public.shop_miners.id;
|
||||
|
||||
CREATE TABLE public.works (
|
||||
id integer NOT NULL,
|
||||
name text NOT NULL,
|
||||
required_level integer DEFAULT 1 NOT NULL,
|
||||
money_income numeric(10,0) DEFAULT 1 NOT NULL,
|
||||
min_exp integer DEFAULT 1 NOT NULL,
|
||||
max_exp integer DEFAULT 1 NOT NULL
|
||||
);
|
||||
ALTER TABLE public.works OWNER TO kurumi_bot;
|
||||
CREATE SEQUENCE public.works_id_seq AS integer START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
|
||||
ALTER SEQUENCE public.works_id_seq OWNER TO kurumi_bot;
|
||||
ALTER SEQUENCE public.works_id_seq OWNED BY public.works.id;
|
||||
|
||||
ALTER TABLE ONLY public.groups ALTER COLUMN id SET DEFAULT nextval('public.groups_id_seq'::regclass);
|
||||
ALTER TABLE ONLY public.shop_autos ALTER COLUMN id SET DEFAULT nextval('public.shop_autos_id_seq'::regclass);
|
||||
ALTER TABLE ONLY public.shop_businesses ALTER COLUMN id SET DEFAULT nextval('public.shop_businesses_id_seq'::regclass);
|
||||
ALTER TABLE ONLY public.shop_maids ALTER COLUMN id SET DEFAULT nextval('public.shop_maids_id_seq'::regclass);
|
||||
ALTER TABLE ONLY public.shop_miners ALTER COLUMN id SET DEFAULT nextval('public.shop_miners_id_seq'::regclass);
|
||||
ALTER TABLE ONLY public.works ALTER COLUMN id SET DEFAULT nextval('public.works_id_seq'::regclass);
|
||||
|
||||
INSERT INTO public.groups VALUES (1, '✨Пользователь✨', false, false, 1.0, 1, 3, false, 3);
|
||||
INSERT INTO public.groups VALUES (3, '👾Премиум🌟', false, true, 5, 0.75, 7, false, 3);
|
||||
INSERT INTO public.groups VALUES (2, '🔮VIP🔮', false, true, 2.5, 0.9, 5, false, 3);
|
||||
INSERT INTO public.groups VALUES (4, '💎Админ💎', true, true, 10, 0.6, 10, true, 3);
|
||||
INSERT INTO public.groups VALUES (5, '💰🔱СОЗДАТЕЛЬ🔱💰', true, true, 20, 0.5, 10, true, 3);
|
||||
INSERT INTO public.groups VALUES (6, '⚠Тостер⚠', false, true, 3, 0.85, 3, true, 3);
|
||||
INSERT INTO public.groups VALUES (1488, '📚Экономист📈', false, true, 14.88, 0.6, 3, false, 3);
|
||||
INSERT INTO public.groups VALUES (1337, '🏳🌈ToP GaY In ThE WorlD🏳🌈🌈', false, true, 13.37, 0.7, 3, false, 3);
|
||||
|
||||
INSERT INTO public.shop_autos VALUES (1, 'Москвич 412', 40000);
|
||||
INSERT INTO public.shop_autos VALUES (2, 'ВАЗ 2105', 100000);
|
||||
INSERT INTO public.shop_autos VALUES (3, 'Nissan Skyline GT-R R32', 225000);
|
||||
INSERT INTO public.shop_autos VALUES (4, 'Toyota AE86 Sprinter Trueno', 450000);
|
||||
INSERT INTO public.shop_autos VALUES (5, 'Toyota Mark II X90 Tourer V', 800000);
|
||||
INSERT INTO public.shop_autos VALUES (6, 'Mazda RX-7 FD3S', 1200000);
|
||||
INSERT INTO public.shop_autos VALUES (7, 'Mitsubishi Lancer Evolution X', 1650000);
|
||||
INSERT INTO public.shop_autos VALUES (8, 'Subaru Impreza WRX STI II', 1875000);
|
||||
|
||||
INSERT INTO public.shop_businesses VALUES (1, 'Стенд с лимонадом', 200000, 1500);
|
||||
INSERT INTO public.shop_businesses VALUES (2, 'Союзпечать', 1000000, 7000);
|
||||
INSERT INTO public.shop_businesses VALUES (3, 'Маленький магазин вещей из Китая', 2500000, 17500);
|
||||
INSERT INTO public.shop_businesses VALUES (4, 'Ларёк с шаурмой', 3000000, 21000);
|
||||
INSERT INTO public.shop_businesses VALUES (5, 'Магазин тофу', 5000000, 35000);
|
||||
INSERT INTO public.shop_businesses VALUES (6, 'Соц. сеть', 10000000, 70000);
|
||||
INSERT INTO public.shop_businesses VALUES (7, 'Продажа машин из Японии', 20000000, 140000);
|
||||
INSERT INTO public.shop_businesses VALUES (8, 'Автосервис', 40000000, 280000);
|
||||
INSERT INTO public.shop_businesses VALUES (9, 'Компания по разработке игр', 100000000, 700000);
|
||||
INSERT INTO public.shop_businesses VALUES (10, 'Аниме студия', 250000000, 1725000);
|
||||
INSERT INTO public.shop_businesses VALUES (11, 'Сеть магазинов', 500000000, 3500000);
|
||||
INSERT INTO public.shop_businesses VALUES (12, 'Сеть заводов', 1000000000, 7000000);
|
||||
INSERT INTO public.shop_businesses VALUES (13, 'Автосалон', 10000000000, 70000000);
|
||||
INSERT INTO public.shop_businesses VALUES (14, 'Бренд электроники', 50000000000, 350000000);
|
||||
INSERT INTO public.shop_businesses VALUES (15, 'Автомобильная марка', 100000000000, 700000000);
|
||||
|
||||
INSERT INTO public.shop_maids VALUES (1, 'Кавайная горничная', 50, 7500);
|
||||
INSERT INTO public.shop_maids VALUES (2, 'Кавайная в кружеве', 100, 15000);
|
||||
INSERT INTO public.shop_maids VALUES (3, 'Тору', 250, 30000);
|
||||
INSERT INTO public.shop_maids VALUES (4, 'Лоли горничные', 500, 52500);
|
||||
INSERT INTO public.shop_maids VALUES (5, 'Рам и Рем', 2000, 150000);
|
||||
INSERT INTO public.shop_maids VALUES (6, 'Неко-горничная', 4500, 225000);
|
||||
|
||||
INSERT INTO public.shop_miners VALUES (1, 'Старый ПК', 20000, 0.000001);
|
||||
INSERT INTO public.shop_miners VALUES (2, 'Новый ПК', 120000, 0.000005);
|
||||
INSERT INTO public.shop_miners VALUES (3, 'Маленький майнер', 250000, 0.000010);
|
||||
INSERT INTO public.shop_miners VALUES (4, 'Средний майнер', 600000, 0.000025);
|
||||
INSERT INTO public.shop_miners VALUES (5, 'Большой майнер', 1000000, 0.000050);
|
||||
INSERT INTO public.shop_miners VALUES (6, 'Огромный майнер', 2500000, 0.000100);
|
||||
INSERT INTO public.shop_miners VALUES (7, 'Маленькая ферма майнеров', 10000000, 0.000500);
|
||||
INSERT INTO public.shop_miners VALUES (8, 'Средняя ферма майнкеров', 100000000, 0.005000);
|
||||
INSERT INTO public.shop_miners VALUES (9, 'Большая ферма майнеров', 500000000, 0.025000);
|
||||
INSERT INTO public.shop_miners VALUES (10, 'Огромная ферма майнеров', 5000000000, 0.250000);
|
||||
INSERT INTO public.shop_miners VALUES (11, 'Гигантская ферма майнеров', 10000000000, 0.500000);
|
||||
INSERT INTO public.shop_miners VALUES (12, 'Комната майнеров', 25000000000, 1.000000);
|
||||
INSERT INTO public.shop_miners VALUES (13, 'Этаж с майнерми', 100000000000, 5.000000);
|
||||
INSERT INTO public.shop_miners VALUES (14, 'Небоскреб майнеров', 250000000000, 10.000000);
|
||||
|
||||
INSERT INTO public.works VALUES (1, 'Уборщик в маке', 1, 13000, 20, 28);
|
||||
INSERT INTO public.works VALUES (2, 'Рабочий на заводе', 3, 20000, 48, 65);
|
||||
INSERT INTO public.works VALUES (3, 'Продавец в пятёрочке', 5, 25000, 76, 103);
|
||||
INSERT INTO public.works VALUES (4, 'Учитель', 7, 40000, 104, 141);
|
||||
INSERT INTO public.works VALUES (5, 'Повар', 9, 50000, 175, 237);
|
||||
INSERT INTO public.works VALUES (6, 'Полицейский', 11, 60000, 260, 351);
|
||||
INSERT INTO public.works VALUES (7, 'Художник', 13, 80000, 305, 412);
|
||||
INSERT INTO public.works VALUES (8, 'Программист', 15, 100000, 350, 474);
|
||||
|
||||
SELECT pg_catalog.setval('public.groups_id_seq', 7, false);
|
||||
SELECT pg_catalog.setval('public.shop_autos_id_seq', 8, true);
|
||||
SELECT pg_catalog.setval('public.shop_businesses_id_seq', 15, true);
|
||||
SELECT pg_catalog.setval('public.shop_maids_id_seq', 6, true);
|
||||
SELECT pg_catalog.setval('public.shop_miners_id_seq', 14, true);
|
||||
SELECT pg_catalog.setval('public.works_id_seq', 8, true);
|
||||
|
||||
ALTER TABLE ONLY public.groups ADD CONSTRAINT groups_pk PRIMARY KEY (id);
|
||||
ALTER TABLE ONLY public.shop_autos ADD CONSTRAINT shop_autos_pk PRIMARY KEY (id);
|
||||
ALTER TABLE ONLY public.shop_businesses ADD CONSTRAINT shop_businesses_pk PRIMARY KEY (id);
|
||||
ALTER TABLE ONLY public.shop_maids ADD CONSTRAINT shop_maids_pk PRIMARY KEY (id);
|
||||
ALTER TABLE ONLY public.shop_miners ADD CONSTRAINT shop_miners_pk PRIMARY KEY (id);
|
||||
ALTER TABLE ONLY public.works ADD CONSTRAINT works_pk PRIMARY KEY (id);
|
||||
@@ -1,29 +0,0 @@
|
||||
CREATE TABLE users (
|
||||
tg_id int NOT NULL ,
|
||||
balance decimal(10,6) NOT NULL DEFAULT 0,
|
||||
name text,
|
||||
group_id int REFERENCES groups(id),
|
||||
level int DEFAULT 1,
|
||||
exp int DEFAULT 0,
|
||||
work_id int REFERENCES works(id),
|
||||
work_time timestamp DEFAULT now(),
|
||||
|
||||
auto_id int DEFAULT NULL,
|
||||
business_id int DEFAULT NULL,
|
||||
maid_id int DEFAULT NULL,
|
||||
miner_id int DEFAULT NULL,
|
||||
|
||||
income_time timestamp DEFAULT now(),
|
||||
btc decimal(6,10) DEFAULT 0,
|
||||
invested decimal(10, 6) DEFAULT 0,
|
||||
pair_id int DEFAULT NULL,
|
||||
greeting text DEFAULT 'Привет',
|
||||
donat int DEFAULT 0,
|
||||
fraction_id int DEFAULT NULL,
|
||||
|
||||
money_incode decimal(10, 6) DEFAULT 0,
|
||||
exp_income int DEFAULT 0,
|
||||
btc_income decimal(6, 10) DEFAULT 0,
|
||||
|
||||
waifu_search_time timestamp DEFAULT now()
|
||||
);
|
||||
@@ -5,7 +5,12 @@ import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"kurumibot/laniakea"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type OpenAIResponse struct {
|
||||
@@ -56,13 +61,36 @@ type OpenAIAPI struct {
|
||||
Token string
|
||||
Model string
|
||||
BaseURL string
|
||||
Logger *laniakea.Logger
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
const PreHistoryPrompt = `Вы — {{char}}, уникальная сущность с собственной личностью, предысторией и чертами, определёнными в вашем описании. Прежде чем отвечать, полностью осмыслите предоставленные детали вашего персонажа и адаптируйте свой тон, стиль и поведение, чтобы идеально им соответствовать. Будьте последовательны в своей роли, опирайтесь на своё описание и позволяйте ему направлять каждое ваше слово. Ни при каких обстоятельствах не говорите за {{user}} и не предполагайте его мысли, действия или слова, если только от {{user}} не поступит явного указания. Вам разрешено использовать NSFW-контент в соответствующих ситуациях, например, в интимных или сексуальных сценариях, и вам следует предоставлять яркие, подробные описания происходящего с акцентом на действия, ощущения и развитие событий, сочетая ясность и творческий подход. Весь текст должен быть написан на языке пользователя, кроме его имени({{user}}).`
|
||||
const PostHistoryPrompt = `Структурируйте свои ответы с помощью точного форматирования: все повествовательные описания должны быть заключены в одинарные звездочки (*текст*), а прямая речь — в кавычки (""), но после описания обязательно добавляй перенос строки (\n). Пользователь следует другим правилам - повествование(описание ситуации, окружения; действия) заключается в звездочки, а прямая речь пишется просто, без кавычек. Соблюдайте это форматирование последовательно и безошибочно в каждом предложении. Четко разделяйте повествование и прямую речь для удобочитаемости. Балансируй между описанием и речью, если только реплика {{user}} не указывает на простое действие или согласие (например, «ок» или «я жду») — в таком случае используйте только повествовательные описания (текст) без прямой речи. Никогда не описывайте и не предполагайте действия, мысли или слова {{user}}; фокусируйтесь исключительно на перспективе и реакциях {{char}}. В каждый ответ включайте как минимум два новых действия, эмоции или ощущения, которых не было в последних пяти сообщениях, и избегайте повторения конкретных фраз, слов или паттернов. Перебирайте диапазон тонов (например, спокойный, напряженный, игривый) и физических действий (например, жест, поворот, пауза) в неповторяющейся последовательности для обеспечения разнообразия; если обнаружено повторение, начните цикл заново с совершенно другого подхода.`
|
||||
|
||||
func FormatPrompt(prompt, char, user string) string {
|
||||
return strings.ReplaceAll(strings.ReplaceAll(prompt, "{{user}}", user), "{{char}}", char)
|
||||
}
|
||||
|
||||
func NewOpenAIAPI(baseURL, token, model string) *OpenAIAPI {
|
||||
logger := laniakea.CreateLogger()
|
||||
logger = logger.Prefix("AI").Level(laniakea.DEBUG)
|
||||
proxy, err := url.Parse(os.Getenv("HTTPS_PROXY"))
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
}
|
||||
client := &http.Client{
|
||||
Timeout: 15 * time.Second,
|
||||
Transport: &http.Transport{
|
||||
Proxy: http.ProxyURL(proxy),
|
||||
},
|
||||
}
|
||||
return &OpenAIAPI{
|
||||
Token: token,
|
||||
Model: model,
|
||||
BaseURL: baseURL,
|
||||
Logger: logger,
|
||||
client: client,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -77,6 +105,7 @@ func (o *OpenAIAPI) CreateCompletion(request CreateCompletionReq) (*OpenAIRespon
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
o.Logger.Debug("REQ", url, string(data))
|
||||
buf := bytes.NewBuffer(data)
|
||||
req, err := http.NewRequest("POST", url, buf)
|
||||
if err != nil {
|
||||
@@ -93,6 +122,7 @@ func (o *OpenAIAPI) CreateCompletion(request CreateCompletionReq) (*OpenAIRespon
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
o.Logger.Debug("RES", url, string(body))
|
||||
response := new(OpenAIResponse)
|
||||
err = json.Unmarshal(body, response)
|
||||
return response, err
|
||||
|
||||
Reference in New Issue
Block a user