gorm

type CombinedWeights struct {
	VectorWeight  float64 // 向量相似度权重
	KeywordWeight float64 // 关键词匹配权重
	TimeWeight    float64 // 时间衰减权重
}

type Config struct {
	DBPath              string
	MaxOpenConns        int
	MaxIdleConns        int
	ConnMaxLifetime     time.Duration
	LogLevel            logger.LogLevel
	DisableVectorSearch bool
	EmbeddingDimension  int
	RecallMode          RecallMode
	CombinedWeights     *CombinedWeights
	ChunkSize           int
	ChunkOverlap        int
	IndexCachePath      string

	// HNSW 参数
	HNSW_M        int     // 每节点最大双向链接数，默认 16
	HNSW_Ml       float64 // 层级生成因子，默认 0.25
	HNSW_EfSearch int     // 搜索宽度（越大越精确越慢），默认 200

	// 召回参数
	DefaultTopK           int     // 默认召回数量，默认 3
	CombinedCandidateMult int     // 组合召回候选倍数，默认 10
	TimeDecayHalfLifeMs   float64 // 时间衰减半衰期（毫秒），默认 7 天
	IndexRebuildBatchSize int     // 索引重建批次大小，默认 500
	HybridTimeWeight      float64 // 混合召回时间权重，默认 0.3
	HybridKeywordWeight   float64 // 混合召回关键词权重，默认 0.7
}

type DocumentModel struct {
	ID         string `gorm:"primaryKey;size:128"`
	Collection string `gorm:"index:idx_collection;size:128;not null"`
	Content    string `gorm:"type:text;not null"`
	Embedding  string `gorm:"type:text"`
	MetaData   string `gorm:"type:text"` // JSON 序列化
	Timestamp  int64  `gorm:"not null"`
	CreatedAt  time.Time
}

type EmbeddingChunk struct {
	ID         string `gorm:"primaryKey;size:64"`
	MessageID  string `gorm:"index;size:64;not null"`  // 关联 MessageModel.ID
	SessionID  string `gorm:"index;size:128;not null"` // 冗余字段，加速查询
	ChunkIndex int    `gorm:"not null"`                // 块序号
	Content    string `gorm:"type:text;not null"`      // 块文本
	Embedding  string `gorm:"type:text"`               // 嵌入向量 JSON
}

type EmbeddingFunc func(ctx context.Context, text string) ([]float32, error)

type GORMDocStore struct {
	// contains filtered or unexported fields
}

type GORMStore struct {
	// contains filtered or unexported fields
}

type HNSWRetriever struct {
	// contains filtered or unexported fields
}

type MessageModel struct {
	ID               string `gorm:"primaryKey;size:64"`
	SessionID        string `gorm:"index:idx_session_time,priority:1;size:128"`
	Role             string `gorm:"size:32;not null"`
	Content          string `gorm:"type:text;not null"`
	ReasoningContent string `gorm:"type:text"`
	Embedding        string `gorm:"type:text"` // JSON 序列化的向量（空字符串表示无嵌入）
	Timestamp        int64  `gorm:"index:idx_session_time,priority:2;not null"`
	Metadata         string `gorm:"type:text"` // JSON 序列化的元数据
	CreatedAt        time.Time
}

type RecallMode int