vidspotai-shared 1.0.82 → 1.0.83

package/lib/services/aiGen/providers/google/googleMusic.d.ts

@@ -0,0 +1,15 @@
+import { MusicGenerationParams, MusicGenerationResult } from "../types";
+/**
+ * Lyria 2 (Vertex AI) music generation. Standalone — does NOT use the
+ * GoogleGenAI (Gemini Developer API) client; it auths via ADC and hits the
+ * Vertex predict endpoint directly. Extracted from GoogleService so the class
+ * stays focused on the Gemini-SDK paths.
+ *
+ * Sync — POST {region}-aiplatform.googleapis.com/.../lyria-002:predict returns
+ * base64-encoded WAV audio inline. Auth via ADC on the function service account
+ * (or GOOGLE_APPLICATION_CREDENTIALS locally).
+ *
+ * Required env: GCP_PROJECT_ID. Optional: GCP_LOCATION (default us-central1).
+ */
+export declare function generateGoogleMusic(params: MusicGenerationParams): Promise<MusicGenerationResult>;
+//# sourceMappingURL=googleMusic.d.ts.map

package/lib/services/aiGen/providers/google/googleMusic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"googleMusic.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/google/googleMusic.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,qBAAqB,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAExE;;;;;;;;;;;GAWG;AACH,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC,CAmEhC"}

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

@@ -0,0 +1,77 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateGoogleMusic = generateGoogleMusic;
+const google_auth_library_1 = require("google-auth-library");
+const aiModels_1 = require("../../../../globals/aiModels");
+const firebase_1 = require("../../../../libs/firebase");
+/**
+ * Lyria 2 (Vertex AI) music generation. Standalone — does NOT use the
+ * GoogleGenAI (Gemini Developer API) client; it auths via ADC and hits the
+ * Vertex predict endpoint directly. Extracted from GoogleService so the class
+ * stays focused on the Gemini-SDK paths.
+ *
+ * Sync — POST {region}-aiplatform.googleapis.com/.../lyria-002:predict returns
+ * base64-encoded WAV audio inline. Auth via ADC on the function service account
+ * (or GOOGLE_APPLICATION_CREDENTIALS locally).
+ *
+ * Required env: GCP_PROJECT_ID. Optional: GCP_LOCATION (default us-central1).
+ */
+async function generateGoogleMusic(params) {
+    const modelConfig = aiModels_1.aiModelConfigs[params.modelKey];
+    const modelId = modelConfig?.modelId ?? "lyria-002";
+    const projectId = process.env.GCP_PROJECT_ID;
+    if (!projectId) {
+        throw new Error("Missing GCP_PROJECT_ID env var (required for Vertex Lyria)");
+    }
+    const location = process.env.GCP_LOCATION ?? "us-central1";
+    const auth = new google_auth_library_1.GoogleAuth({
+        scopes: ["https://www.googleapis.com/auth/cloud-platform"],
+    });
+    const client = await auth.getClient();
+    const tokenRes = await client.getAccessToken();
+    const token = tokenRes?.token;
+    if (!token)
+        throw new Error("Failed to obtain GCP access token for Vertex Lyria");
+    const url = `https://${location}-aiplatform.googleapis.com/v1/projects/${projectId}` +
+        `/locations/${location}/publishers/google/models/${modelId}:predict`;
+    const instance = { prompt: params.prompt };
+    const parameters = {
+        sample_count: 1,
+    };
+    if (params.negativePrompt) {
+        instance.negative_prompt = params.negativePrompt;
+    }
+    if (params.seed !== undefined) {
+        parameters.seed = params.seed;
+    }
+    const resp = await fetch(url, {
+        method: "POST",
+        headers: {
+            Authorization: `Bearer ${token}`,
+            "Content-Type": "application/json",
+        },
+        body: JSON.stringify({ instances: [instance], parameters }),
+    });
+    if (!resp.ok) {
+        const errText = await resp.text();
+        throw new Error(`Vertex Lyria failed (${resp.status}): ${errText}`);
+    }
+    const data = await resp.json();
+    const audioB64 = data?.predictions?.[0]?.bytesBase64Encoded;
+    if (!audioB64) {
+        throw new Error("Vertex Lyria returned no audio data");
+    }
+    const buffer = Buffer.from(audioB64, "base64");
+    const path = `music/lyria/${params.outputFilename}.wav`;
+    const file = (0, firebase_1.getBucket)().file(path);
+    await file.save(buffer, { contentType: "audio/wav" });
+    const [signed] = await file.getSignedUrl({
+        action: "read",
+        expires: "03-09-2491",
+    });
+    return {
+        audioUrl: signed,
+        mimeType: "audio/wav",
+        extension: "wav",
+    };
+}

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

