咏唱引擎 API 的 Node.js 客户端

import { createWriteStream } from "node:fs";
import { Readable } from "node:stream";
import { pipeline } from "node:stream/promises";

export interface AriaComputeOptions {
  apiKey: string;
  baseUrl?: string;
}

export class AriaCompute {
  private baseUrl: string;
  private headers: Record<string, string>;

  constructor(opts: AriaComputeOptions) {
    this.baseUrl = (opts.baseUrl ?? "https://ariacompute.cn/api").replace(/\/$/, "");
    this.headers = {
      Authorization: `Bearer ${opts.apiKey}`,
      Accept: "application/json",
    };
  }

  private async request<T>(path: string, init: RequestInit = {}): Promise<T> {
    const res = await fetch(`${this.baseUrl}${path}`, {
      ...init,
      headers: { ...this.headers, ...(init.headers ?? {}) },
    });
    if (!res.ok) throw new Error(`${res.status} ${res.statusText}: ${await res.text()}`);
    return res.json() as Promise<T>;
  }

  listModels() {
    return this.request<{ models: unknown[] }>("/models");
  }

  wallet() {
    return this.request<{ balance: number; currency: string }>("/billing/wallet");
  }

  createPayment(provider: "stripe" | "wechat" | "alipay", amount: number, currency: string) {
    return this.request("/billing/payments", {
      method: "POST",
      headers: { "Content-Type": "application/json" },
      body: JSON.stringify({ provider, amount, currency }),
    });
  }

  async downloadModel(slug: string, quant: string, sdk: string, dest: string) {
    const url = `${this.baseUrl}/models/${slug}/download?quant=${quant}&sdk=${sdk}`;
    const res = await fetch(url, { headers: this.headers, redirect: "follow" });
    if (!res.ok || !res.body) throw new Error(`Download failed: ${res.status}`);
    await pipeline(Readable.fromWeb(res.body as any), createWriteStream(dest));
  }
}