knowledge

type AuditRecord struct {
	Timestamp time.Time `json:"timestamp"`
	Action    string    `json:"action"`
	UserID    string    `json:"user_id"`
	ItemID    string    `json:"item_id"`
	Details   string    `json:"details"`
}

type AuditStrategy interface {
	Record(action, userID, itemID, details string)
}

type ContextStrategy string

type HybridSearchWeight struct {
	TextWeight   float64 `json:"text_weight"`   // 文本搜索权重
	VectorWeight float64 `json:"vector_weight"` // 向量搜索权重
}

type KnowledgeItem struct {
	// 基本信息
	ID          string        `json:"id"`
	Type        KnowledgeType `json:"type"`
	Title       string        `json:"title"`
	Description string        `json:"description"`
	Content     string        `json:"content"`

	// 分类和标签
	Category string   `json:"category"`
	Tags     []string `json:"tags"`

	// 元数据
	Metadata map[string]any `json:"metadata"`

	// 时效性
	CreatedAt time.Time  `json:"created_at"`
	UpdatedAt time.Time  `json:"updated_at"`
	ExpiresAt *time.Time `json:"expires_at,omitempty"`

	// 置信度和质量
	Confidence float64 `json:"confidence"` // 0-1，知识可信度
	Quality    float64 `json:"quality"`    // 0-1，知识质量评分

	// 来源信息
	Source  string `json:"source"`  // 知识来源
	Author  string `json:"author"`  // 作者
	Version string `json:"version"` // 版本

	// 关系信息
	Relations []KnowledgeRelation `json:"relations"` // 与其他知识的关系

	// 索引信息
	Embedding []float32 `json:"embedding,omitempty"` // 向量表示
	Namespace string    `json:"namespace"`           // 命名空间
}

type KnowledgeRelation struct {
	Type     RelationType `json:"type"`
	TargetID string       `json:"target_id"`
	Weight   float64      `json:"weight"` // 关系强度
	Label    string       `json:"label"`  // 关系标签
}

type KnowledgeStats struct {
	TotalItems      int64            `json:"total_items"`
	ItemsByType     map[string]int64 `json:"items_by_type"`
	ItemsByCategory map[string]int64 `json:"items_by_category"`
	AverageQuality  float64          `json:"average_quality"`
	LastUpdated     time.Time        `json:"last_updated"`
}

type KnowledgeType string

type Manager interface {
	// 基础操作
	Add(ctx context.Context, item *KnowledgeItem) error
	Get(ctx context.Context, id string) (*KnowledgeItem, error)
	Update(ctx context.Context, item *KnowledgeItem) error
	Delete(ctx context.Context, id string) error

	// 搜索操作
	Search(ctx context.Context, query *SearchQuery) ([]*SearchResult, error)
	SearchSimilar(ctx context.Context, id string, maxResults int) ([]*SearchResult, error)

	// 关系操作
	AddRelation(ctx context.Context, fromID, toID string, relationType RelationType, weight float64, label string) error
	RemoveRelation(ctx context.Context, fromID, toID string, relationType RelationType) error
	GetRelated(ctx context.Context, id string, relationTypes []RelationType, maxDepth int) ([]*KnowledgeItem, error)

	// 批量操作
	BulkAdd(ctx context.Context, items []*KnowledgeItem) error
	BulkSearch(ctx context.Context, queries []*SearchQuery) ([][]*SearchResult, error)

	// 知识推理
	Reason(ctx context.Context, query string, maxSteps int) ([]*KnowledgeItem, []string, error)

	// 维护操作
	Validate(ctx context.Context, id string) (bool, []string, error)
	Refresh(ctx context.Context, id string) error
	Compress(ctx context.Context, namespace string) error

	// 统计信息
	GetStats(ctx context.Context, namespace string) (*KnowledgeStats, error)

	// 生命周期
	Start(ctx context.Context) error
	Stop(ctx context.Context) error
}

type ManagerConfig struct {
	// 存储配置
	MemoryManager *memory.Manager    `json:"-"` // 长期记忆管理器
	VectorStore   vector.VectorStore `json:"-"` // 向量存储
	Embedder      vector.Embedder    `json:"-"` // 向量嵌入器

	// 基础配置
	Namespace  string `json:"namespace"`   // 默认命名空间
	MaxResults int    `json:"max_results"` // 默认最大结果数
	AutoEmbed  bool   `json:"auto_embed"`  // 自动生成向量

	// 质量控制
	MinConfidence float64 `json:"min_confidence"` // 最小置信度阈值
	MinQuality    float64 `json:"min_quality"`    // 最小质量阈值

	// 缓存配置
	CacheEnabled bool          `json:"cache_enabled"`
	CacheTTL     time.Duration `json:"cache_ttl"`

	// 安全配置
	EnablePII   bool `json:"enable_pii"`   // 启用PII检测
	EnableAudit bool `json:"enable_audit"` // 启用审计日志

	// 轻量核心管线
	UseCorePipeline bool `json:"use_core_pipeline"` // 启用轻量 ingest/search 管线

	// 可选策略注入
	PIIStrategy   PIIStrategy   `json:"-"`
	AuditStrategy AuditStrategy `json:"-"`
}

