@dataclouder/nest-vertex 0.0.45 → 0.0.46

package/comfyui/comfyui.module.js

@@ -10,6 +10,8 @@ exports.ComfyUIModule = void 0;
 const common_1 = require("@nestjs/common");
 const comfy_services_providers_1 = require("./providers/comfy-services.providers");
 const comfy_ui_controller_1 = require("./controllers/comfy-ui.controller");
+const comfy_sdk_controller_1 = require("./controllers/comfy-sdk.controller");
+const comfy_sdk_service_1 = require("./services/comfy-sdk.service");
 const axios_1 = require("@nestjs/axios");
 const nest_storage_1 = require("@dataclouder/nest-storage");
 const mongoose_1 = require("@nestjs/mongoose");
@@ -27,9 +29,9 @@ exports.ComfyUIModule = ComfyUIModule = __decorate([
             mongoose_1.MongooseModule.forFeature([{ name: generated_asset_entity_1.GeneratedAsset.name, schema: generated_asset_entity_1.GeneratedAssetSchema }]),
             nest_mongo_1.DCMongoDBModule,
         ],
-        providers: [...comfy_services_providers_1.comfyServicesProviders, generated_asset_service_1.GeneratedAssetService],
+        providers: [...comfy_services_providers_1.comfyServicesProviders, generated_asset_service_1.GeneratedAssetService, comfy_sdk_service_1.ComfySDKService],
         exports: [...comfy_services_providers_1.comfyServicesProviders],
-        controllers: [comfy_ui_controller_1.ComfyUiController],
+        controllers: [comfy_ui_controller_1.ComfyUiController, comfy_sdk_controller_1.ComfySDKController],
     })
 ], ComfyUIModule);
 //# sourceMappingURL=comfyui.module.js.map

package/comfyui/controllers/comfy-sdk.controller.d.ts

@@ -0,0 +1,10 @@
+import { ComfySDKService } from '../services/comfy-sdk.service';
+import { StatusResponse } from '../models/comfy-sdk.models';
+export declare class ComfySDKController {
+    private readonly comfySDKService;
+    constructor(comfySDKService: ComfySDKService);
+    testEndpoint(): Promise<{
+        status: string;
+    }>;
+    statusEndpoint(): Promise<StatusResponse>;
+}

package/comfyui/controllers/comfy-sdk.controller.js

@@ -0,0 +1,45 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ComfySDKController = void 0;
+const common_1 = require("@nestjs/common");
+const comfy_sdk_service_1 = require("../services/comfy-sdk.service");
+let ComfySDKController = class ComfySDKController {
+    comfySDKService;
+    constructor(comfySDKService) {
+        this.comfySDKService = comfySDKService;
+    }
+    testEndpoint() {
+        return this.comfySDKService.testMethod();
+    }
+    async statusEndpoint() {
+        const status = await this.comfySDKService.getStatus();
+        return status;
+    }
+};
+exports.ComfySDKController = ComfySDKController;
+__decorate([
+    (0, common_1.Get)('test'),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", void 0)
+], ComfySDKController.prototype, "testEndpoint", null);
+__decorate([
+    (0, common_1.Get)('status'),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], ComfySDKController.prototype, "statusEndpoint", null);
+exports.ComfySDKController = ComfySDKController = __decorate([
+    (0, common_1.Controller)('api/comfy-sdk'),
+    __metadata("design:paramtypes", [comfy_sdk_service_1.ComfySDKService])
+], ComfySDKController);
+//# sourceMappingURL=comfy-sdk.controller.js.map

package/comfyui/models/comfy-sdk.models.d.ts

@@ -0,0 +1,14 @@
+export interface ServerStatus {
+    device: string;
+    memory: {
+        used: number;
+        free: number;
+        total: number;
+    };
+    so: string;
+}
+export interface StatusResponse {
+    status: string;
+    numAvailable: number;
+    servers: ServerStatus[];
+}

