OAuth 2.1

import secrets
from chatatp_studio import OAuthClient

oauth = OAuthClient()
verifier, challenge = OAuthClient.create_pkce()
state = secrets.token_urlsafe(32)

authorization_url = oauth.authorization_url(
    client_id="your-client-id",
    redirect_uri="https://your-app.example.com/oauth/callback",
    scopes=["agents:read", "conversations:read", "messages:send"],
    state=state,
    code_challenge=challenge,
    code_challenge_method="S256",
)

# Store state and verifier in the user's server-side session.
# Redirect the user's browser to authorization_url.

from chatatp_studio import ChatATPClient

async with OAuthClient() as oauth:
    tokens = await oauth.exchange_code(
        client_id="your-client-id",
        code=callback_code,
        redirect_uri="https://your-app.example.com/oauth/callback",
        code_verifier=verifier,
    )

async with ChatATPClient.from_oauth(tokens.access_token) as client:
    agents = await client.agents.list()

async with OAuthClient() as oauth:
    identity = await oauth.userinfo(tokens.access_token)
    print(identity["sub"], identity["email"], identity["name"])

async with OAuthClient() as oauth:
    next_tokens = await oauth.refresh(
        client_id="your-client-id",
        refresh_token=tokens.refresh_token,
    )
    await oauth.revoke(next_tokens.refresh_token, client_id="your-client-id")