@brotu/ai 0.3.0 → 0.5.0

package/dist/index.cjs

@@ -30,9 +30,12 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  BROTU_SUPPORTED_CATEGORIES: () => BROTU_SUPPORTED_CATEGORIES,
   BYTEPLUS_CATALOG: () => BYTEPLUS_CATALOG,
   BYTEPLUS_MODELS: () => BYTEPLUS_MODELS,
+  BrotuAdapter: () => BrotuAdapter,
   BytePlusAdapter: () => BytePlusAdapter,
+  DEFAULT_BROTU_API_URL: () => DEFAULT_BROTU_API_URL,
   ELEVENLABS_CATALOG: () => ELEVENLABS_CATALOG,
   ELEVENLABS_MODELS: () => ELEVENLABS_MODELS,
   ELEVENLABS_OUTPUT_FORMATS: () => ELEVENLABS_OUTPUT_FORMATS,
@@ -59,9 +62,10 @@ __export(index_exports, {
   QWEN_TEXT_MODELS: () => QWEN_TEXT_MODELS,
   QWEN_VIDEO_MODELS: () => QWEN_VIDEO_MODELS,
   QwenAdapter: () => QwenAdapter,
-  brotuClient: () => brotuClient,
+  brotu: () => brotu,
   createS3Storage: () => createS3Storage,
   deliverWebhook: () => deliverWebhook,
+  describeModels: () => describeModels,
   fail: () => fail,
   getAvailableModels: () => getAvailableModels,
   getModel: () => getModel,
@@ -78,34 +82,6 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 
-// src/lib/jobs.ts
-var import_node_async_hooks = require("async_hooks");
-var PendingJob = class extends Error {
-  taskId;
-  pollEndpoint;
-  constructor(taskId, pollEndpoint) {
-    super(`Job ${taskId} was submitted and is still running.`);
-    this.name = "PendingJob";
-    this.taskId = taskId;
-    this.pollEndpoint = pollEndpoint;
-  }
-};
-function isPendingJob(error) {
-  return error instanceof PendingJob;
-}
-var submitMode = new import_node_async_hooks.AsyncLocalStorage();
-function runInSubmitMode(fn) {
-  return submitMode.run(true, fn);
-}
-function isSubmitMode() {
-  return submitMode.getStore() === true;
-}
-
-// src/ports/content-generator.port.ts
-function expiresInHours(hours) {
-  return new Date(Date.now() + hours * 36e5).toISOString();
-}
-
 // src/providers/byteplus.models.ts
 var FPS = 24;
 var PIXELS = {
@@ -1267,8 +1243,16 @@ var PROVIDER_BASE_URLS = {
   elevenlabs: "https://api.elevenlabs.io",
   google: "https://generativelanguage.googleapis.com",
   openai: "https://api.openai.com",
-  qwen: "https://dashscope-intl.aliyuncs.com"
+  qwen: "https://dashscope-intl.aliyuncs.com",
+  brotu: "https://api.brotu.app"
 };
+var DEFAULT_BROTU_API_URL = PROVIDER_BASE_URLS.brotu;
+function vendorProviders(options) {
+  return options.providers ?? {};
+}
+function brotuApiKey(options) {
+  return options.apiKey?.trim() || void 0;
+}
 var BUILT_IN = [
   ...KLING_CATALOG,
   ...KLING_AUDIO_CATALOG,
@@ -1324,25 +1308,95 @@ function resolveProvider(modelId, options) {
       `Model "${modelId}" declares no provider, so nothing knows how to run it.`
     );
   }
-  const configured = options.providers[id];
-  if (!configured) {
-    const owned = Object.keys(options.providers).join(", ") || "none";
-    throw new Error(
-      `Model "${modelId}" runs on "${id}", but no API key was given for it (configured: ${owned}).`
-    );
+  const configured = vendorProviders(options)[id];
+  if (configured) {
+    const baseUrl = configured.baseUrl ?? PROVIDER_BASE_URLS[id];
+    if (!baseUrl) {
+      throw new Error(
+        `Provider "${id}" has no known base URL \u2014 pass providers.${id}.baseUrl.`
+      );
+    }
+    return { id, apiKey: configured.apiKey, baseUrl: baseUrl.replace(/\/$/, "") };
   }
-  const baseUrl = configured.baseUrl ?? PROVIDER_BASE_URLS[id];
-  if (!baseUrl) {
-    throw new Error(
-      `Provider "${id}" has no known base URL \u2014 pass providers.${id}.baseUrl.`
-    );
+  const platformKey = brotuApiKey(options);
+  if (platformKey) {
+    return {
+      id: "brotu",
+      apiKey: platformKey,
+      baseUrl: (options.apiUrl ?? DEFAULT_BROTU_API_URL).replace(/\/$/, "")
+    };
   }
-  return { id, apiKey: configured.apiKey, baseUrl: baseUrl.replace(/\/$/, "") };
+  const owned = Object.keys(vendorProviders(options)).join(", ") || "none";
+  throw new Error(
+    `Model "${modelId}" runs on "${id}", but no API key was given for it (configured: ${owned}). Pass a Brotu key to generate on the platform.`
+  );
+}
+var BROTU_SUPPORTED_CATEGORIES = ["image", "video"];
+function describeModels(options) {
+  const vendors = vendorProviders(options);
+  const platformKey = brotuApiKey(options);
+  return catalog.map((model) => {
+    const provider = model.provider;
+    if (!provider) {
+      return {
+        model,
+        runsOn: "(none)",
+        runnable: false,
+        reason: "The model declares no provider."
+      };
+    }
+    if (provider in vendors) {
+      return { model, runsOn: provider, runnable: true };
+    }
+    if (platformKey) {
+      if (BROTU_SUPPORTED_CATEGORIES.includes(model.category)) {
+        return { model, runsOn: "brotu", runnable: true };
+      }
+      return {
+        model,
+        runsOn: provider,
+        runnable: false,
+        reason: `Brotu generates ${BROTU_SUPPORTED_CATEGORIES.join(" and ")} only. Pass providers.${provider} to run this one.`
+      };
+    }
+    return {
+      model,
+      runsOn: provider,
+      runnable: false,
+      reason: `No key for "${provider}". Pass a Brotu key or providers.${provider}.`
+    };
+  });
 }
 function getAvailableModels(options) {
-  return catalog.filter(
-    (model) => model.provider && model.provider in options.providers
-  );
+  return describeModels(options).filter((entry) => entry.runnable).map((entry) => entry.model);
+}
+
+// src/lib/jobs.ts
+var import_node_async_hooks = require("async_hooks");
+var PendingJob = class extends Error {
+  taskId;
+  pollEndpoint;
+  constructor(taskId, pollEndpoint) {
+    super(`Job ${taskId} was submitted and is still running.`);
+    this.name = "PendingJob";
+    this.taskId = taskId;
+    this.pollEndpoint = pollEndpoint;
+  }
+};
+function isPendingJob(error) {
+  return error instanceof PendingJob;
+}
+var submitMode = new import_node_async_hooks.AsyncLocalStorage();
+function runInSubmitMode(fn) {
+  return submitMode.run(true, fn);
+}
+function isSubmitMode() {
+  return submitMode.getStore() === true;
+}
+
+// src/ports/content-generator.port.ts
+function expiresInHours(hours) {
+  return new Date(Date.now() + hours * 36e5).toISOString();
 }
 
 // src/adapters/estimate.ts
@@ -1371,12 +1425,316 @@ function estimateFor(provider, type, params, defaults) {
   };
 }
 
+// src/adapters/brotu.adapter.ts
+var DEFAULT_API_URL = "https://api.brotu.app";
+var POLL_INTERVAL_MS = 3e3;
+var DEFAULT_MAX_POLL_ATTEMPTS = 400;
+var PLATFORM_MODEL_IDS = {
+  "kling/v2-6": "kling-2.6",
+  "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",
+  "gpt-image-1.5": "gpt-image/1.5"
+};
+var UNSUPPORTED_VIA_CREDITS = "Speech and text generate on the vendor. Pass that provider's key.";
+var BrotuAdapter = class {
+  constructor(opts) {
+    this.opts = opts;
+  }
+  opts;
+  providerName = "brotu";
+  supportedTypes = BROTU_SUPPORTED_CATEGORIES;
+  workspaceIdCache;
+  get origin() {
+    return (this.opts.apiUrl ?? DEFAULT_API_URL).replace(/\/$/, "");
+  }
+  headers() {
+    return {
+      Authorization: `Bearer ${this.opts.apiKey}`,
+      "Content-Type": "application/json"
+    };
+  }
+  async workspaceId() {
+    if (this.opts.workspaceId) return this.opts.workspaceId;
+    if (this.workspaceIdCache) return this.workspaceIdCache;
+    const payload = await this.request("/api/v1/studio/default-workspace");
+    const id = payload.workspaceId?.trim();
+    if (!id) {
+      throw new Error(
+        "This Brotu account has no workspace. Open https://brotu.app and create one."
+      );
+    }
+    this.workspaceIdCache = id;
+    return id;
+  }
+  studioPath(path, workspaceId) {
+    const joiner = path.includes("?") ? "&" : "?";
+    return `${path}${joiner}workspaceId=${encodeURIComponent(workspaceId)}`;
+  }
+  async request(path, init) {
+    const response = await fetch(`${this.origin}${path}`, {
+      ...init,
+      headers: { ...this.headers(), ...init?.headers }
+    });
+    const text = await response.text();
+    let body = void 0;
+    if (text) {
+      try {
+        body = JSON.parse(text);
+      } catch {
+        body = text;
+      }
+    }
+    if (!response.ok) {
+      const parsed = body;
+      const message = parsed && typeof parsed === "object" ? parsed.error?.message || parsed.message || text : text;
+      throw new Error(message || `Brotu API ${response.status}`);
+    }
+    return body;
+  }
+  platformModel(modelId) {
+    return PLATFORM_MODEL_IDS[modelId] ?? modelId;
+  }
+  async startGeneration(kind, params) {
+    const modelId = params.model ?? "";
+    const workspaceId = await this.workspaceId();
+    const path = kind === "video" ? "/api/v1/studio/videos" : "/api/v1/studio/images";
+    const body = kind === "video" ? this.videoBody(params, modelId) : this.imageBody(params, modelId);
+    const result = await this.request(
+      this.studioPath(path, workspaceId),
+      { method: "POST", body: JSON.stringify(body) }
+    );
+    const generationId = result.generationId?.trim();
+    if (!generationId) {
+      throw new Error("Brotu accepted the request but returned no generation id.");
+    }
+    return { generationId, workspaceId };
+  }
+  imageBody(params, modelId) {
+    return {
+      prompt: params.prompt,
+      model: this.platformModel(modelId),
+      aspectRatio: params.aspectRatio,
+      resolution: params.resolution,
+      negativePrompt: params.negativePrompt,
+      outputFormat: params.outputFormat,
+      seed: params.seed,
+      referenceImages: params.referenceImages
+    };
+  }
+  videoBody(params, modelId) {
+    return {
+      prompt: params.prompt,
+      model: this.platformModel(modelId),
+      duration: params.duration,
+      resolution: params.resolution,
+      aspectRatio: params.aspectRatio,
+      mode: params.mode,
+      withAudio: params.withAudio,
+      seed: params.seed,
+      imageUrl: params.imageUrl,
+      imageUrls: params.imageUrls,
+      videoUrl: params.videoUrl,
+      videoUrls: params.videoUrls,
+      referenceImages: params.referenceImages
+    };
+  }
+  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 submitted;
+    try {
+      submitted = await this.startGeneration(kind, params);
+    } catch (error) {
+      return failure(error instanceof Error ? error.message : String(error));
+    }
+    const pollEndpoint = this.studioPath(
+      `/api/v1/studio/generations/${submitted.generationId}`,
+      submitted.workspaceId
+    );
+    if (isSubmitMode()) {
+      throw new PendingJob(submitted.generationId, pollEndpoint);
+    }
+    const job = {
+      id: submitted.generationId,
+      provider: this.providerName,
+      model: modelId,
+      kind,
+      pollEndpoint,
+      params,
+      submittedAt: new Date(startedAt).toISOString()
+    };
+    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS;
+    for (let attempt = 0; attempt < maxAttempts; attempt++) {
+      const snapshot = await this.completeJob(job);
+      if (snapshot.status === "failed") {
+        return failure(snapshot.error ?? "Brotu generation failed.");
+      }
+      if (snapshot.status === "succeeded" && snapshot.result) {
+        return { ...snapshot.result, processingTimeMs: Date.now() - startedAt };
+      }
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS));
+    }
+    return failure(
+      `Brotu generation ${submitted.generationId} did not finish after ${maxAttempts} checks.`
+    );
+  }
+  async completeJob(job) {
+    const workspaceId = this.workspaceFromPoll(job.pollEndpoint) ?? await this.workspaceId();
+    const path = this.studioPath(
+      `/api/v1/studio/generations/${job.id}`,
+      workspaceId
+    );
+    const generation = await this.request(path);
+    const status = (generation.status ?? "").toLowerCase();
+    if (status === "failed") {
+      return {
+        status: "failed",
+        error: generation.errorMessage ?? "Brotu generation failed."
+      };
+    }
+    if (status !== "completed") {
+      return { status: "pending" };
+    }
+    const outputs = outputsFrom(generation, job.kind);
+    return {
+      status: "succeeded",
+      result: {
+        success: true,
+        outputs,
+        creditsUsed: generation.creditsUsed ?? 0,
+        provider: this.providerName,
+        model: job.model,
+        processingTimeMs: 0
+      }
+    };
+  }
+  workspaceFromPoll(pollEndpoint) {
+    if (!pollEndpoint) return void 0;
+    try {
+      const url = new URL(pollEndpoint, "https://brotu.invalid");
+      return url.searchParams.get("workspaceId") ?? void 0;
+    } catch {
+      return void 0;
+    }
+  }
+  generateImage(params) {
+    return this.run("image", params);
+  }
+  generateVideo(params) {
+    return this.run("video", params);
+  }
+  async generateText(params) {
+    return {
+      success: false,
+      outputs: [],
+      creditsUsed: 0,
+      provider: this.providerName,
+      model: params.model ?? "(none)",
+      processingTimeMs: 0,
+      error: UNSUPPORTED_VIA_CREDITS
+    };
+  }
+  async generateAudio(params) {
+    return {
+      success: false,
+      outputs: [],
+      creditsUsed: 0,
+      provider: this.providerName,
+      model: params.model ?? "(none)",
+      processingTimeMs: 0,
+      error: UNSUPPORTED_VIA_CREDITS
+    };
+  }
+  async estimateCost(type, params) {
+    if (type === "audio" || type === "text") {
+      const estimate = estimateFor(this.providerName, type, params);
+      return {
+        ...estimate,
+        usd: null,
+        note: UNSUPPORTED_VIA_CREDITS
+      };
+    }
+    try {
+      const workspaceId = await this.workspaceId();
+      const video = params;
+      const image = params;
+      const payload = await this.request(
+        this.studioPath("/api/v1/studio/estimate", workspaceId),
+        {
+          method: "POST",
+          body: JSON.stringify({
+            model: this.platformModel(params.model ?? ""),
+            duration: video.duration,
+            resolution: video.resolution ?? image.resolution,
+            aspectRatio: video.aspectRatio ?? image.aspectRatio,
+            withAudio: video.withAudio,
+            hasReferenceImage: Boolean(
+              video.imageUrl || video.imageUrls?.length || video.referenceImages?.length || image.referenceImages?.length
+            ),
+            hasReferenceVideo: Boolean(video.videoUrl || video.videoUrls?.length),
+            quality: image.quality
+          })
+        }
+      );
+      const credits = payload.estimatedCredits ?? 0;
+      return {
+        unit: type === "video" ? "second" : "image",
+        units: credits,
+        usd: null,
+        note: `${credits} Brotu credit${credits === 1 ? "" : "s"}. A vendor key generates on that provider.`,
+        provider: this.providerName,
+        model: params.model ?? ""
+      };
+    } catch (error) {
+      const estimate = estimateFor(this.providerName, type, params);
+      return {
+        ...estimate,
+        usd: null,
+        note: error instanceof Error ? error.message : "Could not estimate Brotu credits."
+      };
+    }
+  }
+  supportsModel() {
+    return true;
+  }
+  getAvailableModels() {
+    return [];
+  }
+};
+function outputsFrom(generation, kind) {
+  const bucket = generation.outputs;
+  const urls = kind === "video" ? bucket?.videos ?? bucket?.images ?? [] : bucket?.images ?? bucket?.videos ?? [];
+  const copies = bucket?.copies ?? [];
+  const mimeType = kind === "video" ? "video/mp4" : "image/png";
+  const fromUrls = urls.filter(Boolean).map((url) => ({
+    url,
+    mimeType
+  }));
+  if (fromUrls.length > 0) return fromUrls;
+  return copies.filter(Boolean).map((url) => ({
+    url,
+    mimeType: "text/plain",
+    raw: { text: url }
+  }));
+}
+
 // src/adapters/byteplus.adapter.ts
 var DEFAULT_BASE_URL = "https://ark.ap-southeast.bytepluses.com";
 var TASKS_PATH = "/api/v3/contents/generations/tasks";
 var IMAGES_PATH = "/api/v3/images/generations";