package/comfyui/models/comfy-sdk.models.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=comfy-sdk.models.js.map

package/comfyui/models/comfy.models.d.ts

@@ -19,4 +19,5 @@ export interface VideoGenRequest {
     width?: number;
     height?: number;
     seconds?: number;
+    storagePath?: string;
 }

package/comfyui/services/comfy-sdk.service.d.ts

@@ -0,0 +1,24 @@
+import { StatusResponse } from '../models/comfy-sdk.models';
+import { GeneratedAssetService } from '../../services/generated-asset.service';
+import { HttpService } from '@nestjs/axios';
+import { VideoGenRequest } from '../models/comfy.models';
+import { CloudStorageService } from '@dataclouder/nest-storage';
+export declare class ComfySDKService {
+    private readonly generatedAssetService;
+    private readonly httpService;
+    private readonly cloudStorageService;
+    private readonly logger;
+    private outputDirectory;
+    private api;
+    constructor(generatedAssetService: GeneratedAssetService, httpService: HttpService, cloudStorageService: CloudStorageService);
+    private init;
+    testMethod(): Promise<{
+        status: string;
+    }>;
+    getStatus(): Promise<StatusResponse>;
+    runVideoGenerationForAssetId(assetId: string, metadata?: any): Promise<import("../..").GeneratedAssetDocument | {
+        status: string;
+        message: string;
+    }>;
+    runVideoGenerationFromWorkflow(videoRequest: VideoGenRequest, assetId?: string): Promise<any>;
+}

package/comfyui/services/comfy-sdk.service.js

