query

type Cache interface {
	Get(ctx context.Context, key string) (*Response, bool)
	Set(ctx context.Context, key string, value *Response, expiration time.Duration)
}

type ContextCompressor interface {
	Compress(ctx context.Context, question string, results []core.Result) ([]core.Result, error)
}

type Embedder interface {
	Embed(ctx context.Context, texts []string) ([][]float32, error)
}

type HyDE interface {
	EnhanceQuery(ctx context.Context, question string) (string, error)
}

type Logger interface {
	Info(ctx context.Context, message string, fields map[string]interface{})
	Debug(ctx context.Context, message string, fields map[string]interface{})
	Error(ctx context.Context, message string, err error, fields map[string]interface{})
	Warn(ctx context.Context, message string, fields map[string]interface{})
}

type Metrics interface {
	RecordErrorCount(ctx context.Context, errorType string)
	RecordQueryLatency(ctx context.Context, duration time.Duration)
	RecordQueryCount(ctx context.Context, status string)
}

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

type QueryOptions struct {
	TopK              int
	PromptTemplate    string
	Stream            bool
	UseMultiHopRAG    bool   // Use multi-hop RAG for complex questions
	UseAgenticRAG     bool   // Use agentic RAG with autonomous retrieval
	MaxHops           int    // Maximum number of hops for multi-hop RAG
	AgentInstructions string // Instructions for agentic RAG
	RerankMultiplier  int    // Multiplier for topK when fetching candidates for reranking (default: 2)
}

type Response struct {
	Answer  string
	Sources []core.Result
}

type RouteResult struct {
	Type   string
	Params map[string]interface{}
}

type Router interface {
	Route(ctx context.Context, question string) (RouteResult, error)
}

type StreamResponse struct {
	Chunk   string
	Sources []core.Result
	Done    bool
	Error   error
}

type Tracer interface {
	StartSpan(ctx context.Context, name string) (context.Context, observability.Span)
	Extract(ctx context.Context) (observability.Span, bool)
}