Get Azure AI consumption pricing for token usage

azure-ai-consumption-token-pricing.py

#!/usr/bin/env python3

from requests import get
import json

def main():
    api_url = (
        "https://prices.azure.com/api/retail/prices?api-version=2021-10-01-preview"
    )
    query = [
        "armRegionName eq 'eastus'",
        "serviceName eq 'Cognitive Services'",
        "serviceFamily eq 'AI + Machine Learning'",
        "priceType eq 'Consumption'",
        "contains(meterName, 'Tokens')",
        "unitOfMeasure eq '1K'",
    ]
    query = " and ".join(query)

    data = {}
    response = get(api_url, params={"$filter": query})
    while True:
        response.raise_for_status()
        raw = json.loads(response.text)
        next_page = raw.get("NextPageLink")
        for item in raw["Items"]:
            data[item["meterName"]] = item["retailPrice"]
        if not next_page:
            break
        response = get(next_page)

    print(json.dumps(data, indent=4, sort_keys=True))


if __name__ == "__main__":
    main()