from meetkai_mka1 import SDK


with SDK(
    bearer_auth="<YOUR_BEARER_TOKEN_HERE>",
) as sdk:

    res = sdk.llm.budgets.set_api_key(api_key_id="<id>", period="weekly", limit_kind="tokens", limit=3281.35, thresholds=[
        {
            "pct": 793338,
            "action": "alert",
        },
    ])

    # Handle response
    print(res)

{
  "id": "<string>",
  "scope_id": "<string>",
  "limit": 123,
  "thresholds": [
    {
      "pct": 50
    }
  ],
  "alert_webhook_url": "<string>",
  "spend": {
    "window": "<string>",
    "cost_spent_nano": "<string>",
    "cost_spent": 123,
    "tokens_spent": "<string>",
    "pct": 123
  }
}