@apicity/kie 0.6.6 → 0.7.0

package/dist/src/kie.js

@@ -1,5 +1,5 @@
 import { KieError, } from "./types.js";
-import { CreateTaskRequestSchema, DownloadUrlRequestSchema, UploadMediaRequestSchema, FileUrlUploadRequestSchema, FileBase64UploadRequestSchema, GeminiOmniVideoRequestSchema, GeminiOmniAudioCreateRequestSchema, GeminiOmniCharacterCreateRequestSchema, GeminiOmniCharacterCreateResponseSchema, FluxKontextGenerateRequestSchema, Gpt4oImageGenerateRequestSchema, MjGenerateRequestSchema, RunwayGenerateRequestSchema, RunwayExtendRequestSchema, RunwayRecordDetailResponseSchema, FluxKontextRecordInfoRequestSchema, FluxKontextRecordInfoResponseSchema, GrokImageToVideoRequestSchema, RecordInfoRequestSchema, Gpt4oImageRecordInfoResponseSchema, Seedance2MiniRecordInfoResponseSchema, Seedance2MiniRequestSchema, } from "./zod.js";
+import { CreateTaskRequestSchema, DownloadUrlRequestSchema, UploadMediaRequestSchema, FileUrlUploadRequestSchema, FileBase64UploadRequestSchema, GeminiOmniVideoRequestSchema, GeminiOmniAudioCreateRequestSchema, GeminiOmniCharacterCreateRequestSchema, GeminiOmniCharacterCreateResponseSchema, FluxKontextGenerateRequestSchema, Gpt4oImageGenerateRequestSchema, MjGenerateRequestSchema, MjRecordInfoRequestSchema, MjRecordInfoResponseSchema, RunwayGenerateRequestSchema, RunwayExtendRequestSchema, RunwayRecordDetailResponseSchema, FluxKontextRecordInfoRequestSchema, FluxKontextRecordInfoResponseSchema, GrokImageToVideoRequestSchema, RecordInfoRequestSchema, Gpt4oImageRecordInfoResponseSchema, Seedance2MiniRecordInfoResponseSchema, Seedance2MiniRequestSchema, } from "./zod.js";
 import { modelInputSchemas } from "./model-schemas.js";
 import { createVeoProvider } from "./veo.js";
 import { createSunoProvider } from "./suno.js";
@@ -11,7 +11,7 @@ import { createGemini31ProProvider } from "./gemini-31-pro.js";
 import { attachExamples } from "./example.js";
 import { createReplayStore } from "./paygate.js";
 import { withPaidGate } from "./with-paid-gate.js";
