rag

type ChainVisualization struct {
	Nodes []VisualizationNode `json:"nodes"`
	Edges []VisualizationEdge `json:"edges"`
}

type Chunk struct {
	Content    string         `json:"content"`
	StartPos   int            `json:"start_pos"`
	EndPos     int            `json:"end_pos"`
	Metadata   map[string]any `json:"metadata"`
	TokenCount int            `json:"token_count"`
}

type ChunkingConfig struct {
	Strategy     ChunkingStrategy `json:"strategy"`       // 分块策略
	ChunkSize    int              `json:"chunk_size"`     // 块大小（tokens）
	ChunkOverlap int              `json:"chunk_overlap"`  // 重叠大小（tokens）
	MinChunkSize int              `json:"min_chunk_size"` // 最小块大小

	// 语义分块参数
	SimilarityThreshold float64 `json:"similarity_threshold"` // 语义相似度阈值

	// 文档感知参数
	PreserveTables     bool `json:"preserve_tables"`      // 保留表格
	PreserveCodeBlocks bool `json:"preserve_code_blocks"` // 保留代码块
	PreserveHeaders    bool `json:"preserve_headers"`     // 保留标题
}

type ChunkingStrategy string

type Clearable interface {
	ClearAll(ctx context.Context) error
}

type ContextProvider interface {
	// GenerateContext 为 chunk 生成上下文
	GenerateContext(ctx context.Context, doc Document, chunk string) (string, error)
}

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

type ContextualRetrievalConfig struct {
	// 上下文生成
	UseContextPrefix bool   `json:"use_context_prefix"`
	ContextTemplate  string `json:"context_template"`
	MaxContextLength int    `json:"max_context_length"`

	// 检索增强
	UseReranking  bool    `json:"use_reranking"`
	ContextWeight float64 `json:"context_weight"`

	// 缓存
	CacheContexts bool          `json:"cache_contexts"`
	CacheTTL      time.Duration `json:"cache_ttl"` // 缓存过期时间，默认 1h

	// 分块配置（新增）
	ChunkSize     int  `json:"chunk_size"`      // 分块大小，默认 500
	ChunkOverlap  int  `json:"chunk_overlap"`   // 重叠大小，默认 50
	ChunkByTokens bool `json:"chunk_by_tokens"` // 按 token 还是字符分块

	// BM25 参数（新增）
	BM25K1 float64 `json:"bm25_k1"` // BM25 k1 参数，默认 1.2
	BM25B  float64 `json:"bm25_b"`  // BM25 b 参数，默认 0.75
}

type CrossEncoderConfig struct {
	ModelName      string  `json:"model_name"`      // 模型名称
	MaxLength      int     `json:"max_length"`      // 最大输入长度
	BatchSize      int     `json:"batch_size"`      // 批处理大小
	ScoreWeight    float64 `json:"score_weight"`    // 重排序分数权重
	OriginalWeight float64 `json:"original_weight"` // 原始分数权重
}

type CrossEncoderProvider interface {
	// Score 计算查询-文档对的相关性分数
	Score(ctx context.Context, pairs []QueryDocPair) ([]float64, error)
}

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

type DedupStats struct {
	TotalRetrieved    int `json:"total_retrieved"`     // 原始检索结果数
	DedupByID         int `json:"dedup_by_id"`         // 按 ID 去重数量
	DedupBySimilarity int `json:"dedup_by_similarity"` // 按内容相似度去重数量
	FinalCount        int `json:"final_count"`         // 去重后最终数量
}

type Document struct {
	ID        string         `json:"id"`
	Content   string         `json:"content"`
	Metadata  map[string]any `json:"metadata,omitempty"`
	Embedding []float64      `json:"embedding,omitempty"`
}

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

type DocumentLister interface {
	ListDocumentIDs(ctx context.Context, limit int, offset int) ([]string, error)
}

