@broberg/ai-sdk 0.9.1 → 0.10.1

package/dist/index.d.ts

@@ -13,7 +13,7 @@ interface TierSpec {
     transport: Transport;
 }
 /** High-level capability a call exercises. Mirrors the capability layer (F5). */
-type Capability = "chat" | "vision" | "video" | "translate" | "image" | "embedding" | "transcribe" | "ocr" | "moderation" | "podcast" | "tts" | "mockup" | "design" | "extract" | "classify" | "rerank";
+type Capability = "chat" | "vision" | "video" | "translate" | "image" | "embedding" | "transcribe" | "ocr" | "moderation" | "podcast" | "tts" | "trainStyle" | "mockup" | "design" | "extract" | "classify" | "rerank";
 type Role = "system" | "user" | "assistant" | "tool";
 /** A piece of message content. Text everywhere; image parts feed vision. */
 type ContentPart = {
@@ -150,16 +150,44 @@ type ChatStreamEvent = {
     message: string;
     status?: number;
 };
+/** A trained LoRA to merge at inference time (F021). */
+interface LoraWeight {
+    /** URL (or fal path) to the LoRA weights. */
+    path: string;
+    /** Scales the LoRA before merging (default 1). */
+    scale?: number;
+}
 interface ImageRequest {
     prompt: string;
     spec: TierSpec;
     width?: number;
     height?: number;
+    /** LoRAs to merge at inference (F021) — e.g. a trained brand/style LoRA. */
+    loras?: LoraWeight[];
 }
 interface ImageResult {
     url: string;
     usage: Usage;
 }
+/** Style/brand LoRA training (F021) — fal fal-ai/flux-lora-fast-training. */
+interface TrainStyleRequest {
+    /** A hosted archive URL, or an array of image URLs the SDK zips in-memory. */
+    images: string | string[];
+    spec: TierSpec;
+    /** Style LoRA (disables captioning/masks). Default true. */
+    isStyle?: boolean;
+    triggerWord?: string;
+    /** Training steps (~1000 typical). */
+    steps?: number;
+    createMasks?: boolean;
+}
+interface TrainStyleResult {
+    /** URL to the trained LoRA weights — pass to ai.image({ lora }). */
+    loraUrl: string;
+    /** URL to the training config file. */
+    configUrl: string;
+    usage: Usage;
+}
 interface EmbeddingRequest {
     input: string[];
     spec: TierSpec;
@@ -257,6 +285,8 @@ interface ProviderAdapter {
     chatStream?(req: ChatRequest): AsyncIterable<ChatStreamEvent>;
     vision?(req: ChatRequest): Promise<ChatResult>;
     image?(req: ImageRequest): Promise<ImageResult>;
+    /** Train a style/brand LoRA from images (F021). fal. */
+    trainStyle?(req: TrainStyleRequest): Promise<TrainStyleResult>;
     embedding?(req: EmbeddingRequest): Promise<EmbeddingResult>;
     transcribe?(req: TranscribeRequest): Promise<TranscribeResult>;
     ocr?(req: OcrRequest): Promise<OcrResult>;
@@ -886,9 +916,27 @@ declare const imageInputSchema: z.ZodObject<{
     prompt: z.ZodString;
     width: z.ZodOptional<z.ZodNumber>;
     height: z.ZodOptional<z.ZodNumber>;
+    /** LoRAs to merge at inference (F021). */
+    loras: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        path: z.ZodString;
+        scale: z.ZodOptional<z.ZodNumber>;
+    }, "strip", z.ZodTypeAny, {
+        path: string;
+        scale?: number | undefined;
+    }, {
+        path: string;
+        scale?: number | undefined;
+    }>, "many">>;
+    /** Shorthand for a single LoRA — normalized to loras:[{path, scale:1}]. */
+    lora: z.ZodOptional<z.ZodString>;
 }, "strip", z.ZodTypeAny, {
     prompt: string;
     purpose?: string | undefined;
+    loras?: {
+        path: string;
+        scale?: number | undefined;
+    }[] | undefined;
+    lora?: string | undefined;
     tier?: "fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | undefined;
     override?: {
         provider?: string | undefined;
@@ -906,6 +954,11 @@ declare const imageInputSchema: z.ZodObject<{
 }, {
     prompt: string;
     purpose?: string | undefined;
+    loras?: {
+        path: string;
+        scale?: number | undefined;
+    }[] | undefined;
+    lora?: string | undefined;
     tier?: "fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | undefined;
     override?: {
         provider?: string | undefined;
@@ -921,6 +974,82 @@ declare const imageInputSchema: z.ZodObject<{
     width?: number | undefined;
     height?: number | undefined;
 }>;
+declare const trainStyleInputSchema: z.ZodObject<{
+    tier: z.ZodOptional<z.ZodEnum<["fast", "smart", "powerful", "cheap", "vision", "video", "embedding"]>>;
+    override: z.ZodOptional<z.ZodObject<{
+        provider: z.ZodOptional<z.ZodString>;
+        model: z.ZodOptional<z.ZodString>;
+        transport: z.ZodOptional<z.ZodEnum<["http", "subprocess"]>>;
+    }, "strip", z.ZodTypeAny, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        transport?: "http" | "subprocess" | undefined;
+    }, {
+        provider?: string | undefined;
+        model?: string | undefined;
+        transport?: "http" | "subprocess" | undefined;
+    }>>;
+    fallback: z.ZodOptional<z.ZodArray<z.ZodUnion<[z.ZodEnum<["fast", "smart", "powerful", "cheap", "vision", "video", "embedding"]>, z.ZodObject<{
+        provider: z.ZodString;
+        model: z.ZodString;
+        transport: z.ZodEnum<["http", "subprocess"]>;
+    }, "strip", z.ZodTypeAny, {
+        provider: string;
+        model: string;
+        transport: "http" | "subprocess";
+    }, {
+        provider: string;
+        model: string;
+        transport: "http" | "subprocess";
+    }>]>, "many">>;
+    purpose: z.ZodOptional<z.ZodString>;
+    labels: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+    /** A hosted archive URL, or an array of image URLs the SDK zips in-memory. */
+    images: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
+    /** Style LoRA (disables captioning/masks). Default true. */
+    isStyle: z.ZodOptional<z.ZodBoolean>;
+    triggerWord: z.ZodOptional<z.ZodString>;
+    steps: z.ZodOptional<z.ZodNumber>;
+    createMasks: z.ZodOptional<z.ZodBoolean>;
+}, "strip", z.ZodTypeAny, {
+    images: string | string[];
+    purpose?: string | undefined;
+    steps?: number | undefined;
+    tier?: "fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | undefined;
+    override?: {
+        provider?: string | undefined;
+        model?: string | undefined;
+        transport?: "http" | "subprocess" | undefined;
+    } | undefined;
+    fallback?: ("fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | {
+        provider: string;
+        model: string;
+        transport: "http" | "subprocess";
+    })[] | undefined;
+    labels?: Record<string, string> | undefined;
+    isStyle?: boolean | undefined;
+    triggerWord?: string | undefined;
+    createMasks?: boolean | undefined;
+}, {
+    images: string | string[];
+    purpose?: string | undefined;
+    steps?: number | undefined;
+    tier?: "fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | undefined;
+    override?: {
+        provider?: string | undefined;
+        model?: string | undefined;
+        transport?: "http" | "subprocess" | undefined;
+    } | undefined;
+    fallback?: ("fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | {
+        provider: string;
+        model: string;
+        transport: "http" | "subprocess";
+    })[] | undefined;
+    labels?: Record<string, string> | undefined;
+    isStyle?: boolean | undefined;
+    triggerWord?: string | undefined;
+    createMasks?: boolean | undefined;
+}>;
 declare const embeddingInputSchema: z.ZodObject<{
     tier: z.ZodOptional<z.ZodEnum<["fast", "smart", "powerful", "cheap", "vision", "video", "embedding"]>>;
     override: z.ZodOptional<z.ZodObject<{
@@ -1386,6 +1515,7 @@ type VisionInput = z.infer<typeof visionInputSchema>;
 type VideoInput = z.infer<typeof videoInputSchema>;
 type TranslateInput = z.infer<typeof translateInputSchema>;
 type ImageInput = z.infer<typeof imageInputSchema>;
+type TrainStyleInput = z.infer<typeof trainStyleInputSchema>;
 type EmbeddingInput = z.infer<typeof embeddingInputSchema>;
 type TranscribeInput = z.infer<typeof transcribeInputSchema>;
 type OcrInput = z.infer<typeof ocrInputSchema>;
@@ -1404,6 +1534,8 @@ interface AiClient {
     video(input: VideoInput): Promise<ChatResult>;
     translate(input: TranslateInput): Promise<TranslateResult>;
     image(input: ImageInput): Promise<ImageResult>;
+    /** Train a style/brand LoRA from images (F021) → { loraUrl, configUrl }. fal. */
+    trainStyle(input: TrainStyleInput): Promise<TrainStyleResult>;
     embedding(input: EmbeddingInput): Promise<EmbeddingResult>;
     transcribe(input: TranscribeInput): Promise<TranscribeResult>;
     /** OCR (F016.2) — document/image → structured markdown, billed per page. Mistral. */
@@ -1505,9 +1637,13 @@ interface FalAdapterConfig {
     queueBaseUrl?: string;
     pollIntervalMs?: number;
     timeoutMs?: number;
+    /** Deadline for training jobs — they take minutes (default 600000 = 10 min). */
+    trainTimeoutMs?: number;
     fetch?: typeof fetch;
     /** Override the per-image USD price (else a built-in estimate per model, 0 if unknown). */
     pricePerImage?: number;
+    /** Override the flat per-training USD price (else ~$2 estimate). */
+    pricePerTraining?: number;
 }
 declare function falAdapter(config?: FalAdapterConfig): ProviderAdapter;
 
@@ -1544,8 +1680,8 @@ declare const falStubAdapter: ProviderAdapter;
  *  wires the live adapters. */
 declare const stubProviders: Record<string, ProviderAdapter>;
 
-declare const VERSION: "0.9.1";
-declare const SDK_TAG: "@broberg/ai-sdk@0.9.1";
+declare const VERSION: "0.10.1";
+declare const SDK_TAG: "@broberg/ai-sdk@0.10.1";
 
 /** 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
@@ -1738,4 +1874,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 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 RerankInput, type RerankResult, type Role, SDK_TAG, type SqliteBudgetStoreConfig, type SqliteSinkConfig, StreamHttpError, type SubprocessResponse, type Tier, type TierSpec, type Tool, type ToolCall, 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, 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 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 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, resolveTier, resolveVoice, sqliteBudgetStore, sqliteSink, streamTransport, stubProviders, subprocessTransport, tierSpecSchema, toProviderTools, toolSchema, translateInputSchema, upmetricsSink, visionInputSchema };

package/dist/index.js

@@ -1284,12 +1284,15 @@ function elevenlabsAdapter(config = {}) {
 }
 
 // src/providers/fal.ts
+import { deflateRawSync, crc32 } from "zlib";
 var FAL_IMAGE_PRICE_ESTIMATE = {
   "fal-ai/flux/schnell": 3e-3,
   "fal-ai/flux/dev": 0.025,
+  "fal-ai/flux-lora": 0.025,
   "fal-ai/flux-pro": 0.05,
   "fal-ai/flux-pro/v1.1": 0.04
 };
+var FAL_TRAIN_PRICE_ESTIMATE = 2;
 var sleep = (ms) => new Promise((r) => setTimeout(r, ms));
 function falAdapter(config = {}) {
   const doFetch = config.fetch ?? fetch;
@@ -1297,14 +1300,22 @@ function falAdapter(config = {}) {
   const queueBase = config.queueBaseUrl ?? "https://queue.fal.run";
   const pollIntervalMs = config.pollIntervalMs ?? 2e3;
   const timeoutMs = config.timeoutMs ?? 6e4;
+  const resolveKey = () => config.apiKey ?? process.env.FAL_KEY ?? process.env.FAL_API_KEY;
+  const authHeaders = (apiKey) => ({
+    "content-type": "application/json",
+    Authorization: `Key ${apiKey}`
+  });
   async function image(req) {
-    const apiKey = config.apiKey ?? process.env.FAL_KEY;
+    const apiKey = resolveKey();
     if (!apiKey) throw new Error("fal adapter: FAL_KEY not set");
-    const headers = { "content-type": "application/json", Authorization: `Key ${apiKey}` };
+    const headers = authHeaders(apiKey);
     const body = { prompt: req.prompt };
     if (req.width !== void 0 && req.height !== void 0) {
       body.image_size = { width: req.width, height: req.height };
     }
+    if (req.loras && req.loras.length > 0) {
+      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 usage = freshUsage({
@@ -1318,6 +1329,54 @@ function falAdapter(config = {}) {
     usage.costUsd = config.pricePerImage ?? FAL_IMAGE_PRICE_ESTIMATE[req.spec.model] ?? 0;
     return { url, usage };
   }
+  async function trainStyle(req) {
+    const apiKey = resolveKey();
+    if (!apiKey) throw new Error("fal adapter: FAL_KEY not set");
+    const headers = authHeaders(apiKey);
+    const body = {
+      images_data_url: await resolveImagesDataUrl(req.images),
+      is_style: req.isStyle ?? true
+    };
+    if (req.triggerWord !== void 0) body.trigger_word = req.triggerWord;
+    if (req.steps !== void 0) body.steps = req.steps;
+    if (req.createMasks !== void 0) body.create_masks = req.createMasks;
+    const result = await queueResult(
+      req.spec.model,
+      headers,
+      body,
+      config.trainTimeoutMs ?? 6e5
+    );
+    const { loraUrl, configUrl } = extractTrainedFiles(result);
+    if (!loraUrl) {
+      throw new Error(
+        `fal trainStyle: no LoRA file url in result \u2014 fal returned keys [${Object.keys(
+          result ?? {}
+        ).join(", ")}]: ${JSON.stringify(result).slice(0, 800)}`
+      );
+    }
+    const usage = freshUsage({
+      provider: "fal",
+      model: req.spec.model,
+      transport: "http",
+      capability: "trainStyle",
+      inputTokens: 0,
+      outputTokens: 0
+    });
+    usage.costUsd = config.pricePerTraining ?? FAL_TRAIN_PRICE_ESTIMATE;
+    return { loraUrl, configUrl: configUrl ?? "", usage };
+  }
+  async function resolveImagesDataUrl(images) {
+    if (typeof images === "string") return images;
+    const files = await Promise.all(
+      images.map(async (url, i) => {
+        const res = await doFetch(url);
+        if (!res.ok) throw new Error(`fal trainStyle: failed to fetch image ${url} (${res.status})`);
+        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")}`;
+  }
   async function runSync(model, headers, body) {
     const res = await doFetch(`${syncBase}/${model}`, {
       method: "POST",
@@ -1333,34 +1392,133 @@ function falAdapter(config = {}) {
     return out;
   }
   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;
+  }
+  async function queueResult(model, headers, body, deadlineMs) {
     const submitRes = await doFetch(`${queueBase}/${model}`, {
       method: "POST",
       headers,
       body: JSON.stringify(body)
     });
     if (!submitRes.ok) {
-      throw new Error(`fal queue submit ${submitRes.status}`);
+      throw new Error(
+        `fal queue submit ${submitRes.status}: ${(await submitRes.text().catch(() => "")).slice(0, 200)}`
+      );
     }
     const submit = await submitRes.json();
     const statusUrl = submit.status_url;
     const responseUrl = submit.response_url;
     if (!statusUrl || !responseUrl) throw new Error("fal queue: missing status/response url");
-    const deadline = Date.now() + timeoutMs;
+    const deadline = Date.now() + deadlineMs;
     for (; ; ) {
       const statusRes = await doFetch(statusUrl, { headers });
       const status = await statusRes.json();
       if (status.status === "COMPLETED") break;
-      if (status.status === "FAILED") throw new Error("fal queue: generation FAILED");
-      if (Date.now() >= deadline) throw new Error(`fal queue: timed out after ${timeoutMs}ms`);
+      if (status.status === "FAILED") throw new Error("fal queue: job FAILED");
+      if (Date.now() >= deadline) throw new Error(`fal queue: timed out after ${deadlineMs}ms`);
       await sleep(pollIntervalMs);
     }
     const resultRes = await doFetch(responseUrl, { headers });
-    const result = await resultRes.json();
-    const out = result.images?.[0]?.url;
-    if (!out) throw new Error("fal queue: no image url in result");
-    return out;
+    if (!resultRes.ok) {
+      throw new Error(
+        `fal queue result ${resultRes.status}: ${(await resultRes.text().catch(() => "")).slice(0, 300)}`
+      );
+    }
+    return resultRes.json();
+  }
+  return { name: "fal", image, trainStyle };
+}
+function urlOf(v) {
+  if (typeof v === "string") return v;
+  if (v && typeof v === "object" && typeof v.url === "string") {
+    return v.url;
   }
-  return { name: "fal", image };
+  return void 0;
+}
+function deepFindUrl(obj, match) {
+  const stack = [obj];
+  while (stack.length) {
+    const cur = stack.pop();
+    if (typeof cur === "string") {
+      if (match(cur)) return cur;
+      continue;
+    }
+    if (cur && typeof cur === "object") {
+      const u = cur.url;
+      if (typeof u === "string" && match(u)) return u;
+      for (const v of Object.values(cur)) stack.push(v);
+    }
+  }
+  return void 0;
+}
+function extractTrainedFiles(result) {
+  const r = result;
+  const root = r?.data ?? r?.response ?? r?.output ?? r;
+  const loraUrl = urlOf(root?.diffusers_lora_file) ?? urlOf(root?.lora_file) ?? urlOf(root?.safetensors) ?? urlOf(root?.lora) ?? deepFindUrl(root, (u) => /\.safetensors(\?|$)/i.test(u));
+  const configUrl = urlOf(root?.config_file) ?? urlOf(root?.config) ?? deepFindUrl(root, (u) => /config[^/]*\.json(\?|$)/i.test(u));
+  return { loraUrl, configUrl };
+}
+function fileNameFromUrl(url, i) {
+  const base = url.split("?")[0].split("/").pop() || "";
+  return /\.[a-z0-9]+$/i.test(base) ? base : `image_${i}.png`;
+}
+function buildZip(files) {
+  const parts = [];
+  const central = [];
+  let offset = 0;
+  for (const f of files) {
+    const nameBuf = Buffer.from(f.name, "utf8");
+    const data = Buffer.from(f.data);
+    const comp = deflateRawSync(data);
+    const crc = crc32(data) >>> 0;
+    const lfh = Buffer.alloc(30);
+    lfh.writeUInt32LE(67324752, 0);
+    lfh.writeUInt16LE(20, 4);
+    lfh.writeUInt16LE(0, 6);
+    lfh.writeUInt16LE(8, 8);
+    lfh.writeUInt16LE(0, 10);
+    lfh.writeUInt16LE(33, 12);
+    lfh.writeUInt32LE(crc, 14);
+    lfh.writeUInt32LE(comp.length, 18);
+    lfh.writeUInt32LE(data.length, 22);
+    lfh.writeUInt16LE(nameBuf.length, 26);
+    lfh.writeUInt16LE(0, 28);
+    parts.push(lfh, nameBuf, comp);
+    const cdh = Buffer.alloc(46);
+    cdh.writeUInt32LE(33639248, 0);
+    cdh.writeUInt16LE(20, 4);
+    cdh.writeUInt16LE(20, 6);
+    cdh.writeUInt16LE(0, 8);
+    cdh.writeUInt16LE(8, 10);
+    cdh.writeUInt16LE(0, 12);
+    cdh.writeUInt16LE(33, 14);
+    cdh.writeUInt32LE(crc, 16);
+    cdh.writeUInt32LE(comp.length, 20);
+    cdh.writeUInt32LE(data.length, 24);
+    cdh.writeUInt16LE(nameBuf.length, 28);
+    cdh.writeUInt16LE(0, 30);
+    cdh.writeUInt16LE(0, 32);
+    cdh.writeUInt16LE(0, 34);
+    cdh.writeUInt16LE(0, 36);
+    cdh.writeUInt32LE(0, 38);
+    cdh.writeUInt32LE(offset, 42);
+    central.push(cdh, nameBuf);
+    offset += lfh.length + nameBuf.length + comp.length;
+  }
+  const cd = Buffer.concat(central);
+  const eocd = Buffer.alloc(22);
+  eocd.writeUInt32LE(101010256, 0);
+  eocd.writeUInt16LE(0, 4);
+  eocd.writeUInt16LE(0, 6);
+  eocd.writeUInt16LE(files.length, 8);
+  eocd.writeUInt16LE(files.length, 10);
+  eocd.writeUInt32LE(cd.length, 12);
+  eocd.writeUInt32LE(offset, 16);
+  eocd.writeUInt16LE(0, 20);
+  return Buffer.concat([...parts, cd, eocd]);
 }
 
 // src/providers/registry.ts
@@ -1665,10 +1823,28 @@ var translateInputSchema = z.object({
   from: z.string().optional(),
   ...callOptions
 });
+var loraWeightSchema = z.object({
+  path: z.string(),
+  scale: z.number().optional()
+});
 var imageInputSchema = z.object({
   prompt: z.string(),
   width: z.number().int().positive().optional(),
   height: z.number().int().positive().optional(),
+  /** LoRAs to merge at inference (F021). */
+  loras: z.array(loraWeightSchema).optional(),
+  /** Shorthand for a single LoRA — normalized to loras:[{path, scale:1}]. */
+  lora: z.string().optional(),
+  ...callOptions
+});
+var trainStyleInputSchema = z.object({
+  /** A hosted archive URL, or an array of image URLs the SDK zips in-memory. */
+  images: z.union([z.string(), z.array(z.string())]),
+  /** Style LoRA (disables captioning/masks). Default true. */
+  isStyle: z.boolean().optional(),
+  triggerWord: z.string().optional(),
+  steps: z.number().int().positive().optional(),
+  createMasks: z.boolean().optional(),
   ...callOptions
 });
 var embeddingInputSchema = z.object({
@@ -1724,6 +1900,16 @@ var DEFAULT_IMAGE_SPEC = {
   model: "fal-ai/flux/schnell",
   transport: "http"
 };
+var DEFAULT_LORA_IMAGE_SPEC = {
+  provider: "fal",
+  model: "fal-ai/flux-lora",
+  transport: "http"
+};
+var DEFAULT_TRAINSTYLE_SPEC = {
+  provider: "fal",
+  model: "fal-ai/flux-lora-fast-training",
+  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" };
@@ -1956,8 +2142,13 @@ function createAI(config = {}) {
     },
     async image(input) {
       input = imageInputSchema.parse(input);
+      const loras = [
+        ...input.loras ?? [],
+        ...input.lora ? [{ path: input.lora }] : []
+      ];
+      const base = loras.length > 0 ? DEFAULT_LORA_IMAGE_SPEC : DEFAULT_IMAGE_SPEC;
       return runCapability({
-        primary: { ...DEFAULT_IMAGE_SPEC, ...input.override },
+        primary: { ...base, ...input.override },
         fallback: input.fallback,
         capability: "image",
         purpose: input.purpose,
@@ -1968,7 +2159,39 @@ function createAI(config = {}) {
         invoke: async (spec) => {
           const adapter = pickProvider(spec.provider);
           if (!adapter.image) throw new Error(`createAI: provider "${spec.provider}" does not support image`);
-          return adapter.image({ prompt: input.prompt, spec, width: input.width, height: input.height });
+          return adapter.image({
+            prompt: input.prompt,
+            spec,
+            width: input.width,
+            height: input.height,
+            loras: loras.length ? loras : void 0
+          });
+        }
+      });
+    },
+    async trainStyle(input) {
+      input = trainStyleInputSchema.parse(input);
+      return runCapability({
+        primary: { ...DEFAULT_TRAINSTYLE_SPEC, ...input.override },
+        fallback: input.fallback,
+        capability: "trainStyle",
+        purpose: input.purpose,
+        labels: input.labels,
+        estIn: 0,
+        // training is priced flat by fal, not token-based
+        estOut: 0,
+        invoke: async (spec) => {
+          const adapter = pickProvider(spec.provider);
+          if (!adapter.trainStyle)
+            throw new Error(`createAI: provider "${spec.provider}" does not support trainStyle`);
+          return adapter.trainStyle({
+            images: input.images,
+            spec,
+            isStyle: input.isStyle,
+            triggerWord: input.triggerWord,
+            steps: input.steps,
+            createMasks: input.createMasks
+          });
         }
       });
     },
@@ -2196,8 +2419,8 @@ var stubProviders = {
 };
 
 // src/version.ts
-var VERSION = "0.9.1";
-var SDK_TAG = "@broberg/ai-sdk@0.9.1";
+var VERSION = "0.10.1";
+var SDK_TAG = "@broberg/ai-sdk@0.10.1";
 
 // src/cost/budget-store.ts
 function sqliteBudgetStore(config) {