pixelkiln 0.74.0 → 0.76.0

package/dist/index.cjs

@@ -221,6 +221,7 @@ __export(index_exports, {
   paletteDistance: () => paletteDistance,
   parentAssetId: () => parentAssetId,
   parseLock: () => parseLock,
+  parseProFlashQuote: () => parseProFlashQuote,
   parseTerrainDescriptions: () => parseTerrainDescriptions,
   parseWorkspace: () => parseWorkspace,
   pickHistory: () => pickHistory,
@@ -2032,6 +2033,25 @@ function cleanupResults(images, sent, operation) {
   }
   return images.map((image) => Buffer.from(image.base64, "base64"));
 }
+function parseProFlashQuote(raw) {
+  const body = raw && typeof raw === "object" ? raw : {};
+  const nested = [body, body.usage, body.cost, body.estimate].filter((v) => Boolean(v) && typeof v === "object");
+  const pick = (...names) => {
+    for (const obj of nested) {
+      for (const name of names) {
+        const value = obj[name];
+        if (typeof value === "number" && Number.isFinite(value) && value >= 0) return value;
+      }
+    }
+    return null;
+  };
+  const image = pick("image", "image_generations", "first_image", "first_image_generations");
+  const rotations = pick("rotations", "rotation", "rotations_generations", "rotation_generations");
+  const total = pick("total", "total_generations", "generations");
+  if (total !== null) return { image, rotations, total };
+  if (image !== null || rotations !== null) return { image, rotations, total: (image ?? 0) + (rotations ?? 0) };
+  throw new Error(`GET /pro-flash/cost returned no price: ${JSON.stringify(raw).slice(0, 200)}`);
+}
 var PixelLabError = class extends ProviderError {
   constructor(message6, status, body) {
     super("pixellab", message6, { status });
@@ -2090,6 +2110,19 @@ var PixelLabClient = class {
       plan: raw.subscription?.plan ?? "unknown"
     };
   }
+  /**
+   * `GET /pro-flash/cost`: PixelLab's own quote for one Pro Flash job, split
+   * into the first image and the v3 rotations. Free, and provisional by
+   * PixelLab's own description; the billed amount on the finished job is
+   * the real one. The response schema is undocumented (`{}` in the spec),
+   * so this reads it leniently: a total if one is given, else the sum of
+   * the parts, and refuses an answer with no number in it.
+   */
+  async proFlashCost(args) {
+    const query = new URLSearchParams({ operation: args.operation, width: String(args.width), height: String(args.height) });
+    if (args.nDirections !== void 0) query.set("n_directions", String(args.nDirections));
+    return parseProFlashQuote(await this.request(`/pro-flash/cost?${query}`));
+  }
   /**
    * Square objects that persist indefinitely.
    *
@@ -18865,6 +18898,7 @@ async function workspaceStatus(ws, dir) {
   paletteDistance,
   parentAssetId,
   parseLock,
+  parseProFlashQuote,
   parseTerrainDescriptions,
   parseWorkspace,
   pickHistory,