async_run.py

import asyncio

from ai21 import AsyncAI21Client

client = AsyncAI21Client()


async def main():
    run_result = await client.beta.maestro.runs.create_and_poll(
        input="Write a poem about the ocean",
        requirements=[
            {
                "name": "length requirement",
                "description": "The length of the poem should be less than 1000 characters",
            },
            {
                "name": "rhyme requirement",
                "description": "The poem should rhyme",
            },
        ],
    )

    print(run_result)


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