@brotu/ai 0.7.0 → 0.8.0

package/dist/index.js

@@ -42,7 +42,7 @@ import {
   resetCatalog,
   resolveProvider,
   videoPathFor
-} from "./chunk-QQIHMDJC.js";
+} from "./chunk-OISTBZ64.js";
 
 // src/lib/jobs.ts
 import { AsyncLocalStorage } from "async_hooks";
@@ -1394,10 +1394,383 @@ var GoogleAdapter = class {
   }
 };
 
+// src/lib/webhook.ts
+function resolveWebhook(value) {
+  if (!value) return void 0;
+  if (typeof value === "string") {
+    return isHttpUrl(value) ? { url: value } : void 0;
+  }
+  if (!isHttpUrl(value.url)) return void 0;
+  return {
+    url: value.url,
+    secret: value.secret,
+    headers: value.headers
+  };
+}
+function isHttpUrl(value) {
+  try {
+    const parsed = new URL(value);
+    return parsed.protocol === "https:" || parsed.protocol === "http:";
+  } catch {
+    return false;
+  }
+}
+async function deliverWebhook(config, payload) {
+  try {
+    await fetch(config.url, {
+      method: "POST",
+      headers: {
+        "content-type": "application/json",
+        "user-agent": "@brotu/ai",
+        "x-brotu-event": payload.event,
+        ...config.secret ? { "x-brotu-webhook-secret": config.secret } : {},
+        ...config.headers
+      },
+      body: JSON.stringify(payload),
+      signal: AbortSignal.timeout(1e4)
+    });
+  } catch {
+  }
+}
+
+// src/adapters/kie.adapter.ts
+var DEFAULT_BASE_URL4 = "https://api.kie.ai/api/v1";
+var POLL_INTERVAL_MS4 = 3e3;
+var DEFAULT_MAX_POLL_ATTEMPTS4 = 400;
+var KIE_MODEL_IDS = {
+  "kling/v2-6": "kling-2.6/{mode}-to-video",
+  "kling/v3": "kling-3.0/video",
+  "dreamina-seedance-2-5-260628": "bytedance/seedance-2-5",
+  "dreamina-seedance-2-0-260128": "bytedance/seedance-2",
+  "dreamina-seedance-2-0-fast-260128": "bytedance/seedance-2-fast",
+  "dreamina-seedance-2-0-mini-260615": "bytedance/seedance-2-mini",
+  "wan2.6-t2v": "wan/2-6-text-to-video",
+  "wan2.6-i2v": "wan/2-6-image-to-video",
+  "wan2.7-t2v": "wan/2-7-text-to-video",
+  "wan2.7-i2v": "wan/2-7-image-to-video",
+  "gpt-image-1.5": "gpt-image/1.5-{mode}-to-image",
+  "gpt-image-2": "gpt-image-2-{mode}-to-image",
+  "topaz/video-upscale": "topaz/video-upscale"
+};
+var KieAdapter = class {
+  constructor(opts) {
+    this.opts = opts;
+  }
+  opts;
+  providerName = "kie";
+  supportedTypes = [
+    "image",
+    "video",
+    "text",
+    "audio"
+  ];
+  get origin() {
+    return (this.opts.baseUrl ?? DEFAULT_BASE_URL4).replace(/\/$/, "");
+  }
+  async request(path, init) {
+    const response = await fetch(`${this.origin}${path}`, {
+      ...init,
+      headers: {
+        Authorization: `Bearer ${this.opts.apiKey}`,
+        "Content-Type": "application/json",
+        ...init?.headers
+      }
+    });
+    const text = await response.text();
+    let body;
+    try {
+      body = text ? JSON.parse(text) : void 0;
+    } catch {
+      body = void 0;
+    }
+    const code = body?.code;
+    if (!response.ok || typeof code === "number" && code !== 200) {
+      const message = body?.msg || body?.message || text;
+      throw new Error(message || `kie API ${response.status}`);
+    }
+    if (body?.data === void 0) {
+      throw new Error("kie answered without a payload.");
+    }
+    return body.data;
+  }
+  /** Which market model serves this request. */
+  marketModel(modelId, params) {
+    const fromCatalog = getModel(modelId)?.endpoint?.replace(
+      /^\/market\//,
+      ""
+    );
+    const mapped = fromCatalog || KIE_MODEL_IDS[modelId] || modelId;
+    if (!mapped.includes("{mode}")) return mapped;
+    return mapped.replace("{mode}", hasInputImage(params) ? "image" : "text");
+  }
+  callbackFor(params) {
+    const perRequest = resolveWebhook(params.webhook);
+    return perRequest?.url ?? this.opts.callbackUrl;
+  }
+  async createTask(kind, params) {
+    const modelId = params.model ?? "";
+    const body = {
+      model: this.marketModel(modelId, params),
+      input: buildInput(kind, params, modelId)
+    };
+    const callBackUrl = this.callbackFor(params);
+    if (callBackUrl) body.callBackUrl = callBackUrl;
+    const data = await this.request("/jobs/createTask", {
+      method: "POST",
+      body: JSON.stringify(body)
+    });
+    const taskId = data.taskId?.trim();
+    if (!taskId) {
+      throw new Error("kie accepted the task but returned no taskId.");
+    }
+    return taskId;
+  }
+  /** Ask kie about one task. The same shape a callback carries. */
+  async task(taskId) {
+    const record = await this.request(
+      `/jobs/recordInfo?taskId=${encodeURIComponent(taskId)}`
+    );
+    return readTaskRecord(record, taskId);
+  }
+  async run(kind, params) {
+    const startedAt = Date.now();
+    const modelId = params.model ?? "(none)";
+    const failure = (error) => ({
+      success: false,
+      outputs: [],
+      creditsUsed: 0,
+      provider: this.providerName,
+      model: modelId,
+      processingTimeMs: Date.now() - startedAt,
+      error
+    });
+    let taskId;
+    try {
+      taskId = await this.createTask(kind, params);
+    } catch (error) {
+      return failure(error instanceof Error ? error.message : String(error));
+    }
+    const pollEndpoint = `/jobs/recordInfo?taskId=${encodeURIComponent(taskId)}`;
+    if (isSubmitMode()) {
+      throw new PendingJob(taskId, pollEndpoint);
+    }
+    const job = {
+      id: taskId,
+      provider: this.providerName,
+      model: modelId,
+      kind,
+      pollEndpoint,
+      params,
+      submittedAt: new Date(startedAt).toISOString()
+    };
+    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS4;
+    for (let attempt = 0; attempt < maxAttempts; attempt++) {
+      const snapshot = await this.completeJob(job);
+      if (snapshot.status === "failed") {
+        return failure(snapshot.error ?? "kie task failed.");
+      }
+      if (snapshot.status === "succeeded" && snapshot.result) {
+        return { ...snapshot.result, processingTimeMs: Date.now() - startedAt };
+      }
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS4));
+    }
+    return failure(
+      `kie task ${taskId} did not finish after ${maxAttempts} checks.`
+    );
+  }
+  async completeJob(job) {
+    const settled = await this.task(job.id);
+    return snapshotFrom(settled, job.model);
+  }
+  generateImage(params) {
+    return this.run("image", params);
+  }
+  generateVideo(params) {
+    return this.run("video", params);
+  }
+  generateText(params) {
+    return this.run("text", params);
+  }
+  generateAudio(params) {
+    return this.run("audio", params);
+  }
+  async estimateCost(type, params) {
+    return estimateFor(this.providerName, type, params);
+  }
+  supportsModel() {
+    return true;
+  }
+  getAvailableModels() {
+    return [];
+  }
+};
+function parseKieCallback(body) {
+  let parsed = body;
+  if (typeof body === "string") {
+    try {
+      parsed = JSON.parse(body);
+    } catch {
+      throw new Error("kie callback body is not JSON.");
+    }
+  }
+  const root = asRecord(parsed);
+  const record = asRecord(root?.data) ?? root ?? {};
+  const taskId = record.taskId?.trim();
+  if (!taskId) {
+    throw new Error("kie callback carries no taskId.");
+  }
+  return readTaskRecord(record, taskId);
+}
+function kieSnapshot(settled, model) {
+  return snapshotFrom(settled, model ?? settled.model ?? "");
+}
+function snapshotFrom(settled, model) {
+  if (settled.status === "failed") {
+    return { status: "failed", error: settled.error ?? "kie task failed." };
+  }
+  if (settled.status === "pending") return { status: "pending" };
+  return {
+    status: "succeeded",
+    result: {
+      success: true,
+      outputs: settled.outputs,
+      creditsUsed: settled.creditsUsed,
+      provider: "kie",
+      model,
+      processingTimeMs: 0
+    }
+  };
+}
+function readTaskRecord(record, taskId) {
+  const state = (record.state ?? "").toLowerCase();
+  const creditsUsed = record.creditsConsumed ?? 0;
+  if (state === "fail") {
+    return {
+      taskId,
+      model: record.model,
+      status: "failed",
+      outputs: [],
+      creditsUsed,
+      error: record.failMsg || (record.failCode ? `kie failCode ${record.failCode}` : void 0) || "kie task failed."
+    };
+  }
+  if (state !== "success") {
+    return {
+      taskId,
+      model: record.model,
+      status: "pending",
+      outputs: [],
+      creditsUsed
+    };
+  }
+  return {
+    taskId,
+    model: record.model,
+    status: "succeeded",
+    outputs: outputsFrom2(record.resultJson),
+    creditsUsed
+  };
+}
+function outputsFrom2(resultJson) {
+  if (!resultJson) return [];
+  let parsed;
+  try {
+    parsed = JSON.parse(resultJson);
+  } catch {
+    return [];
+  }
+  const record = asRecord(parsed);
+  const urls = Array.isArray(record?.resultUrls) ? record.resultUrls.filter(
+    (url) => typeof url === "string" && url.length > 0
+  ) : [];
+  if (urls.length > 0) return urls.map((url) => ({ url, mimeType: mimeFor(url) }));
+  const object = record?.resultObject;
+  if (object !== void 0) {
+    const text = typeof object === "string" ? object : JSON.stringify(object);
+    return [
+      {
+        url: `data:text/plain;base64,${Buffer.from(text).toString("base64")}`,
+        mimeType: "text/plain",
+        raw: { text }
+      }
+    ];
+  }
+  return [];
+}
+function mimeFor(url) {
+  const path = url.split("?")[0]?.toLowerCase() ?? "";
+  if (/\.(mp4|mov|webm|m4v)$/.test(path)) return "video/mp4";
+  if (/\.(mp3|wav|m4a|aac)$/.test(path)) return "audio/mpeg";
+  if (/\.jpe?g$/.test(path)) return "image/jpeg";
+  if (/\.webp$/.test(path)) return "image/webp";
+  return "image/png";
+}
+function asRecord(value) {
+  return value && typeof value === "object" ? value : void 0;
+}
+function hasInputImage(params) {
+  const p = params;
+  return Boolean(
+    p.imageUrl || p.imageUrls?.length || p.referenceImages?.length
+  );
+}
+function buildInput(kind, params, modelId) {
+  const raw = params.providerOptions?.kie ?? {};
+  const base = modelId.startsWith("kling/") ? klingInput(params) : modelId.includes("seedance") ? seedanceInput(params) : commonInput(kind, params);
+  return prune({ ...base, ...raw });
+}
+function klingInput(params) {
+  const p = params;
+  return {
+    prompt: p.prompt,
+    negative_prompt: p.negativePrompt,
+    aspect_ratio: p.aspectRatio,
+    // kie's kling enum is a string, unlike every other family.
+    duration: p.duration === void 0 ? void 0 : String(p.duration),
+    sound: p.withAudio,
+    image_url: p.imageUrl ?? p.referenceImages?.[0]
+  };
+}
+function seedanceInput(params) {
+  const p = params;
+  return {
+    prompt: p.prompt,
+    duration: p.duration,
+    resolution: p.resolution,
+    aspect_ratio: p.aspectRatio,
+    first_frame_url: p.imageUrl ?? p.referenceImages?.[0],
+    last_frame_url: p.referenceImages?.[1],
+    reference_image_urls: p.imageUrls,
+    reference_video_urls: p.videoUrls ?? (p.videoUrl ? [p.videoUrl] : void 0),
+    generate_audio: p.withAudio
+  };
+}
+function commonInput(kind, params) {
+  const p = params;
+  const images = p.imageUrls ?? p.referenceImages;
+  return {
+    prompt: p.prompt,
+    negative_prompt: p.negativePrompt,
+    aspect_ratio: p.aspectRatio,
+    resolution: p.resolution,
+    duration: kind === "video" ? p.duration : void 0,
+    seed: p.seed,
+    output_format: p.outputFormat,
+    image_url: p.imageUrl ?? images?.[0],
+    image_urls: images && images.length > 1 ? images : void 0,
+    video_url: p.videoUrl
+  };
+}
+function prune(input) {
+  return Object.fromEntries(
+    Object.entries(input).filter(([, value]) => value !== void 0)
+  );
+}
+
 // src/adapters/kling.adapter.ts
