knowledgebase

type Chunk struct {
	ID        int64  `gorm:"primaryKey;autoIncrement"`
	Source    string `gorm:"index"` // 源文件相对路径，如 06-plugins/COMMAND_SYSTEM.md
	Heading   string // 所属 Markdown 标题（检索展示用）
	Content   string `gorm:"type:text"`
	Hash      string `gorm:"index"` // content sha256（增量更新检测）
	Vector    []byte // float32 LE 序列；空表示未嵌入
	Dim       int    // 向量维度
	Model     string // 生成向量的模型名
	UpdatedAt int64
}

type Config struct {
	// Enabled 是否启用知识库检索（默认 false）。
	Enabled bool `yaml:"enabled"`
	// Markdown 平台支持 MD 时用 Markdown 回复（/kb status、/kb rebuild 等），
	// 否则回退纯文本（默认 true，与 ai 插件一致）。
	Markdown bool `yaml:"markdown"`
	// SourceDir 数据源根目录（相对工作目录，默认 docs）。
	SourceDir string `yaml:"source_dir"`
	// ExcludeDirs 扫描时排除的子目录（默认 ["notes"]：设计笔记多为历史/计划内容）。
	ExcludeDirs []string `yaml:"exclude_dirs"`
	// DBPath 索引持久化路径（默认 data/db/knowledgebase.db）。
	DBPath string `yaml:"db_path"`
	// EmbeddingBaseURL OpenAI 兼容 /embeddings 端点根地址（如 http://host:8080/v1）。
	// 为空时禁用语义检索，仅使用关键词兜底。
	EmbeddingBaseURL string `yaml:"embedding_base_url"`
	// EmbeddingAPIKey 嵌入服务 API Key（自托管服务通常为空）。
	EmbeddingAPIKey string `yaml:"embedding_api_key"`
	// EmbeddingModel 嵌入模型名。
	EmbeddingModel string `yaml:"embedding_model"`
	// ChunkSize 分块目标长度（rune，默认 600）。
	ChunkSize int `yaml:"chunk_size"`
	// ChunkOverlap 相邻分块重叠长度（rune，默认 100）。
	ChunkOverlap int `yaml:"chunk_overlap"`
	// MaxResults kb_search 单次返回条数上限（默认 5）。
	MaxResults int `yaml:"max_results"`
	// ToolDescription kb_search 工具的自定义描述（告诉 LLM 知识库包含什么内容）。
	// 为空时使用通用描述（自动带上数据源目录）。
	ToolDescription string `yaml:"tool_description"`
}

type Hit struct {
	Source  string
	Heading string
	Content string
	Score   float32
}

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

type SourceFile struct {
	Source     string `gorm:"primaryKey"`
	Hash       string // 文件内容 sha256
	ChunkCount int
	UpdatedAt  int64
}

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