type PIIStrategy interface {
	Sanitize(item *KnowledgeItem) *KnowledgeItem
}

type QualityFilter struct {
	MinConfidence float64 `json:"min_confidence"`
	MinQuality    float64 `json:"min_quality"`
}

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

type RAGConfig struct {
	// 检索配置
	MaxRetrievalResults int     `json:"max_retrieval_results"`
	Reranking           bool    `json:"reranking"`
	DiversityThreshold  float64 `json:"diversity_threshold"`
	RelevanceThreshold  float64 `json:"relevance_threshold"`

	// 混合搜索权重
	TextSearchWeight   float64 `json:"text_search_weight"`
	VectorSearchWeight float64 `json:"vector_search_weight"`
	GraphSearchWeight  float64 `json:"graph_search_weight"`

	// 重排序配置
	RerankerType       RerankerType `json:"reranker_type"`
	CrossEncoderWeight float64      `json:"cross_encoder_weight"`
	BM25Weight         float64      `json:"bm25_weight"`
	EmbeddingWeight    float64      `json:"embedding_weight"`

	// 上下文配置
	MaxContextLength   int             `json:"max_context_length"`
	ContextCompression bool            `json:"context_compression"`
	ContextStrategy    ContextStrategy `json:"context_strategy"`

	// 高级检索
	EnableMultiHop       bool `json:"enable_multi_hop"`
	MaxHops              int  `json:"max_hops"`
	EnableTemporalFilter bool `json:"enable_temporal_filter"`
	EnableQualityFilter  bool `json:"enable_quality_filter"`
}

type RAGOption func(*RAGOptions)

type RAGOptions struct {
	MaxResults     int            `json:"max_results"`
	KnowledgeType  KnowledgeType  `json:"knowledge_type"`
	Category       string         `json:"category"`
	Tags           []string       `json:"tags"`
	Namespace      string         `json:"namespace"`
	Metadata       map[string]any `json:"metadata"`
	TimeRange      *TimeRange     `json:"time_range"`
	QualityFilter  *QualityFilter `json:"quality_filter"`
	DiversityBoost bool           `json:"diversity_boost"`
}

type RAGResult struct {
	Query          string           `json:"query"`
	Results        []*RetrievalItem `json:"results"`
	EnhancedQuery  string           `json:"enhanced_query"`
	Context        string           `json:"context"`
	ContextItems   []*KnowledgeItem `json:"context_items"`
	Confidence     float64          `json:"confidence"`
	RetrievalTime  time.Duration    `json:"retrieval_time"`
	ProcessingTime time.Duration    `json:"processing_time"`
	TotalTime      time.Duration    `json:"total_time"`
}

type RelationType string

type RerankerType string

type RetrievalItem struct {
	Item        *KnowledgeItem `json:"item"`
	Score       float64        `json:"score"`
	Source      string         `json:"source"`      // 检索来源：text, vector, graph
	Explanation string         `json:"explanation"` // 检索解释
	Rank        int            `json:"rank"`        // 排名
}

type SearchQuery struct {
	// 基础查询
	Query    string        `json:"query"`    // 文本查询
	Vector   []float32     `json:"vector"`   // 向量查询
	Type     KnowledgeType `json:"type"`     // 知识类型过滤
	Category string        `json:"category"` // 分类过滤
	Tags     []string      `json:"tags"`     // 标签过滤

	// 高级查询
	Relations []RelationType `json:"relations"` // 关系过滤
	Sources   []string       `json:"sources"`   // 来源过滤
	Authors   []string       `json:"authors"`   // 作者过滤

	// 时效性查询
	After      *time.Time `json:"after,omitempty"`  // 创建时间之后
	Before     *time.Time `json:"before,omitempty"` // 创建时间之前
	NotExpired bool       `json:"not_expired"`      // 只查询未过期的

	// 质量过滤
	MinConfidence float64 `json:"min_confidence"` // 最小置信度
	MinQuality    float64 `json:"min_quality"`    // 最小质量

	// 检索配置
	MaxResults int            `json:"max_results"` // 最大结果数
	Namespace  string         `json:"namespace"`   // 命名空间
	Filters    map[string]any `json:"filters"`     // 自定义过滤

	// 搜索策略
	Strategy     SearchStrategy     `json:"strategy"`      // 搜索策略
	HybridWeight HybridSearchWeight `json:"hybrid_weight"` // 混合搜索权重
}

type SearchResult struct {
	Item        KnowledgeItem `json:"item"`
	Score       float64       `json:"score"`       // 相关性评分
	Explanation string        `json:"explanation"` // 结果解释
	Path        []string      `json:"path"`        // 知识路径（图搜索时）
}

type SearchStrategy string

type TimeRange struct {
	Start time.Time `json:"start"`
	End   time.Time `json:"end"`
}