Method Query

pkg/rag/manager.go:239–425

Query searches for relevant documents using all configured strategies If multiple strategies are configured, results are combined using the fusion strategy

(ctx context.Context, query string)

// Query searches for relevant documents using all configured strategies
// If multiple strategies are configured, results are combined using the fusion strategy
func (m *Manager) Query(ctx context.Context, query string) (results []database.SearchResult, err error) {
	// Start a `retrieval {rag_name}` span per the OTel GenAI semconv.
	// The query text itself is sensitive so we never capture it on the
	// span here — content capture is gated by a separate environment
	// variable in a later commit and emitted via a span event then.
	ctx, retSpan := genai.StartRetrieval(ctx, "rag", m.name, false, "")
	defer func() {
		if err != nil {
			retSpan.RecordError(err, "")
		}
		retSpan.SetResultCount(len(results))
		retSpan.End()
	}()

	slog.DebugContext(ctx, "[RAG Manager] Starting query",
		"rag_name", m.name,
		"num_strategies", len(m.strategies),
		"query_length", len(query))

	// Single retrieval strategy
	if len(m.strategies) == 1 {
		for strategyName, strategyImpl := range m.strategies {
			strategyCfg := m.strategyConfigs[strategyName]

			slog.DebugContext(ctx, "[RAG Manager] Single strategy query",
				"rag_name", m.name,
				"strategy", strategyName,
				"strategy_limit", strategyCfg.Limit,
				"strategy_threshold", strategyCfg.Threshold)

			// Assign to the function's named returns (note `=`, not
			// `:=`) so the deferred span closure sees the live values
			// even if a future change replaces the explicit
			// `return X, Y` form below with a bare `return`.
			results, err = strategyImpl.Query(ctx, query, strategyCfg.Limit, strategyCfg.Threshold)
			if err != nil {
				slog.ErrorContext(ctx, "[RAG Manager] Strategy query failed",
					"rag_name", m.name,
					"strategy", strategyName,
					"error", err)
				return nil, err
			}

			slog.DebugContext(ctx, "[RAG Manager] Single strategy results",
				"rag_name", m.name,
				"strategy", strategyName,
				"num_results", len(results))

			// Apply reranking if configured
			results = m.rerank(ctx, query, results)

			if limit := m.config.Results.Limit; limit > 0 && len(results) > limit {
				slog.DebugContext(ctx, "[RAG Manager] Truncating to global result limit",
					"rag_name", m.name,
					"strategy", strategyName,
					"before", len(results),
					"after", limit)
				results = results[:limit]