type Edge struct {
	ID         string         `json:"id"`
	Source     string         `json:"source"`
	Target     string         `json:"target"`
	Type       string         `json:"type"`
	Properties map[string]any `json:"properties,omitempty"`
	Weight     float64        `json:"weight"`
}

type EmbeddingProvider interface {
	EmbedQuery(ctx context.Context, query string) ([]float64, error)
	EmbedDocuments(ctx context.Context, documents []string) ([][]float64, error)
	Name() string
}

type EmbeddingProviderType string

type EnhancedRetriever struct {
	*HybridRetriever
	// contains filtered or unexported fields
}

type EnhancedRetrieverConfig struct {
	HybridConfig      HybridRetrievalConfig
	EmbeddingProvider EmbeddingProvider
	RerankProvider    RerankProvider
}

type EnhancedTokenizer struct{}

type Entity struct {
	ID   string `json:"id"`
	Name string `json:"name"`
	Type string `json:"type"`
}

type ExpansionResult struct {
	Original   string      `json:"original"`
	Expansions []string    `json:"expansions"`
	Keywords   []string    `json:"keywords"`
	Intent     QueryIntent `json:"intent"`
}

type GraphDocument struct {
	ID       string         `json:"id"`
	Title    string         `json:"title"`
	Content  string         `json:"content"`
	Metadata map[string]any `json:"metadata,omitempty"`
	Entities []Entity       `json:"entities,omitempty"`
}

type GraphEmbedder interface {
	Embed(ctx context.Context, text string) ([]float64, error)
}

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

type GraphRAGConfig struct {
	GraphWeight   float64 `json:"graph_weight"`    // Weight for graph results
	VectorWeight  float64 `json:"vector_weight"`   // Weight for vector results
	MaxGraphDepth int     `json:"max_graph_depth"` // Max traversal depth
	MaxResults    int     `json:"max_results"`
	MinScore      float64 `json:"min_score"`
}

type GraphRetrievalResult struct {
	ID           string         `json:"id"`
	Content      string         `json:"content"`
	Score        float64        `json:"score"`
	GraphScore   float64        `json:"graph_score"`
	VectorScore  float64        `json:"vector_score"`
	Source       string         `json:"source"` // "graph", "vector", "hybrid"
	Metadata     map[string]any `json:"metadata,omitempty"`
	RelatedNodes []*Node        `json:"related_nodes,omitempty"`
}

type HNSWConfig struct {
	M              int     `json:"m"`               // 每层最大连接数（12-48）
	EfConstruction int     `json:"ef_construction"` // 构建时搜索宽度（100-200）
	EfSearch       int     `json:"ef_search"`       // 搜索时宽度（50-200）
	MaxLevel       int     `json:"max_level"`       // 最大层数
	Ml             float64 `json:"ml"`              // 层数归一化因子
}

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

type HopType string

type HybridRetrievalConfig struct {
	// BM25 配置
	UseBM25    bool    `json:"use_bm25"`
	BM25Weight float64 `json:"bm25_weight"`
	BM25K1     float64 `json:"bm25_k1"` // BM25 参数 k1 (1.2-2.0)
	BM25B      float64 `json:"bm25_b"`  // BM25 参数 b (0.75)

	// 向量检索配置
	UseVector    bool    `json:"use_vector"`
	VectorWeight float64 `json:"vector_weight"`

	// Reranking 配置
	UseReranking bool `json:"use_reranking"`
	RerankTopK   int  `json:"rerank_top_k"`

	// 检索参数
	TopK     int     `json:"top_k"`
	MinScore float64 `json:"min_score"`
}

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

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

type IndexType string

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

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

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

type LLMRerankerConfig struct {
	MaxCandidates  int     `json:"max_candidates"`  // 最大候选数
	Temperature    float64 `json:"temperature"`     // 温度
	PromptTemplate string  `json:"prompt_template"` // 提示模板
}

type LLMRerankerProvider interface {
	// ScoreRelevance 评估相关性
	ScoreRelevance(ctx context.Context, query, document string) (float64, error)
}

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

