@broberg/ai-sdk 0.10.2 → 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.2";
-declare const SDK_TAG: "@broberg/ai-sdk@0.10.2";
+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) {
@@ -1408,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}`, {
@@ -1854,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({
@@ -2183,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
           });
         }
       });
@@ -2438,8 +2447,8 @@ var stubProviders = {
 };
 
 // src/version.ts
-var VERSION = "0.10.2";
-var SDK_TAG = "@broberg/ai-sdk@0.10.2";
+var VERSION = "0.10.3";
+var SDK_TAG = "@broberg/ai-sdk@0.10.3";
 
 // src/cost/budget-store.ts
 function sqliteBudgetStore(config) {