vidspotai-shared 1.0.29 → 1.0.31

package/lib/services/aiGen/providers/bytedance/bytedance.service.js

@@ -0,0 +1,188 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ByteDanceService = void 0;
+// bytedance.service.ts
+const baseAiGenProvider_service_1 = require("../baseAiGenProvider.service");
+const helpers_1 = require("../../helpers");
+const aiModels_1 = require("../../../../globals/aiModels");
+const types_1 = require("../../../../globals/types");
+const firebase_1 = require("../../../../libs/firebase");
+const helpers_2 = require("../../../../utils/helpers");
+const logger_1 = require("../../../../utils/logger");
+const crypto_1 = __importDefault(require("crypto"));
+class ByteDanceService extends baseAiGenProvider_service_1.BaseAiGenProviderService {
+    constructor() {
+        super(...arguments);
+        this.baseUrl = "https://ark.ap-southeast.bytepluses.com/api/v3/contents/generations/tasks";
+    }
+    async generateVideo(params) {
+        (0, helpers_1.validateParams)(params);
+        const modelConfig = aiModels_1.aiModelConfigs[params.modelKey];
+        const modelId = modelConfig.modelId;
+        const content = [
+            {
+                type: "text",
+                text: params.prompt,
+            },
+        ];
+        // Optional image → Image-to-Video (first frame)
+        if (params.inputImageUrl) {
+            content.push({
+                type: "image",
+                image_url: params.inputImageUrl,
+            });
+        }
+        const body = {
+            model: modelId,
+            content,
+            generate_audio: params.audio || false,
+            ratio: params.aspectRatio, // e.g. 16:9
+            resolution: params.resolution, // e.g. 720p
+            duration: params.duration, // seconds
+            watermark: false,
+            service_tier: "default",
+            execution_expires_after: 172800, // 48h
+        };
+        const resp = await fetch(this.baseUrl, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${process.env.BYTEDANCE_API_KEY}`,
+            },
+            body: JSON.stringify(body),
+        });
+        if (!resp.ok) {
+            const errText = await resp.text();
+            throw new Error(`ByteDance generateVideo failed (${resp.status}): ${errText}`);
+        }
+        const data = await resp.json();
+        if (!data.id) {
+            throw new Error("ByteDance API did not return task id");
+        }
+        return {
+            task: data.id,
+            status: types_1.EVideoSceneStatus.TRIGGERED,
+        };
+    }
+    async checkVideoStatus({ task, outputFilename, outputFilePath = "videos", }) {
+        const url = `https://ark.ap-southeast.bytepluses.com/api/v3/contents/generations/tasks/${task}`;
+        const resp = await fetch(url, {
+            method: "GET",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${process.env.BYTEDANCE_API_KEY}`,
+            },
+        });
+        if (!resp.ok) {
+            const errText = await resp.text();
+            throw new Error(`ByteDance checkVideoStatus failed (${resp.status}): ${errText}`);
+        }
+        const data = await resp.json();
+        // ---- Status mapping ----
+        switch (data.status) {
+            case "queued":
+            case "running":
+                return { status: types_1.EVideoSceneStatus.PENDING };
+            case "failed":
+            case "expired":
+            case "cancelled":
+                return {
+                    status: types_1.EVideoSceneStatus.FAILED,
+                    errorMessage: data.error
+                        ? JSON.stringify(data.error)
+                        : "ByteDance video generation failed",
+                };
+            case "succeeded":
+                break;
+            default:
+                return {
+                    status: types_1.EVideoSceneStatus.FAILED,
+                    errorMessage: `Unknown ByteDance task status: ${data.status}`,
+                };
+        }
+        // ---- Success path ----
+        const sourceVideoUrl = data.content?.video_url;
+        if (!sourceVideoUrl) {
+            return {
+                status: types_1.EVideoSceneStatus.FAILED,
+                errorMessage: "ByteDance response missing video_url",
+            };
+        }
+        // Download generated video
+        const videoResp = await fetch(sourceVideoUrl);
+        if (!videoResp.ok) {
+            throw new Error(`Failed to download ByteDance video (${videoResp.status})`);
+        }
+        const buffer = Buffer.from(await videoResp.arrayBuffer());
+        // ---- Upload to Firebase Storage (PUBLIC URL) ----
+        const filePath = `${outputFilePath}/${outputFilename}.mp4`;
+        const bucket = (0, firebase_1.getBucket)();
+        const file = bucket.file(filePath);
+        const downloadToken = crypto_1.default.randomUUID();
+        await file.save(buffer, {
+            contentType: "video/mp4",
+            resumable: false,
+            metadata: {
+                cacheControl: "public, max-age=31536000",
+                metadata: {
+                    firebaseStorageDownloadTokens: downloadToken,
+                },
+            },
+        });
+        const publicVideoUrl = `https://firebasestorage.googleapis.com/v0/b/${bucket.name}` +
+            `/o/${encodeURIComponent(filePath)}?alt=media&token=${downloadToken}`;
+        return {
+            videoUrl: publicVideoUrl,
+            status: types_1.EVideoSceneStatus.COMPLETED,
+        };
+    }
+    getCreditUsed(params) {
+        const modelConfig = aiModels_1.aiModelConfigs[params.modelKey];
+        const per1KTokens = params.audio
+            ? modelConfig.cost?.audio?.per1KTokens
+            : modelConfig.cost?.noAudio?.per1KTokens ?? modelConfig.cost?.per1KTokens;
+        if (!per1KTokens || !params.resolution || !params.aspectRatio || !params.duration) {
+            logger_1.logger.warn(`Missing cost or parameters for credit calculation. Returning default credits. Params: ${JSON.stringify(params)}, ModelConfig: ${JSON.stringify(modelConfig)}`);
+            return 10; // default fallback
+        }
+        const dims = dimensionsMap[params.resolution]?.[params.aspectRatio];
+        if (!dims) {
+            logger_1.logger.warn(`Unsupported resolution/aspectRatio for credit calculation. Returning default credits. Params: ${JSON.stringify(params)}`);
+            return 10; // default fallback
+        }
+        const FRAME_RATE = 24;
+        const tokensConsumed = Math.ceil((dims.width * dims.height * FRAME_RATE * params.duration) / 1024);
+        const cost = (per1KTokens * tokensConsumed) / 1000;
+        return (0, helpers_2.getCreditsFromCost)(cost);
+    }
+}
+exports.ByteDanceService = ByteDanceService;
+const dimensionsMap = {
+    "480p": {
+        "16:9": { width: 848, height: 480 },
+        "9:16": { width: 480, height: 848 },
+        "1:1": { width: 480, height: 480 },
+        "4:3": { width: 640, height: 480 },
+        "3:4": { width: 480, height: 640 },
+        "21:9": { width: 1008, height: 432 },
+    },
+    "720p": {
+        "16:9": { width: 1280, height: 720 },
+        "9:16": { width: 720, height: 1280 },
+        "1:1": { width: 720, height: 720 },
+        "4:3": { width: 960, height: 720 },
+        "3:4": { width: 720, height: 960 },
+        "21:9": { width: 1680, height: 720 },
+    },
+    "1080p": {
+        "16:9": { width: 1920, height: 1080 },
+        "9:16": { width: 1080, height: 1920 },
+        "1:1": { width: 1080, height: 1080 },
+        "4:3": { width: 1440, height: 1080 },
+        "3:4": { width: 1080, height: 1440 },
+        "21:9": { width: 2520, height: 1080 },
+    },
+};

package/lib/services/aiGen/providers/bytedance/constants.d.ts

@@ -0,0 +1,2 @@
+declare const dimensionsMap: Record<string, Record<string, VideoDimensions>>;
+//# sourceMappingURL=constants.d.ts.map

package/lib/services/aiGen/providers/bytedance/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/bytedance/constants.ts"],"names":[],"mappings":"AAEA,QAAA,MAAM,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,CAyBlE,CAAC"}

package/lib/services/aiGen/providers/bytedance/constants.js

@@ -0,0 +1,27 @@
+"use strict";
+const dimensionsMap = {
+    "480p": {
+        "21:9": { width: 960, height: 416 },
+        "16:9": { width: 864, height: 480 },
+        "9:16": { width: 480, height: 864 },
+        "1:1": { width: 640, height: 640 },
+        "4:3": { width: 736, height: 544 },
+        "3:4": { width: 544, height: 736 },
+    },
+    "720p": {
+        "21:9": { width: 1504, height: 640 },
+        "16:9": { width: 1248, height: 704 },
+        "9:16": { width: 704, height: 1248 },
+        "1:1": { width: 960, height: 960 },
+        "4:3": { width: 1120, height: 832 },
+        "3:4": { width: 832, height: 1120 },
+    },
+    "1080p": {
+        "21:9": { width: 2176, height: 928 },
+        "16:9": { width: 1920, height: 1088 },
+        "9:16": { width: 1088, height: 1920 },
+        "1:1": { width: 1440, height: 1440 },
+        "4:3": { width: 1664, height: 1248 },
+        "3:4": { width: 1248, height: 1664 },
+    },
+};

package/lib/services/aiGen/providers/bytedance/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./bytedance.service";
+//# sourceMappingURL=index.d.ts.map

package/lib/services/aiGen/providers/bytedance/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/bytedance/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC"}

package/lib/services/aiGen/providers/bytedance/index.js

@@ -0,0 +1,17 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./bytedance.service"), exports);

package/lib/services/aiGen/providers/bytedance/types.d.ts

@@ -0,0 +1,5 @@
+type VideoDimensions = {
+    width: number;
+    height: number;
+};
+//# sourceMappingURL=types.d.ts.map

package/lib/services/aiGen/providers/bytedance/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/bytedance/types.ts"],"names":[],"mappings":"AAAA,KAAK,eAAe,GAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,CAAC"}

package/lib/services/aiGen/providers/bytedance/types.js

@@ -0,0 +1 @@
+"use strict";

package/lib/services/aiGen/providers/google/google.service.js

@@ -24,7 +24,7 @@ class GoogleService extends baseAiGenProvider_service_1.BaseAiGenProviderService
         (0, helpers_2.validateParams)(params);
         const modelConfig = aiModels_1.aiModelConfigs[params.modelKey];
         const modelId = modelConfig.modelId;
-        console.log("GoogleService.generateVideo - modelId:", modelId, params);
+        // console.log("GoogleService.generateVideo - modelId:", modelId, params);
         const request = {
             model: modelId,
             prompt: params.prompt,

package/lib/services/aiGen/providers/kling/kling.service.d.ts

@@ -2,11 +2,16 @@ import { BaseAiGenProviderService } from "../baseAiGenProvider.service";
 import { CreditUsageParams, VideoGenerationParams, VideoGenerationResult, VideoStatusParams, VideoStatusResult } from "../types";
 export declare class KlingService extends BaseAiGenProviderService {
     private readonly baseUrl;
-    private readonly timeout;
+    private readonly generateTimeout;
+    private readonly statusTimeout;
     constructor();
     private request;
     generateVideo(params: VideoGenerationParams): Promise<VideoGenerationResult>;
-    checkVideoStatus({ task, outputFilename, outputFilePath, }: VideoStatusParams): Promise<VideoStatusResult>;
+    private generateAvatarVideo;
+    private requestWithTimeoutRetry;
+    checkVideoStatus(params: VideoStatusParams): Promise<VideoStatusResult>;
+    private checkRegularVideoStatus;
+    private checkAvatarVideoStatus;
     encodeJwtToken(): string;
     getCreditUsed({ modelKey, mode, duration, }: CreditUsageParams): number;
 }

package/lib/services/aiGen/providers/kling/kling.service.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"kling.service.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/kling/kling.service.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EACL,iBAAiB,EACjB,qBAAqB,EACrB,qBAAqB,EACrB,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,UAAU,CAAC;AAWlB,qBAAa,YAAa,SAAQ,wBAAwB;IACxD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAuC;IAC/D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;;YAWnB,OAAO;IA6Bf,aAAa,CACjB,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC;IA8B3B,gBAAgB,CAAC,EACrB,IAAI,EACJ,cAAc,EACd,cAAyB,GAC1B,EAAE,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA8DjD,cAAc,IAAI,MAAM;IAexB,aAAa,CAAC,EACZ,QAAQ,EACR,IAA8B,EAC9B,QAAY,GACb,EAAE,iBAAiB,GAAG,MAAM;CAK9B"}
+{"version":3,"file":"kling.service.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/kling/kling.service.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EACL,iBAAiB,EACjB,qBAAqB,EACrB,qBAAqB,EACrB,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,UAAU,CAAC;AAYlB,qBAAa,YAAa,SAAQ,wBAAwB;IACxD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAuC;IAC/D,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAW;IAC3C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAU;;YAW1B,OAAO;IA8Bf,aAAa,CACjB,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC;YAqCnB,mBAAmB;YAqCnB,uBAAuB;IAmC/B,gBAAgB,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;YAO/D,uBAAuB;YAgEvB,sBAAsB;IA6DpC,cAAc,IAAI,MAAM;IAexB,aAAa,CAAC,EACZ,QAAQ,EACR,IAA8B,EAC9B,QAAY,GACb,EAAE,iBAAiB,GAAG,MAAM;CAc9B"}

package/lib/services/aiGen/providers/kling/kling.service.js

@@ -10,6 +10,7 @@ const jsonwebtoken_1 = __importDefault(require("jsonwebtoken"));
 const aiModels_1 = require("../../../../globals/aiModels");
 const types_1 = require("../../../../globals/types");
 const firebase_1 = require("../../../../libs/firebase");
+const logger_1 = require("../../../../utils/logger");
 const helpers_1 = require("../../helpers");
 const types_2 = require("../../types");
 const baseAiGenProvider_service_1 = require("../baseAiGenProvider.service");
@@ -18,12 +19,13 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
     constructor() {
         super();
         this.baseUrl = "https://api-singapore.klingai.com";
-        this.timeout = 60000; // 1 min
+        this.generateTimeout = 120000; // 2 min — submit endpoint can be slow under Kling load
+        this.statusTimeout = 60000; // 1 min — status/poll should be fast
         if (!process.env.KLING_SECRET_KEY || !process.env.KLING_ACCESS_KEY) {
             throw new Error("Missing KLING_SECRET_KEY or KLING_ACCESS_KEY in environment variables");
         }
     }
-    async request(endpoint, method, body) {
+    async request(endpoint, method, body, timeout = this.generateTimeout) {
         const config = {
             method,
             url: `${this.baseUrl}${endpoint}`,
@@ -31,7 +33,7 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
                 Authorization: `Bearer ${this.encodeJwtToken()}`,
                 "Content-Type": "application/json",
             },
-            timeout: this.timeout,
+            timeout,
             data: body,
         };
         const response = await axios_1.default.request(config);
@@ -42,6 +44,9 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
     }
     async generateVideo(params) {
         (0, helpers_1.validateParams)(params);
+        if (params.modelKey === "kling-avatar") {
+            return this.generateAvatarVideo(params);
+        }
         const modelId = aiModels_1.aiModelConfigs[params.modelKey]?.modelId;
         if (!modelId)
             throw new Error(`Unknown modelKey: ${params.modelKey}`);
@@ -53,14 +58,75 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
             mode: params.mode || types_2.EVideoMode.STANDARD,
         };
         const starttime = Date.now();
-        const task = await this.request("/v1/videos/text2video", "POST", request);
-        console.log(`Kling generateVideo API call took ${Date.now() - starttime} ms`, task ? `(task ID: ${task.data.task_id})` : "(failed)");
+        const task = await this.requestWithTimeoutRetry("/v1/videos/text2video", "POST", request, 4, this.generateTimeout);
+        logger_1.logger.info("Kling generateVideo submitted", {
+            modelKey: params.modelKey,
+            taskId: task.data.task_id,
+            durationMs: Date.now() - starttime,
+        });
+        return { task: task.data.task_id, status: types_1.EVideoSceneStatus.TRIGGERED };
+    }
+    async generateAvatarVideo(params) {
+        if (!params.inputImageUrl) {
+            throw new Error("inputImageUrl (avatar face image) is required for Kling Avatar generation");
+        }
+        if (!params.inputAudioUrl) {
+            throw new Error("inputAudioUrl (audio file) is required for Kling Avatar generation");
+        }
+        const request = {
+            image: params.inputImageUrl,
+            sound_file: params.inputAudioUrl,
+            prompt: params.prompt,
+            mode: params.mode || "std",
+        };
+        const starttime = Date.now();
+        // Use requestWithTimeoutRetry so transient Kling server slowness doesn't fail the job outright.
+        // generateTimeout (2 min) gives enough headroom; up to 4 attempts before giving up.
+        const task = await this.requestWithTimeoutRetry("/v1/videos/avatar/image2video", "POST", request, 4, this.generateTimeout);
+        logger_1.logger.info("Kling generateAvatarVideo submitted", {
+            taskId: task.data.task_id,
+            durationMs: Date.now() - starttime,
+        });
         return { task: task.data.task_id, status: types_1.EVideoSceneStatus.TRIGGERED };
     }
-    async checkVideoStatus({ task, outputFilename, outputFilePath = "videos", }) {
+    async requestWithTimeoutRetry(endpoint, method, body, maxAttempts = 4, timeout = this.statusTimeout) {
+        for (let attempt = 1; attempt <= maxAttempts; attempt++) {
+            try {
+                return await this.request(endpoint, method, body, timeout);
+            }
+            catch (err) {
+                const isTimeout = err.code === "ECONNABORTED" || err.message?.includes("timeout");
+                const isNetworkError = err.code === "ECONNRESET" ||
+                    err.code === "ECONNREFUSED" ||
+                    err.code === "ENOTFOUND" ||
+                    err.code === "ERR_NETWORK" ||
+                    err?.name === "AggregateError";
+                const isRetryable = isTimeout || isNetworkError;
+                if (!isRetryable || attempt === maxAttempts)
+                    throw err;
+                logger_1.logger.warn("Kling request failed with transient error — retrying", {
+                    endpoint,
+                    attempt,
+                    maxAttempts,
+                    timeoutMs: timeout,
+                    errorCode: err.code,
+                    errorName: err?.name,
+                });
+                await new Promise((r) => setTimeout(r, 1000));
+            }
+        }
+        throw new Error("Unreachable");
+    }
+    async checkVideoStatus(params) {
+        if (params.modelKey === "kling-avatar") {
+            return this.checkAvatarVideoStatus(params);
+        }
+        return this.checkRegularVideoStatus(params);
+    }
+    async checkRegularVideoStatus({ task, outputFilename, outputFilePath = "videos", }) {
         const startTime = Date.now();
-        const result = await this.request(`/v1/videos/text2video/${task}`, "GET");
-        console.log(`Kling checkVideoStatus API call took ${Date.now() - startTime} ms`);
+        const result = await this.requestWithTimeoutRetry(`/v1/videos/text2video/${task}`, "GET");
+        logger_1.logger.info("Kling checkVideoStatus polled", { task, durationMs: Date.now() - startTime, status: result?.data?.task_status });
         if (!result) {
             return {
                 status: types_1.EVideoSceneStatus.FAILED,
@@ -78,10 +144,53 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
             // console.log("Kling video URL:", videoUrl);
             const filePath = `${outputFilePath}/${outputFilename}.mp4`;
             const file = (0, firebase_1.getBucket)().file(filePath);
-            // Download video using axios
+            // Download video using axios — use generateTimeout since video files can be large
             const videoResp = await axios_1.default.get(videoUrl, {
                 responseType: "arraybuffer",
-                timeout: this.timeout,
+                timeout: this.generateTimeout,
+            });
+            const buffer = Buffer.from(videoResp.data);
+            await file.save(buffer, { contentType: "video/mp4" });
+            const [signedUrl] = await file.getSignedUrl({
+                action: "read",
+                expires: "03-09-2491",
+            });
+            return {
+                videoUrl: signedUrl,
+                status: types_1.EVideoSceneStatus.COMPLETED,
+            };
+        }
+        if (result.data.task_status === "failed") {
+            return {
+                status: types_1.EVideoSceneStatus.FAILED,
+                errorMessage: `${result.code}: ${result.message}`,
+            };
+        }
+        return { status: types_1.EVideoSceneStatus.PENDING };
+    }
+    async checkAvatarVideoStatus({ task, outputFilename, outputFilePath = "videos", }) {
+        const startTime = Date.now();
+        const result = await this.requestWithTimeoutRetry(`/v1/videos/avatar/image2video/${task}`, "GET");
+        logger_1.logger.info("Kling checkAvatarVideoStatus polled", { task, durationMs: Date.now() - startTime, status: result?.data?.task_status });
+        if (!result) {
+            return {
+                status: types_1.EVideoSceneStatus.FAILED,
+                errorMessage: "No response from Kling Avatar API",
+            };
+        }
+        if (result.data.task_status === "succeed") {
+            const videoUrl = result.data.task_result?.videos?.[0]?.url;
+            if (!videoUrl) {
+                return {
+                    status: types_1.EVideoSceneStatus.FAILED,
+                    errorMessage: "No video URL in avatar response",
+                };
+            }
+            const filePath = `${outputFilePath}/${outputFilename}.mp4`;
+            const file = (0, firebase_1.getBucket)().file(filePath);
+            const videoResp = await axios_1.default.get(videoUrl, {
+                responseType: "arraybuffer",
+                timeout: this.generateTimeout,
             });
             const buffer = Buffer.from(videoResp.data);
             await file.save(buffer, { contentType: "video/mp4" });
@@ -116,6 +225,13 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
     }
     getCreditUsed({ modelKey, mode = types_2.EVideoMode.PROFESSIONAL, duration = 5, }) {
         const modelConfig = aiModels_1.aiModelConfigs[modelKey];
+        // Avatar: per-second pricing based on mode
+        if (modelKey === "kling-avatar") {
+            const perSecond = modelConfig?.cost?.[mode]?.perSecond
+                ?? modelConfig?.cost?.["std"]?.perSecond
+                ?? 0.014;
+            return (0, helpers_2.getCreditsFromCost)(perSecond * duration);
+        }
         const cost = modelConfig?.cost?.table?.[mode]?.[duration];
         return (0, helpers_2.getCreditsFromCost)(cost ?? 1);
     }

package/lib/services/aiGen/providers/kling/types.d.ts

@@ -68,6 +68,19 @@ export interface TaskResult {
 export interface VideoResult {
     id: string;
     url: string;
+    watermark_url?: string;
     duration: string;
 }
+export interface KlingAvatarGenerateRequest {
+    image: string;
+    audio_id?: string;
+    sound_file?: string;
+    prompt?: string;
+    mode?: "std" | "pro";
+    watermark_info?: {
+        enabled: boolean;
+    };
+    callback_url?: string;
+    external_task_id?: string;
+}
 //# sourceMappingURL=types.d.ts.map

package/lib/services/aiGen/providers/kling/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/kling/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,MAAM,WAAW,0BAA0B;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,aAAa,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,WAAW,GAAG,YAAY,GAAG,SAAS,GAAG,QAAQ,CAAC;IAC/D,SAAS,EAAE,QAAQ,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,QAAQ;IACvB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,MAAM,cAAc,GACtB,UAAU,GACV,YAAY,GACZ,iBAAiB,GACjB,mBAAmB,CAAC;AAExB,MAAM,MAAM,UAAU,GAClB,QAAQ,GACR,WAAW,GACX,YAAY,GACZ,oBAAoB,GACpB,mBAAmB,CAAC;AAExB,MAAM,MAAM,gBAAgB,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC;AAEvD,MAAM,MAAM,aAAa,GAAG,CAAC,GAAG,EAAE,CAAC;AAEnC,MAAM,WAAW,YAAY;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,EAAE,UAAU,CAAC;IAClB,MAAM,CAAC,EAAE,YAAY,CAAC;CACvB;AAED,MAAM,WAAW,yBAAyB;IACxC,UAAU,CAAC,EAAE,cAAc,CAAC;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,UAAU,CAAC;IAClB,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,YAAY,CAAC,EAAE,gBAAgB,CAAC;IAChC,QAAQ,CAAC,EAAE,aAAa,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAGD,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,aAAa,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,UAAU,CAAC;IACxB,eAAe,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,QAAQ,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,UAAU,CAAC;CAC1B;AAED,MAAM,MAAM,UAAU,GAAG,WAAW,GAAG,YAAY,GAAG,SAAS,GAAG,QAAQ,CAAC;AAE3E,MAAM,WAAW,QAAQ;IACvB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,UAAU;IACzB,MAAM,EAAE,WAAW,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;CAClB"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/kling/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,MAAM,WAAW,0BAA0B;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,aAAa,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,WAAW,GAAG,YAAY,GAAG,SAAS,GAAG,QAAQ,CAAC;IAC/D,SAAS,EAAE,QAAQ,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,QAAQ;IACvB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,MAAM,cAAc,GACtB,UAAU,GACV,YAAY,GACZ,iBAAiB,GACjB,mBAAmB,CAAC;AAExB,MAAM,MAAM,UAAU,GAClB,QAAQ,GACR,WAAW,GACX,YAAY,GACZ,oBAAoB,GACpB,mBAAmB,CAAC;AAExB,MAAM,MAAM,gBAAgB,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC;AAEvD,MAAM,MAAM,aAAa,GAAG,CAAC,GAAG,EAAE,CAAC;AAEnC,MAAM,WAAW,YAAY;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,EAAE,UAAU,CAAC;IAClB,MAAM,CAAC,EAAE,YAAY,CAAC;CACvB;AAED,MAAM,WAAW,yBAAyB;IACxC,UAAU,CAAC,EAAE,cAAc,CAAC;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,UAAU,CAAC;IAClB,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,YAAY,CAAC,EAAE,gBAAgB,CAAC;IAChC,QAAQ,CAAC,EAAE,aAAa,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAGD,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,aAAa,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,UAAU,CAAC;IACxB,eAAe,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,QAAQ,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,UAAU,CAAC;CAC1B;AAED,MAAM,MAAM,UAAU,GAAG,WAAW,GAAG,YAAY,GAAG,SAAS,GAAG,QAAQ,CAAC;AAE3E,MAAM,WAAW,QAAQ;IACvB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,UAAU;IACzB,MAAM,EAAE,WAAW,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAID,MAAM,WAAW,0BAA0B;IACzC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC;IACrB,cAAc,CAAC,EAAE;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IACtC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B"}

package/lib/services/aiGen/providers/pixverse/pixverse.service.d.ts

@@ -0,0 +1,9 @@
+import { BaseAiGenProviderService } from "../baseAiGenProvider.service";
+import { CreditUsageParams, VideoGenerationParams, VideoGenerationResult, VideoStatusParams, VideoStatusResult } from "../types";
+export declare class PixVerseService extends BaseAiGenProviderService {
+    private readonly baseUrl;
+    generateVideo(params: VideoGenerationParams): Promise<VideoGenerationResult>;
+    checkVideoStatus({ task, outputFilename, outputFilePath, }: VideoStatusParams): Promise<VideoStatusResult>;
+    getCreditUsed(params: CreditUsageParams): number;
+}
+//# sourceMappingURL=pixverse.service.d.ts.map

package/lib/services/aiGen/providers/pixverse/pixverse.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pixverse.service.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/pixverse/pixverse.service.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EACL,iBAAiB,EACjB,qBAAqB,EACrB,qBAAqB,EACrB,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,UAAU,CAAC;AASlB,qBAAa,eAAgB,SAAQ,wBAAwB;IAC3D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAkD;IAKpE,aAAa,CACjB,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC;IAqE3B,gBAAgB,CAAC,EACrB,IAAI,EACJ,cAAc,EACd,cAAyB,GAC1B,EAAE,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAmGjD,aAAa,CAAC,MAAM,EAAE,iBAAiB,GAAG,MAAM;CA+CjD"}