type LowLevelSearchResult struct {
	ID       string         `json:"id"`
	Score    float64        `json:"score"`
	Metadata map[string]any `json:"metadata"`
}

type LowLevelVectorStore interface {
	// Store stores a vector with its ID and metadata.
	Store(ctx context.Context, id string, vector []float64, metadata map[string]any) error

	// Search finds the top-K most similar vectors, optionally filtered by metadata.
	Search(ctx context.Context, query []float64, topK int, filter map[string]any) ([]LowLevelSearchResult, error)

	// Delete removes a vector by ID.
	Delete(ctx context.Context, id string) error
}

type MilvusConfig struct {
	// 连接设置
	Host    string `json:"host"`
	Port    int    `json:"port"`
	BaseURL string `json:"base_url,omitempty"` // Override host:port if set

	// 认证
	Username string `json:"username,omitempty"`
	Password string `json:"password,omitempty"`
	Token    string `json:"token,omitempty"` // For Zilliz Cloud

	// 收藏设置
	Collection string `json:"collection"`
	Database   string `json:"database,omitempty"` // Default: "default"

	// Schema 设置
	VectorDimension int    `json:"vector_dimension,omitempty"` // Required for auto-create
	PrimaryField    string `json:"primary_field,omitempty"`    // Default: "id"
	VectorField     string `json:"vector_field,omitempty"`     // Default: "vector"
	ContentField    string `json:"content_field,omitempty"`    // Default: "content"
	MetadataField   string `json:"metadata_field,omitempty"`   // Default: "metadata"

	// 索引设置
	IndexType   MilvusIndexType  `json:"index_type,omitempty"`   // Default: IVF_FLAT
	MetricType  MilvusMetricType `json:"metric_type,omitempty"`  // Default: COSINE
	IndexParams map[string]any   `json:"index_params,omitempty"` // Index-specific params

	// 搜索设置
	SearchParams map[string]any `json:"search_params,omitempty"` // Search-specific params

	// 行为设置
	AutoCreateCollection bool          `json:"auto_create_collection,omitempty"`
	Timeout              time.Duration `json:"timeout,omitempty"`
	BatchSize            int           `json:"batch_size,omitempty"` // For batch operations

	// 一致性水平:强、会、会、会、终
	ConsistencyLevel string `json:"consistency_level,omitempty"`
}

type MilvusIndexType string

type MilvusMetricType string

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

type MultiHopConfig struct {
	// 跳跃限制
	MaxHops      int           `json:"max_hops"`      // Maximum number of hops (2-5)
	MinHops      int           `json:"min_hops"`      // Minimum hops before stopping
	HopTimeout   time.Duration `json:"hop_timeout"`   // Timeout per hop
	TotalTimeout time.Duration `json:"total_timeout"` // Total reasoning timeout

	// 检索设置
	ResultsPerHop     int     `json:"results_per_hop"`     // Documents per hop
	MinConfidence     float64 `json:"min_confidence"`      // Minimum confidence to continue
	ContextWindowSize int     `json:"context_window_size"` // Max context tokens

	// 原因设置
	EnableLLMReasoning    bool    `json:"enable_llm_reasoning"`    // Use LLM for reasoning
	EnableQueryRefinement bool    `json:"enable_query_refinement"` // Refine queries between hops
	EnableVerification    bool    `json:"enable_verification"`     // Verify answers
	ConfidenceThreshold   float64 `json:"confidence_threshold"`    // Stop if confidence exceeds

	// 复制
	DeduplicateResults  bool    `json:"deduplicate_results"`  // Remove duplicate documents
	SimilarityThreshold float64 `json:"similarity_threshold"` // Threshold for deduplication

	// 缓存
	EnableCache bool          `json:"enable_cache"`
	CacheTTL    time.Duration `json:"cache_ttl"`
}

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

