package model

// Embedding 向量嵌入模型，用于知识库检索
type Embedding struct {
	ID          string `json:"id"`           // 向量唯一标识
	ContentHash string `json:"content_hash"` // 内容哈希（去重）
	Embedding   string `json:"embedding"`    // 向量嵌入（JSON数组）
	SourceType  string `json:"source_type"`  // 来源类型（document/resume/text）
	SourceID    string `json:"source_id"`    // 来源ID
	SourceName  string `json:"source_name"`  // 来源名称
	UserID      string `json:"user_id"`      // 关联的用户ID
	CreatedAt   string `json:"created_at"`   // 创建时间
}

// ChunkWithScore 带相似度评分的文档片段，用于向量检索排序
type ChunkWithScore struct {
	Content    string  `json:"content"`     // 片段内容
	Score      float64 `json:"score"`       // 相似度评分
	Metadata   string  `json:"metadata"`    // 元数据（含向量JSON）
	SourceID   string  `json:"source_id"`   // 来源ID
	SourceName string  `json:"source_name"` // 来源名称
	SourceType string  `json:"source_type"` // 来源类型
}

// KnowledgeBaseStats 知识库统计信息
type KnowledgeBaseStats struct {
	TotalDocuments  int `json:"total_documents"`  // 文档总数
	TotalChunks     int `json:"total_chunks"`     // 切片总数
	TotalEmbeddings int `json:"total_embeddings"` // 向量总数
}