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 {
	Field string
	Value interface{}
}

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 {
	Strategy          string  // 路由策略名称
	Confidence        float32 // 置信度 (0.0-1.0)
	TimeRange         *TimeRange
	SemanticQuery     string            // 清理后的语义查询
	NeedsReranker     bool              // 是否需要重排序
	ScheduleQueryMode ScheduleQueryMode // 日程查询模式（P1 新增）
}

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 {
	// 最大允许的未来时间（天数）
	MaxFutureDays int `json:"maxFutureDays" yaml:"maxFutureDays"`
	// 最大时间范围（天数）
	MaxRangeDays int `json:"maxRangeDays" yaml:"maxRangeDays"`
	// 时区（使用 UTC）
	Timezone string `json:"timezone" yaml:"timezone"`
}