Как создать асинхронный генератор с неопределенным числом итераций?

import asyncio

async def async_generator():
    i = 0
    while True:
        yield i
        await asyncio.sleep(1)
        i += 1

async def main():
    async for number in async_generator():
        print(number)
        if number >= 10:
            break

asyncio.run(main())