type MultiStrategyDecision struct {
	Query      string               `json:"query"`
	Strategies []StrategyWithWeight `json:"strategies"`
	Reasoning  string               `json:"reasoning,omitempty"`
	Timestamp  time.Time            `json:"timestamp"`
}

type Node struct {
	ID         string         `json:"id"`
	Type       string         `json:"type"`
	Label      string         `json:"label"`
	Properties map[string]any `json:"properties,omitempty"`
	Embedding  []float64      `json:"embedding,omitempty"`
	CreatedAt  time.Time      `json:"created_at"`
}

type PineconeConfig struct {
	APIKey    string        `json:"api_key"`
	Index     string        `json:"index,omitempty"`    // Used to resolve BaseURL if BaseURL is empty
	BaseURL   string        `json:"base_url,omitempty"` // Data-plane base URL (preferred if known)
	Namespace string        `json:"namespace,omitempty"`
	Timeout   time.Duration `json:"timeout,omitempty"`

	ControllerBaseURL string `json:"controller_base_url,omitempty"` // Default: https://api.pinecone.io

	// 存储在元数据中的有效载荷字段 。
	MetadataContentField string `json:"metadata_content_field,omitempty"` // Default: "content"
}

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

type QdrantConfig struct {
	Host       string        `json:"host"`
	Port       int           `json:"port"`
	BaseURL    string        `json:"base_url,omitempty"`
	APIKey     string        `json:"api_key,omitempty"`
	Collection string        `json:"collection"`
	Timeout    time.Duration `json:"timeout,omitempty"`

	AutoCreateCollection bool   `json:"auto_create_collection,omitempty"`
	Distance             string `json:"distance,omitempty"`     // Cosine (default), Dot, Euclid
	VectorSize           int    `json:"vector_size,omitempty"`  // Optional override; defaults to len(embedding)
	Wait                 *bool  `json:"wait,omitempty"`         // Wait for operation completion (default true)
	PayloadContentField  string `json:"payload_content_field"`  // Payload key for document content (default "content")
	PayloadMetadataField string `json:"payload_metadata_field"` // Payload key for document metadata (default "metadata")
	PayloadIDField       string `json:"payload_id_field"`       // Payload key for original document ID (default "doc_id")
}

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

type QueryDocPair struct {
	Query    string
	Document string
}

type QueryFeatures struct {
	Intent      QueryIntent `json:"intent"`
	Complexity  string      `json:"complexity"` // "low", "medium", "high"
	Length      string      `json:"length"`     // "short", "medium", "long"
	HasEntities bool        `json:"has_entities"`
	HasKeywords bool        `json:"has_keywords"`
	IsQuestion  bool        `json:"is_question"`
	Keywords    []string    `json:"keywords"`
	Entities    []string    `json:"entities"`
	WordCount   int         `json:"word_count"`
}

type QueryIntent string

type QueryLLMProvider interface {
	// 完整生成给定快件的补全
	Complete(ctx context.Context, prompt string) (string, error)
}

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

type QueryRouterConfig struct {
	// 战略配置
	Strategies []StrategyConfig `json:"strategies"`

	// 默认策略
	DefaultStrategy RetrievalStrategy `json:"default_strategy"`

	// 运行设置
	EnableLLMRouting      bool    `json:"enable_llm_routing"`      // Use LLM for routing decisions
	EnableAdaptiveRouting bool    `json:"enable_adaptive_routing"` // Learn from feedback
	ConfidenceThreshold   float64 `json:"confidence_threshold"`    // Min confidence for routing

	// 后退设置
	EnableFallback   bool              `json:"enable_fallback"`
	FallbackStrategy RetrievalStrategy `json:"fallback_strategy"`

	// 缓存
	EnableCache bool          `json:"enable_cache"`
	CacheTTL    time.Duration `json:"cache_ttl"`

	// 日志
	LogDecisions bool `json:"log_decisions"`
}

