@dataclouder/nest-vertex 0.0.29 → 0.0.32

package/audio.utils.js

@@ -1,8 +1,41 @@
 "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 __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;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.saveAudioToFile = saveAudioToFile;
-const path = require("path");
-const fs = require("fs/promises");
+const path = __importStar(require("path"));
+const fs = __importStar(require("fs/promises"));
 const common_1 = require("@nestjs/common");
 const logger = new common_1.Logger('AudioUtils');
 async function saveAudioToFile(audioBuffer, filename) {

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

@@ -1,10 +1,43 @@
 "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);
 };
@@ -18,8 +51,8 @@ const common_1 = require("@nestjs/common");
 const comfy_models_1 = require("../models/comfy.models");
 const nest_core_1 = require("@dataclouder/nest-core");
 const video_url_request_dto_1 = require("../dto/video-url-request.dto");
-const fs = require("fs/promises");
-const path = require("path");
+const fs = __importStar(require("fs/promises"));
+const path = __importStar(require("path"));
 const comfy_image_service_1 = require("../services/comfy-image.service");
 const comfy_video_service_1 = require("../services/comfy-video.service");
 const generated_asset_service_1 = require("../../services/generated-asset.service");

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

@@ -1,4 +1,4 @@
-export { GeneratedImageAdapter, ImageGenAdapterResponse, ImageGenRequestAdapter } from '../../models/adapter.models';
+export { type GeneratedImageAdapter, ImageGenAdapterResponse, ImageGenRequestAdapter } from '../../models/adapter.models';
 export interface ComfyUIGenerateImagesResponse {
     images: {
         filename: string;

package/comfyui/providers/comfy-services.providers.js

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.comfyServicesProviders = void 0;
-const config_1 = require("@nestjs/config");
 const comfy_connection_service_1 = require("../services/comfy-connection.service");
 const comfy_image_service_1 = require("../services/comfy-image.service");
 const comfy_video_service_1 = require("../services/comfy-video.service");
@@ -12,13 +11,13 @@ const event_emitter_1 = require("@nestjs/event-emitter");
 exports.comfyServicesProviders = [
     {
         provide: comfy_connection_service_1.ComfyConnectionService,
-        useFactory: (configService, httpService, generatedAssetService, cloudStorageService, eventEmitter) => {
-            if (!configService.get('COMFYUI_API_HOST')) {
+        useFactory: (httpService, generatedAssetService, cloudStorageService, eventEmitter) => {
+            if (!process.env.COMFYUI_API_HOST) {
                 return null;
             }
-            return new comfy_connection_service_1.ComfyConnectionService(configService, httpService, generatedAssetService, cloudStorageService, eventEmitter);
+            return new comfy_connection_service_1.ComfyConnectionService(httpService, generatedAssetService, cloudStorageService, eventEmitter);
         },
-        inject: [config_1.ConfigService, axios_1.HttpService, generated_asset_service_1.GeneratedAssetService, nest_storage_1.CloudStorageService, event_emitter_1.EventEmitter2],
+        inject: [axios_1.HttpService, generated_asset_service_1.GeneratedAssetService, nest_storage_1.CloudStorageService, event_emitter_1.EventEmitter2],
     },
     {
         provide: comfy_image_service_1.ComfyImageService,

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

@@ -1,12 +1,10 @@
 import { OnModuleInit, OnModuleDestroy } from '@nestjs/common';
-import { ConfigService } from '@nestjs/config';
 import { Client as ComfyUIClient } from '@stable-canvas/comfyui-client';
 import { HttpService } from '@nestjs/axios';
 import { GeneratedAssetService } from '../../services/generated-asset.service';
 import { CloudStorageService } from '@dataclouder/nest-storage';
 import { EventEmitter2 } from '@nestjs/event-emitter';
 export declare class ComfyConnectionService implements OnModuleInit, OnModuleDestroy {
-    private readonly configService;
     private readonly httpService;
     private readonly generatedAssetService;
     private readonly cloudStorageService;
@@ -17,7 +15,7 @@ export declare class ComfyConnectionService implements OnModuleInit, OnModuleDes
     private comfyApiHost;
     private clientId;
     private pendingPrompts;
-    constructor(configService: ConfigService, httpService: HttpService, generatedAssetService: GeneratedAssetService, cloudStorageService: CloudStorageService, eventEmitter: EventEmitter2);
+    constructor(httpService: HttpService, generatedAssetService: GeneratedAssetService, cloudStorageService: CloudStorageService, eventEmitter: EventEmitter2);
     onModuleInit(): Promise<void>;
     onModuleDestroy(): Promise<void>;
     private checkApiStatus;

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

@@ -11,11 +11,13 @@ var __metadata = (this && this.__metadata) || function (k, v) {
 var __param = (this && this.__param) || function (paramIndex, decorator) {
     return function (target, key) { decorator(target, key, paramIndex); }
 };
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 var ComfyConnectionService_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ComfyConnectionService = void 0;
 const common_1 = require("@nestjs/common");
-const config_1 = require("@nestjs/config");
 const comfyui_client_1 = require("@stable-canvas/comfyui-client");
 const ws_1 = require("ws");
 const axios_1 = require("@nestjs/axios");
@@ -23,9 +25,8 @@ const rxjs_1 = require("rxjs");
 const generated_asset_service_1 = require("../../services/generated-asset.service");
 const nest_storage_1 = require("@dataclouder/nest-storage");
 const event_emitter_1 = require("@nestjs/event-emitter");
-const FormData = require("form-data");
+const form_data_1 = __importDefault(require("form-data"));
 let ComfyConnectionService = ComfyConnectionService_1 = class ComfyConnectionService {
-    configService;
     httpService;
     generatedAssetService;
     cloudStorageService;
@@ -36,8 +37,7 @@ let ComfyConnectionService = ComfyConnectionService_1 = class ComfyConnectionSer
     comfyApiHost;
     clientId;
     pendingPrompts = new Map();
-    constructor(configService, httpService, generatedAssetService, cloudStorageService, eventEmitter) {
-        this.configService = configService;
+    constructor(httpService, generatedAssetService, cloudStorageService, eventEmitter) {
         this.httpService = httpService;
         this.generatedAssetService = generatedAssetService;
         this.cloudStorageService = cloudStorageService;
@@ -45,7 +45,7 @@ let ComfyConnectionService = ComfyConnectionService_1 = class ComfyConnectionSer
         this.clientId = `nest-comfy-client-${Date.now()}`;
     }
     async onModuleInit() {
-        this.comfyApiHost = this.configService.get('COMFYUI_API_HOST');
+        this.comfyApiHost = process.env.COMFYUI_API_HOST || '';
         if (!this.comfyApiHost) {
             this.logger.warn('COMFYUI_API_HOST environment variable not set. ComfyConnectionService will not be initialized.');
             return;
@@ -118,7 +118,7 @@ let ComfyConnectionService = ComfyConnectionService_1 = class ComfyConnectionSer
         this.logger.log(`Workflow ${prompt_id} executed. Processing output...`);
         if (this.pendingPrompts.has(prompt_id)) {
             const pending = this.pendingPrompts.get(prompt_id);
-            pending.resolve(output);
+            pending?.resolve(output);
             this.pendingPrompts.delete(prompt_id);
             this.logger.log(`Resolved pending promise for prompt ${prompt_id}.`);
             return;
@@ -137,9 +137,9 @@ let ComfyConnectionService = ComfyConnectionService_1 = class ComfyConnectionSer
             videoFilename = video.filename;
             this.logger.log(`Found video output: ${videoFilename}`);
             if (videoBuffer) {
-                const uploadResult = await this.cloudStorageService.uploadFileAndMakePublic(process.env.STORAGE_BUCKET, `generated-videos/${generatedAsset.id}.mp4`, videoBuffer, 'video/mp4');
+                const uploadResult = await this.cloudStorageService.uploadFileAndMakePublic(process.env.STORAGE_BUCKET || '', `generated-videos/${generatedAsset.id}.mp4`, videoBuffer, 'video/mp4');
                 this.logger.log(`Successfully uploaded video to: ${uploadResult.url}`);
-                generatedAsset.url = uploadResult.url;
+                generatedAsset.url = uploadResult?.url || '';
                 generatedAsset.result = uploadResult;
                 generatedAsset.status = 'completed';
                 await this.generatedAssetService.update(generatedAsset.id, generatedAsset);
@@ -181,7 +181,7 @@ let ComfyConnectionService = ComfyConnectionService_1 = class ComfyConnectionSer
     }
     async uploadImage(imageBuffer, filename) {
         const url = `http://${this.comfyApiHost}/upload/image`;
-        const form = new FormData();
+        const form = new form_data_1.default();
         form.append('image', imageBuffer, { filename });
         form.append('overwrite', 'true');
         const response = await (0, rxjs_1.firstValueFrom)(this.httpService.post(url, form, {
@@ -219,9 +219,8 @@ let ComfyConnectionService = ComfyConnectionService_1 = class ComfyConnectionSer
 exports.ComfyConnectionService = ComfyConnectionService;
 exports.ComfyConnectionService = ComfyConnectionService = ComfyConnectionService_1 = __decorate([
     (0, common_1.Injectable)(),
-    __param(1, (0, common_1.Inject)(axios_1.HttpService)),
-    __metadata("design:paramtypes", [config_1.ConfigService,
-        axios_1.HttpService,
+    __param(0, (0, common_1.Inject)(axios_1.HttpService)),
+    __metadata("design:paramtypes", [axios_1.HttpService,
         generated_asset_service_1.GeneratedAssetService,
         nest_storage_1.CloudStorageService,
         event_emitter_1.EventEmitter2])

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

@@ -1,10 +1,43 @@
 "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);
 };
@@ -13,8 +46,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ComfyImageService = void 0;
 const common_1 = require("@nestjs/common");
 const comfyui_client_1 = require("@stable-canvas/comfyui-client");
-const fs = require("fs/promises");
-const path = require("path");
+const fs = __importStar(require("fs/promises"));
+const path = __importStar(require("path"));
 const comfy_connection_service_1 = require("./comfy-connection.service");
 let ComfyImageService = ComfyImageService_1 = class ComfyImageService {
     connectionService;
@@ -47,7 +80,7 @@ let ComfyImageService = ComfyImageService_1 = class ComfyImageService {
         const { model, negativePrompt, width, height, steps, cfg, seed, samplerName, scheduler } = imageDto;
         try {
             const client = this.connectionService.getClient();
-            const pipe = new comfyui_client_1.BasePipe().with(client).model(model).prompt(imageDto.prompt).size(width, height);
+            const pipe = new comfyui_client_1.BasePipe().with(client).model(model || '').prompt(imageDto.prompt).size(width || 512, height || 512);
             if (negativePrompt)
                 pipe.negative(negativePrompt);
             if (steps)

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

@@ -1,10 +1,43 @@
 "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);
 };
@@ -12,8 +45,8 @@ var ComfyVideoService_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ComfyVideoService = void 0;
 const common_1 = require("@nestjs/common");
-const fs = require("fs/promises");
-const path = require("path");
+const fs = __importStar(require("fs/promises"));
+const path = __importStar(require("path"));
 const comfy_connection_service_1 = require("./comfy-connection.service");
 const generated_asset_service_1 = require("../../services/generated-asset.service");
 const axios_1 = require("@nestjs/axios");
@@ -43,7 +76,7 @@ let ComfyVideoService = ComfyVideoService_1 = class ComfyVideoService {
     }
     async runVideoGenerationFromWorkflow(videoRequest, assetId) {
         const { inputImage, positivePrompt, negativePrompt } = videoRequest;
-        const workflow = await this.updateWorkflow(videoRequest.width, videoRequest.height, videoRequest.seconds);
+        const workflow = await this.updateWorkflow(videoRequest.width || 512, videoRequest.height || 512, videoRequest.seconds || 15);
         workflow['6'].inputs.text = positivePrompt;
         workflow['7'].inputs.text = negativePrompt;
         const uploadResponse = await this.comfyConnectionService.uploadImage(inputImage, 'input_image.jpg');

package/controllers/vertex-comfy.controller.d.ts

@@ -1,5 +1,5 @@
 import { GeneratedAssetService } from '../services/generated-asset.service';
-import { Response } from 'express';
+import { type Response } from 'express';
 import { ComfyStatusDto } from '../dto/comfy-status.dto';
 import { ComfyImageService } from '../comfyui/services/comfy-image.service';
 import { ComfyVideoService } from '../comfyui/services/comfy-video.service';

package/controllers/vertex-image-adapter.controller.d.ts

@@ -9,6 +9,6 @@ export declare class VertexImageVideoAdapterController {
     listModels(provider?: string): Promise<{
         name: string;
         id: string;
-    }[]>;
-    generateImage(generateImageDto: ImageGenRequestAdapter): Promise<ImageGenAdapterResponse>;
+    }[] | undefined>;
+    generateImage(generateImageDto: ImageGenRequestAdapter): Promise<ImageGenAdapterResponse | null>;
 }

package/controllers/vertex-llm-adapter.controller.d.ts

@@ -1,4 +1,4 @@
-import { ChatLLMRequestAdapter, DescribeImageRequestAdapter } from '../models/adapter.models';
+import { type ChatLLMRequestAdapter, DescribeImageRequestAdapter } from '../models/adapter.models';
 import { LLMAdapterService } from '../services/adapter-llm.service';
 export declare class VertexAdapterLLMController {
     private readonly llmAdapterService;

package/controllers/vertex-tts-adapter.controller.d.ts

@@ -1,10 +1,10 @@
 import { google } from '@google-cloud/text-to-speech/build/protos/protos';
-import { FastifyReply } from 'fastify';
-import { NestTtsService, SynthesizeSpeechInput } from '../services/vertex-tts.service';
+import { type FastifyReply } from 'fastify';
+import { NestTtsService, type SynthesizeSpeechInput } from '../services/vertex-tts.service';
 export declare class VertexTtsAdapterController {
     private readonly nestTtsService;
     private readonly logger;
     constructor(nestTtsService: NestTtsService);
-    synthesizeSpeech(body: SynthesizeSpeechInput, res: FastifyReply): Promise<Buffer>;
+    synthesizeSpeech(body: SynthesizeSpeechInput, res: FastifyReply): Promise<Buffer<ArrayBufferLike>>;
     listVoices(languageCode?: string): Promise<google.cloud.texttospeech.v1.IVoice[]>;
 }

package/controllers/vertex-veo-video.controller.d.ts

@@ -0,0 +1,7 @@
+import { VertexVeoGenaiService } from '../services/vertex-veo-genai.service';
+export declare class VertexVeoGenerationController {
+    private readonly vertexVeoGcpService;
+    private readonly logger;
+    constructor(vertexVeoGcpService: VertexVeoGenaiService);
+    generateVeoVideo(): Promise<any>;
+}

package/controllers/vertex-veo-video.controller.js

@@ -0,0 +1,47 @@
+"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);
+};
+var VertexVeoGenerationController_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VertexVeoGenerationController = void 0;
+const common_1 = require("@nestjs/common");
+const swagger_1 = require("@nestjs/swagger");
+const vertex_veo_genai_service_1 = require("../services/vertex-veo-genai.service");
+let VertexVeoGenerationController = VertexVeoGenerationController_1 = class VertexVeoGenerationController {
+    vertexVeoGcpService;
+    logger = new common_1.Logger(VertexVeoGenerationController_1.name);
+    constructor(vertexVeoGcpService) {
+        this.vertexVeoGcpService = vertexVeoGcpService;
+    }
+    async generateVeoVideo() {
+        this.logger.log(`Received request to generate VEO video`);
+        try {
+            const assetGenerated = await this.vertexVeoGcpService.generateVideo();
+            return assetGenerated;
+        }
+        catch (error) {
+            this.logger.error('Error in generateVeoVideo endpoint:', error);
+            throw error;
+        }
+    }
+};
+exports.VertexVeoGenerationController = VertexVeoGenerationController;
+__decorate([
+    (0, common_1.Get)('generate-veo-video'),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], VertexVeoGenerationController.prototype, "generateVeoVideo", null);
+exports.VertexVeoGenerationController = VertexVeoGenerationController = VertexVeoGenerationController_1 = __decorate([
+    (0, swagger_1.ApiTags)('Vertex Image/video Adapter'),
+    (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

package/controllers/vertex-video-adapter.controller.d.ts

@@ -4,16 +4,25 @@ import { GetVeoVideoStatusDto } from '../dto/get-veo-video-status.dto';
 import { GenerateVideosOperation } from '@google/genai';
 import { VertexVeoGcpService } from '../services/vertex-veo-gcp.service';
 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';
 export declare class VertexVideoImageAdapterController {
     private readonly vertexImageVideoService;
     private readonly vertexVeoGcpService;
+    private readonly generatedAssetService;
+    private readonly comfyVideoService;
+    private readonly vertexVeoGenaiService;
     private readonly logger;
-    constructor(vertexImageVideoService: VertexImageVideoService, vertexVeoGcpService: VertexVeoGcpService);
+    constructor(vertexImageVideoService: VertexImageVideoService, vertexVeoGcpService: VertexVeoGcpService, generatedAssetService: GeneratedAssetService, comfyVideoService: ComfyVideoService, vertexVeoGenaiService: VertexVeoGenaiService);
     listModels(provider?: string): Promise<{
         name: string;
         id: string;
-    }[]>;
+    }[] | undefined>;
     generateVideo(generateVideoDto: GenerateVideoDto, provider?: string): Promise<any>;
+    generateVideoFromAsset(id: string): Promise<{
+        prompt_id: string;
+    } | null>;
     listModelsVertex(): Promise<any>;
     testVideo(): Promise<any>;
     generateVeoVideo(generateVideoDto: GenerateVideoDto): Promise<GeneratedAsset>;

package/controllers/vertex-video-adapter.controller.js

@@ -20,13 +20,22 @@ const generate_video_dto_1 = require("../dto/generate-video.dto");
 const get_veo_video_status_dto_1 = require("../dto/get-veo-video-status.dto");
 const vertex_veo_gcp_service_1 = require("../services/vertex-veo-gcp.service");
 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");
 let VertexVideoImageAdapterController = VertexVideoImageAdapterController_1 = class VertexVideoImageAdapterController {
     vertexImageVideoService;
     vertexVeoGcpService;
+    generatedAssetService;
+    comfyVideoService;
+    vertexVeoGenaiService;
     logger = new common_1.Logger(VertexVideoImageAdapterController_1.name);
-    constructor(vertexImageVideoService, vertexVeoGcpService) {
+    constructor(vertexImageVideoService, vertexVeoGcpService, generatedAssetService, comfyVideoService, vertexVeoGenaiService) {
         this.vertexImageVideoService = vertexImageVideoService;
         this.vertexVeoGcpService = vertexVeoGcpService;
+        this.generatedAssetService = generatedAssetService;
+        this.comfyVideoService = comfyVideoService;
+        this.vertexVeoGenaiService = vertexVeoGenaiService;
     }
     async listModels(provider) {
         this.logger.log(`List models with provider: ${provider}`);
@@ -48,6 +57,29 @@ let VertexVideoImageAdapterController = VertexVideoImageAdapterController_1 = cl
             throw error;
         }
     }
+    async generateVideoFromAsset(id) {
+        this.logger.log(`Received request to generate video from asset with id: ${id}`);
+        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.`);
+            }
+            if (asset.provider === 'comfy') {
+                const videoResponse = await this.comfyVideoService.runVideoGenerationForAssetId(id);
+                this.logger.log(`ComfyUI video generation started. Response: ${JSON.stringify(videoResponse)}`);
+                return videoResponse;
+            }
+            else {
+                this.logger.log(`Starting Generation with VEO.`);
+                this.vertexVeoGenaiService.generateVideoFromAssetId(id);
+                return null;
+            }
+        }
+        catch (error) {
+            this.logger.error('Error in generateVideoFromAsset endpoint:', error);
+            throw error;
+        }
+    }
     async listModelsVertex() {
         return this.vertexImageVideoService.listModelsVertex();
     }
@@ -121,6 +153,13 @@ __decorate([
     __metadata("design:paramtypes", [generate_video_dto_1.GenerateVideoDto, String]),
     __metadata("design:returntype", Promise)
 ], VertexVideoImageAdapterController.prototype, "generateVideo", null);
+__decorate([
+    (0, common_1.Post)('generate-video-from-asset'),
+    __param(0, (0, common_1.Body)('id')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [String]),
+    __metadata("design:returntype", Promise)
+], VertexVideoImageAdapterController.prototype, "generateVideoFromAsset", null);
 __decorate([
     (0, common_1.Get)('list-models-vertex'),
     (0, swagger_1.ApiQuery)({ name: 'provider', required: false, type: String, description: 'The provider for the service (e.g., "local", "google")' }),
@@ -162,6 +201,9 @@ exports.VertexVideoImageAdapterController = VertexVideoImageAdapterController =
     (0, swagger_1.ApiTags)('Vertex Image/video Adapter'),
     (0, common_1.Controller)('api/vertex-adapter/video'),
     __metadata("design:paramtypes", [vertex_image_video_service_1.VertexImageVideoService,
-        vertex_veo_gcp_service_1.VertexVeoGcpService])
+        vertex_veo_gcp_service_1.VertexVeoGcpService,
+        generated_asset_service_1.GeneratedAssetService,
+        comfy_video_service_1.ComfyVideoService,
+        vertex_veo_genai_service_1.VertexVeoGenaiService])
 ], VertexVideoImageAdapterController);
 //# sourceMappingURL=vertex-video-adapter.controller.js.map

package/dto/generate-video.dto.d.ts

@@ -13,7 +13,7 @@ export declare class GenerateVideoDto {
     personGeneration?: string;
     sampleCount?: number;
     seed?: number;
-    storageUri?: string;
+    storageUri: string;
     enhancePrompt?: boolean;
 }
 export {};

package/dto/generate-video.dto.js

@@ -41,14 +41,14 @@ __decorate([
 class GenerateVideoDto {
     prompt = 'A cinamatic shot of siamese cat ridding a horse in the galaxy';
     model = gemini_models_1.GeminiVideoModels.Veo2;
-    image = null;
+    image = undefined;
     durationSeconds = 8;
     aspectRatio = '16:9';
     negativePrompt;
     personGeneration = 'allow_all';
     sampleCount = 1;
     seed;
-    storageUri = null;
+    storageUri;
     enhancePrompt = true;
 }
 exports.GenerateVideoDto = GenerateVideoDto;

package/models/generated-asset.entity.d.ts

@@ -1,7 +1,7 @@
-import { Document } from 'mongoose';
-import { IAuditable } from '@dataclouder/nest-core';
-import { IAssetsGeneration, IGeneratedAsset } from './generated-asset.model';
-import { CloudFileStorage } from '@dataclouder/nest-storage';
+import type { Document } from 'mongoose';
+import { type IAuditable } from '@dataclouder/nest-core';
+import type { IAssetsGeneration, IGeneratedAsset } from './generated-asset.model';
+import type { CloudFileStorage } from '@dataclouder/nest-storage';
 export type GeneratedAssetDocument = GeneratedAsset & Document;
 export declare class GeneratedAsset implements IGeneratedAsset {
     _id?: string;
@@ -12,6 +12,7 @@ export declare class GeneratedAsset implements IGeneratedAsset {
     auditable: IAuditable;
     url: string;
     model: string;
+    provider: 'vertex' | 'comfy';
     request: any;
     response: any;
     status: 'pending' | 'completed' | 'failed';
@@ -21,11 +22,11 @@ export declare class GeneratedAsset implements IGeneratedAsset {
     description: string;
     metadata: any;
 }
-export declare const GeneratedAssetSchema: import("mongoose").Schema<GeneratedAsset, import("mongoose").Model<GeneratedAsset, any, any, any, Document<unknown, any, GeneratedAsset, any> & GeneratedAsset & Required<{
+export declare const GeneratedAssetSchema: import("mongoose").Schema<GeneratedAsset, import("mongoose").Model<GeneratedAsset, any, any, any, Document<unknown, any, GeneratedAsset, any, {}> & GeneratedAsset & Required<{
     _id: string;
 }> & {
     __v: number;
-}, any>, {}, {}, {}, {}, import("mongoose").DefaultSchemaOptions, GeneratedAsset, Document<unknown, {}, import("mongoose").FlatRecord<GeneratedAsset>, {}> & import("mongoose").FlatRecord<GeneratedAsset> & Required<{
+}, any>, {}, {}, {}, {}, import("mongoose").DefaultSchemaOptions, GeneratedAsset, Document<unknown, {}, import("mongoose").FlatRecord<GeneratedAsset>, {}, import("mongoose").ResolveSchemaOptions<import("mongoose").DefaultSchemaOptions>> & import("mongoose").FlatRecord<GeneratedAsset> & Required<{
     _id: string;
 }> & {
     __v: number;