@broberg/ai-sdk 0.14.0 → 0.16.0

package/dist/index.d.ts

@@ -170,6 +170,16 @@ interface ImageRequest {
     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;
@@ -952,6 +962,15 @@ declare const imageInputSchema: z.ZodObject<{
     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, {
@@ -961,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;
@@ -974,10 +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;
@@ -986,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;
@@ -999,10 +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<{
@@ -1706,6 +1733,19 @@ interface BflAdapterConfig {
     /** Override the per-image USD price (else the built-in finetuned-inference estimate). */
     pricePerImage?: number;
 }
+/** Remaining BFL account credits + their USD value (1 credit = $0.01). */
+interface BflCredits {
+    credits: number;
+    usd: number;
+}
+/** F023.6 — check the BFL account's remaining credit balance (`GET /v1/credits`).
+ *  Account-level (region-independent); EU-pinned by default for consistency. Use it
+ *  for budget-gating before a generate call: `if ((await bflCredits()).usd < 1) …`. */
+declare function bflCredits(opts?: {
+    apiKey?: string;
+    baseUrl?: string;
+    fetch?: typeof fetch;
+}): Promise<BflCredits>;
 declare function bflAdapter(config?: BflAdapterConfig): ProviderAdapter;
 
 interface OpenAICompatibleConfig {
@@ -1741,8 +1781,8 @@ declare const falStubAdapter: ProviderAdapter;
  *  wires the live adapters. */
 declare const stubProviders: Record<string, ProviderAdapter>;
 
-declare const VERSION: "0.14.0";
-declare const SDK_TAG: "@broberg/ai-sdk@0.14.0";
+declare const VERSION: "0.16.0";
+declare const SDK_TAG: "@broberg/ai-sdk@0.16.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
@@ -1968,4 +2008,4 @@ interface StreamTransportRequest extends TransportRequest {
  */
 declare function streamTransport(req: StreamTransportRequest): AsyncIterable<string>;
 
-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 };
+export { type AiClient, type AiConfig, type BatchJob, type BatchRequestItem, type BatchResultItem, type BflAdapterConfig, type BflCredits, 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, bflCredits, 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

@@ -1575,11 +1575,24 @@ function buildZip(files) {
 
 // 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);
 }
+async function bflCredits(opts = {}) {
+  const apiKey = opts.apiKey ?? process.env.BFL_API_KEY;
+  if (!apiKey) throw new Error("bflCredits: BFL_API_KEY not set");
+  const doFetch = opts.fetch ?? fetch;
+  const res = await doFetch(`${opts.baseUrl ?? EU_BASE}/v1/credits`, { headers: { "x-key": apiKey } });
+  if (!res.ok) {
+    throw new Error(`bflCredits ${res.status}: ${(await res.text().catch(() => "")).slice(0, 200)}`);
+  }
+  const data = await res.json();
+  const credits = typeof data.credits === "number" ? data.credits : 0;
+  return { credits, usd: credits * BFL_CREDIT_USD };
+}
 function bflAdapter(config = {}) {
   const doFetch = config.fetch ?? fetch;
   const base = config.baseUrl ?? EU_BASE;
@@ -1589,22 +1602,30 @@ function bflAdapter(config = {}) {
   async function image(req) {
     const apiKey = resolveKey();
     if (!apiKey) throw new Error("bfl adapter: BFL_API_KEY not set");
-    if (!req.finetune) {
+    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 a finetune id \u2014 call ai.image({ finetune, override: { provider: 'bfl' } }). Train the subject once in the BFL dashboard (dashboard.bfl.ai) \u2014 finetune-create is not in the public API."
+        "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 headers = { "content-type": "application/json", "x-key": apiKey };
-    const body = {
-      finetune_id: req.finetune,
-      prompt: req.prompt
-    };
-    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}`;
-    }
-    const submitRes = await doFetch(`${base}/v1/flux-pro-1.1-ultra-finetuned`, {
+    const submitRes = await doFetch(`${base}/v1/${req.spec.model}`, {
       method: "POST",
       headers,
       body: JSON.stringify(body)
@@ -1623,9 +1644,15 @@ function bflAdapter(config = {}) {
       inputTokens: 0,
       outputTokens: 0
     });
-    usage.costUsd = config.pricePerImage ?? BFL_IMAGE_PRICE;
+    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;
@@ -1979,6 +2006,15 @@ var imageInputSchema = z.object({
   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
@@ -2066,6 +2102,11 @@ var DEFAULT_BFL_FINETUNE_SPEC = {
   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" };
@@ -2312,7 +2353,7 @@ function createAI(config = {}) {
         ...input.loras ?? [],
         ...input.lora ? [{ path: input.lora }] : []
       ];
-      const base = input.finetune ? DEFAULT_BFL_FINETUNE_SPEC : 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,
@@ -2333,6 +2374,10 @@ function createAI(config = {}) {
             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
           });
         }
@@ -2588,8 +2633,8 @@ var stubProviders = {
 };
 
 // src/version.ts
-var VERSION = "0.14.0";
-var SDK_TAG = "@broberg/ai-sdk@0.14.0";
+var VERSION = "0.16.0";
+var SDK_TAG = "@broberg/ai-sdk@0.16.0";
 
 // src/availability/refresh.ts
 var NOT_REFRESHED = { refreshed: false, checked: 0, markedUnavailable: [] };
@@ -2876,6 +2921,7 @@ export {
   anthropicApiAdapter,
   anthropicSubprocessAdapter,
   bflAdapter,
+  bflCredits,
   chatInputSchema,
   computeCost,
   createAI,