type QueryTransformConfig struct {
	// 扩展设置
	EnableExpansion    bool    `json:"enable_expansion"`
	MaxExpansions      int     `json:"max_expansions"`      // Max expanded queries (3-5)
	ExpansionDiversity float64 `json:"expansion_diversity"` // 0-1, higher = more diverse

	// 重写设置
	EnableRewriting     bool `json:"enable_rewriting"`
	RewriteForRetrieval bool `json:"rewrite_for_retrieval"` // Optimize for retrieval

	// 分解设置
	EnableDecomposition bool    `json:"enable_decomposition"`
	MaxSubQueries       int     `json:"max_sub_queries"`     // Max sub-queries (2-5)
	DecomposeThreshold  float64 `json:"decompose_threshold"` // Complexity threshold

	// 有意检测
	EnableIntentDetection bool `json:"enable_intent_detection"`

	// HyDE( 嵌入式文档)
	EnableHyDE        bool `json:"enable_hyde"`
	HyDEDocumentCount int  `json:"hyde_document_count"` // Number of hypothetical docs

	// 后退提示
	EnableStepBack bool `json:"enable_step_back"`

	// 缓存
	EnableCache bool          `json:"enable_cache"`
	CacheTTL    time.Duration `json:"cache_ttl"`

	// LLM 设置
	UseLLM      bool    `json:"use_llm"`     // Use LLM for transformations
	Temperature float64 `json:"temperature"` // LLM temperature
}

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

type ReasoningChain struct {
	ID              string          `json:"id"`
	OriginalQuery   string          `json:"original_query"`
	Hops            []ReasoningHop  `json:"hops"`
	FinalAnswer     string          `json:"final_answer,omitempty"`
	FinalContext    string          `json:"final_context"`
	Status          ReasoningStatus `json:"status"`
	TotalDuration   time.Duration   `json:"total_duration"`
	TotalRetrieval  int             `json:"total_retrieval"`  // Total documents retrieved
	UniqueDocuments int             `json:"unique_documents"` // Unique documents
	Metadata        map[string]any  `json:"metadata,omitempty"`
	CreatedAt       time.Time       `json:"created_at"`
	CompletedAt     time.Time       `json:"completed_at,omitempty"`

	// 全局去重统计（新增）
	TotalDedupByID         int `json:"total_dedup_by_id"`
	TotalDedupBySimilarity int `json:"total_dedup_by_similarity"`
}

type ReasoningHop struct {
	ID               string            `json:"id"`
	HopNumber        int               `json:"hop_number"`
	Type             HopType           `json:"type"`
	Query            string            `json:"query"`
	TransformedQuery string            `json:"transformed_query,omitempty"`
	Results          []RetrievalResult `json:"results"`
	Context          string            `json:"context,omitempty"`   // Accumulated context
	Reasoning        string            `json:"reasoning,omitempty"` // LLM reasoning for this hop
	Confidence       float64           `json:"confidence"`
	Duration         time.Duration     `json:"duration"`
	Metadata         map[string]any    `json:"metadata,omitempty"`
	Timestamp        time.Time         `json:"timestamp"`

	// 去重统计（新增）
	DedupStats *DedupStats `json:"dedup_stats,omitempty"`
}

type ReasoningStatus string

type RerankProvider interface {
	RerankSimple(ctx context.Context, query string, documents []string, topN int) ([]rerank.RerankResult, error)
	Name() string
}

type RerankProviderType string

type Reranker interface {
	// Rerank 重排序结果
	Rerank(ctx context.Context, query string, results []RetrievalResult) ([]RetrievalResult, error)
}

type RerankerType string

type RetrievalResult struct {
	Document    Document `json:"document"`
	BM25Score   float64  `json:"bm25_score"`
	VectorScore float64  `json:"vector_score"`
	HybridScore float64  `json:"hybrid_score"`
	RerankScore float64  `json:"rerank_score,omitempty"`
	FinalScore  float64  `json:"final_score"`
}

type RetrievalStrategy string

type RetrieverOption func(*retrieverOptions)