-var POLL_INTERVAL_MS = 5e3;
-var DEFAULT_MAX_POLL_ATTEMPTS = 240;
+var POLL_INTERVAL_MS2 = 5e3;
+var DEFAULT_MAX_POLL_ATTEMPTS2 = 240;
 var BytePlusAdapter = class {
   providerName = "byteplus";
   supportedTypes = ["image", "video"];
@@ -1576,7 +1934,7 @@ var BytePlusAdapter = class {
       params,
       submittedAt: new Date(startedAt).toISOString()
     };
-    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS;
+    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS2;
     for (let attempt = 0; attempt < maxAttempts; attempt++) {
       const snapshot = await this.completeJob(job);
       if (snapshot.status === "failed") {
@@ -1585,7 +1943,7 @@ var BytePlusAdapter = class {
       if (snapshot.status === "succeeded" && snapshot.result) {
         return { ...snapshot.result, processingTimeMs: Date.now() - startedAt };
       }
-      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS));
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS2));
     }
     return failure(
       `Ark task ${taskId} did not finish after ${maxAttempts} checks.`
@@ -1862,8 +2220,8 @@ var ElevenLabsAdapter = class {
 // src/adapters/google.adapter.ts
 var DEFAULT_BASE_URL3 = "https://generativelanguage.googleapis.com";
 var INTERACTIONS_PATH = "/v1beta/interactions";
-var POLL_INTERVAL_MS2 = 1e4;
-var DEFAULT_MAX_POLL_ATTEMPTS2 = 120;
+var POLL_INTERVAL_MS3 = 1e4;
+var DEFAULT_MAX_POLL_ATTEMPTS3 = 120;
 var GoogleAdapter = class {
   providerName = "google";
   supportedTypes = [
@@ -2135,7 +2493,7 @@ var GoogleAdapter = class {
       params,
       submittedAt: new Date(startedAt).toISOString()
     };
-    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS2;
+    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS3;
     for (let attempt = 0; attempt < maxAttempts; attempt++) {
       const snapshot = await this.completeJob(job);
       if (snapshot.status === "failed") {
@@ -2144,7 +2502,7 @@ var GoogleAdapter = class {
       if (snapshot.status === "succeeded" && snapshot.result) {
         return { ...snapshot.result, processingTimeMs: Date.now() - startedAt };
       }
-      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS2));
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS3));
     }
     return failure(`Veo operation ${operation} did not finish in time.`);
   }
@@ -2359,8 +2717,8 @@ var GoogleAdapter = class {
 
 // src/adapters/kling.adapter.ts
 var DEFAULT_BASE_URL4 = "https://api-singapore.klingai.com";
-var POLL_INTERVAL_MS3 = 5e3;
-var DEFAULT_MAX_POLL_ATTEMPTS3 = 240;
+var POLL_INTERVAL_MS4 = 5e3;
+var DEFAULT_MAX_POLL_ATTEMPTS4 = 240;
 function stateOf(task) {
   const raw = (task.status ?? task.task_status ?? "").toLowerCase();
   if (raw === "failed") return "failed";
@@ -2557,7 +2915,7 @@ var KlingAdapter = class {
       params,
       submittedAt: new Date(startedAt).toISOString()
     };
-    const maxAttempts = this.opts.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS3;
+    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") {
@@ -2566,7 +2924,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_MS3));
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS4));
     }
     return failure(
       `Kling task ${submitted.taskId} did not finish after ${maxAttempts} checks.`
@@ -2937,8 +3295,8 @@ var OpenAIAdapter = class {
 // src/adapters/qwen.adapter.ts
 var DEFAULT_BASE_URL6 = "https://dashscope-intl.aliyuncs.com";
 var TASKS_PATH2 = "/api/v1/tasks";
-var POLL_INTERVAL_MS4 = 5e3;
-var DEFAULT_MAX_POLL_ATTEMPTS4 = 240;
+var POLL_INTERVAL_MS5 = 5e3;
+var DEFAULT_MAX_POLL_ATTEMPTS5 = 240;
 var QwenAdapter = class {
   providerName = "qwen";
   supportedTypes = ["image", "video"];
@@ -3145,7 +3503,7 @@ var QwenAdapter = 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") {
@@ -3154,7 +3512,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_MS4));
+      await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS5));
     }
     return failure(
       `DashScope task ${taskId} did not finish after ${maxAttempts} checks.`
@@ -3414,6 +3772,22 @@ function failFrom(code, cause, context) {
   });
 }
 
