getaiapi 1.3.1 → 2.0.0

package/dist/index.js

@@ -26,9 +26,531 @@ import {
   submit,
   submitAndPoll,
   uploadAsset
-} from "./chunk-CN4FJ4FW.js";
+} from "./chunk-H6MMJNJX.js";
 
-// src/configure.ts
+// src/providers/kling/errors.ts
+var KlingError = class extends Error {
+  code;
+  constructor(code, message) {
+    super(message);
+    this.name = "KlingError";
+    this.code = code;
+  }
+};
+var KlingAuthError = class extends KlingError {
+  constructor(message = "Missing Kling credentials. Call kling.configure() or set KLING_ACCESS_KEY + KLING_SECRET_KEY.") {
+    super("AUTH_ERROR", message);
+    this.name = "KlingAuthError";
+  }
+};
+var KlingRateLimitError = class extends KlingError {
+  retryAfterMs;
+  constructor(retryAfterMs) {
+    super("RATE_LIMIT", `Rate limited by Kling. Retry after ${retryAfterMs}ms.`);
+    this.name = "KlingRateLimitError";
+    this.retryAfterMs = retryAfterMs;
+  }
+};
+var KlingApiError = class extends KlingError {
+  statusCode;
+  raw;
+  constructor(endpoint, statusCode, raw) {
+    super("API_ERROR", `Kling API error ${statusCode} on ${endpoint}.`);
+    this.name = "KlingApiError";
+    this.statusCode = statusCode;
+    this.raw = raw;
+  }
+};
+var KlingTimeoutError = class extends KlingError {
+  timeoutMs;
+  constructor(timeoutMs) {
+    super("TIMEOUT", `Kling task timed out after ${timeoutMs}ms.`);
+    this.name = "KlingTimeoutError";
+    this.timeoutMs = timeoutMs;
+  }
+};
+var KlingTaskFailedError = class extends KlingError {
+  taskId;
+  constructor(taskId, message) {
+    super("TASK_FAILED", `Kling task ${taskId} failed: ${message}`);
+    this.name = "KlingTaskFailedError";
+    this.taskId = taskId;
+  }
+};
+
+// src/providers/kling/auth.ts
+import { createHmac } from "crypto";
+function generateJwt(accessKey, secretKey) {
+  const header = Buffer.from(JSON.stringify({ alg: "HS256", typ: "JWT" })).toString("base64url");
+  const now = Math.floor(Date.now() / 1e3);
+  const payload = Buffer.from(JSON.stringify({
+    iss: accessKey,
+    exp: now + 1800,
+    nbf: now - 5
+  })).toString("base64url");
+  const signature = createHmac("sha256", secretKey).update(`${header}.${payload}`).digest("base64url");
+  return `${header}.${payload}.${signature}`;
+}
+function buildAuthHeaders(accessKey, secretKey) {
+  return {
+    Authorization: `Bearer ${generateJwt(accessKey, secretKey)}`,
+    "Content-Type": "application/json"
+  };
+}
+
+// src/providers/kling/client.ts
+var API_BASE = "https://api-singapore.klingai.com";
+var DEFAULT_FETCH_TIMEOUT_MS = 3e4;
+var DEFAULT_POLL_TIMEOUT_MS = 3e5;
+var DEFAULT_POLL_INTERVAL_MS = 2e3;
+var POLL_BACKOFF_MULTIPLIER = 1.5;
+var MAX_POLL_INTERVAL_MS = 1e4;
+var SYNC_ENDPOINTS = /* @__PURE__ */ new Set(["v1/audio/tts", "v1/videos/image-recognize", "v1/videos/identify-face"]);
+function cleanBase64(value) {
+  if (typeof value === "string" && value.startsWith("data:")) {
+    return value.split(",")[1] ?? value;
+  }
+  if (Array.isArray(value)) {
+    return value.map(cleanBase64);
+  }
+  if (value !== null && typeof value === "object") {
+    const result = {};
+    for (const [k, v] of Object.entries(value)) {
+      result[k] = cleanBase64(v);
+    }
+    return result;
+  }
+  return value;
+}
+var KlingClient = class {
+  accessKey = null;
+  secretKey = null;
+  fetchTimeoutMs = DEFAULT_FETCH_TIMEOUT_MS;
+  /** Update credentials and settings. */
+  configure(config) {
+    this.accessKey = config.accessKey;
+    this.secretKey = config.secretKey;
+    if (config.fetchTimeoutMs !== void 0) {
+      this.fetchTimeoutMs = config.fetchTimeoutMs;
+    }
+  }
+  resolveAuth() {
+    if (this.accessKey && this.secretKey) {
+      return { accessKey: this.accessKey, secretKey: this.secretKey };
+    }
+    const ak = process.env.KLING_ACCESS_KEY?.trim();
+    const sk = process.env.KLING_SECRET_KEY?.trim();
+    if (ak && sk) return { accessKey: ak, secretKey: sk };
+    throw new KlingAuthError();
+  }
+  // ── HTTP ─────────────────────────────────────────────────────────────────
+  async handleErrors(response, endpoint) {
+    if (response.ok) return;
+    if (response.status === 401) throw new KlingAuthError("Kling returned 401 Unauthorized.");
+    if (response.status === 429) {
+      const retryAfter = response.headers.get("retry-after");
+      throw new KlingRateLimitError(retryAfter ? parseInt(retryAfter, 10) * 1e3 : 6e4);
+    }
+    let raw;
+    try {
+      raw = await response.json();
+    } catch {
+      raw = await response.text().catch(() => null);
+    }
+    throw new KlingApiError(endpoint, response.status, raw);
+  }
+  handleBodyErrors(json, endpoint) {
+    if (json.code === 0) return;
+    if (json.code >= 1e3 && json.code <= 1004) {
+      throw new KlingAuthError(`Kling auth error code ${json.code}: ${json.message}`);
+    }
+    if (json.code >= 1100 && json.code <= 1102 || json.code >= 1302 && json.code <= 1304) {
+      throw new KlingRateLimitError(5e3);
+    }
+    throw new KlingApiError(endpoint, json.code, json);
+  }
+  async httpSubmit(endpoint, body) {
+    const auth = this.resolveAuth();
+    const url = `${API_BASE}/${endpoint}`;
+    const response = await fetch(url, {
+      method: "POST",
+      headers: buildAuthHeaders(auth.accessKey, auth.secretKey),
+      body: JSON.stringify(body),
+      signal: AbortSignal.timeout(this.fetchTimeoutMs)
+    });
+    await this.handleErrors(response, endpoint);
+    const json = await response.json();
+    this.handleBodyErrors(json, endpoint);
+    return json;
+  }
+  async httpPoll(endpoint, taskId) {
+    const auth = this.resolveAuth();
+    const url = `${API_BASE}/${endpoint}/${taskId}`;
+    const response = await fetch(url, {
+      headers: buildAuthHeaders(auth.accessKey, auth.secretKey),
+      signal: AbortSignal.timeout(this.fetchTimeoutMs)
+    });
+    await this.handleErrors(response, endpoint);
+    const json = await response.json();
+    this.handleBodyErrors(json, endpoint);
+    return json;
+  }
+  // ── Execute ──────────────────────────────────────────────────────────────
+  async execute(endpoint, defaults, input, extractor, sync) {
+    const { timeout, pollInterval, options, ...params } = input;
+    const timeoutMs = timeout ?? DEFAULT_POLL_TIMEOUT_MS;
+    const initialIntervalMs = pollInterval ?? DEFAULT_POLL_INTERVAL_MS;
+    const body = { ...params };
+    if (options && typeof options === "object") {
+      for (const [key, val] of Object.entries(options)) {
+        if (!(key in body)) body[key] = val;
+      }
+    }
+    for (const [key, val] of Object.entries(defaults)) {
+      if (!(key in body)) body[key] = val;
+    }
+    const cleanedBody = cleanBase64(body);
+    const submitResult = await this.httpSubmit(endpoint, cleanedBody);
+    const isSyncEndpoint = sync ?? SYNC_ENDPOINTS.has(endpoint);
+    if (isSyncEndpoint) {
+      return extractor(submitResult.data);
+    }
+    const taskId = submitResult.data.task_id;
+    const deadline = Date.now() + timeoutMs;
+    let intervalMs = initialIntervalMs;
+    while (Date.now() < deadline) {
+      await new Promise((resolve) => setTimeout(resolve, intervalMs));
+      intervalMs = Math.min(intervalMs * POLL_BACKOFF_MULTIPLIER, MAX_POLL_INTERVAL_MS);
+      const pollResult = await this.httpPoll(endpoint, taskId);
+      const status = pollResult.data.task_status;
+      if (status === "failed") {
+        throw new KlingTaskFailedError(taskId, pollResult.data.task_status_msg ?? "Unknown error");
+      }
+      if (status === "succeed") {
+        return extractor(pollResult.data);
+      }
+    }
+    throw new KlingTimeoutError(timeoutMs);
+  }
+};
+
+// src/providers/kling/extract.ts
+function extractVideos(data) {
+  const taskResult = data.task_result;
+  const videos = taskResult?.videos ?? [];
+  return { task_id: data.task_id, videos };
+}
+function extractImages(data) {
+  const taskResult = data.task_result;
+  const images = taskResult?.images ?? [];
+  return { task_id: data.task_id, images };
+}
+function extractAudios(data) {
+  const taskResult = data.task_result;
+  const rawAudios = taskResult?.audios ?? [];
+  const audios = rawAudios.map((a) => ({
+    id: a.id,
+    url: a.url ?? a.url_mp3 ?? a.url_wav ?? "",
+    url_mp3: a.url_mp3,
+    url_wav: a.url_wav,
+    duration: a.duration ?? a.duration_mp3,
+    duration_mp3: a.duration_mp3,
+    duration_wav: a.duration_wav
+  }));
+  return { task_id: data.task_id, audios };
+}
+function extractJson(data) {
+  return { task_id: data.task_id, data: data.task_result ?? data };
+}
+function extractFace(data) {
+  return {
+    session_id: data.session_id,
+    face_data: data.face_data ?? []
+  };
+}
+function extractMultiShot(data) {
+  const taskResult = data.task_result;
+  const rawImages = taskResult?.images ?? [];
+  const images = rawImages.map((img, i) => ({
+    index: img.index ?? i,
+    url_1: img.url_1,
+    url_2: img.url_2,
+    url_3: img.url_3
+  }));
+  return { task_id: data.task_id, images };
+}
+function extractVoices(data) {
+  const taskResult = data.task_result;
+  const voices = taskResult?.voices ?? [];
+  return { task_id: data.task_id, voices };
+}
+function extractVideoAudio(data) {
+  const taskResult = data.task_result;
+  const videos = taskResult?.videos ?? [];
+  const rawAudios = taskResult?.audios ?? [];
+  const audios = rawAudios.map((a) => ({
+    id: a.id,
+    url_mp3: a.url_mp3,
+    url_wav: a.url_wav,
+    duration_mp3: a.duration_mp3,
+    duration_wav: a.duration_wav
+  }));
+  return { task_id: data.task_id, videos, audios };
+}
+
+// src/providers/kling/models.ts
+function createModels(client) {
+  return {
+    // ── text2video (9 models) ──────────────────────────────────────────────
+    textToVideoV1Standard(input) {
+      return client.execute("v1/videos/text2video", { model_name: "kling-v1", mode: "std" }, input, extractVideos);
+    },
+    textToVideoV1_6Pro(input) {
+      return client.execute("v1/videos/text2video", { model_name: "kling-v1-6", mode: "pro" }, input, extractVideos);
+    },
+    textToVideoV1_6Standard(input) {
+      return client.execute("v1/videos/text2video", { model_name: "kling-v1-6", mode: "std" }, input, extractVideos);
+    },
+    textToVideoV2Master(input) {
+      return client.execute("v1/videos/text2video", { model_name: "kling-v2-master" }, input, extractVideos);
+    },
+    textToVideoV2_1Master(input) {
+      return client.execute("v1/videos/text2video", { model_name: "kling-v2-1-master" }, input, extractVideos);
+    },
+    textToVideoV2_5TurboPro(input) {
+      return client.execute("v1/videos/text2video", { model_name: "kling-v2-5-turbo", mode: "pro" }, input, extractVideos);
+    },
+    textToVideoV2_6Pro(input) {
+      return client.execute("v1/videos/text2video", { model_name: "kling-v2-6", mode: "pro" }, input, extractVideos);
+    },
+    textToVideoV3Pro(input) {
+      return client.execute("v1/videos/text2video", { model_name: "kling-v3", mode: "pro" }, input, extractVideos);
+    },
+    textToVideoV3Standard(input) {
+      return client.execute("v1/videos/text2video", { model_name: "kling-v3", mode: "std" }, input, extractVideos);
+    },
+    // ── image2video (13 models) ────────────────────────────────────────────
+    imageToVideoV1Standard(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v1", mode: "std" }, input, extractVideos);
+    },
+    imageToVideoV1_5Pro(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v1-5", mode: "pro" }, input, extractVideos);
+    },
+    imageToVideoV1_6Pro(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v1-6", mode: "pro" }, input, extractVideos);
+    },
+    imageToVideoV1_6Standard(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v1-6", mode: "std" }, input, extractVideos);
+    },
+    imageToVideoV2Master(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v2-master" }, input, extractVideos);
+    },
+    imageToVideoV2_1Master(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v2-1-master" }, input, extractVideos);
+    },
+    imageToVideoV2_1Pro(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v2-1", mode: "pro" }, input, extractVideos);
+    },
+    imageToVideoV2_1Standard(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v2-1", mode: "std" }, input, extractVideos);
+    },
+    imageToVideoV2_5TurboPro(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v2-5-turbo", mode: "pro" }, input, extractVideos);
+    },
+    imageToVideoV2_5TurboStandard(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v2-5-turbo", mode: "std" }, input, extractVideos);
+    },
+    imageToVideoV2_6Pro(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v2-6", mode: "pro" }, input, extractVideos);
+    },
+    imageToVideoV3Pro(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v3", mode: "pro" }, input, extractVideos);
+    },
+    imageToVideoV3Standard(input) {
+      return client.execute("v1/videos/image2video", { model_name: "kling-v3", mode: "std" }, input, extractVideos);
+    },
+    // ── omni-video (17 models) ─────────────────────────────────────────────
+    omniVideoO1ImageToVideo(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-video-o1" }, input, extractVideos);
+    },
+    omniVideoO1ReferenceToVideo(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-video-o1" }, input, extractVideos);
+    },
+    omniVideoO1StandardImageToVideo(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-video-o1", mode: "std" }, input, extractVideos);
+    },
+    omniVideoO1StandardReferenceToVideo(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-video-o1", mode: "std" }, input, extractVideos);
+    },
+    omniVideoO1StandardVideoEdit(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-video-o1", mode: "std" }, input, extractVideos);
+    },
+    omniVideoO1StandardVideoReference(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-video-o1", mode: "std" }, input, extractVideos);
+    },
+    omniVideoO1VideoEdit(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-video-o1" }, input, extractVideos);
+    },
+    omniVideoO1VideoReference(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-video-o1" }, input, extractVideos);
+    },
+    omniVideoO3ProImageToVideo(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-v3-omni", mode: "pro" }, input, extractVideos);
+    },
+    omniVideoO3ProReferenceToVideo(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-v3-omni", mode: "pro" }, input, extractVideos);
+    },
+    omniVideoO3ProTextToVideo(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-v3-omni", mode: "pro" }, input, extractVideos);
+    },
+    omniVideoO3ProVideoEdit(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-v3-omni", mode: "pro" }, input, extractVideos);
+    },
+    omniVideoO3ProVideoReference(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-v3-omni", mode: "pro" }, input, extractVideos);
+    },
+    omniVideoO3StandardReferenceToVideo(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-v3-omni", mode: "std" }, input, extractVideos);
+    },
+    omniVideoO3StandardTextToVideo(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-v3-omni", mode: "std" }, input, extractVideos);
+    },
+    omniVideoO3StandardVideoEdit(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-v3-omni", mode: "std" }, input, extractVideos);
+    },
+    omniVideoO3StandardVideoReference(input) {
+      return client.execute("v1/videos/omni-video", { model_name: "kling-v3-omni", mode: "std" }, input, extractVideos);
+    },
+    // ── images/generations (2 models) ──────────────────────────────────────
+    imageV3TextToImage(input) {
+      return client.execute("v1/images/generations", { model_name: "kling-v3" }, input, extractImages);
+    },
+    imageV3ImageToImage(input) {
+      return client.execute("v1/images/generations", { model_name: "kling-v3" }, input, extractImages);
+    },
+    // ── images/omni-image (3 models) ───────────────────────────────────────
+    imageO1(input) {
+      return client.execute("v1/images/omni-image", { model_name: "kling-image-o1" }, input, extractImages);
+    },
+    imageO3TextToImage(input) {
+      return client.execute("v1/images/omni-image", { model_name: "kling-v3-omni" }, input, extractImages);
+    },
+    imageO3ImageToImage(input) {
+      return client.execute("v1/images/omni-image", { model_name: "kling-v3-omni" }, input, extractImages);
+    },
+    // ── virtual-try-on (1 model) ───────────────────────────────────────────
+    virtualTryOn(input) {
+      return client.execute("v1/images/kolors-virtual-try-on", { model_name: "kolors-virtual-try-on-v1-5" }, input, extractImages);
+    },
+    // ── avatar (4 models) ──────────────────────────────────────────────────
+    avatarV2Pro(input) {
+      return client.execute("v1/videos/avatar/image2video", { mode: "pro" }, input, extractVideos);
+    },
+    avatarV2Standard(input) {
+      return client.execute("v1/videos/avatar/image2video", { mode: "std" }, input, extractVideos);
+    },
+    avatarV1Pro(input) {
+      return client.execute("v1/videos/avatar/image2video", { mode: "pro" }, input, extractVideos);
+    },
+    avatarV1Standard(input) {
+      return client.execute("v1/videos/avatar/image2video", { mode: "std" }, input, extractVideos);
+    },
+    // ── lip-sync (2 models) ────────────────────────────────────────────────
+    lipSyncAudioToVideo(input) {
+      return client.execute("v1/videos/advanced-lip-sync", {}, input, extractVideos);
+    },
+    lipSyncTextToVideo(input) {
+      return client.execute("v1/videos/advanced-lip-sync", {}, input, extractVideos);
+    },
+    // ── effects (4 models) ─────────────────────────────────────────────────
+    effectsV1Standard(input) {
+      return client.execute("v1/videos/effects", {}, input, extractVideos);
+    },
+    effectsV1_5Pro(input) {
+      return client.execute("v1/videos/effects", {}, input, extractVideos);
+    },
+    effectsV1_6Pro(input) {
+      return client.execute("v1/videos/effects", {}, input, extractVideos);
+    },
+    effectsV1_6Standard(input) {
+      return client.execute("v1/videos/effects", {}, input, extractVideos);
+    },
+    // ── motion-control (4 models) ──────────────────────────────────────────
+    motionControlV2_6Pro(input) {
+      return client.execute("v1/videos/motion-control", { model_name: "kling-v2-6", mode: "pro" }, input, extractVideos);
+    },
+    motionControlV2_6Standard(input) {
+      return client.execute("v1/videos/motion-control", { model_name: "kling-v2-6", mode: "std" }, input, extractVideos);
+    },
+    motionControlV3Pro(input) {
+      return client.execute("v1/videos/motion-control", { model_name: "kling-v3", mode: "pro" }, input, extractVideos);
+    },
+    motionControlV3Standard(input) {
+      return client.execute("v1/videos/motion-control", { model_name: "kling-v3", mode: "std" }, input, extractVideos);
+    },
+    // ── tts (1 model, sync) ────────────────────────────────────────────────
+    tts(input) {
+      return client.execute("v1/audio/tts", {}, input, extractAudios, true);
+    },
+    // ── video-to-audio (1 model) ───────────────────────────────────────────
+    videoToAudio(input) {
+      return client.execute("v1/audio/video-to-audio", {}, input, extractVideoAudio);
+    },
+    // ── text-to-audio (1 model) ────────────────────────────────────────────
+    textToAudio(input) {
+      return client.execute("v1/audio/text-to-audio", {}, input, extractAudios);
+    },
+    // ── create-voice (1 model) ─────────────────────────────────────────────
+    createVoice(input) {
+      return client.execute("v1/general/custom-voices", {}, input, extractVoices);
+    },
+    // ── multi-shot (1 model) ───────────────────────────────────────────────
+    multiShot(input) {
+      return client.execute("v1/general/ai-multi-shot", {}, input, extractMultiShot);
+    },
+    // ── reference-to-image (1 model) ───────────────────────────────────────
+    referenceToImage(input) {
+      return client.execute("v1/images/multi-image2image", { model_name: "kling-v2-1" }, input, extractImages);
+    },
+    // ── expand-image (1 model) ─────────────────────────────────────────────
+    expandImage(input) {
+      return client.execute("v1/images/editing/expand", {}, input, extractImages);
+    },
+    // ── extend-video (1 model) ─────────────────────────────────────────────
+    extendVideo(input) {
+      return client.execute("v1/videos/video-extend", {}, input, extractVideos);
+    },
+    // ── identify-face (1 model) ────────────────────────────────────────────
+    identifyFace(input) {
+      return client.execute("v1/videos/identify-face", {}, input, extractFace, true);
+    },
+    // ── image-recognize (1 model, sync) ────────────────────────────────────
+    imageRecognize(input) {
+      return client.execute("v1/videos/image-recognize", {}, input, extractJson, true);
+    },
+    // ── reference-to-video (1 model) ──────────────────────────────────────
+    referenceToVideo(input) {
+      return client.execute("v1/videos/multi-image2video", { model_name: "kling-v1-6" }, input, extractVideos);
+    }
+  };
+}
+
+// src/providers/kling/index.ts
+function createClient(config) {
+  const client = new KlingClient();
+  client.configure(config);
+  return {
+    configure: (c) => client.configure(c),
+    ...createModels(client)
+  };
+}
+var defaultClient = new KlingClient();
+var kling = {
+  configure: (config) => defaultClient.configure(config),
+  ...createModels(defaultClient)
+};
+
+// deprecated/src/configure.ts
 function configure(options) {
   if (options.keys) {
     configureAuth(options.keys);
@@ -43,6 +565,13 @@ function configure(options) {
 export {
   AuthError,
   GetAIApiError,
+  KlingApiError,
+  KlingAuthError,
+  KlingClient,
+  KlingError,
+  KlingRateLimitError,
+  KlingTaskFailedError,
+  KlingTimeoutError,
   ModelNotFoundError,
   NoProviderError,
   ProviderError,
@@ -55,9 +584,12 @@ export {
   configureAuth,
   configureFetch,
   configureStorage,
+  createClient,
   deleteAsset,
   deriveCategory,
   generate,
+  generateJwt,
+  kling,
   klingAdapter,
   listModels,
   loadRegistry,