Support all reader implementations

The original implementation only supports `FileCacheReader`. Widen the allowed reader types.