@broberg/ai-sdk 0.10.0 → 0.10.1

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.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

package/dist/index.js

@@ -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,7 +1363,7 @@ 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) {
     if (typeof images === "string") return images;
@@ -1416,10 +1422,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 +2419,8 @@ var stubProviders = {
 };
 
 // src/version.ts
-var VERSION = "0.10.0";
-var SDK_TAG = "@broberg/ai-sdk@0.10.0";
+var VERSION = "0.10.1";
+var SDK_TAG = "@broberg/ai-sdk@0.10.1";
 
 // src/cost/budget-store.ts
 function sqliteBudgetStore(config) {