@@ -0,0 +1,226 @@
+"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 __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var ComfySDKService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ComfySDKService = void 0;
+const common_1 = require("@nestjs/common");
+const comfyui_sdk_1 = require("@saintno/comfyui-sdk");
+const generated_asset_service_1 = require("../../services/generated-asset.service");
+const axios_1 = require("@nestjs/axios");
+const path = __importStar(require("path"));
+const fs = __importStar(require("fs/promises"));
+const nest_storage_1 = require("@dataclouder/nest-storage");
+let ComfySDKService = ComfySDKService_1 = class ComfySDKService {
+    generatedAssetService;
+    httpService;
+    cloudStorageService;
+    logger = new common_1.Logger(ComfySDKService_1.name);
+    outputDirectory = './comfyui_generated_images';
+    api;
+    constructor(generatedAssetService, httpService, cloudStorageService) {
+        this.generatedAssetService = generatedAssetService;
+        this.httpService = httpService;
+        this.cloudStorageService = cloudStorageService;
+        this.api = new comfyui_sdk_1.ComfyApi('http://192.168.2.224:8188').init();
+        console.log('what is in the comfy api?', this.api.apiHost);
+    }
+    init() {
+    }
+    async testMethod() {
+        this.logger.log('testMethod called');
+        const queue = await this.api.getQueue();
+        const settings = await this.api.getSettings();
+        const stats = await this.api.getSystemStats();
+        const pullStatus = await this.api.pollStatus();
+        const user = await this.api.getUserConfig();
+        console.log('what is in the comfy api?', queue, settings, stats);
+        console.log('user', user);
+        return { status: 'ok' };
+    }
+    async getStatus() {
+        const stats = await this.api.getSystemStats();
+        const devices = stats.devices.map(device => {
+            const match = device.name.match(/RTX\s(\d{4})/);
+            const model = match ? match[0] : device.name;
+            const freeVram = Number(Number(device.vram_free / 1073741824).toFixed(2));
+            const totalVram = Number(Number(device.vram_total / 1073741824).toFixed(2));
+            return {
+                device: model,
+                memory: {
+                    used: totalVram - freeVram,
+                    free: freeVram,
+                    total: totalVram,
+                },
+                so: 'linux hardcoded',
+            };
+        });
+        console.log('Sending Stats');
+        return { status: 'ok', numAvailable: devices.length, servers: devices };
+    }
+    async runVideoGenerationForAssetId(assetId, metadata = null) {
+        try {
+            const stats = await this.api.getSystemStats();
+        }
+        catch (error) {
+            this.logger.error('Error in runVideoGenerationForAssetId endpoint:', error);
+            return { status: 'error', message: 'There is not Comfy server running or is not accessible.' };
+        }
+        console.log('Running video generation for asset ID:', assetId);
+        const genAsset = await this.generatedAssetService.findOne(assetId);
+        this.logger.verbose(`Generating video for asset ${assetId} with url ${genAsset.assets.firstFrame.url} and metadata ${JSON.stringify(genAsset.request)}`);
+        if (metadata) {
+            console.log('Updating asset metadata for future retrieval:', metadata);
+            await this.generatedAssetService.partialUpdate(assetId, metadata);
+        }
+        if (!genAsset || !genAsset.assets?.firstFrame?.url) {
+            throw new common_1.NotFoundException(`Asset with id ${assetId} not found or does not have a valid image url.`);
+        }
+        const imageUrl = genAsset.assets.firstFrame.url;
+        const response = await this.httpService.axiosRef.get(imageUrl, { responseType: 'arraybuffer' });
+        const imageBuffer = Buffer.from(response.data, 'binary');
+        const videoRequest = {
+            workflowPath: path.join(process.cwd(), 'comfy_workflows', 'video-15_seconds.json'),
+            inputImage: imageBuffer,
+            positivePrompt: genAsset.prompt || genAsset?.description || 'Random movement for video',
+            negativePrompt: 'low quality, blurry, static',
+            width: genAsset.request?.width || null,
+            height: genAsset.request?.height || null,
+            seconds: genAsset.request?.seconds || null,
+        };
+        try {
+            const result = await this.runVideoGenerationFromWorkflow(videoRequest, assetId);
+            this.logger.log(`Video generation for asset ${assetId} completed successfully.`, result);
+            return await this.generatedAssetService.findOne(assetId);
+        }
+        catch (error) {
+            this.logger.error(`Video generation for asset ${assetId} failed.`, error);
+            await this.generatedAssetService.partialUpdate(assetId, { status: 'failed' });
+            throw error;
+        }
+    }
+    async runVideoGenerationFromWorkflow(videoRequest, assetId) {
+        return new Promise(async (resolve, reject) => {
+            const { inputImage, positivePrompt, negativePrompt, width, height, seconds } = videoRequest;
+            const workflowPath = path.join(process.cwd(), 'comfy_workflows', 'video-15_seconds.json');
+            let videoWorkflowJson;
+            try {
+                const workflowData = await fs.readFile(workflowPath, 'utf-8');
+                videoWorkflowJson = JSON.parse(workflowData);
+            }
+            catch (error) {
+                this.logger.error(`Failed to read or parse workflow file at ${workflowPath}`, error);
+                return reject(new Error('Could not load the video generation workflow.'));
+            }
+            const inputKeys = ['positive', 'negative', 'image', 'width', 'height', 'seconds'];
+            const outputKeys = ['video'];
+            const workflow = new comfyui_sdk_1.PromptBuilder(videoWorkflowJson, inputKeys, outputKeys)
+                .setInputNode('positive', '6.inputs.text')
+                .setInputNode('negative', '7.inputs.text')
+                .setInputNode('image', '62.inputs.image')
+                .setInputNode('width', '84.inputs.width')
+                .setInputNode('height', '84.inputs.height')
+                .setInputNode('seconds', '63.inputs.length')
+                .setOutputNode('video', '90');
+            const uploadedImage = await this.api.uploadImage(inputImage, 'input_image.jpg');
+            if (!uploadedImage) {
+                this.logger.error('Image upload failed.');
+                return reject(new Error('Image upload failed.'));
+            }
+            workflow
+                .input('positive', positivePrompt)
+                .input('negative', negativePrompt)
+                .input('image', uploadedImage.info.filename)
+                .input('width', width || 512)
+                .input('height', height || 512)
+                .input('seconds', (seconds || 15) * 16 + 1);
+            new comfyui_sdk_1.CallWrapper(this.api, workflow)
+                .onFinished(async (data) => {
+                this.logger.log('Workflow finished:', data);
+                if (assetId && data.video?.gifs?.[0]) {
+                    const videoInfo = data.video.gifs[0];
+                    const videoBlob = await this.api.getImage(videoInfo);
+                    const videoBuffer = Buffer.from(await videoBlob.arrayBuffer());
+                    if (videoBuffer) {
+                        const storageFolder = videoRequest?.storagePath || `generated-videos`;
+                        const storageFinalPath = `${storageFolder}/${assetId}.mp4`;
+                        console.log('Saving video to storage path:', storageFinalPath);
+                        const uploadResult = await this.cloudStorageService.uploadFileAndMakePublic(process.env.STORAGE_BUCKET || '', storageFinalPath, videoBuffer, 'video/mp4');
+                        this.logger.log(`Successfully uploaded video to: ${uploadResult.url}`);
+                        await this.generatedAssetService.partialUpdate(assetId, {
+                            url: uploadResult.url,
+                            result: uploadResult,
+                            status: 'completed',
+                            assets: {
+                                video: {
+                                    url: uploadResult.url,
+                                    ...videoInfo,
+                                },
+                            },
+                        });
+                        resolve(data);
+                    }
+                    else {
+                        reject(new Error('Video buffer is empty.'));
+                    }
+                }
+                else {
+                    reject(new Error('Asset ID or video data is missing.'));
+                }
+            })
+                .onFailed(err => {
+                this.logger.error('Workflow failed:', err);
+                reject(err);
+            })
+                .run();
+        });
+    }
+};
+exports.ComfySDKService = ComfySDKService;
+exports.ComfySDKService = ComfySDKService = ComfySDKService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [generated_asset_service_1.GeneratedAssetService,
+        axios_1.HttpService,
+        nest_storage_1.CloudStorageService])
+], ComfySDKService);
+//# sourceMappingURL=comfy-sdk.service.js.map