@@ -1,5 +1,11 @@
 import { BaseAiGenProviderService } from "../baseAiGenProvider.service";
 import { CreditUsageParams, VideoGenerationParams, VideoGenerationResult, VideoStatusParams, VideoStatusResult } from "../types";
+/**
+ * Kling provider. This class is the transport (JWT auth, IPv4-pinned agents,
+ * timeout/retry) plus the per-endpoint submit dispatch. Request-body building
+ * lives in ./klingRequests, status polling + result stashing in ./klingStatus,
+ * and credit math in ./klingCredits.
+ */
 export declare class KlingService extends BaseAiGenProviderService {
     private readonly baseUrl;
     private readonly generateTimeout;
@@ -21,9 +27,7 @@ export declare class KlingService extends BaseAiGenProviderService {
     private generateAvatarVideo;
     private requestWithTimeoutRetry;
     checkVideoStatus(params: VideoStatusParams): Promise<VideoStatusResult>;
-    private checkRegularVideoStatus;
-    private checkAvatarVideoStatus;
     encodeJwtToken(): string;
-    getCreditUsed({ modelKey, mode, duration, multiClip, audio, inputType, }: CreditUsageParams): number;
+    getCreditUsed(params: CreditUsageParams): number;
 }
 //# sourceMappingURL=kling.service.d.ts.map

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":"AAYA,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EACL,iBAAiB,EACjB,qBAAqB,EACrB,qBAAqB,EACrB,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,UAAU,CAAC;AAoBlB,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;IASxC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAkD;IAC9E,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAmD;;YAWlE,OAAO;IAgCf,aAAa,CACjB,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC;IAsGjC;;;;OAIG;YACW,WAAW;YAkCX,iBAAiB;YAuEjB,qBAAqB;YAyCrB,eAAe;YAiEf,mBAAmB;YAuCnB,uBAAuB;IAwF/B,gBAAgB,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;YAO/D,uBAAuB;YAiFvB,sBAAsB;IA2GpC,cAAc,IAAI,MAAM;IAexB,aAAa,CAAC,EACZ,QAAQ,EACR,IAA8B,EAC9B,QAAY,EACZ,SAAiB,EACjB,KAAa,EACb,SAAS,GACV,EAAE,iBAAiB,GAAG,MAAM;CAwD9B"}
+{"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;AAsBlB;;;;;GAKG;AACH,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;IASxC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAkD;IAC9E,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAmD;;YAWlE,OAAO;IAgCf,aAAa,CACjB,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC;IAoDjC;;;;OAIG;YACW,WAAW;YA6BX,iBAAiB;YA6BjB,qBAAqB;YAgCrB,eAAe;YAkDf,mBAAmB;YAgCnB,uBAAuB;IAwF/B,gBAAgB,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAW7E,cAAc,IAAI,MAAM;IAexB,aAAa,CAAC,MAAM,EAAE,iBAAiB,GAAG,MAAM;CAGjD"}

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

@@ -11,14 +11,19 @@ const https_1 = __importDefault(require("https"));
 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");
-const helpers_2 = require("../../../../utils/helpers");
 const errors_1 = require("../../../../utils/errors");
-const cameraAdapter_1 = require("./cameraAdapter");
+const klingRequests_1 = require("./klingRequests");
+const klingCredits_1 = require("./klingCredits");
+const klingStatus_1 = require("./klingStatus");
+/**
+ * Kling provider. This class is the transport (JWT auth, IPv4-pinned agents,
+ * timeout/retry) plus the per-endpoint submit dispatch. Request-body building
+ * lives in ./klingRequests, status polling + result stashing in ./klingStatus,
+ * and credit math in ./klingCredits.
+ */
 class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService {
     constructor() {
         super();
@@ -78,67 +83,9 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
         const modelId = aiModels_1.aiModelConfigs[params.modelKey]?.modelId;
         if (!modelId)
             throw new Error(`Unknown modelKey: ${params.modelKey}`);
-        // Common optional fields shared by text2video and image2video endpoints.
-        const optional = {};
-        if (params.negativePrompt)
-            optional.negative_prompt = params.negativePrompt;
-        if (params.cfgScale !== undefined)
-            optional.cfg_scale = params.cfgScale;
-        if (params.callbackUrl)
-            optional.callback_url = params.callbackUrl;
-        if (params.externalTaskId)
-            optional.external_task_id = params.externalTaskId;
-        if (params.cameraControl) {
-            const cc = (0, cameraAdapter_1.toKlingCameraControl)(params.cameraControl);
-            if (cc)
-                optional.camera_control = cc;
-        }
-        // v2.6 uses `sound: "on"|"off"` as a pricing-tier toggle. v3 uses
-        // `generate_audio: boolean` for native audio synthesis. Route by model.
-        if (params.audio !== undefined) {
-            if (params.modelKey === "kling-v3.0") {
-                optional.generate_audio = Boolean(params.audio);
-            }
-            else if (params.modelKey === "kling-v2.6") {
-                optional.sound = params.audio ? "on" : "off";
-            }
-        }
         const isImageToVideo = !!params.inputImageUrl;
         const endpoint = isImageToVideo ? "/v1/videos/image2video" : "/v1/videos/text2video";
-        let request;
-        if (isImageToVideo) {
-            const i2v = {
-                model_name: modelId,
-                prompt: params.prompt,
-                image: params.inputImageUrl,
-                duration: params.duration || 5,
-                mode: params.mode || types_2.EVideoMode.STANDARD,
-                ...optional,
-            };
-            if (params.aspectRatio)
-                i2v.aspect_ratio = params.aspectRatio;
-            if (params.lastFrameImageUrl)
-                i2v.image_tail = params.lastFrameImageUrl;
-            if (params.staticMaskUrl)
-                i2v.static_mask = params.staticMaskUrl;
-            if (params.dynamicMasks?.length) {
-                i2v.dynamic_masks = params.dynamicMasks.slice(0, 6).map((d) => ({
-                    mask: d.maskUrl,
-                    trajectories: d.trajectories,
-                }));
-            }
-            request = i2v;
-        }
-        else {
-            request = {
-                model_name: modelId,
-                prompt: params.prompt,
-                aspect_ratio: params.aspectRatio || "16:9",
-                duration: params.duration || 5,
-                mode: params.mode || types_2.EVideoMode.STANDARD,
-                ...optional,
-            };
-        }
+        const request = (0, klingRequests_1.buildVideoGenerateRequest)(params, modelId);
         const starttime = Date.now();
         const task = await this.requestWithTimeoutRetry(endpoint, "POST", request, 4, this.generateTimeout);
         logger_1.logger.info("Kling generateVideo submitted", {
@@ -157,17 +104,7 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
         if (!params.sourceTaskId) {
             throw new Error("sourceTaskId (prior Kling task id) is required for video extension");
         }
-        const body = { video_id: params.sourceTaskId };
-        if (params.prompt)
-            body.prompt = params.prompt;
-        if (params.negativePrompt)
-            body.negative_prompt = params.negativePrompt;
-        if (params.cfgScale !== undefined)
-            body.cfg_scale = params.cfgScale;
-        if (params.callbackUrl)
-            body.callback_url = params.callbackUrl;
-        if (params.externalTaskId)
-            body.external_task_id = params.externalTaskId;
+        const body = (0, klingRequests_1.buildExtendBody)(params);
         const starttime = Date.now();
         const task = await this.requestWithTimeoutRetry("/v1/videos/video-extend", "POST", body, 4, this.generateTimeout);
         logger_1.logger.info("Kling extendVideo submitted", {
@@ -182,51 +119,7 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
         const modelId = aiModels_1.aiModelConfigs[params.modelKey]?.modelId;
         if (!modelId)
             throw new Error(`Unknown modelKey: ${params.modelKey}`);
-        const body = {
-            model_name: modelId,
-            mode: params.mode || types_2.EVideoMode.STANDARD,
-            duration: String(params.duration ?? 5),
-        };
-        if (params.prompt)
-            body.prompt = params.prompt;
-        if (params.aspectRatio)
-            body.aspect_ratio = params.aspectRatio;
-        if (params.callbackUrl)
-            body.callback_url = params.callbackUrl;
-        if (params.externalTaskId)
-            body.external_task_id = params.externalTaskId;
-        if (params.audio !== undefined)
-            body.sound = params.audio ? "on" : "off";
-        // Build image_list: start frame, end frame, plain reference images.
-        const imageList = [];
-        if (params.inputImageUrl) {
-            imageList.push({ image_url: params.inputImageUrl, type: "first_frame" });
-        }
-        if (params.lastFrameImageUrl) {
-            imageList.push({ image_url: params.lastFrameImageUrl, type: "end_frame" });
-        }
-        for (const url of params.referenceImageUrls ?? []) {
-            imageList.push({ image_url: url });
-        }
-        if (imageList.length)
-            body.image_list = imageList;
-        // Build video_list: reference videos (feature type) or base videos (editing).
-        if (params.referenceVideoUrls?.length) {
-            body.video_list = params.referenceVideoUrls.map((url) => ({
-                video_url: url,
-                refer_type: "feature",
-            }));
-        }
-        // Multi-shot support. MultiShot is Shot[] directly (not { shots: Shot[] }).
-        if (params.multiShot?.length) {
-            body.multi_shot = true;
-            body.shot_type = "customize";
-            body.multi_prompt = params.multiShot.map((s, i) => ({
-                index: i + 1,
-                prompt: s.prompt,
-                ...(s.duration !== undefined ? { duration: String(s.duration) } : {}),
-            }));
-        }
+        const body = (0, klingRequests_1.buildOmniVideoBody)(params, modelId);
         const starttime = Date.now();
         const task = await this.requestWithTimeoutRetry("/v1/videos/omni-video", "POST", body, 4, this.generateTimeout);
         logger_1.logger.info("Kling generateOmniVideo submitted", {
@@ -246,19 +139,7 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
             throw new Error("inputVideoUrl (motion reference video) is required for motion control");
         }
         const modelId = aiModels_1.aiModelConfigs[params.modelKey]?.modelId ?? "kling-v3";
-        const body = {
-            model_name: modelId,
-            image_url: params.inputImageUrl,
-            video_url: params.inputVideoUrl,
-        };
-        if (params.prompt)
-            body.prompt = params.prompt;
-        if (params.mode)
-            body.mode = params.mode;
-        if (params.callbackUrl)
-            body.callback_url = params.callbackUrl;
-        if (params.externalTaskId)
-            body.external_task_id = params.externalTaskId;
+        const body = (0, klingRequests_1.buildMotionControlBody)(params, modelId);
         const starttime = Date.now();
         const task = await this.requestWithTimeoutRetry("/v1/videos/motion-control", "POST", body, 4, this.generateTimeout);
         logger_1.logger.info("Kling generateMotionControl submitted", {
@@ -286,21 +167,7 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
         }
         // Auto-select the first detected face.
         const face = faceResp.data.face_data[0];
-        const audioDurationMs = params.audioDurationMs ?? 60000;
-        const lipSyncBody = {
-            session_id: faceResp.data.session_id,
-            face_choose: [{
-                    face_id: face.face_id,
-                    sound_file: params.inputAudioUrl,
-                    sound_insert_time: face.start_time,
-                    sound_start_time: 0,
-                    sound_end_time: audioDurationMs,
-                    sound_volume: 1,
-                    original_audio_volume: 1,
-                }],
-            ...(params.callbackUrl ? { callback_url: params.callbackUrl } : {}),
-            ...(params.externalTaskId ? { external_task_id: params.externalTaskId } : {}),
-        };
+        const lipSyncBody = (0, klingRequests_1.buildLipSyncBody)(params, faceResp.data.session_id, face);
         const starttime = Date.now();
         const task = await this.requestWithTimeoutRetry("/v1/videos/advanced-lip-sync", "POST", lipSyncBody, 4, this.generateTimeout);
         logger_1.logger.info("Kling generateLipSync submitted", {
@@ -318,14 +185,7 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
         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",
-            ...(params.callbackUrl ? { callback_url: params.callbackUrl } : {}),
-            ...(params.externalTaskId ? { external_task_id: params.externalTaskId } : {}),
-        };
+        const request = (0, klingRequests_1.buildAvatarRequest)(params);
         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.
@@ -408,171 +268,14 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
         throw new Error("Unreachable");
     }
     async checkVideoStatus(params) {
+        const deps = {
+            request: this.requestWithTimeoutRetry.bind(this),
+            generateTimeout: this.generateTimeout,
+        };
         if (params.modelKey === "kling-avatar") {
-            return this.checkAvatarVideoStatus(params);
-        }
-        return this.checkRegularVideoStatus(params);
-    }
-    async checkRegularVideoStatus({ task, outputFilename, outputFilePath = "videos", }) {
-        const startTime = Date.now();
-        // Try each Kling status endpoint in order until one returns non-404.
-        // We don't persist which endpoint was used at submit time, so we probe all of them.
-        const statusEndpoints = [
-            `/v1/videos/text2video/${task}`,
-            `/v1/videos/image2video/${task}`,
-            `/v1/videos/video-extend/${task}`,
-            `/v1/videos/omni-video/${task}`,
-            `/v1/videos/motion-control/${task}`,
-            `/v1/videos/advanced-lip-sync/${task}`,
-        ];
-        let result;
-        for (const ep of statusEndpoints) {
-            try {
-                result = await this.requestWithTimeoutRetry(ep, "GET");
-                break; // found it
-            }
-            catch (err) {
-                if (err.response?.status !== 404)
-                    throw err;
-                // 404 → try next endpoint
-            }
-        }
-        if (!result) {
-            return {
-                status: types_1.EVideoSceneStatus.FAILED,
-                errorMessage: `Task ${task} not found on any Kling status endpoint`,
-            };
+            return (0, klingStatus_1.checkAvatarVideoStatus)(deps, params);
         }
-        logger_1.logger.info("Kling checkVideoStatus polled", { task, durationMs: Date.now() - startTime, status: result.data.task_status });
-        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 response",
-                };
-            }
-            // console.log("Kling video URL:", videoUrl);
-            const filePath = `${outputFilePath}/${outputFilename}.mp4`;
-            const file = (0, firebase_1.getBucket)().file(filePath);
-            // Download video using axios — use generateTimeout since video files can be large
-            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" });
-            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.data.task_status_msg || result.message || "Kling task failed (no reason provided)",
-            };
-        }
-        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, statusMsg: result?.data?.task_status_msg || undefined });
-        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",
-                };
-            }
-            // Kling task is done. Now download from Kling's CDN, push to GCS, and
-            // mint a signed URL. Any of these three can hiccup (Kling CDN slow,
-            // GCS 503, signedURL service blip) — historically the whole call threw
-            // and the outer TaskPoller burned a poll attempt re-asking Kling for a
-            // status that's already known. Wrap the post-success work in its own
-            // bounded retry so a transient blip doesn't poison the poll loop.
-            const filePath = `${outputFilePath}/${outputFilename}.mp4`;
-            const file = (0, firebase_1.getBucket)().file(filePath);
-            const maxAttempts = 3;
-            let lastErr;
-            // Force IPv4 on the download. The Kling-returned video URL frequently
-            // resolves to a CDN host where IPv6 is broken or routes us to a
-            // geo-blocked PoP, producing an AggregateError(ETIMEDOUT) where both
-            // happy-eyeballs sockets fail. Pinning v4 avoids the IPv6 dead-end.
-            const v4Http = new http_1.default.Agent({ family: 4, keepAlive: true });
-            const v4Https = new https_1.default.Agent({ family: 4, keepAlive: true });
-            // Parse host so we can log which CDN we couldn't reach.
-            let downloadHost = "unknown";
-            try {
-                downloadHost = new URL(videoUrl).host;
-            }
-            catch {
-                /* leave as unknown */
-            }
-            for (let attempt = 1; attempt <= maxAttempts; attempt++) {
-                try {
-                    const videoResp = await axios_1.default.get(videoUrl, {
-                        responseType: "arraybuffer",
-                        timeout: this.generateTimeout,
-                        httpAgent: v4Http,
-                        httpsAgent: v4Https,
-                    });
-                    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,
-                    };
-                }
-                catch (err) {
-                    lastErr = err;
-                    const isErr = err instanceof Error;
-                    logger_1.logger.warn("Kling avatar post-success stash failed", {
-                        task,
-                        attempt,
-                        maxAttempts,
-                        downloadHost,
-                        errName: isErr ? err.name : typeof err,
-                        errCode: isErr ? err.code : undefined,
-                        httpStatus: isErr ? err.response?.status : undefined,
-                        msg: isErr ? err.message : String(err),
-                    });
-                    if (attempt < maxAttempts) {
-                        await new Promise((r) => setTimeout(r, 2000 * 2 ** (attempt - 1)));
-                    }
-                }
-            }
-            // All internal retries exhausted — surface as pending so the outer
-            // TaskPoller retries the WHOLE checkVideoStatus call (with its own
-            // budget). Throwing here would burn an outer attempt with no extra
-            // signal beyond what we already logged.
-            throw lastErr instanceof Error
-                ? lastErr
-                : new Error(`Kling avatar stash failed after ${maxAttempts} attempts`);
-        }
-        if (result.data.task_status === "failed") {
-            return {
-                status: types_1.EVideoSceneStatus.FAILED,
-                errorMessage: result.data.task_status_msg || result.message || "Kling avatar task failed (no reason provided)",
-            };
-        }
-        return { status: types_1.EVideoSceneStatus.PENDING };
+        return (0, klingStatus_1.checkRegularVideoStatus)(deps, params);
     }
     encodeJwtToken() {
         const payload = {
@@ -586,55 +289,8 @@ class KlingService extends baseAiGenProvider_service_1.BaseAiGenProviderService
         };
         return jsonwebtoken_1.default.sign(payload, process.env.KLING_SECRET_KEY, options);
     }
