Vectara REST API

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

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

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

export class VectaraRESTAPI {
  /** @param token Your OmniStream key (one key for every API). */
  constructor(private token: string, private opts: VectaraRESTAPIOptions = {}) {
    if (!token) throw new Error("VectaraRESTAPI: 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 VectaraRESTAPIError(res.status, data?.error?.code, data?.error?.message);
    return data.data ?? data;
  }

  /** Delete */
  delete(body: unknown, params: { "customer-id": number }): Promise<any> {
    return this._request("POST", "/v1/delete-doc", { body, params });
  }

  /** Index */
  index(body: unknown, params: { "customer-id": number }): Promise<any> {
    return this._request("POST", "/v1/index", { body, params });
  }

  /** Query */
  query(params: { "customer-id": number }): Promise<any> {
    return this._request("POST", "/v1/query", { params });
  }

  /** Stream Query */
  streamQuery(params: { "customer-id": number }): Promise<any> {
    return this._request("POST", "/v1/stream-query", { params });
  }

  /** File Upload */
  fileUpload(params: { "c": number; "o": number; "d"?: boolean }): Promise<any> {
    return this._request("POST", "/v1/upload", { params });
  }
}

export default VectaraRESTAPI;