queryengine

type Config struct {
	// 时间范围配置
	TimeRange TimeRangeConfig `json:"timeRange" yaml:"timeRange"`

	// 查询限制配置
	QueryLimits QueryLimitsConfig `json:"queryLimits" yaml:"queryLimits"`

	// 检索配置
	Retrieval RetrievalConfig `json:"retrieval" yaml:"retrieval"`

	// 评分配置
	Scoring ScoringConfig `json:"scoring" yaml:"scoring"`
}

type ErrInvalidConfig struct {
	Value interface{}
	Field string
}

type QueryLimitsConfig struct {
	// 最大查询长度（字符数）
	MaxQueryLength int `json:"maxQueryLength" yaml:"maxQueryLength"`
	// 最大结果数量
	MaxResults int `json:"maxResults" yaml:"maxResults"`
	// 最小分数阈值
	MinScore float32 `json:"minScore" yaml:"minScore"`
}

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

type RetrievalConfig struct {
	// 向量检索限制
	VectorLimit int `json:"vectorLimit" yaml:"vectorLimit"`
	// 混合检索限制
	HybridLimit int `json:"hybridLimit" yaml:"hybridLimit"`
	// 扩展检索限制
	ExpandLimit int `json:"expandLimit" yaml:"expandLimit"`
	// 是否启用 Reranker
	EnableReranker bool `json:"enableReranker" yaml:"enableReranker"`
	// 最大文档长度（字符数）
	MaxDocLength int `json:"maxDocLength" yaml:"maxDocLength"`
}

type RouteDecision struct {
	TimeRange         *TimeRange
	Strategy          string
	SemanticQuery     string
	Confidence        float32
	ScheduleQueryMode ScheduleQueryMode
	NeedsReranker     bool
}

type ScheduleQueryMode int32

type ScoringConfig struct {
	// BM25 权重范围
	BM25WeightMin float32 `json:"bm25WeightMin" yaml:"bm25WeightMin"`
	BM25WeightMax float32 `json:"bm25WeightMax" yaml:"bm25WeightMax"`
	// 语义权重
	SemanticWeight float32 `json:"semanticWeight" yaml:"semanticWeight"`
	// 高质量阈值
	HighQualityThreshold float32 `json:"highQualityThreshold" yaml:"highQualityThreshold"`
	// 中等质量阈值
	MediumQualityThreshold float32 `json:"mediumQualityThreshold" yaml:"mediumQualityThreshold"`
	// 分数差距阈值（用于判断是否需要重排）
	ScoreGapThreshold float32 `json:"scoreGapThreshold" yaml:"scoreGapThreshold"`
	// 最小重排结果数
	MinRerankResults int `json:"minRerankResults" yaml:"minRerankResults"`
}

type TimeRange struct {
	Start time.Time
	End   time.Time
	Label string
}

type TimeRangeConfig struct {
	Timezone      string `json:"timezone" yaml:"timezone"`
	MaxFutureDays int    `json:"maxFutureDays" yaml:"maxFutureDays"`
	MaxRangeDays  int    `json:"maxRangeDays" yaml:"maxRangeDays"`
}