interrupt_event = asyncio.Event()

...
        await interrupt_event.wait()
...

if __name__ == "__main__":
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    try:
        loop.run_until_complete(main())
    except KeyboardInterrupt:
        interrupt_event.set()
        loop.run_until_complete(loop.shutdown_asyncgens())

...
    async with Worker(
        client,
        task_queue=TASK_QUEUE,
        workflows=[Workflow],
    ):
        # Wait forever
        await asyncio.Future()
        ...


if __name__ == "__main__":
    asyncio.run(main())