type RoutingCondition struct {
	Type     string  `json:"type"`     // "intent", "keyword", "length", "complexity"
	Value    string  `json:"value"`    // Value to match
	Weight   float64 `json:"weight"`   // Weight adjustment when matched
	Operator string  `json:"operator"` // "equals", "contains", "greater", "less"
}

type RoutingDecision struct {
	Query            string                        `json:"query"`
	SelectedStrategy RetrievalStrategy             `json:"selected_strategy"`
	Confidence       float64                       `json:"confidence"`
	Scores           map[RetrievalStrategy]float64 `json:"scores"`
	Reasoning        string                        `json:"reasoning,omitempty"`
	Metadata         map[string]any                `json:"metadata,omitempty"`
	Timestamp        time.Time                     `json:"timestamp"`
}

type RoutingFeedback struct {
	Query            string            `json:"query"`
	SelectedStrategy RetrievalStrategy `json:"selected_strategy"`
	Success          bool              `json:"success"`
	Score            float64           `json:"score"`
	Timestamp        time.Time         `json:"timestamp"`
}

type SearchResult struct {
	ID       string
	Distance float64
	Score    float64 // 1 - distance (for cosine)
}

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

type SemanticCacheConfig struct {
	SimilarityThreshold float64 `json:"similarity_threshold"` // 相似度阈值（0.9-0.95）
}

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

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

type SimpleGraphEmbedderConfig struct {
	// Dimension 嵌入向量维度，默认 128。
	Dimension int
}

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

type SimpleTokenizer struct{}

type StrategyConfig struct {
	Strategy   RetrievalStrategy  `json:"strategy"`
	Enabled    bool               `json:"enabled"`
	Weight     float64            `json:"weight"`     // Base weight for this strategy
	MinScore   float64            `json:"min_score"`  // Minimum score to use this strategy
	MaxTokens  int                `json:"max_tokens"` // Max query tokens for this strategy
	Conditions []RoutingCondition `json:"conditions"` // Conditions that favor this strategy
}

type StrategyStats struct {
	Strategy     RetrievalStrategy `json:"strategy"`
	TotalCalls   int               `json:"total_calls"`
	SuccessRate  float64           `json:"success_rate"`
	AverageScore float64           `json:"average_score"`
}

type StrategyWithWeight struct {
	Strategy RetrievalStrategy `json:"strategy"`
	Weight   float64           `json:"weight"`
}

type StructuralBlock struct {
	Type     string // code, table, text, header
	Content  string
	StartPos int
	EndPos   int
}

type Tokenizer interface {
	CountTokens(text string) int
	Encode(text string) []int
}

type TransformationType string

type TransformedQuery struct {
	Original    string             `json:"original"`
	Transformed string             `json:"transformed"`
	Type        TransformationType `json:"type"`
	Intent      QueryIntent        `json:"intent,omitempty"`
	Confidence  float64            `json:"confidence"`
	SubQueries  []string           `json:"sub_queries,omitempty"`
	Keywords    []string           `json:"keywords,omitempty"`
	Entities    []string           `json:"entities,omitempty"`
	Metadata    map[string]any     `json:"metadata,omitempty"`
}

type Triple struct {
	Subject   string `json:"subject"`
	Predicate string `json:"predicate"`
	Object    string `json:"object"`
}

type VectorIndex interface {
	// Build 构建索引
	Build(vectors [][]float64, ids []string) error

	// Search 搜索最近邻
	Search(query []float64, k int) ([]SearchResult, error)

	// Add 添加向量
	Add(vector []float64, id string) error

	// Delete 删除向量
	Delete(id string) error

	// Size 索引大小
	Size() int
}

type VectorSearchResult struct {
	Document Document `json:"document"`
	Score    float64  `json:"score"`
	Distance float64  `json:"distance"`
}

