66 lines
4.1 KiB
Go
66 lines
4.1 KiB
Go
package app
|
||
|
||
import (
|
||
"git.echol.cn/loser/st/server/global"
|
||
"github.com/lib/pq"
|
||
"gorm.io/datatypes"
|
||
)
|
||
|
||
// AIWorldInfo 世界书（World Info）表
|
||
type AIWorldInfo struct {
|
||
global.GVA_MODEL
|
||
UserID uint `json:"userId" gorm:"not null;index;comment:所属用户ID"`
|
||
User *AppUser `json:"user" gorm:"foreignKey:UserID"`
|
||
BookName string `json:"bookName" gorm:"type:varchar(500);not null;index;comment:世界书名称"`
|
||
IsGlobal bool `json:"isGlobal" gorm:"default:false;index;comment:是否全局世界书"`
|
||
Entries datatypes.JSON `json:"entries" gorm:"type:jsonb;not null;comment:世界书条目列表"`
|
||
LinkedChars pq.StringArray `json:"linkedChars" gorm:"type:text[];comment:关联的角色ID列表"`
|
||
}
|
||
|
||
func (AIWorldInfo) TableName() string {
|
||
return "ai_world_info"
|
||
}
|
||
|
||
// AIWorldInfoEntry 世界书条目（存储在 AIWorldInfo.Entries 中）
|
||
type AIWorldInfoEntry struct {
|
||
UID string `json:"uid"` // 条目唯一ID
|
||
Keys []string `json:"keys"` // 主要关键词
|
||
SecondaryKeys []string `json:"secondary_keys"` // 次要关键词（可选匹配）
|
||
Content string `json:"content"` // 条目内容
|
||
Comment string `json:"comment"` // 备注
|
||
Enabled bool `json:"enabled"` // 是否启用
|
||
Constant bool `json:"constant"` // 永远激活
|
||
Selective bool `json:"selective"` // 选择性激活（需要主键+次键都匹配）
|
||
Order int `json:"order"` // 插入顺序（越小越靠前）
|
||
Position string `json:"position"` // 插入位置：before_char, after_char
|
||
Depth int `json:"depth"` // 扫描深度（从最近消息往前扫描几条）
|
||
Probability int `json:"probability"` // 激活概率（0-100）
|
||
UseProbability bool `json:"use_probability"` // 是否使用概率
|
||
Group string `json:"group"` // 分组（同组只激活一个）
|
||
GroupOverride bool `json:"group_override"` // 分组覆盖
|
||
GroupWeight int `json:"group_weight"` // 分组权重
|
||
PreventRecursion bool `json:"prevent_recursion"` // 防止递归激活
|
||
DelayUntilRecursion bool `json:"delay_until_recursion"` // 延迟到递归时激活
|
||
ScanDepth *int `json:"scan_depth"` // 扫描深度（nil=使用全局设置）
|
||
CaseSensitive *bool `json:"case_sensitive"` // 大小写敏感（nil=使用全局设置）
|
||
MatchWholeWords *bool `json:"match_whole_words"` // 匹配整个单词
|
||
UseRegex *bool `json:"use_regex"` // 使用正则表达式
|
||
Automation string `json:"automation_id"` // 自动化ID
|
||
Role string `json:"role"` // 角色（system/user/assistant）
|
||
VectorizedContent string `json:"vectorized"` // 向量化的内容ID
|
||
Extensions map[string]interface{} `json:"extensions"` // 扩展数据
|
||
}
|
||
|
||
// AICharacterWorldInfo 角色关联的世界书（中间表）
|
||
type AICharacterWorldInfo struct {
|
||
global.GVA_MODEL
|
||
CharacterID uint `json:"characterId" gorm:"not null;index:idx_char_world,unique;comment:角色ID"`
|
||
Character *AICharacter `json:"character" gorm:"foreignKey:CharacterID"`
|
||
WorldInfoID uint `json:"worldInfoId" gorm:"not null;index:idx_char_world,unique;comment:世界书ID"`
|
||
WorldInfo *AIWorldInfo `json:"worldInfo" gorm:"foreignKey:WorldInfoID"`
|
||
}
|
||
|
||
func (AICharacterWorldInfo) TableName() string {
|
||
return "ai_character_world_info"
|
||
}
|