-    getCreditUsed({ modelKey, mode = types_2.EVideoMode.PROFESSIONAL, duration = 5, multiClip = false, audio = false, inputType, }) {
-        const modelConfig = aiModels_1.aiModelConfigs[modelKey];
-        const applyFloor = !multiClip;
-        // Avatar: per-second pricing based on mode. Kling lipsync bills in
-        // 5-second increments (a 7s clip is billed as 10s).
-        if (modelKey === "kling-avatar") {
-            const perSecond = modelConfig?.cost?.perMode?.[mode]?.perSecond
-                ?? modelConfig?.cost?.perMode?.["std"]?.perSecond
-                ?? 0.014;
-            const billedSeconds = Math.ceil(duration / 5) * 5;
-            return (0, helpers_2.getCreditsFromCost)(perSecond * billedSeconds, applyFloor);
-        }
-        // v2.6 splits cost by audio/noAudio tier (audio is 2× upcharge).
-        if (modelKey === "kling-v2.6") {
-            const branch = audio ? modelConfig?.cost?.audio : modelConfig?.cost?.noAudio;
-            const cost = branch?.table?.[mode]?.[duration];
-            if (cost === undefined || cost === null) {
-                logger_1.logger.warn(`Kling getCreditUsed: no v2.6 cost entry for mode="${mode}" duration=${duration} audio=${audio} — returning fallback`, { modelKey, mode, duration, audio });
-                return (0, helpers_2.getCreditsFromCost)(1.0, applyFloor);
-            }
-            return (0, helpers_2.getCreditsFromCost)(cost, applyFloor);
-        }
-        const baseCost = modelConfig?.cost?.table?.[mode]?.[duration];
-        if (baseCost === undefined || baseCost === null) {
-            logger_1.logger.warn(`Kling getCreditUsed: no cost entry for modelKey="${modelKey}" mode="${mode}" duration=${duration} — returning fallback`, { modelKey, mode, duration });
-            return (0, helpers_2.getCreditsFromCost)(1.0, applyFloor);
-        }
-        // Deferred-table upcharges. The cost tables hold T2V / no-audio BASE rates;
-        // Kling bills more for two cases that don't have separate table keys yet
-        // (see config comments in globals/aiModels/providers/kling.ts):
-        //   • I2V (image- or video-conditioned) → 1.5× on kling-video-o1 and
-        //     kling-v3-omni-video (verified 2026-05-20: std $0.084→$0.126/s).
-        //   • generate_audio=true → 1.5× on kling-v3.0.
-        // Applied here so the reconciled ACTUAL cost reflects them; without this the
-        // multi-clip (agent) path under-bills these calls (single clips are covered
-        // by the per-call floor regardless).
-        const I2V_UPCHARGE_MODELS = [
-            "kling-video-o1",
-            "kling-v3-omni-video",
-        ];
-        let cost = baseCost;
-        if (I2V_UPCHARGE_MODELS.includes(modelKey) &&
-            (inputType === "image" || inputType === "video")) {
-            cost *= 1.5;
-        }
-        if (modelKey === "kling-v3.0" && audio) {
-            cost *= 1.5;
-        }
-        return (0, helpers_2.getCreditsFromCost)(cost, applyFloor);
+    getCreditUsed(params) {
+        return (0, klingCredits_1.getKlingCreditUsed)(params);
     }
 }
 exports.KlingService = KlingService;

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

