pareta 1.1.0 → 1.3.0

package/dist/index.cjs

@@ -174,7 +174,7 @@ function parseSSE(reader, opts) {
 }
 
 // src/version.ts
-var VERSION = "1.1.0";
+var VERSION = "1.3.0";
 
 // src/money.ts
 var MICRO_PER_CENT = 10000n;
@@ -486,6 +486,79 @@ var Embeddings = class extends BaseModel {
     return (this.raw.data ?? []).length;
   }
 };
+function base64ToBytes(b64) {
+  if (typeof Buffer !== "undefined") return new Uint8Array(Buffer.from(b64, "base64"));
+  const bin = atob(b64);
+  const bytes = new Uint8Array(bin.length);
+  for (let i = 0; i < bin.length; i++) bytes[i] = bin.charCodeAt(i);
+  return bytes;
+}
+var Transcription = class extends BaseModel {
+  get text() {
+    return this.raw.text ?? null;
+  }
+  get language() {
+    return this.raw.language ?? null;
+  }
+  get durationS() {
+    return this.raw.duration_s ?? null;
+  }
+  toString() {
+    return this.text ?? "";
+  }
+};
+var Speech = class extends BaseModel {
+  /** The synthesized audio, base64-decoded to raw bytes. */
+  get audio() {
+    const b64 = this.raw.audio_base64 || "";
+    return b64 ? base64ToBytes(b64) : new Uint8Array(0);
+  }
+  get audioBase64() {
+    return this.raw.audio_base64 ?? null;
+  }
+  get sampleRate() {
+    return this.raw.sample_rate ?? null;
+  }
+  get durationS() {
+    return this.raw.duration_s ?? null;
+  }
+  /** Container/codec of the returned audio (e.g. "wav"). */
+  get format() {
+    return this.raw.format ?? null;
+  }
+  /** Write the decoded audio to `path` (Node only — lazy node:fs). */
+  async save(path) {
+    const { writeFile } = await import('fs/promises');
+    await writeFile(path, this.audio);
+    return this;
+  }
+};
+var ImageGeneration = class extends BaseModel {
+  /** The generated image, base64-decoded to raw PNG bytes. */
+  get image() {
+    const b64 = this.b64Json ?? "";
+    return b64 ? base64ToBytes(b64) : new Uint8Array(0);
+  }
+  get b64Json() {
+    const data = this.raw.data ?? [];
+    return data[0]?.b64_json ?? null;
+  }
+  get size() {
+    return this.raw.size ?? null;
+  }
+  get model() {
+    return this.raw.model ?? null;
+  }
+  get created() {
+    return this.raw.created ?? null;
+  }
+  /** Write the decoded PNG to `path` (Node only — lazy node:fs). */
+  async save(path) {
+    const { writeFile } = await import('fs/promises');
+    await writeFile(path, this.image);
+    return this;
+  }
+};
 
 // src/resources/chat.ts
 var PATH = "/v1/chat/completions";
@@ -788,6 +861,82 @@ var Auto = class {
   }
 };
 