type VectorStore interface {
	// 添加文档
	AddDocuments(ctx context.Context, docs []Document) error

	// 搜索相似文档
	Search(ctx context.Context, queryEmbedding []float64, topK int) ([]VectorSearchResult, error)

	// 删除文档
	DeleteDocuments(ctx context.Context, ids []string) error

	// 更新文档
	UpdateDocument(ctx context.Context, doc Document) error

	// 获取文档数量
	Count(ctx context.Context) (int, error)
}

type VectorStoreType string

type VisualizationEdge struct {
	Source string  `json:"source"`
	Target string  `json:"target"`
	Label  string  `json:"label,omitempty"`
	Weight float64 `json:"weight,omitempty"`
}

type VisualizationNode struct {
	ID       string         `json:"id"`
	Type     string         `json:"type"` // "query", "hop", "document", "answer"
	Label    string         `json:"label"`
	Metadata map[string]any `json:"metadata,omitempty"`
}

type WeaviateConfig struct {
	// 连接设置
	Host    string `json:"host"`               // Weaviate host (default: localhost)
	Port    int    `json:"port"`               // Weaviate port (default: 8080)
	Scheme  string `json:"scheme,omitempty"`   // http or https (default: http)
	BaseURL string `json:"base_url,omitempty"` // Full base URL (overrides host/port/scheme)

	// 认证
	APIKey string `json:"api_key,omitempty"` // API key for authentication

	// 类/收集设置
	ClassName string `json:"class_name"` // Weaviate class name (required)

	// Schema 设置
	AutoCreateSchema bool   `json:"auto_create_schema,omitempty"` // Auto-create class if not exists
	VectorIndexType  string `json:"vector_index_type,omitempty"`  // hnsw (default), flat
	Distance         string `json:"distance,omitempty"`           // cosine (default), dot, l2, hamming, manhattan

	// 矢量设置
	VectorSize int `json:"vector_size,omitempty"` // Vector dimension (optional, auto-detected)

	// 混合搜索设置
	HybridAlpha float64 `json:"hybrid_alpha,omitempty"` // Alpha for hybrid search (0=BM25, 1=vector, default: 0.5)

	// 超时设置
	Timeout time.Duration `json:"timeout,omitempty"` // Request timeout (default: 30s)

	// 属性字段名称
	ContentProperty  string `json:"content_property,omitempty"`  // Property for document content (default: content)
	MetadataProperty string `json:"metadata_property,omitempty"` // Property for document metadata (default: metadata)
	DocIDProperty    string `json:"doc_id_property,omitempty"`   // Property for original document ID (default: docId)
}

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

type WebRetrievalResult struct {
	URL     string  `json:"url"`
	Title   string  `json:"title"`
	Content string  `json:"content"`
	Score   float64 `json:"score"`
}

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

type WebRetrieverConfig struct {
	// 地方和网络成果之间的重量分配
	LocalWeight float64 `json:"local_weight"` // Weight for local RAG results (0-1)
	WebWeight   float64 `json:"web_weight"`   // Weight for web search results (0-1)

	// 网络搜索设置
	MaxWebResults    int           `json:"max_web_results"`    // Maximum web results to fetch
	WebSearchTimeout time.Duration `json:"web_search_timeout"` // Timeout for web search
	ParallelSearch   bool          `json:"parallel_search"`    // Search local and web in parallel

	// 结果合并
	TopK             int     `json:"top_k"`              // Final number of results to return
	MinScore         float64 `json:"min_score"`          // Minimum score threshold
	DeduplicateByURL bool    `json:"deduplicate_by_url"` // Remove duplicate URLs

	// 缓存
	EnableCache bool          `json:"enable_cache"` // Cache web results
	CacheTTL    time.Duration `json:"cache_ttl"`    // Cache time-to-live

	// 退后行为
	FallbackToLocal bool `json:"fallback_to_local"` // Use local-only if web fails
	FallbackToWeb   bool `json:"fallback_to_web"`   // Use web-only if local fails
}

type WebSearchFunc func(ctx context.Context, query string, maxResults int) ([]WebRetrievalResult, error)