Wolfram

/**
 * Wolfram - generated by OmniStream from Wolfram's OpenAPI spec.
 * One typed method per endpoint. A single Omni key reaches the API.
 */
const DEFAULT_BASE = "https://grid.skinvaults.online/v1/proxy/wolframalpha";

export class WolframError extends Error {
  status: number;
  code?: string;
  constructor(status: number, code: string | undefined, message?: string) {
    super(message || `Wolfram error ${status}`);
    this.name = "WolframError";
    this.status = status;
    this.code = code;
  }
}

export interface WolframOptions {
  baseUrl?: string;
  fetch?: typeof fetch;
  timeoutMs?: number;
}

export class Wolfram {
  /** @param token Your OmniStream key (one key for every API). */
  constructor(private token: string, private opts: WolframOptions = {}) {
    if (!token) throw new Error("Wolfram: token is required");
  }
  rateLimit: { remainingMinute?: number; remainingDay?: number } | null = null;

  private async _request(method: string, path: string, { params, body }: { params?: Record<string, unknown>; body?: unknown } = {}): Promise<any> {
    const f = this.opts.fetch ?? globalThis.fetch;
    const url = new URL((this.opts.baseUrl ?? DEFAULT_BASE).replace(/\/+$/, "") + path);
    if (params) for (const [k, v] of Object.entries(params)) if (v != null) url.searchParams.set(k, String(v));
    const headers: Record<string, string> = { "x-omni-key": this.token };
    if (body !== undefined) headers["content-type"] = "application/json";
    const res = await f(url, { method, headers, body: body !== undefined ? JSON.stringify(body) : undefined });
    this.rateLimit = {
      remainingMinute: Number(res.headers.get("x-ratelimit-remaining-minute")) || undefined,
      remainingDay: Number(res.headers.get("x-ratelimit-remaining-day")) || undefined,
    };
    const data = await res.json().catch(() => ({}));
    if (!res.ok) throw new WolframError(res.status, data?.error?.code, data?.error?.message);
    return data.data ?? data;
  }

  /** Evaluate Wolfram Language code */
  getWolframCloudResults(params: { "input": string }): Promise<any> {
    return this._request("GET", "/api/v1/cloud-plugin", { params });
  }

  /** Get Wolfram|Alpha results */
  getWolframAlphaResults(params: { "input": string }): Promise<any> {
    return this._request("GET", "/api/v1/llm-api", { params });
  }
}

export default Wolfram;