pareta 1.2.0 → 1.4.0

package/dist/index.cjs

@@ -174,7 +174,7 @@ function parseSSE(reader, opts) {
 }
 
 // src/version.ts
-var VERSION = "1.2.0";
+var VERSION = "1.4.0";
 
 // src/money.ts
 var MICRO_PER_CENT = 10000n;
@@ -533,6 +533,32 @@ var Speech = class extends BaseModel {
     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";
@@ -889,6 +915,56 @@ var Audio = class {
   }
 };
 
+// src/resources/images.ts
+var PATH3 = "/v1/images/generations";
+var EDIT_PATH = "/v1/images/edits";
+async function toBase642(image) {
+  if (typeof image === "string") {
+    const { readFile } = await import('fs/promises');
+    return bytesToBase64(new Uint8Array(await readFile(image)));
+  }
+  if (typeof image === "object" && image !== null && "base64" in image) {
+    if (!image.base64 || !image.base64.trim()) throw new ParetaError("image.base64 is empty");
+    return image.base64;
+  }
+  if (image instanceof Blob) return bytesToBase64(new Uint8Array(await image.arrayBuffer()));
+  const bytes = image instanceof Uint8Array ? image : new Uint8Array(image);
+  if (bytes.byteLength === 0) throw new ParetaError("image is empty");
+  return bytesToBase64(bytes);
+}
+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)
+    });
+  }
+  /** Edit one reference image with a plain-language instruction (no mask).
+   * `image` is a file path, raw bytes, a Blob, or `{ base64 }` (≤25MB
+   * decoded). The output keeps the reference's aspect ratio. Billed flat
+   * per edit. */
+  async edit(image, prompt, opts = {}) {
+    if (!prompt || !prompt.trim()) throw new ParetaError("prompt is required");
+    const body = { prompt, image: await toBase642(image) };
+    if (opts.seed !== void 0) body.seed = opts.seed;
+    return this.client.request("POST", EDIT_PATH, {
+      body,
+      cast: (raw) => new ImageGeneration(raw)
+    });
+  }
+};
+
 // src/resources/retrieval.ts
 function makeRerank(client) {
   return (query, documents, opts = {}) => {
@@ -955,6 +1031,8 @@ var Pareta = class _Pareta {
   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(
@@ -978,6 +1056,7 @@ var Pareta = class _Pareta {
     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 = {}) {
@@ -1178,6 +1257,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;