Create and manage runs

python -m pip install papermap-air
export AIR_BASE_URL="https://dev.airapi.papermap.ai"
export AIR_APP_KEY="<your-app-key>"

import asyncio
import os

from air import AirClient


async def main() -> None:
    async with AirClient(
        os.environ["AIR_BASE_URL"],
        os.environ["AIR_APP_KEY"],
    ) as client:
        run = await client.runs.create(
            agent_id="support-agent",
            input={"message": "Where is order 1042?"},
            context={"tenant_id": "acme"},
            webhook_url="https://example.com/hooks/air",
        )
        result = await client.runs.wait(
            run.id,
            timeout=120,
            poll_interval=0.5,
        )
        print(result.status, result.output)


asyncio.run(main())

result = await client.runs.wait(run.id, timeout=120)

if result.status == "failed":
    raise RuntimeError(result.error)

answer = result.output["answer"]
needs_review = result.output["needs_human_review"]

run = await client.runs.create(
    agent_id="support-agent",
    input={"message": "Write a detailed account summary."},
)

text = ""
async for event in client.runs.stream(run.id):
    if event.output_delta is not None:
        text += event.output_delta
        print(event.output_delta, end="", flush=True)
    elif event.type == "model_output_reset":
        # AIR retried or used a fallback after a partial response.
        text = ""
        # Clear the partial text in your UI before showing the new attempt.

final = await client.runs.get(run.id)
print(f"\nRun status: {final.status}")

async for event in client.runs.stream(run.id, after_seq=last_seq):
    last_seq = event.seq

current = await client.runs.get(run.id)

if current.status not in {"succeeded", "failed", "cancelled"}:
    await client.runs.cancel(run.id)

from air import AirProtocolError, RunTimeoutError

try:
    result = await client.runs.wait(run.id, timeout=30)
except RunTimeoutError:
    # The server-side run may still be active.
    result = await client.runs.get(run.id)
except AirProtocolError as exc:
    print(exc.status_code)
    raise