API Key 分组管理

POST /v1/apikey_groups

{
  "name": "生产环境",
  "description": "用于生产环境的 API Key 分组"
}

curl -X POST https://api.nonelinear.com/v1/apikey_groups \
  -H "Authorization: Bearer sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" \
  -H "Content-Type: application/json" \
  -d '{
    "name": "生产环境",
    "description": "用于生产环境的 API Key 分组"
  }'

{
  "success": true,
  "message": "分组\"生产环境\"创建成功",
  "group_id": "6a06f7db38b3bf1609a9c200",
  "group_name": "生产环境"
}

GET /v1/apikey_groups

curl -H "Authorization: Bearer sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" \
  https://api.nonelinear.com/v1/apikey_groups

{
  "object": "list",
  "data": [
    {
      "id": "6a06f7db38b3bf1609a9c200",
      "name": "生产环境",
      "description": "用于生产环境的 API Key 分组",
      "created_at": 1774081270,
      "updated_at": 1774081270,
      "apikey_count": 5
    },
    {
      "id": "6a06f7db38b3bf1609a9c201",
      "name": "测试环境",
      "description": "用于测试环境的 API Key 分组",
      "created_at": 1774081280,
      "updated_at": 1774081280,
      "apikey_count": 3
    }
  ]
}

POST /v1/apikey_groups/<group_id>/batch_budget

{
  "budget": 1000.0,
  "daily_limit": 100.0,
  "monthly_limit": 500.0,
  "expiry_date": "2026-12-31",
  "input_price_limit": 50.0,
  "output_price_limit": 200.0
}

curl -X POST https://api.nonelinear.com/v1/apikey_groups/6a06f7db38b3bf1609a9c200/batch_budget \
  -H "Authorization: Bearer sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx" \
  -H "Content-Type: application/json" \
  -d '{
    "budget": 1000.0,
    "daily_limit": 100.0,
    "monthly_limit": 500.0,
    "expiry_date": "2026-12-31"
  }'

{
  "success": true,
  "message": "批量设置限额成功，共更新 5 个API Key",
  "updated_count": 5,
  "failed_count": 0
}

{
  "success": false,
  "error": "参数 budget 类型错误，期望 int 或 float，实际收到 str"
}

{
  "success": false,
  "error": "参数 daily_limit 不能为负数"
}

{
  "success": false,
  "error": "参数 expiry_date 格式错误，应为 YYYY-MM-DD"
}

{
  "success": false,
  "error": "至少需要指定一项限额参数"
}

import requests
import json

BASE_URL = 'https://api.nonelinear.com'

def create_apikey_group(apikey: str, name: str, description: str = '') -> dict:
    """创建 API Key 分组"""
    resp = requests.post(
        f'{BASE_URL}/v1/apikey_groups',
        headers={
            'Authorization': f'Bearer {apikey}',
            'Content-Type': 'application/json'
        },
        json={
            'name': name,
            'description': description
        }
    )
    resp.raise_for_status()
    return resp.json()


def query_apikey_groups(apikey: str) -> list:
    """查询所有 API Key 分组"""
    resp = requests.get(
        f'{BASE_URL}/v1/apikey_groups',
        headers={'Authorization': f'Bearer {apikey}'}
    )
    resp.raise_for_status()
    return resp.json()['data']


def set_group_apikey_budget(apikey: str, group_id: str, **kwargs) -> dict:
    """
    批量设置分组内 API Key 的限额
    
    kwargs 可包含：
      - budget: 预算限额
      - daily_limit: 每日限额
      - monthly_limit: 每月限额
      - expiry_date: 有效期
      - input_price_limit: 输入价格限额
      - output_price_limit: 输出价格限额
    """
    resp = requests.post(
        f'{BASE_URL}/v1/apikey_groups/{group_id}/batch_budget',
        headers={
            'Authorization': f'Bearer {apikey}',
            'Content-Type': 'application/json'
        },
        json=kwargs
    )
    resp.raise_for_status()
    return resp.json()


if __name__ == '__main__':
    API_KEY = '<你的默认APIKey>'

    # 创建分组
    result = create_apikey_group(API_KEY, '生产环境', '用于生产环境的 API Key')
    print(f"创建分组: {result}")
    group_id = result.get('group_id')

    # 查询所有分组
    groups = query_apikey_groups(API_KEY)
    print(f"\n所有分组:")
    for group in groups:
        print(f"  {group['name']}: {group['apikey_count']} 个 API Key")

    # 批量设置分组内 API Key 的限额
    if group_id:
        budget_result = set_group_apikey_budget(
            API_KEY,
            group_id,
            budget=1000.0,
            daily_limit=100.0,
            monthly_limit=500.0,
            expiry_date='2026-12-31'
        )
        print(f"\n批量设置限额结果: {budget_result}")