@broberg/ai-sdk 0.10.0 → 0.10.2

package/dist/index.d.ts

@@ -936,6 +936,7 @@ declare const imageInputSchema: z.ZodObject<{
         path: string;
         scale?: number | undefined;
     }[] | undefined;
+    lora?: string | undefined;
     tier?: "fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | undefined;
     override?: {
         provider?: string | undefined;
@@ -950,7 +951,6 @@ declare const imageInputSchema: z.ZodObject<{
     labels?: Record<string, string> | undefined;
     width?: number | undefined;
     height?: number | undefined;
-    lora?: string | undefined;
 }, {
     prompt: string;
     purpose?: string | undefined;
@@ -958,6 +958,7 @@ declare const imageInputSchema: z.ZodObject<{
         path: string;
         scale?: number | undefined;
     }[] | undefined;
+    lora?: string | undefined;
     tier?: "fast" | "smart" | "powerful" | "cheap" | "vision" | "video" | "embedding" | undefined;
     override?: {
         provider?: string | undefined;
@@ -972,7 +973,6 @@ declare const imageInputSchema: z.ZodObject<{
     labels?: Record<string, string> | undefined;
     width?: number | undefined;
     height?: number | undefined;
-    lora?: string | undefined;
 }>;
 declare const trainStyleInputSchema: z.ZodObject<{
     tier: z.ZodOptional<z.ZodEnum<["fast", "smart", "powerful", "cheap", "vision", "video", "embedding"]>>;
@@ -1680,8 +1680,8 @@ declare const falStubAdapter: ProviderAdapter;
  *  wires the live adapters. */
 declare const stubProviders: Record<string, ProviderAdapter>;
 
-declare const VERSION: "0.10.0";
-declare const SDK_TAG: "@broberg/ai-sdk@0.10.0";
+declare const VERSION: "0.10.2";
+declare const SDK_TAG: "@broberg/ai-sdk@0.10.2";
 
 /** 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

@@ -1334,7 +1334,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;
@@ -1346,8 +1346,14 @@ function falAdapter(config = {}) {
       body,
       config.trainTimeoutMs ?? 6e5
     );
-    const loraUrl = result.diffusers_lora_file?.url;
-    if (!loraUrl) throw new Error("fal trainStyle: no diffusers_lora_file.url in result");
+    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,
@@ -1357,9 +1363,9 @@ function falAdapter(config = {}) {
       outputTokens: 0
     });
     usage.costUsd = config.pricePerTraining ?? FAL_TRAIN_PRICE_ESTIMATE;
-    return { loraUrl, configUrl: result.config_file?.url ?? "", usage };
+    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) => {
@@ -1368,8 +1374,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}`, {
@@ -1416,10 +1441,45 @@ function falAdapter(config = {}) {
       await sleep(pollIntervalMs);
     }
     const resultRes = await doFetch(responseUrl, { headers });
+    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 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`;
@@ -2378,8 +2438,8 @@ var stubProviders = {
 };
 
 // src/version.ts
-var VERSION = "0.10.0";
-var SDK_TAG = "@broberg/ai-sdk@0.10.0";
+var VERSION = "0.10.2";
+var SDK_TAG = "@broberg/ai-sdk@0.10.2";
 
 // src/cost/budget-store.ts
 function sqliteBudgetStore(config) {