@sanity/migrate/src/it-utils/take.ts

Tooling for running data migrations on Sanity.io projects

export async function* take<T>(it: AsyncIterableIterator<T>, count: number) {
  let i = 0
  for await (const chunk of it) {
    if (i++ >= count) return
    yield chunk
  }
}