Управление token API

export APIYI_SYS_TOKEN='nap_YOUR_PERSONAL_ACCESS_TOKEN'   # personal access token

curl --compressed -s -X POST 'https://api.apiyi.com/api/token/' \
  -H "Authorization: $APIYI_SYS_TOKEN" \
  -H 'Content-Type: application/json' \
  -d '{
    "name": "team-alice",
    "remain_quota": 500000,
    "unlimited_quota": false,
    "group": "default",
    "models": "gpt-5.6,deepseek-chat",
    "expired_time": -1
  }'

{
  "success": true,
  "message": "",
  "data": {
    "id": 119431,
    "user_id": 80778,
    "key": "K1RP**********4b75",
    "status": 1,
    "name": "team-alice",
    "group": "default",
    "models": "gpt-5.6,deepseek-chat",
    "remain_quota": 500000,
    "unlimited_quota": false,
    "used_quota": 0,
    "expired_time": -1,
    "created_time": 1785599000
  }
}

curl --compressed -s -X POST 'https://api.apiyi.com/api/token/119431/key' \
  -H "Authorization: $APIYI_SYS_TOKEN"

{
  "success": true,
  "message": "",
  "data": {
    "key": "K1RPzapuXLfBU4kDC5D9C0E70b1841AeAa542186B2F54b75"
  }
}

{
  "error": {
    "message": "This token is not authorized to use model: deepseek-chat"
  }
}

import time
expired_time = int(time.time()) + 30 * 86400

curl --compressed -s 'https://api.apiyi.com/api/token/?p=0&page_size=100' \
  -H "Authorization: $APIYI_SYS_TOKEN" | jq '.data[] | {id, name, status, remain_quota, used_quota, models}'

import os
import requests

BASE = "https://api.apiyi.com"
HEADERS = {"Authorization": os.environ["APIYI_SYS_TOKEN"],
           "Content-Type": "application/json"}

# 1. fetch the complete object
token = requests.get(f"{BASE}/api/token/119431", headers=HEADERS, timeout=30).json()["data"]

# 2. change only what you need, leaving everything else as-is
token["remain_quota"] = 2_500_000     # raise the cap to $5
token["models"] = "gpt-5.6,gemini-3-pro"

# 3. PUT the whole object back
resp = requests.put(f"{BASE}/api/token/", headers=HEADERS, json=token, timeout=30)
print(resp.json()["success"])

"""Run once a day at 00:00 (UTC+8): reset the listed tokens' quota to the daily cap."""
import json
import os
from datetime import datetime, timedelta, timezone
from pathlib import Path

import requests

BASE = "https://api.apiyi.com"
HEADERS = {"Authorization": os.environ["APIYI_SYS_TOKEN"],
           "Content-Type": "application/json"}
QUOTA_PER_USD = 500_000

# token ID -> daily cap (USD)
DAILY_BUDGET_USD = {
    119431: 5,     # team-alice
    119432: 5,     # team-bob
    119433: 20,    # workflow-prod
}
CARRY_OVERDRAFT = True   # deduct yesterday's overspend from today's budget

TZ = timezone(timedelta(hours=8))   # a "day" is measured in UTC+8
STATE = Path(__file__).with_name("daily_quota_state.json")


def main():
    today = datetime.now(TZ).date().isoformat()
    state = json.loads(STATE.read_text()) if STATE.exists() else {}

    for token_id, usd in DAILY_BUDGET_USD.items():
        last = state.get(str(token_id), {})
        if last.get("date") == today:
            continue                     # already reset today; don't hand out a second day's budget

        # 1. fetch the complete object
        token = requests.get(f"{BASE}/api/token/{token_id}",
                             headers=HEADERS, timeout=30).json()["data"]
        remain = token["remain_quota"]
        if "quota" in last:
            spent = (last["quota"] - remain) / QUOTA_PER_USD
            print(f"{token['name']}: spent ${spent:.2f} last period")

        # 2. change only the quota fields, leaving everything else as-is
        quota = int(usd * QUOTA_PER_USD)
        if CARRY_OVERDRAFT and remain < 0:
            quota = max(quota + remain, 0)
        token["remain_quota"] = quota
        token["unlimited_quota"] = False

        # 3. PUT the whole object back
        resp = requests.put(f"{BASE}/api/token/", headers=HEADERS,
                            json=token, timeout=30).json()
        if not resp.get("success"):
            raise RuntimeError(f"failed to reset token {token_id}: {resp.get('message')}")

        # save after each token, so a rerun after a mid-way failure skips the ones already done
        state[str(token_id)] = {"date": today, "quota": quota}
        STATE.write_text(json.dumps(state, indent=1))
        print(f"{token['name']}: today's budget ${quota / QUOTA_PER_USD:.2f}")


if __name__ == "__main__":
    main()

# /opt/apiyi-quota/env (chmod 600)
export APIYI_SYS_TOKEN='nap_YOUR_PERSONAL_ACCESS_TOKEN'

# crontab -e
# server time zone is UTC+8:
0 0 * * * . /opt/apiyi-quota/env && python3 /opt/apiyi-quota/reset_daily_quota.py >> /opt/apiyi-quota/reset.log 2>&1
# server time zone is UTC: 00:00 (UTC+8) = 16:00 (UTC), so start the line with 0 16 * * *

token = requests.get(f"{BASE}/api/token/119431", headers=HEADERS, timeout=30).json()["data"]
token["status"] = 2      # 1 = enabled, 2 = disabled

requests.put(f"{BASE}/api/token/", headers=HEADERS,
             params={"status_only": "true"}, json=token, timeout=30)

curl --compressed -s -X DELETE 'https://api.apiyi.com/api/token/119431' \
  -H "Authorization: $APIYI_SYS_TOKEN"

for token_id in [119431, 119432, 119433]:
    requests.delete(f"{BASE}/api/token/{token_id}", headers=HEADERS, timeout=30)