@@ -0,0 +1,9 @@
+import { CreditUsageParams } from "../types";
+/**
+ * Kling credit calculation. Pure (config lookup + arithmetic) — broken out of
+ * KlingService.getCreditUsed. Handles the three pricing shapes: avatar
+ * (per-second, 5s-rounded), v2.6 (audio/noAudio tier table), and the base
+ * table with deferred I2V / generate_audio upcharges.
+ */
+export declare function getKlingCreditUsed({ modelKey, mode, duration, multiClip, audio, inputType, }: CreditUsageParams): number;
+//# sourceMappingURL=klingCredits.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"klingCredits.d.ts","sourceRoot":"","sources":["../../../../../src/services/aiGen/providers/kling/klingCredits.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAE7C;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAAC,EACjC,QAAQ,EACR,IAA8B,EAC9B,QAAY,EACZ,SAAiB,EACjB,KAAa,EACb,SAAS,GACV,EAAE,iBAAiB,GAAG,MAAM,CAuD5B"}

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

@@ -0,0 +1,63 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getKlingCreditUsed = getKlingCreditUsed;
+const aiModels_1 = require("../../../../globals/aiModels");
+const logger_1 = require("../../../../utils/logger");
+const helpers_1 = require("../../../../utils/helpers");
+const types_1 = require("../../types");
+/**
+ * Kling credit calculation. Pure (config lookup + arithmetic) — broken out of
+ * KlingService.getCreditUsed. Handles the three pricing shapes: avatar
+ * (per-second, 5s-rounded), v2.6 (audio/noAudio tier table), and the base
+ * table with deferred I2V / generate_audio upcharges.
+ */
+function getKlingCreditUsed({ modelKey, mode = types_1.EVideoMode.PROFESSIONAL, duration = 5, multiClip = false, audio = false, inputType, }) {
+    const modelConfig = aiModels_1.aiModelConfigs[modelKey];
+    const applyFloor = !multiClip;
+    // Avatar: per-second pricing based on mode. Kling lipsync bills in
+    // 5-second increments (a 7s clip is billed as 10s).
+    if (modelKey === "kling-avatar") {
+        const perSecond = modelConfig?.cost?.perMode?.[mode]?.perSecond
+            ?? modelConfig?.cost?.perMode?.["std"]?.perSecond
+            ?? 0.014;
+        const billedSeconds = Math.ceil(duration / 5) * 5;
+        return (0, helpers_1.getCreditsFromCost)(perSecond * billedSeconds, applyFloor);
+    }
+    // v2.6 splits cost by audio/noAudio tier (audio is 2× upcharge).
+    if (modelKey === "kling-v2.6") {
+        const branch = audio ? modelConfig?.cost?.audio : modelConfig?.cost?.noAudio;
+        const cost = branch?.table?.[mode]?.[duration];
+        if (cost === undefined || cost === null) {
+            logger_1.logger.warn(`Kling getCreditUsed: no v2.6 cost entry for mode="${mode}" duration=${duration} audio=${audio} — returning fallback`, { modelKey, mode, duration, audio });
+            return (0, helpers_1.getCreditsFromCost)(1.0, applyFloor);
+        }
+        return (0, helpers_1.getCreditsFromCost)(cost, applyFloor);
+    }
+    const baseCost = modelConfig?.cost?.table?.[mode]?.[duration];
+    if (baseCost === undefined || baseCost === null) {
+        logger_1.logger.warn(`Kling getCreditUsed: no cost entry for modelKey="${modelKey}" mode="${mode}" duration=${duration} — returning fallback`, { modelKey, mode, duration });
+        return (0, helpers_1.getCreditsFromCost)(1.0, applyFloor);
+    }
+    // Deferred-table upcharges. The cost tables hold T2V / no-audio BASE rates;
+    // Kling bills more for two cases that don't have separate table keys yet
+    // (see config comments in globals/aiModels/providers/kling.ts):
+    //   • I2V (image- or video-conditioned) → 1.5× on kling-video-o1 and
+    //     kling-v3-omni-video (verified 2026-05-20: std $0.084→$0.126/s).
+    //   • generate_audio=true → 1.5× on kling-v3.0.
+    // Applied here so the reconciled ACTUAL cost reflects them; without this the
+    // multi-clip (agent) path under-bills these calls (single clips are covered
+    // by the per-call floor regardless).
+    const I2V_UPCHARGE_MODELS = [
+        "kling-video-o1",
+        "kling-v3-omni-video",
+    ];
+    let cost = baseCost;
+    if (I2V_UPCHARGE_MODELS.includes(modelKey) &&
+        (inputType === "image" || inputType === "video")) {
+        cost *= 1.5;
+    }
+    if (modelKey === "kling-v3.0" && audio) {
+        cost *= 1.5;
+    }
+    return (0, helpers_1.getCreditsFromCost)(cost, applyFloor);
+}