This commit is contained in:
2026-01-19 22:14:23 +03:00
parent d0c62f4ec5
commit fec89b5305
2 changed files with 7 additions and 1 deletions

View File

@@ -21,6 +21,7 @@ type RPScenarios struct {
} }
type RPUser struct { type RPUser struct {
UserID int64 `db:"user_id"` UserID int64 `db:"user_id"`
UserPrompt string `db:"user_prompt"`
SelectedPreset string `db:"selected_preset"` SelectedPreset string `db:"selected_preset"`
UsedTokens int64 `db:"used_tokens"` UsedTokens int64 `db:"used_tokens"`
} }

View File

@@ -135,12 +135,17 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return return
} }
userPrompt := ""
if rpUser.UserPrompt != "" {
userPrompt = fmt.Sprintf("Вот описание моего персонажа %s ", rpUser.UserPrompt)
}
beforeHistory := ai.Message{ beforeHistory := ai.Message{
Role: "system", Role: "system",
Content: fmt.Sprintf( Content: fmt.Sprintf(
"%s %s %s", "%s %s %s %s",
ai.FormatPrompt(preset.PreHistory, waifu.Name, ctx.Msg.From.FirstName), ai.FormatPrompt(preset.PreHistory, waifu.Name, ctx.Msg.From.FirstName),
fmt.Sprintf("Вот краткое описание твоего персонажа: %s", waifu.RpPrompt), fmt.Sprintf("Вот краткое описание твоего персонажа: %s", waifu.RpPrompt),
userPrompt,
red.RPGetChatPrompt(db, ctx.FromID, waifuId), red.RPGetChatPrompt(db, ctx.FromID, waifuId),
), ),
} }