package/comfyui/services/comfy-video.service.js

@@ -91,6 +91,7 @@ let ComfyVideoService = ComfyVideoService_1 = class ComfyVideoService {
         return response;
     }
     async runVideoGenerationForAssetId(assetId, metadata = null) {
+        console.log('Running video generation for asset ID:', assetId);
         const genAsset = await this.generatedAssetService.findOne(assetId);
         if (metadata) {
             console.log('Updating asset metadata for future retrieval:', metadata);

package/controllers/groq-stt.controller.js

@@ -18,6 +18,7 @@ const common_1 = require("@nestjs/common");
 const nest_fastify_file_upload_1 = require("@webundsoehne/nest-fastify-file-upload");
 const swagger_1 = require("@nestjs/swagger");
 const groq_service_1 = require("../services/whisper/groq.service");
+const nest_core_1 = require("@dataclouder/nest-core");
 let GroqSttController = GroqSttController_1 = class GroqSttController {
     groqService;
     logger = new common_1.Logger(GroqSttController_1.name);
@@ -55,6 +56,7 @@ __decorate([
 exports.GroqSttController = GroqSttController = GroqSttController_1 = __decorate([
     (0, swagger_1.ApiTags)('Speech-to-Text'),
     (0, common_1.Controller)('api/groq-stt'),
+    (0, common_1.UseFilters)(nest_core_1.AllExceptionsHandler),
     __metadata("design:paramtypes", [groq_service_1.GroqService])
 ], GroqSttController);
 //# sourceMappingURL=groq-stt.controller.js.map

package/controllers/{vertex-veo-video.controller.js → veo-video.controller.js}

@@ -44,4 +44,4 @@ exports.VertexVeoGenerationController = VertexVeoGenerationController = VertexVe
     (0, common_1.Controller)('api/vertex-veo/video'),
     __metadata("design:paramtypes", [vertex_veo_genai_service_1.VertexVeoGenaiService])
 ], VertexVeoGenerationController);
-//# sourceMappingURL=vertex-veo-video.controller.js.map
+//# sourceMappingURL=veo-video.controller.js.map

package/controllers/vertex-comfy.controller.js

@@ -21,6 +21,7 @@ const comfy_status_dto_1 = require("../dto/comfy-status.dto");
 const comfy_image_service_1 = require("../comfyui/services/comfy-image.service");
 const comfy_video_service_1 = require("../comfyui/services/comfy-video.service");
 const comfy_models_1 = require("../comfyui/models/comfy.models");
+const nest_core_1 = require("@dataclouder/nest-core");
 let VertexComfyController = VertexComfyController_1 = class VertexComfyController {
     comfyImageService;
     comfyVideoService;
@@ -106,6 +107,7 @@ __decorate([
 exports.VertexComfyController = VertexComfyController = VertexComfyController_1 = __decorate([
     (0, swagger_1.ApiTags)('Vertex AI'),
     (0, common_1.Controller)('api/vertex-comfy'),
+    (0, common_1.UseFilters)(nest_core_1.AllExceptionsHandler),
     __metadata("design:paramtypes", [comfy_image_service_1.ComfyImageService,
         comfy_video_service_1.ComfyVideoService,
         generated_asset_service_1.GeneratedAssetService])

package/controllers/vertex-gemini-chat.controller.js

@@ -18,6 +18,7 @@ const common_1 = require("@nestjs/common");
 const vertex_gemini_chat_service_1 = require("../services/vertex-gemini-chat.service");
 const swagger_1 = require("@nestjs/swagger");
 const generate_text_dto_1 = require("../dto/generate-text.dto");
+const nest_core_1 = require("@dataclouder/nest-core");
 let GeminiChatController = GeminiChatController_1 = class GeminiChatController {
     geminiChatService;
     logger = new common_1.Logger(GeminiChatController_1.name);
@@ -47,6 +48,7 @@ __decorate([
 exports.GeminiChatController = GeminiChatController = GeminiChatController_1 = __decorate([
     (0, swagger_1.ApiTags)('LLM Gemini '),
     (0, common_1.Controller)('api/ai-services/gemini'),
+    (0, common_1.UseFilters)(nest_core_1.AllExceptionsHandler),
     __metadata("design:paramtypes", [vertex_gemini_chat_service_1.GeminiChatService])
 ], GeminiChatController);
 //# sourceMappingURL=vertex-gemini-chat.controller.js.map

package/controllers/{vertex-video-adapter.controller.d.ts → video-gen-adapter.controller.d.ts}

@@ -7,21 +7,24 @@ import { GeneratedAsset } from '../models/generated-asset.entity';
 import { GeneratedAssetService } from '../services/generated-asset.service';
 import { ComfyVideoService } from '../comfyui/services/comfy-video.service';
 import { VertexVeoGenaiService } from '../services/vertex-veo-genai.service';
+import { ComfySDKService } from '../comfyui/services/comfy-sdk.service';
 export declare class VertexVideoImageAdapterController {
     private readonly vertexImageVideoService;
     private readonly vertexVeoGcpService;
     private readonly generatedAssetService;
     private readonly comfyVideoService;
     private readonly vertexVeoGenaiService;
+    private readonly comfySDKService;
     private readonly logger;
-    constructor(vertexImageVideoService: VertexImageVideoService, vertexVeoGcpService: VertexVeoGcpService, generatedAssetService: GeneratedAssetService, comfyVideoService: ComfyVideoService, vertexVeoGenaiService: VertexVeoGenaiService);
+    constructor(vertexImageVideoService: VertexImageVideoService, vertexVeoGcpService: VertexVeoGcpService, generatedAssetService: GeneratedAssetService, comfyVideoService: ComfyVideoService, vertexVeoGenaiService: VertexVeoGenaiService, comfySDKService: ComfySDKService);
     listModels(provider?: string): Promise<{
         name: string;
         id: string;
     }[] | undefined>;
     generateVideo(generateVideoDto: GenerateVideoDto, provider?: string): Promise<any>;
-    generateVideoFromAsset(id: string): Promise<{
-        prompt_id: string;
+    generateVideoFromAsset(id: string): Promise<import("../models/generated-asset.entity").GeneratedAssetDocument | {
+        status: string;
+        message: string;
     } | null>;
     listModelsVertex(): Promise<any>;
     testVideo(): Promise<any>;

package/controllers/{vertex-video-adapter.controller.js → video-gen-adapter.controller.js}

@@ -23,19 +23,23 @@ const swagger_1 = require("@nestjs/swagger");
 const generated_asset_service_1 = require("../services/generated-asset.service");
 const comfy_video_service_1 = require("../comfyui/services/comfy-video.service");
 const vertex_veo_genai_service_1 = require("../services/vertex-veo-genai.service");
+const comfy_sdk_service_1 = require("../comfyui/services/comfy-sdk.service");
+const nest_core_1 = require("@dataclouder/nest-core");
 let VertexVideoImageAdapterController = VertexVideoImageAdapterController_1 = class VertexVideoImageAdapterController {
     vertexImageVideoService;
     vertexVeoGcpService;
     generatedAssetService;
     comfyVideoService;
     vertexVeoGenaiService;
+    comfySDKService;
     logger = new common_1.Logger(VertexVideoImageAdapterController_1.name);
-    constructor(vertexImageVideoService, vertexVeoGcpService, generatedAssetService, comfyVideoService, vertexVeoGenaiService) {
+    constructor(vertexImageVideoService, vertexVeoGcpService, generatedAssetService, comfyVideoService, vertexVeoGenaiService, comfySDKService) {
         this.vertexImageVideoService = vertexImageVideoService;
         this.vertexVeoGcpService = vertexVeoGcpService;
         this.generatedAssetService = generatedAssetService;
         this.comfyVideoService = comfyVideoService;
         this.vertexVeoGenaiService = vertexVeoGenaiService;
+        this.comfySDKService = comfySDKService;
     }
     async listModels(provider) {
         this.logger.log(`List models with provider: ${provider}`);
@@ -62,10 +66,10 @@ let VertexVideoImageAdapterController = VertexVideoImageAdapterController_1 = cl
         try {
             const asset = await this.generatedAssetService.findOne(id);
             if (!asset || !asset.assets?.firstFrame?.url) {
-                throw new common_1.NotFoundException(`Asset with id ${id} not found or does not have a valid image url.`);
+                throw new common_1.NotFoundException(`Asset with id ${id} Does not have FIRST FRAME image url.`);
             }
             if (asset.provider === 'comfy') {
-                const videoResponse = await this.comfyVideoService.runVideoGenerationForAssetId(id);
+                const videoResponse = await this.comfySDKService.runVideoGenerationForAssetId(id);
                 this.logger.log(`ComfyUI video generation started. Response: ${JSON.stringify(videoResponse)}`);
                 return videoResponse;
             }
@@ -200,10 +204,12 @@ __decorate([
 exports.VertexVideoImageAdapterController = VertexVideoImageAdapterController = VertexVideoImageAdapterController_1 = __decorate([
     (0, swagger_1.ApiTags)('Vertex Image/video Adapter'),
     (0, common_1.Controller)('api/vertex-adapter/video'),
+    (0, common_1.UseFilters)(nest_core_1.AllExceptionsHandler),
     __metadata("design:paramtypes", [vertex_image_video_service_1.VertexImageVideoService,
         vertex_veo_gcp_service_1.VertexVeoGcpService,
         generated_asset_service_1.GeneratedAssetService,
         comfy_video_service_1.ComfyVideoService,
-        vertex_veo_genai_service_1.VertexVeoGenaiService])
+        vertex_veo_genai_service_1.VertexVeoGenaiService,
+        comfy_sdk_service_1.ComfySDKService])
 ], VertexVideoImageAdapterController);
-//# sourceMappingURL=vertex-video-adapter.controller.js.map
+//# sourceMappingURL=video-gen-adapter.controller.js.map

package/nest-vertex.module.js

@@ -19,7 +19,7 @@ const ai_sdk_chat_service_1 = require("./services/ai-sdk-chat.service");
 const adapter_llm_service_1 = require("./services/adapter-llm.service");
 const adapter_image_gen_service_1 = require("./services/adapter-image-gen.service");
 const adapter_audio_gen_service_1 = require("./services/adapter-audio-gen.service");
-const vertex_video_adapter_controller_1 = require("./controllers/vertex-video-adapter.controller");
+const video_gen_adapter_controller_1 = require("./controllers/video-gen-adapter.controller");
 const vertex_llm_adapter_controller_1 = require("./controllers/vertex-llm-adapter.controller");
 const vertex_veo_gcp_service_1 = require("./services/vertex-veo-gcp.service");
 const axios_1 = require("@nestjs/axios");
@@ -33,10 +33,11 @@ const vertex_comfy_controller_1 = require("./controllers/vertex-comfy.controller
 const key_balancer_api_service_1 = require("./services/key-balancer-api.service");
 const comfyui_module_1 = require("./comfyui/comfyui.module");
 const vertex_veo_genai_service_1 = require("./services/vertex-veo-genai.service");
-const vertex_veo_video_controller_1 = require("./controllers/vertex-veo-video.controller");
+const veo_video_controller_1 = require("./controllers/veo-video.controller");
 const google_genai_service_1 = require("./services/google-genai.service");
 const groq_stt_controller_1 = require("./controllers/groq-stt.controller");
 const groq_service_1 = require("./services/whisper/groq.service");
+const comfy_sdk_service_1 = require("./comfyui/services/comfy-sdk.service");
 let NestVertexModule = class NestVertexModule {
 };
 exports.NestVertexModule = NestVertexModule;
@@ -64,6 +65,7 @@ exports.NestVertexModule = NestVertexModule = __decorate([
             vertex_veo_genai_service_1.VertexVeoGenaiService,
             google_genai_service_1.GoogleGenaiService,
             groq_service_1.GroqService,
+            comfy_sdk_service_1.ComfySDKService
         ],
         exports: [
             vertex_image_video_service_1.VertexImageVideoService,
@@ -81,16 +83,17 @@ exports.NestVertexModule = NestVertexModule = __decorate([
             google_genai_service_1.GoogleGenaiService,
             groq_service_1.GroqService,
             comfyui_module_1.ComfyUIModule,
+            comfy_sdk_service_1.ComfySDKService
         ],
         controllers: [
             vertex_image_adapter_controller_1.VertexImageVideoAdapterController,
             vertex_gemini_chat_controller_1.GeminiChatController,
-            vertex_video_adapter_controller_1.VertexVideoImageAdapterController,
+            video_gen_adapter_controller_1.VertexVideoImageAdapterController,
             vertex_tts_adapter_controller_1.VertexTtsAdapterController,
             vertex_llm_adapter_controller_1.VertexAdapterLLMController,
             generated_assets_controller_1.GeneratedAssetsController,
             vertex_comfy_controller_1.VertexComfyController,
-            vertex_veo_video_controller_1.VertexVeoGenerationController,
+            veo_video_controller_1.VertexVeoGenerationController,
             groq_stt_controller_1.GroqSttController,
         ],
     })

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dataclouder/nest-vertex",
-  "version": "0.0.45",
+  "version": "0.0.46",
   "description": "NestJS Vertex AI library for Dataclouder",
   "author": "dataclouder",
   "license": "MIT",