+// src/resources/audio.ts
+var BASE3 = "/v1/audio";
+function bytesToBase64(bytes) {
+  if (typeof Buffer !== "undefined") return Buffer.from(bytes).toString("base64");
+  let bin = "";
+  const CHUNK = 32768;
+  for (let i = 0; i < bytes.length; i += CHUNK) {
+    bin += String.fromCharCode(...bytes.subarray(i, i + CHUNK));
+  }
+  return btoa(bin);
+}
+async function toBase64(audio) {
+  if (typeof audio === "string") {
+    const { readFile } = await import('fs/promises');
+    return bytesToBase64(new Uint8Array(await readFile(audio)));
+  }
+  if (typeof audio === "object" && audio !== null && "base64" in audio) {
+    if (!audio.base64 || !audio.base64.trim()) throw new ParetaError("audio.base64 is empty");
+    return audio.base64;
+  }
+  if (audio instanceof Blob) return bytesToBase64(new Uint8Array(await audio.arrayBuffer()));
+  const bytes = audio instanceof Uint8Array ? audio : new Uint8Array(audio);
+  if (bytes.byteLength === 0) throw new ParetaError("audio is empty");
+  return bytesToBase64(bytes);
+}
+var Audio = class {
+  constructor(client) {
+    this.client = client;
+  }
+  client;
+  /** Speech-to-text (the `asr` lane). `audio` is a file path, raw bytes, a
+   * Blob, or `{ base64 }`. Metered per minute of input audio. */
+  async transcriptions(audio, opts = {}) {
+    const body = { audio_base64: await toBase64(audio) };
+    if (opts.language) body.language = opts.language;
+    return this.client.request("POST", `${BASE3}/transcriptions`, {
+      body,
+      cast: (raw) => new Transcription(raw)
+    });
+  }
+  /** Text-to-speech (the `tts` lane). Returns a `Speech` whose `.audio` is
+   * decoded bytes (`.save(path)` writes a file in Node). Metered per minute
+   * of output audio. */
+  speech(text, opts = {}) {
+    if (!text || !text.trim()) throw new ParetaError("text is required");
+    const body = { text };
+    if (opts.voice) body.voice = opts.voice;
+    return this.client.request("POST", `${BASE3}/speech`, {
+      body,
+      cast: (raw) => new Speech(raw)
+    });
+  }
+};
+
+// src/resources/images.ts
+var PATH3 = "/v1/images/generations";
+var Images = class {
+  constructor(client) {
+    this.client = client;
+  }
+  client;
+  /** Generate one image from a text prompt. Returns an `ImageGeneration`
+   * whose `.image` is decoded PNG bytes (`.save(path)` writes a file in
+   * Node). Billed flat per image. */
+  generate(prompt, opts = {}) {
+    if (!prompt || !prompt.trim()) throw new ParetaError("prompt is required");
+    const body = { prompt };
+    if (opts.size) body.size = opts.size;
+    if (opts.seed !== void 0) body.seed = opts.seed;
+    return this.client.request("POST", PATH3, {
+      body,
+      cast: (raw) => new ImageGeneration(raw)
+    });
+  }
+};
+
 // src/resources/retrieval.ts
 function makeRerank(client) {
   return (query, documents, opts = {}) => {
@@ -848,10 +997,14 @@ var Pareta = class _Pareta {
   tasks;
   evals;
   auto;
+  /** The Speech lanes — `pa.audio.transcriptions(...)` (ASR) + `pa.audio.speech(...)` (TTS). */
+  audio;
   /** Document reranking (the Retrieval precision lane) — `pa.rerank(query, docs, {topN})`. */
   rerank;
   /** Text embeddings (the Retrieval recall lane) — `pa.embeddings(input, {inputType})`. */
   embeddings;
+  /** Image generation — `pa.images.generate(prompt, {size, seed})`, flat price per image. */
+  images;
   constructor(options = {}) {
     if (!options.apiKey) {
       throw new ParetaError(
@@ -872,8 +1025,10 @@ var Pareta = class _Pareta {
     this.tasks = new Tasks(this);
     this.evals = new Evals(this);
     this.auto = new Auto(this);
+    this.audio = new Audio(this);
     this.rerank = makeRerank(this);
     this.embeddings = makeEmbeddings(this);
+    this.images = new Images(this);
   }
   /** Build from PARETA_API_KEY (+ optional PARETA_BASE_URL); explicit opts win. */
   static fromEnv(options = {}) {
@@ -1057,6 +1212,7 @@ var Pareta = class _Pareta {
 exports.APIConnectionError = APIConnectionError;
 exports.APIStatusError = APIStatusError;
 exports.APITimeoutError = APITimeoutError;
+exports.Audio = Audio;
 exports.AuthenticationError = AuthenticationError;
 exports.BadRequestError = BadRequestError;
 exports.BaseModel = BaseModel;
@@ -1073,6 +1229,8 @@ exports.EvalRuns = EvalRuns;
 exports.EvalSet = EvalSet;
 exports.EvalSets = EvalSets;
 exports.FrontierModel = FrontierModel;
+exports.ImageGeneration = ImageGeneration;
+exports.Images = Images;
 exports.InsufficientCreditsError = InsufficientCreditsError;
 exports.Message = Message;
 exports.Model = Model;
@@ -1084,9 +1242,11 @@ exports.PermissionDeniedError = PermissionDeniedError;
 exports.RateLimitError = RateLimitError;
 exports.Rerank = Rerank;
 exports.RerankResult = RerankResult;
+exports.Speech = Speech;
 exports.Task = Task;
 exports.TaskMatch = TaskMatch;
 exports.TaskMatchCandidate = TaskMatchCandidate;
+exports.Transcription = Transcription;
 exports.Usage = Usage;
 exports.VERSION = VERSION;
 //# sourceMappingURL=index.cjs.map