-var DEFAULT_BASE_URL4 = "https://api-singapore.klingai.com";
-var POLL_INTERVAL_MS4 = 5e3;
-var DEFAULT_MAX_POLL_ATTEMPTS4 = 240;
+var DEFAULT_BASE_URL5 = "https://api-singapore.klingai.com";
+var POLL_INTERVAL_MS5 = 5e3;
+var DEFAULT_MAX_POLL_ATTEMPTS5 = 240;
 function stateOf(task) {
   const raw = (task.status ?? task.task_status ?? "").toLowerCase();
   if (raw === "failed") return "failed";
@@ -1412,7 +1785,7 @@ var KlingAdapter = class {
     this.opts = opts;
   }
   get baseUrl() {
-    return (this.opts.baseUrl ?? DEFAULT_BASE_URL4).replace(/\/$/, "");
+    return (this.opts.baseUrl ?? DEFAULT_BASE_URL5).replace(/\/$/, "");
   }
   async request(path, init) {
     const response = await fetch(`${this.baseUrl}${path}`, {
@@ -1594,7 +1967,7 @@ var KlingAdapter = class {
       params,
       submittedAt: new Date(startedAt).toISOString()
     };
-    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS4;
+    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS5;
     for (let attempt = 0; attempt < maxAttempts; attempt++) {
       const snapshot = await this.completeJob(job);
       if (snapshot.status === "failed") {
@@ -1603,7 +1976,7 @@ var KlingAdapter = class {
       if (snapshot.status === "succeeded" && snapshot.result) {
         return { ...snapshot.result, processingTimeMs: Date.now() - startedAt };
       }
-      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS4));
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS5));
     }
     return failure(
       `Kling task ${submitted.taskId} did not finish after ${maxAttempts} checks.`
@@ -1740,7 +2113,7 @@ var KlingAdapter = class {
 };
 
 // src/adapters/openai.adapter.ts
-var DEFAULT_BASE_URL5 = "https://api.openai.com";
+var DEFAULT_BASE_URL6 = "https://api.openai.com";
 var IMAGES_PATH2 = "/v1/images/generations";
 var OpenAIAdapter = class {
   providerName = "openai";
@@ -1751,7 +2124,7 @@ var OpenAIAdapter = class {
     this.opts = opts;
   }
   get baseUrl() {
-    return (this.opts.baseUrl ?? DEFAULT_BASE_URL5).replace(/\/$/, "");
+    return (this.opts.baseUrl ?? DEFAULT_BASE_URL6).replace(/\/$/, "");
   }
   binding(modelId) {
     const id = modelId ?? "";
@@ -1972,10 +2345,10 @@ var OpenAIAdapter = class {
 };
 
 // src/adapters/qwen.adapter.ts
-var DEFAULT_BASE_URL6 = "https://dashscope-intl.aliyuncs.com";
+var DEFAULT_BASE_URL7 = "https://dashscope-intl.aliyuncs.com";
 var TASKS_PATH2 = "/api/v1/tasks";
-var POLL_INTERVAL_MS5 = 5e3;
-var DEFAULT_MAX_POLL_ATTEMPTS5 = 240;
+var POLL_INTERVAL_MS6 = 5e3;
+var DEFAULT_MAX_POLL_ATTEMPTS6 = 240;
 var QwenAdapter = class {
   providerName = "qwen";
   supportedTypes = ["image", "video"];
@@ -1984,7 +2357,7 @@ var QwenAdapter = class {
     this.opts = opts;
   }
   get baseUrl() {
-    return (this.opts.baseUrl ?? DEFAULT_BASE_URL6).replace(/\/$/, "");
+    return (this.opts.baseUrl ?? DEFAULT_BASE_URL7).replace(/\/$/, "");
   }
   async request(path, init) {
     const headers = {
@@ -2182,7 +2555,7 @@ var QwenAdapter = class {
       params,
       submittedAt: new Date(startedAt).toISOString()
     };
-    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS5;
+    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS6;
     for (let attempt = 0; attempt < maxAttempts; attempt++) {
       const snapshot = await this.completeJob(job);
       if (snapshot.status === "failed") {
@@ -2191,7 +2564,7 @@ var QwenAdapter = class {
       if (snapshot.status === "succeeded" && snapshot.result) {
         return { ...snapshot.result, processingTimeMs: Date.now() - startedAt };
       }
-      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS5));
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS6));
     }
     return failure(
       `DashScope task ${taskId} did not finish after ${maxAttempts} checks.`
@@ -2436,9 +2809,9 @@ var QwenAdapter = class {
 };
 
 // src/adapters/topaz.adapter.ts
-var DEFAULT_BASE_URL7 = "https://api.topazlabs.com";
-var POLL_INTERVAL_MS6 = 5e3;
-var DEFAULT_MAX_POLL_ATTEMPTS6 = 240;
+var DEFAULT_BASE_URL8 = "https://api.topazlabs.com";
+var POLL_INTERVAL_MS7 = 5e3;
+var DEFAULT_MAX_POLL_ATTEMPTS7 = 240;
 var DEFAULT_FPS = 30;
 var PIXELS = {
   "720p": { width: 1280, height: 720 },
@@ -2458,7 +2831,7 @@ var TopazAdapter = class {
     this.opts = opts;
   }
   get baseUrl() {
-    return (this.opts.baseUrl ?? DEFAULT_BASE_URL7).replace(/\/$/, "");
+    return (this.opts.baseUrl ?? DEFAULT_BASE_URL8).replace(/\/$/, "");
   }
   get headers() {
     return {
@@ -2671,7 +3044,7 @@ var TopazAdapter = class {
       params,
       submittedAt: new Date(startedAt).toISOString()
     };
-    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS6;
+    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS7;
     for (let attempt = 0; attempt < maxAttempts; attempt++) {
       const snapshot = await this.completeJob(job);
       if (snapshot.status === "failed") {
@@ -2680,7 +3053,7 @@ var TopazAdapter = class {
       if (snapshot.status === "succeeded" && snapshot.result) {
         return { ...snapshot.result, processingTimeMs: Date.now() - startedAt };
       }
-      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS6));
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS7));
     }
     return failure(
       `Topaz request ${submitted.taskId} did not finish after ${maxAttempts} checks.`
@@ -2846,7 +3219,7 @@ var TopazAdapter = class {
       params,
       submittedAt: new Date(startedAt).toISOString()
     };
-    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS6;
+    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS7;
     for (let attempt = 0; attempt < maxAttempts; attempt++) {
       const snapshot = await this.completeJob(job);
       if (snapshot.status === "failed") {
@@ -2855,7 +3228,7 @@ var TopazAdapter = class {
       if (snapshot.status === "succeeded" && snapshot.result) {
         return { ...snapshot.result, processingTimeMs: Date.now() - startedAt };
       }
-      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS6));
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS7));
     }
     return failure(
       `Topaz image ${submitted.taskId} did not finish after ${maxAttempts} checks.`
@@ -3058,50 +3431,12 @@ async function persistOutputs(storage, outputs) {
   );
 }
 
-// src/lib/webhook.ts
-function resolveWebhook(value) {
-  if (!value) return void 0;
-  if (typeof value === "string") {
-    return isHttpUrl(value) ? { url: value } : void 0;
-  }
-  if (!isHttpUrl(value.url)) return void 0;
-  return {
-    url: value.url,
-    secret: value.secret,
-    headers: value.headers
-  };
-}
-function isHttpUrl(value) {
-  try {
-    const parsed = new URL(value);
-    return parsed.protocol === "https:" || parsed.protocol === "http:";
-  } catch {
-    return false;
-  }
-}
-async function deliverWebhook(config, payload) {
-  try {
-    await fetch(config.url, {
-      method: "POST",
-      headers: {
-        "content-type": "application/json",
-        "user-agent": "@brotu/ai",
-        "x-brotu-event": payload.event,
-        ...config.secret ? { "x-brotu-webhook-secret": config.secret } : {},
-        ...config.headers
-      },
-      body: JSON.stringify(payload),
-      signal: AbortSignal.timeout(1e4)
-    });
-  } catch {
-  }
-}
-
 // src/client.ts
 var NATIVE_PROVIDERS = [
   "byteplus",
   "elevenlabs",
   "google",
+  "kie",
   "kling",
   "openai",
   "qwen",
@@ -3167,6 +3502,16 @@ function brotu(options) {
         workspaceId: optionsWithKey.workspaceId
       });
     }
+    if (provider.id === "kie") {
+      return new KieAdapter({
+        apiKey: provider.apiKey,
+        baseUrl: provider.baseUrl,
+        // The client-level webhook doubles as kie's callback, so a submit
+        // never has to be polled. kie posts its own payload there — read it
+        // with `parseKieCallback`.
+        callbackUrl: registeredWebhook?.url
+      });
+    }
     if (provider.id === "kling") {
       return new KlingAdapter({
         apiKey: provider.apiKey,
@@ -3697,6 +4042,7 @@ export {
   KLING_CAPABILITIES,
   KLING_CATALOG,
   KLING_MODELS,
+  KieAdapter,
   KlingAdapter,
   OPENAI_CATALOG,
   OPENAI_IMAGE_MODELS,
@@ -3725,7 +4071,9 @@ export {
   getProviders,
   hasModel,
   isPendingJob,
+  kieSnapshot,
   ok,
+  parseKieCallback,
   persistOutputs,
   registerModels,
   resetCatalog,

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@brotu/ai",
-	"version": "0.7.0",
+	"version": "0.8.0",
 	"type": "module",
 	"main": "./dist/index.cjs",
 	"exports": {