fix: Pass query to pipeline wrappers from the AST (#12305)

pkg/ingester/instance.go

@@ -443,7 +443,7 @@ func (i *instance) Query(ctx context.Context, req logql.SelectLogParams) (iter.E
 			return nil, err
 		}
 
-		pipeline = i.pipelineWrapper.Wrap(ctx, pipeline, expr.String(), userID, shards)
+		pipeline = i.pipelineWrapper.Wrap(ctx, pipeline, req.Plan.String(), userID, shards)
 	}
 
 	stats := stats.FromContext(ctx)
@@ -498,7 +498,7 @@ func (i *instance) QuerySample(ctx context.Context, req logql.SelectSampleParams
 			return nil, err
 		}
 
-		extractor = i.extractorWrapper.Wrap(ctx, extractor, expr.String(), userID, shards)
+		extractor = i.extractorWrapper.Wrap(ctx, extractor, req.Plan.String(), userID, shards)
 	}
 
 	stats := stats.FromContext(ctx)

pkg/storage/store.go

@@ -514,7 +514,7 @@ func (s *LokiStore) SelectLogs(ctx context.Context, req logql.SelectLogParams) (
 			return nil, err
 		}
 
-		pipeline = s.pipelineWrapper.Wrap(ctx, pipeline, expr.String(), userID, shards)
+		pipeline = s.pipelineWrapper.Wrap(ctx, pipeline, req.Plan.String(), userID, shards)
 	}
 
 	var chunkFilterer chunk.Filterer
@@ -562,7 +562,7 @@ func (s *LokiStore) SelectSamples(ctx context.Context, req logql.SelectSamplePar
 			return nil, err
 		}
 
-		extractor = s.extractorWrapper.Wrap(ctx, extractor, expr.String(), userID, shards)
+		extractor = s.extractorWrapper.Wrap(ctx, extractor, req.Plan.String(), userID, shards)
 	}
 
 	var chunkFilterer chunk.Filterer