@broberg/ai-sdk 0.13.1 → 0.15.0

package/dist/index.d.ts

@@ -165,6 +165,21 @@ interface ImageRequest {
     height?: number;
     /** LoRAs to merge at inference (F021) — e.g. a trained brand/style LoRA. */
     loras?: LoraWeight[];
+    /** F023 — a BFL finetune id (a subject trained in the BFL dashboard). Routes to the
+     *  EU-resident BFL finetuned-portrait endpoint. */
+    finetune?: string;
+    /** F023 — how strongly the finetune is applied (BFL finetune_strength, ~0–2). */
+    finetuneStrength?: number;
+    /** F023.5 — 1–8 reference photos of a subject (URL or raw bytes). Routes to the
+     *  EU-resident BFL FLUX 2 multi-reference endpoint — generate a likeness with NO
+     *  training step. Bytes are base64-inlined into the EU call (no cross-region fetch). */
+    referenceImages?: (string | Uint8Array)[];
+    /** F023.5 — fixed seed for reproducible output (BFL). */
+    seed?: number;
+    /** F023.5 — output container (BFL FLUX 2): "jpeg" | "png" | "webp". Default jpeg. */
+    outputFormat?: "jpeg" | "png" | "webp";
+    /** F023.5 — BFL content-moderation strictness, 0 (strict) … 6 (lax). Default 2. */
+    safetyTolerance?: number;
     /** 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;
@@ -942,6 +957,20 @@ declare const imageInputSchema: z.ZodObject<{
     }>, "many">>;
     /** Shorthand for a single LoRA — normalized to loras:[{path, scale:1}]. */
     lora: z.ZodOptional<z.ZodString>;
+    /** F023 — a BFL finetune id (subject trained in the BFL dashboard). Routes to the
+     *  EU-resident BFL finetuned-portrait endpoint. */
+    finetune: z.ZodOptional<z.ZodString>;
+    /** F023 — BFL finetune_strength (~0–2; higher = stronger likeness). */
+    finetuneStrength: z.ZodOptional<z.ZodNumber>;
+    /** F023.5 — 1–8 reference photos (URL or raw bytes) → EU-resident BFL FLUX 2
+     *  multi-reference generation (likeness, no training step). */
+    referenceImages: z.ZodOptional<z.ZodArray<z.ZodUnion<[z.ZodString, z.ZodType<Uint8Array<ArrayBuffer>, z.ZodTypeDef, Uint8Array<ArrayBuffer>>]>, "many">>;
+    /** F023.5 — fixed seed for reproducible output (BFL). */
+    seed: z.ZodOptional<z.ZodNumber>;
+    /** F023.5 — output container (BFL FLUX 2). Default jpeg. */
+    outputFormat: z.ZodOptional<z.ZodEnum<["jpeg", "png", "webp"]>>;
+    /** F023.5 — BFL content-moderation strictness, 0 (strict) … 6 (lax). Default 2. */
+    safetyTolerance: z.ZodOptional<z.ZodNumber>;
     /** F021.4 — re-roll once if fal returns a black image (NSFW false-positive). */
     retryOnBlack: z.ZodOptional<z.ZodBoolean>;
 }, "strip", z.ZodTypeAny, {
@@ -951,7 +980,10 @@ declare const imageInputSchema: z.ZodObject<{
         path: string;
         scale?: number | undefined;
     }[] | undefined;
+    seed?: number | undefined;
     lora?: string | undefined;
+    width?: number | undefined;
+    height?: number | undefined;
     tier?: "fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | undefined;
     override?: {
         provider?: string | undefined;
@@ -964,8 +996,11 @@ declare const imageInputSchema: z.ZodObject<{
         transport: "http" | "subprocess";
     })[] | undefined;
     labels?: Record<string, string> | undefined;
-    width?: number | undefined;
-    height?: number | undefined;
+    finetune?: string | undefined;
+    finetuneStrength?: number | undefined;
+    referenceImages?: (string | Uint8Array<ArrayBuffer>)[] | undefined;
+    outputFormat?: "jpeg" | "png" | "webp" | undefined;
+    safetyTolerance?: number | undefined;
     retryOnBlack?: boolean | undefined;
 }, {
     prompt: string;
@@ -974,7 +1009,10 @@ declare const imageInputSchema: z.ZodObject<{
         path: string;
         scale?: number | undefined;
     }[] | undefined;
+    seed?: number | undefined;
     lora?: string | undefined;
+    width?: number | undefined;
+    height?: number | undefined;
     tier?: "fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | undefined;
     override?: {
         provider?: string | undefined;
@@ -987,8 +1025,11 @@ declare const imageInputSchema: z.ZodObject<{
         transport: "http" | "subprocess";
     })[] | undefined;
     labels?: Record<string, string> | undefined;
-    width?: number | undefined;
-    height?: number | undefined;
+    finetune?: string | undefined;
+    finetuneStrength?: number | undefined;
+    referenceImages?: (string | Uint8Array<ArrayBuffer>)[] | undefined;
+    outputFormat?: "jpeg" | "png" | "webp" | undefined;
+    safetyTolerance?: number | undefined;
     retryOnBlack?: boolean | undefined;
 }>;
 declare const trainStyleInputSchema: z.ZodObject<{
@@ -1682,6 +1723,18 @@ interface FalAdapterConfig {
 }
 declare function falAdapter(config?: FalAdapterConfig): ProviderAdapter;
 
+interface BflAdapterConfig {
+    apiKey?: string;
+    /** EU-resident base. Do NOT point this at api.bfl.ai (global, US-failover) — see header. */
+    baseUrl?: string;
+    pollIntervalMs?: number;
+    timeoutMs?: number;
+    fetch?: typeof fetch;
+    /** Override the per-image USD price (else the built-in finetuned-inference estimate). */
+    pricePerImage?: number;
+}
+declare function bflAdapter(config?: BflAdapterConfig): ProviderAdapter;
+
 interface OpenAICompatibleConfig {
     /** Provider name recorded on Usage (e.g. "openai", "deepinfra", "openrouter"). */
     name: string;
@@ -1715,8 +1768,8 @@ declare const falStubAdapter: ProviderAdapter;
  *  wires the live adapters. */
 declare const stubProviders: Record<string, ProviderAdapter>;
 
-declare const VERSION: "0.13.1";
-declare const SDK_TAG: "@broberg/ai-sdk@0.13.1";
+declare const VERSION: "0.15.0";
+declare const SDK_TAG: "@broberg/ai-sdk@0.15.0";
 
 /** 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
@@ -1942,4 +1995,4 @@ interface StreamTransportRequest extends TransportRequest {
  */
 declare function streamTransport(req: StreamTransportRequest): AsyncIterable<string>;
 
-export { type AiClient, type AiConfig, type BatchJob, type BatchRequestItem, type BatchResultItem, type BudgetConfig, BudgetExceededError, BudgetGuard, type BudgetStore, type CallOptions, type Capability, type ChatInput, type ChatRequest, type ChatResult, type ChatStreamEvent, type ClassifyInput, type ClassifyResult, type ContentPart, type Contracts, type CostSink, type CostSummary, DEFAULT_TIER_MAP, type DesignInput, type DesignResult, type DialogueRequest, type DialogueTurn, type DiscordSinkConfig, ELEVENLABS_DANISH_VOICES, type EmbeddingInput, type EmbeddingRequest, type EmbeddingResult, type ExtractInput, type ExtractResult, type FalAdapterConfig, type HttpResponse, type ImageInput, type ImageRequest, type ImageResult, type LoraWeight, type Message, type MockupInput, type MockupResult, type ModerationInput, type ModerationItem, type ModerationRequest, type ModerationResult, type OcrInput, type OcrPage, type OcrRequest, type OcrResult, type OpenAICompatibleConfig, type PodcastInput, type PodcastResult, type PricingEntry, type ProviderAdapter, type RefreshOptions, type RefreshResult, type RerankInput, type RerankResult, type Role, SDK_TAG, type SqliteBudgetStoreConfig, type SqliteSinkConfig, StreamHttpError, type SubprocessResponse, type Tier, type TierSpec, type Tool, type ToolCall, type TrainStyleInput, type TrainStyleRequest, type TrainStyleResult, type TranscribeInput, type TranscribeRequest, type TranscribeResult, type TranslateInput, type TranslateResult, type Transport, type TransportRequest, type TransportResponse, type TtsInput, type TtsRequest, type UpmetricsSinkConfig, type Usage, VERSION, type VideoInput, type VisionInput, aiConfigSchema, anthropicAdapter, anthropicApiAdapter, anthropicSubprocessAdapter, chatInputSchema, computeCost, createAI, deepinfraAdapter, defaultProviders, discordSink, elevenlabsAdapter, embeddingInputSchema, falAdapter, falStubAdapter, freshUsage, fromProviderToolCall, geminiAdapter, getCostSummary, getPrice, httpTransport, imageInputSchema, makeContracts, makeOpenAICompatibleAdapter, messageSchema, mistralAdapter, multiSink, noopSink, openaiAdapter, openaiStubAdapter, openrouterAdapter, parseClaudeCliJson, parseJsonLoose, refreshAvailability, resetRefreshState, resetRegistry, resolveTier, resolveVoice, sqliteBudgetStore, sqliteSink, streamTransport, stubProviders, subprocessTransport, tierSpecSchema, toProviderTools, toolSchema, translateInputSchema, upmetricsSink, visionInputSchema };
+export { type AiClient, type AiConfig, type BatchJob, type BatchRequestItem, type BatchResultItem, type BflAdapterConfig, type BudgetConfig, BudgetExceededError, BudgetGuard, type BudgetStore, type CallOptions, type Capability, type ChatInput, type ChatRequest, type ChatResult, type ChatStreamEvent, type ClassifyInput, type ClassifyResult, type ContentPart, type Contracts, type CostSink, type CostSummary, DEFAULT_TIER_MAP, type DesignInput, type DesignResult, type DialogueRequest, type DialogueTurn, type DiscordSinkConfig, ELEVENLABS_DANISH_VOICES, type EmbeddingInput, type EmbeddingRequest, type EmbeddingResult, type ExtractInput, type ExtractResult, type FalAdapterConfig, type HttpResponse, type ImageInput, type ImageRequest, type ImageResult, type LoraWeight, type Message, type MockupInput, type MockupResult, type ModerationInput, type ModerationItem, type ModerationRequest, type ModerationResult, type OcrInput, type OcrPage, type OcrRequest, type OcrResult, type OpenAICompatibleConfig, type PodcastInput, type PodcastResult, type PricingEntry, type ProviderAdapter, type RefreshOptions, type RefreshResult, type RerankInput, type RerankResult, type Role, SDK_TAG, type SqliteBudgetStoreConfig, type SqliteSinkConfig, StreamHttpError, type SubprocessResponse, type Tier, type TierSpec, type Tool, type ToolCall, type TrainStyleInput, type TrainStyleRequest, type TrainStyleResult, type TranscribeInput, type TranscribeRequest, type TranscribeResult, type TranslateInput, type TranslateResult, type Transport, type TransportRequest, type TransportResponse, type TtsInput, type TtsRequest, type UpmetricsSinkConfig, type Usage, VERSION, type VideoInput, type VisionInput, aiConfigSchema, anthropicAdapter, anthropicApiAdapter, anthropicSubprocessAdapter, bflAdapter, chatInputSchema, computeCost, createAI, deepinfraAdapter, defaultProviders, discordSink, elevenlabsAdapter, embeddingInputSchema, falAdapter, falStubAdapter, freshUsage, fromProviderToolCall, geminiAdapter, getCostSummary, getPrice, httpTransport, imageInputSchema, makeContracts, makeOpenAICompatibleAdapter, messageSchema, mistralAdapter, multiSink, noopSink, openaiAdapter, openaiStubAdapter, openrouterAdapter, parseClaudeCliJson, parseJsonLoose, refreshAvailability, resetRefreshState, resetRegistry, resolveTier, resolveVoice, sqliteBudgetStore, sqliteSink, streamTransport, stubProviders, subprocessTransport, tierSpecSchema, toProviderTools, toolSchema, translateInputSchema, upmetricsSink, visionInputSchema };

package/dist/index.js

@@ -1573,6 +1573,100 @@ function buildZip(files) {
   return Buffer.concat([...parts, cd, eocd]);
 }
 
+// src/providers/bfl.ts
+var EU_BASE = "https://api.eu.bfl.ai";
+var BFL_CREDIT_USD = 0.01;
+var BFL_IMAGE_PRICE = 0.06;
+var sleep2 = (ms) => new Promise((r) => setTimeout(r, ms));
+function gcd(a, b) {
+  return b === 0 ? a : gcd(b, a % b);
+}
+function bflAdapter(config = {}) {
+  const doFetch = config.fetch ?? fetch;
+  const base = config.baseUrl ?? EU_BASE;
+  const pollIntervalMs = config.pollIntervalMs ?? 1500;
+  const timeoutMs = config.timeoutMs ?? 12e4;
+  const resolveKey = () => config.apiKey ?? process.env.BFL_API_KEY;
+  async function image(req) {
+    const apiKey = resolveKey();
+    if (!apiKey) throw new Error("bfl adapter: BFL_API_KEY not set");
+    const headers = { "content-type": "application/json", "x-key": apiKey };
+    const body = { prompt: req.prompt };
+    if (req.referenceImages?.length) {
+      req.referenceImages.forEach((img, i) => {
+        body[i === 0 ? "input_image" : `input_image_${i + 1}`] = toBflImage(img);
+      });
+      if (req.width) body.width = req.width;
+      if (req.height) body.height = req.height;
+      if (req.seed !== void 0) body.seed = req.seed;
+      body.output_format = req.outputFormat ?? "jpeg";
+      body.safety_tolerance = req.safetyTolerance ?? 2;
+    } else if (req.finetune) {
+      body.finetune_id = req.finetune;
+      if (req.finetuneStrength !== void 0) body.finetune_strength = req.finetuneStrength;
+      if (req.width && req.height) {
+        const g = gcd(req.width, req.height) || 1;
+        body.aspect_ratio = `${req.width / g}:${req.height / g}`;
+      }
+    } else {
+      throw new Error(
+        "bfl adapter: requires referenceImages (FLUX 2 multi-reference) or a finetune id. ai.image({ referenceImages: [...] }) needs no training; ai.image({ finetune }) uses a subject trained once in the BFL dashboard (dashboard.bfl.ai \u2014 finetune-create is not in the public API)."
+      );
+    }
+    const submitRes = await doFetch(`${base}/v1/${req.spec.model}`, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(body)
+    });
+    if (!submitRes.ok) {
+      throw new Error(`bfl ${submitRes.status}: ${(await submitRes.text().catch(() => "")).slice(0, 200)}`);
+    }
+    const submit = await submitRes.json();
+    if (!submit.id) throw new Error("bfl: no task id in submit response");
+    const sample = await poll(submit.id, apiKey);
+    const usage = freshUsage({
+      provider: "bfl",
+      model: req.spec.model,
+      transport: "http",
+      capability: "image",
+      inputTokens: 0,
+      outputTokens: 0
+    });
+    usage.costUsd = typeof submit.cost === "number" ? submit.cost * BFL_CREDIT_USD : config.pricePerImage ?? BFL_IMAGE_PRICE;
+    return { url: sample, usage };
+  }
+  function toBflImage(img) {
+    if (typeof img !== "string") return Buffer.from(img).toString("base64");
+    if (/^https?:\/\//i.test(img)) return img;
+    const comma = img.startsWith("data:") ? img.indexOf(",") : -1;
+    return comma >= 0 ? img.slice(comma + 1) : img;
+  }
+  async function poll(id, apiKey) {
+    const headers = { "x-key": apiKey };
+    const deadline = Date.now() + timeoutMs;
+    for (; ; ) {
+      const res = await doFetch(`${base}/v1/get_result?id=${encodeURIComponent(id)}`, { headers });
+      if (!res.ok) throw new Error(`bfl get_result ${res.status}`);
+      const data = await res.json();
+      const status = data.status ?? "";
+      if (status === "Ready") {
+        const sample = data.result?.sample;
+        if (!sample) throw new Error("bfl: status Ready but no result.sample url");
+        return sample;
+      }
+      if (status === "Error" || status === "Failed" || status === "Task not found") {
+        throw new Error(`bfl: generation ${status}`);
+      }
+      if (status.includes("Moderated")) {
+        throw new Error(`bfl: ${status} \u2014 prompt or output flagged by BFL safety`);
+      }
+      if (Date.now() >= deadline) throw new Error(`bfl: timed out after ${timeoutMs}ms`);
+      await sleep2(pollIntervalMs);
+    }
+  }
+  return { name: "bfl", image };
+}
+
 // src/providers/registry.ts
 var defaultProviders = {
   anthropic: anthropicAdapter(),
@@ -1582,7 +1676,8 @@ var defaultProviders = {
   openrouter: openrouterAdapter(),
   mistral: mistralAdapter(),
   elevenlabs: elevenlabsAdapter(),
-  fal: falAdapter()
+  fal: falAdapter(),
+  bfl: bflAdapter()
 };
 
 // src/cost/budget.ts
@@ -1894,6 +1989,20 @@ var imageInputSchema = z.object({
   loras: z.array(loraWeightSchema).optional(),
   /** Shorthand for a single LoRA — normalized to loras:[{path, scale:1}]. */
   lora: z.string().optional(),
+  /** F023 — a BFL finetune id (subject trained in the BFL dashboard). Routes to the
+   *  EU-resident BFL finetuned-portrait endpoint. */
+  finetune: z.string().optional(),
+  /** F023 — BFL finetune_strength (~0–2; higher = stronger likeness). */
+  finetuneStrength: z.number().min(0).max(2).optional(),
+  /** F023.5 — 1–8 reference photos (URL or raw bytes) → EU-resident BFL FLUX 2
+   *  multi-reference generation (likeness, no training step). */
+  referenceImages: z.array(z.union([z.string(), z.instanceof(Uint8Array)])).min(1).max(8).optional(),
+  /** F023.5 — fixed seed for reproducible output (BFL). */
+  seed: z.number().int().optional(),
+  /** F023.5 — output container (BFL FLUX 2). Default jpeg. */
+  outputFormat: z.enum(["jpeg", "png", "webp"]).optional(),
+  /** F023.5 — BFL content-moderation strictness, 0 (strict) … 6 (lax). Default 2. */
+  safetyTolerance: z.number().int().min(0).max(6).optional(),
   /** F021.4 — re-roll once if fal returns a black image (NSFW false-positive). */
   retryOnBlack: z.boolean().optional(),
   ...callOptions
@@ -1976,6 +2085,16 @@ var DEFAULT_TRAINSTYLE_SPEC = {
   model: "fal-ai/flux-lora-fast-training",
   transport: "http"
 };
+var DEFAULT_BFL_FINETUNE_SPEC = {
+  provider: "bfl",
+  model: "flux-pro-1.1-ultra-finetuned",
+  transport: "http"
+};
+var DEFAULT_BFL_REFERENCE_SPEC = {
+  provider: "bfl",
+  model: "flux-2-max",
+  transport: "http"
+};
 var DEFAULT_OCR_SPEC = { provider: "mistral", model: "mistral-ocr-latest", transport: "http" };
 var DEFAULT_MODERATION_SPEC = { provider: "mistral", model: "mistral-moderation-latest", transport: "http" };
 var DEFAULT_PODCAST_SPEC = { provider: "elevenlabs", model: "eleven_v3", transport: "http" };
@@ -2222,7 +2341,7 @@ function createAI(config = {}) {
         ...input.loras ?? [],
         ...input.lora ? [{ path: input.lora }] : []
       ];
-      const base = loras.length > 0 ? DEFAULT_LORA_IMAGE_SPEC : DEFAULT_IMAGE_SPEC;
+      const base = input.referenceImages?.length ? DEFAULT_BFL_REFERENCE_SPEC : input.finetune ? DEFAULT_BFL_FINETUNE_SPEC : loras.length > 0 ? DEFAULT_LORA_IMAGE_SPEC : DEFAULT_IMAGE_SPEC;
       return runCapability({
         primary: { ...base, ...input.override },
         fallback: input.fallback,
@@ -2241,6 +2360,12 @@ function createAI(config = {}) {
             width: input.width,
             height: input.height,
             loras: loras.length ? loras : void 0,
+            finetune: input.finetune,
+            finetuneStrength: input.finetuneStrength,
+            referenceImages: input.referenceImages,
+            seed: input.seed,
+            outputFormat: input.outputFormat,
+            safetyTolerance: input.safetyTolerance,
             retryOnBlack: input.retryOnBlack
           });
         }
@@ -2496,8 +2621,8 @@ var stubProviders = {
 };
 
 // src/version.ts
-var VERSION = "0.13.1";
-var SDK_TAG = "@broberg/ai-sdk@0.13.1";
+var VERSION = "0.15.0";
+var SDK_TAG = "@broberg/ai-sdk@0.15.0";
 
 // src/availability/refresh.ts
 var NOT_REFRESHED = { refreshed: false, checked: 0, markedUnavailable: [] };
@@ -2783,6 +2908,7 @@ export {
   anthropicAdapter,
   anthropicApiAdapter,
   anthropicSubprocessAdapter,
+  bflAdapter,
   chatInputSchema,
   computeCost,
   createAI,