diff --git a/database/psql/rp.go b/database/psql/rp.go index c034b58..dd80fce 100644 --- a/database/psql/rp.go +++ b/database/psql/rp.go @@ -21,6 +21,7 @@ type RPScenarios struct { } type RPUser struct { UserID int64 `db:"user_id"` + UserPrompt string `db:"user_prompt"` SelectedPreset string `db:"selected_preset"` UsedTokens int64 `db:"used_tokens"` } diff --git a/plugins/rp.go b/plugins/rp.go index 8146edd..0f9e962 100644 --- a/plugins/rp.go +++ b/plugins/rp.go @@ -135,12 +135,17 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) { return } + userPrompt := "" + if rpUser.UserPrompt != "" { + userPrompt = fmt.Sprintf("Вот описание моего персонажа %s ", rpUser.UserPrompt) + } beforeHistory := ai.Message{ Role: "system", Content: fmt.Sprintf( - "%s %s %s", + "%s %s %s %s", ai.FormatPrompt(preset.PreHistory, waifu.Name, ctx.Msg.From.FirstName), fmt.Sprintf("Вот краткое описание твоего персонажа: %s", waifu.RpPrompt), + userPrompt, red.RPGetChatPrompt(db, ctx.FromID, waifuId), ), }