+// src/lib/hooks.ts
+function hookName(kind, stage) {
+  return `on${kind[0].toUpperCase()}${kind.slice(1)}${stage}`;
+}
+function findHook(hooks, kind, stage) {
+  if (!hooks || !kind) return void 0;
+  return hooks[hookName(kind, stage)];
+}
+async function runHook(hook, event) {
+  if (!hook) return;
+  try {
+    await hook(event);
+  } catch {
+  }
+}
+
 // src/lib/storage.ts
 async function loadS3(config) {
   let s3;
@@ -3584,10 +3958,15 @@ var NATIVE_PROVIDERS = [
   "openai",
   "qwen"
 ];
-function brotuClient(options) {
-  if (Object.keys(options.providers).length === 0) {
-    throw new Error("brotuClient requires at least one provider API key.");
+function brotu(options) {
+  const apiKey = options.apiKey?.trim();
+  if (!apiKey) {
+    throw new Error(
+      "Pass a Brotu API key (brotu_sk_\u2026). Get one at https://brotu.app."
+    );
   }
+  const optionsWithKey = { ...options, apiKey };
+  const vendors = optionsWithKey.providers ?? {};
   registerModels(options.models);
   let registeredWebhook = resolveWebhook(options.webhook);
   const notifiedJobIds = /* @__PURE__ */ new Set();
@@ -3611,7 +3990,7 @@ function brotuClient(options) {
     }
     let provider;
     try {
-      provider = resolveProvider(modelId, options);
+      provider = resolveProvider(modelId, optionsWithKey);
     } catch (error) {
       return failFrom("missing_key", error, { model: modelId });
     }
@@ -3632,6 +4011,13 @@ function brotuClient(options) {
     return ok({ adapter, provider: provider.id, model: modelId });
   }
   function buildAdapter(provider) {
+    if (provider.id === "brotu") {
+      return new BrotuAdapter({
+        apiKey,
+        apiUrl: optionsWithKey.apiUrl,
+        workspaceId: optionsWithKey.workspaceId
+      });
+    }
     if (provider.id === "kling") {
       return new KlingAdapter({
         apiKey: provider.apiKey,
@@ -3685,7 +4071,10 @@ function brotuClient(options) {
   }
   async function notifySettled(input) {
     const config = webhookFor(input.params);
-    if (!config) return;
+    const kind = input.kind ?? input.job?.kind;
+    const stage = input.event === "generation.succeeded" ? "Success" : "Error";
+    const hook = findHook(options.hooks, kind, stage);
+    if (!config && !hook) return;
     const jobId = input.job?.id;
     if (jobId) {
       if (notifiedJobIds.has(jobId)) return;
@@ -3696,14 +4085,38 @@ function brotuClient(options) {
       jobId,
       provider: input.provider ?? input.job?.provider,
       model: input.model ?? input.job?.model,
-      kind: input.kind ?? input.job?.kind,
+      kind,
       outputs: input.outputs,
       error: input.error ? { code: input.error.code, message: input.error.message } : void 0,
       metadata: input.metadata ?? input.job?.metadata,
       processingTimeMs: input.processingTimeMs,
       completedAt: (/* @__PURE__ */ new Date()).toISOString()
     };
-    await deliverWebhook(config, payload);
+    if (hook && kind) {
+      await runHook(hook, {
+        kind,
+        stage,
+        provider: payload.provider ?? "",
+        model: payload.model ?? "",
+        jobId: payload.jobId,
+        outputs: payload.outputs,
+        error: payload.error,
+        metadata: payload.metadata,
+        processingTimeMs: payload.processingTimeMs,
+        at: payload.completedAt
+      });
+    }
+    if (config) await deliverWebhook(config, payload);
+  }
+  function notifyLoading(kind, routed, params) {
+    return runHook(findHook(options.hooks, kind, "Loading"), {
+      kind,
+      stage: "Loading",
+      provider: routed.provider,
+      model: routed.model,
+      metadata: params.metadata,
+      at: (/* @__PURE__ */ new Date()).toISOString()
+    });
   }
   async function toGeneration(raw, metadata) {
     if (!raw.success) {
@@ -3726,6 +4139,7 @@ function brotuClient(options) {
   async function generate(kind, params) {
     const routed = route(params.model);
     if (routed.error) return fail(routed.error);
+    await notifyLoading(kind, routed.data, params);
     try {
       const result = await toGeneration(
         await generateWith(routed.data.adapter, kind, params),
@@ -3776,6 +4190,7 @@ function brotuClient(options) {
   async function submit(kind, params) {
     const routed = route(params.model);
     if (routed.error) return fail(routed.error);
+    await notifyLoading(kind, routed.data, params);
     const base = {
       provider: routed.data.provider,
       model: routed.data.model,
@@ -3906,7 +4321,7 @@ function brotuClient(options) {
     }
   }
   function klingNamespace() {
-    const configured = options.providers.kling;
+    const configured = vendors.kling;
     if (!configured) return void 0;
     const adapter = new KlingAdapter({
       apiKey: configured.apiKey,
@@ -3931,7 +4346,7 @@ function brotuClient(options) {
     };
   }
   function googleNamespace() {
-    const configured = options.providers.google;
+    const configured = vendors.google;
     if (!configured) return void 0;
     const adapter = new GoogleAdapter({
       apiKey: configured.apiKey,
@@ -3947,24 +4362,33 @@ function brotuClient(options) {
       }
     };
   }
+  function listFor(category) {
+    return describeModels(optionsWithKey).filter(
+      (entry) => entry.model.category === category
+    );
+  }
   return {
     google: googleNamespace(),
     kling: klingNamespace(),
     image: {
       submit: (params) => submit("image", params),
-      generate: (params) => generate("image", params)
+      generate: (params) => generate("image", params),
+      list: () => listFor("image")
     },
     video: {
       submit: (params) => submit("video", params),
-      generate: (params) => generate("video", params)
+      generate: (params) => generate("video", params),
+      list: () => listFor("video")
     },
     text: {
       submit: (params) => submit("text", params),
-      generate: (params) => generate("text", params)
+      generate: (params) => generate("text", params),
+      list: () => listFor("text")
     },
     audio: {
       submit: (params) => submit("audio", params),
-      generate: (params) => generate("audio", params)
+      generate: (params) => generate("audio", params),
+      list: () => listFor("audio")
     },
     webhook: {
       set(value) {
@@ -4014,7 +4438,7 @@ function brotuClient(options) {
         }
       }
     },
-    models: () => getAvailableModels(options),
+    models: () => getAvailableModels(optionsWithKey),
     estimateCost: async (type, params) => {
       const routed = route(params.model);
       if (routed.error) return fail(routed.error);
@@ -4031,9 +4455,12 @@ function brotuClient(options) {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  BROTU_SUPPORTED_CATEGORIES,
   BYTEPLUS_CATALOG,
   BYTEPLUS_MODELS,
+  BrotuAdapter,
   BytePlusAdapter,
+  DEFAULT_BROTU_API_URL,
   ELEVENLABS_CATALOG,
   ELEVENLABS_MODELS,
   ELEVENLABS_OUTPUT_FORMATS,
@@ -4060,9 +4487,10 @@ function brotuClient(options) {
   QWEN_TEXT_MODELS,
   QWEN_VIDEO_MODELS,
   QwenAdapter,
-  brotuClient,
+  brotu,
   createS3Storage,
   deliverWebhook,
+  describeModels,
   fail,
   getAvailableModels,
   getModel,