@broberg/ai-sdk 0.10.1 → 0.10.3

package/dist/index.d.ts

@@ -164,6 +164,9 @@ interface ImageRequest {
     height?: number;
     /** LoRAs to merge at inference (F021) — e.g. a trained brand/style LoRA. */
     loras?: LoraWeight[];
+    /** F021.4 — re-roll once with a fresh seed if fal's safety-checker false-positives
+     *  and returns a black image (has_nsfw_concepts). fal only. */
+    retryOnBlack?: boolean;
 }
 interface ImageResult {
     url: string;
@@ -929,6 +932,8 @@ declare const imageInputSchema: z.ZodObject<{
     }>, "many">>;
     /** Shorthand for a single LoRA — normalized to loras:[{path, scale:1}]. */
     lora: z.ZodOptional<z.ZodString>;
+    /** F021.4 — re-roll once if fal returns a black image (NSFW false-positive). */
+    retryOnBlack: z.ZodOptional<z.ZodBoolean>;
 }, "strip", z.ZodTypeAny, {
     prompt: string;
     purpose?: string | undefined;
@@ -951,6 +956,7 @@ declare const imageInputSchema: z.ZodObject<{
     labels?: Record<string, string> | undefined;
     width?: number | undefined;
     height?: number | undefined;
+    retryOnBlack?: boolean | undefined;
 }, {
     prompt: string;
     purpose?: string | undefined;
@@ -973,6 +979,7 @@ declare const imageInputSchema: z.ZodObject<{
     labels?: Record<string, string> | undefined;
     width?: number | undefined;
     height?: number | undefined;
+    retryOnBlack?: boolean | undefined;
 }>;
 declare const trainStyleInputSchema: z.ZodObject<{
     tier: z.ZodOptional<z.ZodEnum<["fast", "smart", "powerful", "cheap", "vision", "video", "embedding"]>>;
@@ -1680,8 +1687,8 @@ declare const falStubAdapter: ProviderAdapter;
  *  wires the live adapters. */
 declare const stubProviders: Record<string, ProviderAdapter>;
 
-declare const VERSION: "0.10.1";
-declare const SDK_TAG: "@broberg/ai-sdk@0.10.1";
+declare const VERSION: "0.10.3";
+declare const SDK_TAG: "@broberg/ai-sdk@0.10.3";
 
 /** Built-in defaults. Every entry is overridable via AiConfig.defaults or a
  *  per-call override. Model IDs are current at scaffold time; callers pin their

package/dist/index.js

@@ -1317,7 +1317,13 @@ function falAdapter(config = {}) {
       body.loras = req.loras.map((l) => ({ path: l.path, scale: l.scale ?? 1 }));
     }
     const mode = config.mode ?? "sync";
-    const url = await (mode === "sync" ? runSync(req.spec.model, headers, body) : runQueue(req.spec.model, headers, body));
+    const run = (b) => mode === "sync" ? runSync(req.spec.model, headers, b) : runQueue(req.spec.model, headers, b);
+    let calls = 1;
+    let { url, flagged } = await run(body);
+    if (flagged && req.retryOnBlack) {
+      calls++;
+      ({ url, flagged } = await run({ ...body, seed: Math.floor(Math.random() * 1e9) }));
+    }
     const usage = freshUsage({
       provider: "fal",
       model: req.spec.model,
@@ -1326,7 +1332,7 @@ function falAdapter(config = {}) {
       inputTokens: 0,
       outputTokens: 0
     });
-    usage.costUsd = config.pricePerImage ?? FAL_IMAGE_PRICE_ESTIMATE[req.spec.model] ?? 0;
+    usage.costUsd = (config.pricePerImage ?? FAL_IMAGE_PRICE_ESTIMATE[req.spec.model] ?? 0) * calls;
     return { url, usage };
   }
   async function trainStyle(req) {
@@ -1334,7 +1340,7 @@ function falAdapter(config = {}) {
     if (!apiKey) throw new Error("fal adapter: FAL_KEY not set");
     const headers = authHeaders(apiKey);
     const body = {
-      images_data_url: await resolveImagesDataUrl(req.images),
+      images_data_url: await resolveImagesUrl(req.images, apiKey),
       is_style: req.isStyle ?? true
     };
     if (req.triggerWord !== void 0) body.trigger_word = req.triggerWord;
@@ -1365,7 +1371,7 @@ function falAdapter(config = {}) {
     usage.costUsd = config.pricePerTraining ?? FAL_TRAIN_PRICE_ESTIMATE;
     return { loraUrl, configUrl: configUrl ?? "", usage };
   }
-  async function resolveImagesDataUrl(images) {
+  async function resolveImagesUrl(images, apiKey) {
     if (typeof images === "string") return images;
     const files = await Promise.all(
       images.map(async (url, i) => {
@@ -1374,8 +1380,27 @@ function falAdapter(config = {}) {
         return { name: fileNameFromUrl(url, i), data: new Uint8Array(await res.arrayBuffer()) };
       })
     );
-    const zip = buildZip(files);
-    return `data:application/zip;base64,${Buffer.from(zip).toString("base64")}`;
+    return uploadToFalStorage(buildZip(files), "application/zip", "styleset.zip", apiKey);
+  }
+  async function uploadToFalStorage(bytes, contentType, fileName, apiKey) {
+    const initiate = await doFetch("https://rest.alpha.fal.ai/storage/upload/initiate", {
+      method: "POST",
+      headers: { Authorization: `Key ${apiKey}`, "content-type": "application/json" },
+      body: JSON.stringify({ content_type: contentType, file_name: fileName })
+    });
+    if (!initiate.ok) {
+      throw new Error(
+        `fal storage initiate ${initiate.status}: ${(await initiate.text().catch(() => "")).slice(0, 200)}`
+      );
+    }
+    const { upload_url, file_url } = await initiate.json();
+    const put = await doFetch(upload_url, {
+      method: "PUT",
+      headers: { "content-type": contentType },
+      body: bytes
+    });
+    if (!put.ok) throw new Error(`fal storage upload PUT ${put.status}`);
+    return file_url;
   }
   async function runSync(model, headers, body) {
     const res = await doFetch(`${syncBase}/${model}`, {
@@ -1389,13 +1414,13 @@ function falAdapter(config = {}) {
     const data = await res.json();
     const out = data.images?.[0]?.url;
     if (!out) throw new Error(`fal: no image url in response`);
-    return out;
+    return { url: out, flagged: data.has_nsfw_concepts?.[0] === true };
   }
   async function runQueue(model, headers, body) {
     const result = await queueResult(model, headers, body, timeoutMs);
     const out = result.images?.[0]?.url;
     if (!out) throw new Error("fal queue: no image url in result");
-    return out;
+    return { url: out, flagged: result.has_nsfw_concepts?.[0] === true };
   }
   async function queueResult(model, headers, body, deadlineMs) {
     const submitRes = await doFetch(`${queueBase}/${model}`, {
@@ -1835,6 +1860,8 @@ var imageInputSchema = z.object({
   loras: z.array(loraWeightSchema).optional(),
   /** Shorthand for a single LoRA — normalized to loras:[{path, scale:1}]. */
   lora: z.string().optional(),
+  /** F021.4 — re-roll once if fal returns a black image (NSFW false-positive). */
+  retryOnBlack: z.boolean().optional(),
   ...callOptions
 });
 var trainStyleInputSchema = z.object({
@@ -2164,7 +2191,8 @@ function createAI(config = {}) {
             spec,
             width: input.width,
             height: input.height,
-            loras: loras.length ? loras : void 0
+            loras: loras.length ? loras : void 0,
+            retryOnBlack: input.retryOnBlack
           });
         }
       });
@@ -2419,8 +2447,8 @@ var stubProviders = {
 };
 
 // src/version.ts
-var VERSION = "0.10.1";
-var SDK_TAG = "@broberg/ai-sdk@0.10.1";
+var VERSION = "0.10.3";
+var SDK_TAG = "@broberg/ai-sdk@0.10.3";
 
 // src/cost/budget-store.ts
 function sqliteBudgetStore(config) {