pareta 1.0.0 → 1.2.0

package/dist/index.cjs

@@ -174,7 +174,7 @@ function parseSSE(reader, opts) {
 }
 
 // src/version.ts
-var VERSION = "1.0.0";
+var VERSION = "1.2.0";
 
 // src/money.ts
 var MICRO_PER_CENT = 10000n;
@@ -445,6 +445,94 @@ var EvalRun = class extends BaseModel {
     return (this.raw.results ?? []).map((r) => new EvalResult(r));
   }
 };
+var RerankResult = class extends BaseModel {
+  /** Position of this document in YOUR request's documents array. */
+  get index() {
+    return Number(this.raw.index ?? -1);
+  }
+  /** Calibrated P(relevant) in (0, 1) — thresholdable, not just ordinal. */
+  get relevanceScore() {
+    return Number(this.raw.relevance_score ?? 0);
+  }
+};
+var Rerank = class extends BaseModel {
+  get results() {
+    return (this.raw.results ?? []).map((r) => new RerankResult(r));
+  }
+  get model() {
+    return this.raw.model ?? null;
+  }
+  get pairs() {
+    return this.raw.pairs ?? null;
+  }
+  /** Map the ranked indices back onto the documents you sent — best first. */
+  topDocuments(documents) {
+    return this.results.filter((r) => r.index >= 0 && r.index < documents.length).map((r) => documents[r.index]);
+  }
+};
+var Embeddings = class extends BaseModel {
+  get vectors() {
+    const rows = [...this.raw.data ?? []];
+    rows.sort((a, b) => Number(a.index ?? 0) - Number(b.index ?? 0));
+    return rows.map((r) => r.embedding ?? []);
+  }
+  get model() {
+    return this.raw.model ?? null;
+  }
+  get promptTokens() {
+    return this.raw.usage?.prompt_tokens ?? null;
+  }
+  get length() {
+    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;
+  }
+};
 
 // src/resources/chat.ts
 var PATH = "/v1/chat/completions";
@@ -747,6 +835,93 @@ 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/retrieval.ts
+function makeRerank(client) {
+  return (query, documents, opts = {}) => {
+    if (!query || !query.trim()) throw new ParetaError("query is required");
+    if (!Array.isArray(documents) || documents.length === 0) {
+      throw new ParetaError("documents must be a non-empty array of strings");
+    }
+    const body = { query, documents };
+    if (opts.topN !== void 0) body.top_n = opts.topN;
+    return client.request("POST", "/v1/rerank", {
+      body,
+      cast: (raw) => new Rerank(raw)
+    });
+  };
+}
+function makeEmbeddings(client) {
+  return (input, opts = {}) => {
+    const texts = typeof input === "string" ? [input] : input;
+    if (!Array.isArray(texts) || texts.length === 0 || texts.some((t) => typeof t !== "string" || !t.trim())) {
+      throw new ParetaError("input must be a non-empty string or array of non-empty strings");
+    }
+    if (opts.inputType !== void 0 && opts.inputType !== "query" && opts.inputType !== "document") {
+      throw new ParetaError('inputType must be "query" or "document"');
+    }
+    const body = { input: texts };
+    if (opts.inputType !== void 0) body.input_type = opts.inputType;
+    return client.request("POST", "/v1/embeddings", {
+      body,
+      cast: (raw) => new Embeddings(raw)
+    });
+  };
+}
+
 // src/client.ts
 var DEFAULT_BASE_URL = "https://api.pareta.ai";
 function randomKey() {
@@ -774,6 +949,12 @@ 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;
   constructor(options = {}) {
     if (!options.apiKey) {
       throw new ParetaError(
@@ -794,6 +975,9 @@ 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);
   }
   /** Build from PARETA_API_KEY (+ optional PARETA_BASE_URL); explicit opts win. */
   static fromEnv(options = {}) {
@@ -977,6 +1161,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;
@@ -984,6 +1169,7 @@ exports.ChatCompletion = ChatCompletion;
 exports.ChatCompletionChunk = ChatCompletionChunk;
 exports.Choice = Choice;
 exports.ConflictError = ConflictError;
+exports.Embeddings = Embeddings;
 exports.EndpointNotReadyError = EndpointNotReadyError;
 exports.EvalItemResult = EvalItemResult;
 exports.EvalResult = EvalResult;
@@ -1001,9 +1187,13 @@ exports.Pareta = Pareta;
 exports.ParetaError = ParetaError;
 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