-import { kieRequest } from "./request.js";
+import { createKieTransport, kieRequest } from "./request.js";
 const MIME_TYPES = {
     jpg: "image/jpeg",
     jpeg: "image/jpeg",
@@ -94,6 +94,22 @@ export function createKie(opts) {
     const uploadBaseURL = opts.uploadBaseURL ?? "https://kieai.redpandaai.co";
     const doFetch = opts.fetch ?? fetch;
     const timeout = opts.timeout ?? 30000;
+    const transport = createKieTransport({
+        baseURL,
+        apiKey: opts.apiKey,
+        doFetch,
+        timeout,
+        requestFailedPrefix: "Kie request failed",
+    });
+    const uploadTransport = createKieTransport({
+        baseURL: uploadBaseURL,
+        apiKey: opts.apiKey,
+        doFetch,
+        timeout,
+        errorPrefix: "Kie upload error",
+        requestFailedPrefix: "Kie upload failed",
+        jsonContentType: false,
+    });
     const paygate = opts.paygate
         ? {
             ...opts.paygate,
@@ -106,323 +122,65 @@ export function createKie(opts) {
         validateGrokImageToVideoRequest(req);
         validateSeedance2MiniRequest(req);
         validateGeminiOmniVideoRequest(req);
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), timeout);
-        try {
-            const res = await doFetch(`${baseURL}/api/v1/jobs/createTask`, {
-                method: "POST",
-                headers: {
-                    Authorization: `Bearer ${opts.apiKey}`,
-                    "Content-Type": "application/json",
-                },
-                body: JSON.stringify(req),
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-            if (!res.ok) {
-                let message = `Kie API error: ${res.status}`;
-                let body = null;
-                try {
-                    body = await res.json();
-                    if (typeof body === "object" &&
-                        body !== null &&
-                        "msg" in body &&
-                        typeof body.msg === "string") {
-                        message = `Kie API error ${res.status}: ${body.msg}`;
-                    }
-                }
-                catch {
-                    // ignore parse errors
-                }
-                throw new KieError(message, res.status, body);
-            }
-            return (await res.json());
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof KieError)
-                throw error;
-            throw new KieError(`Failed to create task: ${error}`, 500);
-        }
+        return await transport.postJson("/api/v1/jobs/createTask", req);
     }
     // GET https://api.kie.ai/api/v1/jobs/recordInfo?taskId={taskId}
     // Docs: https://docs.kie.ai/market/common/get-task-detail
     async function recordInfo(taskId) {
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), timeout);
-        try {
-            const res = await doFetch(`${baseURL}/api/v1/jobs/recordInfo?taskId=${encodeURIComponent(taskId)}`, {
-                method: "GET",
-                headers: {
-                    Authorization: `Bearer ${opts.apiKey}`,
-                    "Content-Type": "application/json",
-                },
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-            if (!res.ok) {
-                let message = `Kie API error: ${res.status}`;
-                let body = null;
-                try {
-                    body = await res.json();
-                    if (typeof body === "object" &&
-                        body !== null &&
-                        "msg" in body &&
-                        typeof body.msg === "string") {
-                        message = `Kie API error ${res.status}: ${body.msg}`;
-                    }
-                }
-                catch {
-                    // ignore parse errors
-                }
-                throw new KieError(message, res.status, body);
-            }
-            return (await res.json());
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof KieError)
-                throw error;
-            throw new KieError(`Failed to get task: ${error}`, 500);
-        }
+        return await transport.getJson(`/api/v1/jobs/recordInfo?taskId=${encodeURIComponent(taskId)}`);
     }
     // GET https://api.kie.ai/api/v1/gpt4o-image/record-info?taskId={taskId}
     // Docs: https://docs.kie.ai/4o-image-api/get-4-o-image-details
     async function gpt4oImageRecordInfo(taskId) {
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), timeout);
-        try {
-            const res = await doFetch(`${baseURL}/api/v1/gpt4o-image/record-info?taskId=${encodeURIComponent(taskId)}`, {
-                method: "GET",
-                headers: {
-                    Authorization: `Bearer ${opts.apiKey}`,
-                    "Content-Type": "application/json",
-                },
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-            if (!res.ok) {
-                let message = `Kie API error: ${res.status}`;
-                let body = null;
-                try {
-                    body = await res.json();
-                    if (typeof body === "object" &&
-                        body !== null &&
-                        "msg" in body &&
-                        typeof body.msg === "string") {
-                        message = `Kie API error ${res.status}: ${body.msg}`;
-                    }
-                }
-                catch {
-                    // ignore parse errors
-                }
-                throw new KieError(message, res.status, body);
-            }
-            return (await res.json());
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof KieError)
-                throw error;
-            throw new KieError(`Failed to get 4o image task: ${error}`, 500);
-        }
+        return await transport.getJson(`/api/v1/gpt4o-image/record-info?taskId=${encodeURIComponent(taskId)}`);
     }
     // POST https://kieai.redpandaai.co/api/file-stream-upload
     // Docs: https://docs.kie.ai/file-upload-api/upload-file-stream
     async function fileStreamUpload(req) {
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), timeout);
-        try {
-            const mimeType = req.mimeType ?? inferMimeType(req.filename);
-            if (!mimeType) {
-                throw new KieError(`Cannot determine MIME type for: ${req.filename}`, 400);
-            }
-            const formData = new FormData();
-            const file = new File([req.file], req.filename, { type: mimeType });
-            formData.append("file", file);
-            formData.append("uploadPath", req.uploadPath);
-            if (req.fileName) {
-                formData.append("fileName", req.fileName);
-            }
-            const res = await doFetch(`${uploadBaseURL}/api/file-stream-upload`, {
-                method: "POST",
-                headers: {
-                    Authorization: `Bearer ${opts.apiKey}`,
-                },
-                body: formData,
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-            if (!res.ok) {
-                let message = `Kie upload error: ${res.status}`;
-                let body = null;
-                try {
-                    body = await res.json();
-                    if (typeof body === "object" &&
-                        body !== null &&
-                        "msg" in body &&
-                        typeof body.msg === "string") {
-                        message = `Kie upload error ${res.status}: ${body.msg}`;
-                    }
-                }
-                catch {
-                    // ignore parse errors
-                }
-                throw new KieError(message, res.status, body);
-            }
-            return (await res.json());
+        const mimeType = req.mimeType ?? inferMimeType(req.filename);
+        if (!mimeType) {
+            throw new KieError(`Cannot determine MIME type for: ${req.filename}`, 400);
         }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof KieError)
-                throw error;
-            throw new KieError(`Failed to upload media: ${error}`, 500);
+        const formData = new FormData();
+        const file = new File([req.file], req.filename, { type: mimeType });
+        formData.append("file", file);
+        formData.append("uploadPath", req.uploadPath);
+        if (req.fileName) {
+            formData.append("fileName", req.fileName);
         }
+        return await uploadTransport.postForm("/api/file-stream-upload", formData);
     }
     // POST https://kieai.redpandaai.co/api/file-url-upload
     // Docs: https://docs.kie.ai/file-upload-api/upload-file-url
     async function fileUrlUpload(req) {
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), timeout);
-        try {
-            const res = await doFetch(`${uploadBaseURL}/api/file-url-upload`, {
-                method: "POST",
-                headers: {
-                    Authorization: `Bearer ${opts.apiKey}`,
-                    "Content-Type": "application/json",
-                },
-                body: JSON.stringify({
-                    fileUrl: req.fileUrl,
-                    uploadPath: req.uploadPath,
-                    ...(req.fileName ? { fileName: req.fileName } : {}),
-                }),
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-            if (!res.ok) {
-                let message = `Kie upload error: ${res.status}`;
-                let body = null;
-                try {
-                    body = await res.json();
-                    if (typeof body === "object" &&
-                        body !== null &&
-                        "msg" in body &&
-                        typeof body.msg === "string") {
-                        message = `Kie upload error ${res.status}: ${body.msg}`;
-                    }
-                }
-                catch {
-                    // ignore parse errors
-                }
-                throw new KieError(message, res.status, body);
-            }
-            return (await res.json());
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof KieError)
-                throw error;
-            throw new KieError(`Failed to upload from URL: ${error}`, 500);
-        }
+        return await uploadTransport.postJson("/api/file-url-upload", {
+            fileUrl: req.fileUrl,
+            uploadPath: req.uploadPath,
+            ...(req.fileName ? { fileName: req.fileName } : {}),
+        });
     }
     // POST https://kieai.redpandaai.co/api/file-base64-upload
     // Docs: https://docs.kie.ai/file-upload-api/upload-file-base-64
     async function fileBase64Upload(req) {
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), timeout);
-        try {
-            const res = await doFetch(`${uploadBaseURL}/api/file-base64-upload`, {
-                method: "POST",
-                headers: {
-                    Authorization: `Bearer ${opts.apiKey}`,
-                    "Content-Type": "application/json",
-                },
-                body: JSON.stringify({
-                    base64Data: req.base64Data,
-                    uploadPath: req.uploadPath,
-                    ...(req.fileName ? { fileName: req.fileName } : {}),
-                    ...(req.mimeType ? { mimeType: req.mimeType } : {}),
-                }),
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-            if (!res.ok) {
-                let message = `Kie upload error: ${res.status}`;
-                let body = null;
-                try {
-                    body = await res.json();
-                    if (typeof body === "object" &&
-                        body !== null &&
-                        "msg" in body &&
-                        typeof body.msg === "string") {
-                        message = `Kie upload error ${res.status}: ${body.msg}`;
-                    }
-                }
-                catch {
-                    // ignore parse errors
-                }
-                throw new KieError(message, res.status, body);
-            }
-            return (await res.json());
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof KieError)
-                throw error;
-            throw new KieError(`Failed to upload base64 file: ${error}`, 500);
-        }
+        return await uploadTransport.postJson("/api/file-base64-upload", {
+            base64Data: req.base64Data,
+            uploadPath: req.uploadPath,
+            ...(req.fileName ? { fileName: req.fileName } : {}),
+            ...(req.mimeType ? { mimeType: req.mimeType } : {}),
+        });
     }
     // POST https://api.kie.ai/api/v1/common/download-url
     // Docs: https://docs.kie.ai/common-api/download-url
     async function downloadUrl(req) {
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), timeout);
-        try {
-            const res = await doFetch(`${baseURL}/api/v1/common/download-url`, {
-                method: "POST",
-                headers: {
-                    Authorization: `Bearer ${opts.apiKey}`,
-                    "Content-Type": "application/json",
-                },
-                body: JSON.stringify(req),
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-            if (!res.ok) {
-                let message = `Kie API error: ${res.status}`;
-                let body = null;
-                try {
-                    body = await res.json();
-                    if (typeof body === "object" &&
-                        body !== null &&
-                        "msg" in body &&
-                        typeof body.msg === "string") {
-                        message = `Kie API error ${res.status}: ${body.msg}`;
-                    }
-                }
-                catch {
-                    // ignore parse errors
-                }
-                throw new KieError(message, res.status, body);
-            }
-            return (await res.json());
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof KieError)
-                throw error;
-            throw new KieError(`Failed to get download URL: ${error}`, 500);
-        }
+        return await transport.postJson("/api/v1/common/download-url", req);
     }
     // POST https://api.kie.ai/api/v1/omni/audio/create
     // Docs: https://docs.kie.ai/market/gemini-omni-audio
     async function omniAudioCreate(req) {
-        const response = await kieRequest(`${baseURL}/api/v1/omni/audio/create`, {
+        const response = await kieRequest(transport, {
             method: "POST",
+            path: "/api/v1/omni/audio/create",
             body: req,
-            apiKey: opts.apiKey,
-            doFetch,
-            timeout,
         });
         if (!response.data) {
             return response;
@@ -443,42 +201,44 @@ export function createKie(opts) {
     // POST https://api.kie.ai/api/v1/omni/character/create
     // Docs: https://docs.kie.ai/market/gemini-omni-character
     async function omniCharacterCreate(req) {
-        return kieRequest(`${baseURL}/api/v1/omni/character/create`, {
+        return kieRequest(transport, {
             method: "POST",
+            path: "/api/v1/omni/character/create",
             body: req,
-            apiKey: opts.apiKey,
-            doFetch,
-            timeout,
         });
     }
     // POST https://api.kie.ai/api/v1/flux/kontext/generate
     // Docs: https://docs.kie.ai/flux-kontext-api/generate-or-edit-image
     async function fluxKontextGenerate(req) {
-        return kieRequest(`${baseURL}/api/v1/flux/kontext/generate`, {
+        return kieRequest(transport, {
             method: "POST",
+            path: "/api/v1/flux/kontext/generate",
             body: req,
-            apiKey: opts.apiKey,
-            doFetch,
-            timeout,
         });
     }
     // POST https://api.kie.ai/api/v1/gpt4o-image/generate
     // Docs: https://docs.kie.ai/4o-image-api/generate-4-o-image
     async function gpt4oImageGenerate(req) {
-        return kieRequest(`${baseURL}/api/v1/gpt4o-image/generate`, {
+        return kieRequest(transport, {
             method: "POST",
+            path: "/api/v1/gpt4o-image/generate",
             body: req,
-            apiKey: opts.apiKey,
-            doFetch,
-            timeout,
         });
     }
     // POST https://api.kie.ai/api/v1/mj/generate
     // Docs: https://docs.kie.ai/mj-api/generate-mj-image
     async function mjGenerate(req) {
-        return kieRequest(`${baseURL}/api/v1/mj/generate`, {
+        return kieRequest(transport, {
             method: "POST",
+            path: "/api/v1/mj/generate",
             body: req,
+        });
+    }
+    // GET https://api.kie.ai/api/v1/mj/record-info?taskId={taskId}
+    // Docs: https://docs.kie.ai/mj-api/get-mj-task-details
+    async function mjRecordInfo(taskId) {
+        return kieRequest(`${baseURL}/api/v1/mj/record-info?taskId=${encodeURIComponent(taskId)}`, {
+            method: "GET",
             apiKey: opts.apiKey,
             doFetch,
             timeout,
@@ -487,128 +247,43 @@ export function createKie(opts) {
     // POST https://api.kie.ai/api/v1/runway/generate
     // Docs: https://docs.kie.ai/runway-api/generate-ai-video
     async function runwayGenerate(req) {
-        return kieRequest(`${baseURL}/api/v1/runway/generate`, {
+        return kieRequest(transport, {
             method: "POST",
+            path: "/api/v1/runway/generate",
             body: req,
-            apiKey: opts.apiKey,
-            doFetch,
-            timeout,
         });
     }
     // POST https://api.kie.ai/api/v1/runway/extend
     // Docs: https://docs.kie.ai/runway-api/extend-ai-video
     async function runwayExtend(req) {
-        return kieRequest(`${baseURL}/api/v1/runway/extend`, {
+        return kieRequest(transport, {
             method: "POST",
+            path: "/api/v1/runway/extend",
             body: req,
-            apiKey: opts.apiKey,
-            doFetch,
-            timeout,
         });
     }
     // GET https://api.kie.ai/api/v1/runway/record-detail?taskId={taskId}
     // Docs: https://docs.kie.ai/runway-api/get-ai-video-details
     async function runwayRecordDetail(taskId) {
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), timeout);
-        try {
-            const res = await doFetch(`${baseURL}/api/v1/runway/record-detail?taskId=${encodeURIComponent(taskId)}`, {
-                method: "GET",
-                headers: {
-                    Authorization: `Bearer ${opts.apiKey}`,
-                    "Content-Type": "application/json",
-                },
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-            if (!res.ok) {
-                let message = `Kie API error: ${res.status}`;
-                let body = null;
-                try {
-                    body = await res.json();
-                    if (typeof body === "object" &&
-                        body !== null &&
-                        "msg" in body &&
-                        typeof body.msg === "string") {
-                        message = `Kie API error ${res.status}: ${body.msg}`;
-                    }
-                }
-                catch {
-                    // ignore parse errors
-                }
-                throw new KieError(message, res.status, body);
-            }
-            return (await res.json());
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof KieError)
-                throw error;
-            throw new KieError(`Failed to get Runway task: ${error}`, 500);
-        }
+        return await transport.getJson(`/api/v1/runway/record-detail?taskId=${encodeURIComponent(taskId)}`);
     }
     // GET https://api.kie.ai/api/v1/flux/kontext/record-info?taskId={taskId}
     // Docs: https://docs.kie.ai/flux-kontext-api/get-image-details
     async function fluxKontextRecordInfo(taskId) {
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), timeout);
-        try {
-            const res = await doFetch(`${baseURL}/api/v1/flux/kontext/record-info?taskId=${encodeURIComponent(taskId)}`, {
-                method: "GET",
-                headers: {
-                    Authorization: `Bearer ${opts.apiKey}`,
-                    "Content-Type": "application/json",
-                },
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-            if (!res.ok) {
-                let message = `Kie API error: ${res.status}`;
-                let body = null;
-                try {
-                    body = await res.json();
-                    if (typeof body === "object" &&
-                        body !== null &&
-                        "msg" in body &&
-                        typeof body.msg === "string") {
-                        message = `Kie API error ${res.status}: ${body.msg}`;
-                    }
-                }
-                catch {
-                    // ignore parse errors
-                }
-                throw new KieError(message, res.status, body);
-            }
-            return (await res.json());
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof KieError)
-                throw error;
-            throw new KieError(`Failed to get Flux Kontext task: ${error}`, 500);
-        }
+        return await transport.getJson(`/api/v1/flux/kontext/record-info?taskId=${encodeURIComponent(taskId)}`);
     }
     // GET https://api.kie.ai/api/v1/chat/credit
     // Docs: https://docs.kie.ai/common-api/get-account-credits
     async function credit() {
-        const res = await doFetch(`${baseURL}/api/v1/chat/credit`, {
-            method: "GET",
-            headers: {
-                Authorization: `Bearer ${opts.apiKey}`,
-                "Content-Type": "application/json",
-            },
-        });
-        if (!res.ok) {
-            let body = null;
-            try {
-                body = await res.json();
-            }
-            catch {
-                // ignore parse errors
+        try {
+            return await transport.getJson("/api/v1/chat/credit");
+        }
+        catch (error) {
+            if (error instanceof KieError) {
+                throw new KieError(`Failed to get credits: ${error.status}`, error.status, error.body, error.code);
             }
-            throw new KieError(`Failed to get credits: ${res.status}`, res.status, body);
+            throw error;
         }
-        return (await res.json());
     }
     return attachExamples(withPaidGate("kie", {
         veo: withPaidGate("kie", createVeoProvider(baseURL, opts.apiKey, doFetch, timeout), { config: paygate }),
@@ -697,6 +372,12 @@ export function createKie(opts) {
                             responseSchema: Gpt4oImageRecordInfoResponseSchema,
                         }),
                     },
+                    mj: {
+                        recordInfo: Object.assign(mjRecordInfo, {
+                            schema: MjRecordInfoRequestSchema,
+                            responseSchema: MjRecordInfoResponseSchema,
+                        }),
+                    },
                     runway: {
                         recordDetail: Object.assign(runwayRecordDetail, {
                             schema: RecordInfoRequestSchema,