Вы можете изучить использование источника BoundedReadFromUnbounded, который позволяет преобразовывать неограниченную коллекцию в ограниченную.
UnboundedSource<String> unboundedSource = ...;
PCollection<String> boundedCollection=
p.apply(Read.from(unboundedSource).withMaxNumRecords(10));
Вы также можете использовать с MaxReadTime