@dataclouder/nest-vertex 0.0.24 → 0.0.26

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

@@ -18,15 +18,15 @@ export declare class ComfyUiController {
     generateImage(generateImageDto: ImageGenRequestAdapter): Promise<ImageGenAdapterResponse>;
     generateVideoFromLocalImage(positivePrompt: string, negativePrompt: string): Promise<{
         message: string;
-        asset: import("../../models/generated-asset.entity").GeneratedAsset;
+        asset: import("../..").GeneratedAsset;
     }>;
     generateVideoFromAsset(id: string): Promise<{
         message: string;
-        asset: import("../../models/generated-asset.entity").GeneratedAsset;
+        asset: import("../..").GeneratedAsset;
     }>;
     generateVideoFromUrl(videoUrlRequestDto: VideoUrlRequestDto): Promise<{
         message: string;
-        asset: import("../../models/generated-asset.entity").GeneratedAsset;
+        asset: import("../..").GeneratedAsset;
     }>;
     sse(): Observable<MessageEvent>;
 }

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

@@ -104,20 +104,7 @@ let ComfyUiController = ComfyUiController_1 = class ComfyUiController {
             const newAsset = await this.generatedAssetService.save({
                 assets: { firstFrame: { url } },
             });
-            const asset = await this.generatedAssetService.findOne(newAsset.id);
-            if (!asset || !asset.assets?.firstFrame?.url) {
-                throw new common_1.NotFoundException(`Asset with id ${newAsset.id} not found or does not have a valid image url.`);
-            }
-            const imageUrl = asset.assets.firstFrame.url;
-            const response = await this.httpService.axiosRef.get(imageUrl, { responseType: 'arraybuffer' });
-            const imageBuffer = Buffer.from(response.data, 'binary');
-            const videoRequest = {
-                workflowPath: path.resolve('./comfy_workflows/video-15_seconds.json'),
-                inputImage: imageBuffer,
-                positivePrompt: positivePrompt || 'Faces are smiling',
-                negativePrompt: negativePrompt || 'low quality, blurry, static',
-            };
-            const videoResponse = await this.comfyVideoService.runVideoGenerationFromWorkflow(videoRequest, newAsset.id);
+            const videoResponse = await this.comfyVideoService.runVideoGenerationForAssetId(newAsset.id, positivePrompt, negativePrompt);
             this.logger.log(`ComfyUI video generation started. Response: ${JSON.stringify(videoResponse)}`);
             return videoResponse;
         }

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

@@ -15,4 +15,8 @@ export declare class ComfyVideoService {
         message: string;
         asset: GeneratedAsset;
     }>;
+    runVideoGenerationForAssetId(assetId: string, positivePrompt: string, negativePrompt: string): Promise<{
+        message: string;
+        asset: GeneratedAsset;
+    }>;
 }

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

@@ -13,6 +13,7 @@ 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 comfy_connection_service_1 = require("./comfy-connection.service");
 const axios_1 = require("@nestjs/axios");
 const generated_asset_service_1 = require("../../services/generated-asset.service");
@@ -80,6 +81,23 @@ let ComfyVideoService = ComfyVideoService_1 = class ComfyVideoService {
             throw new common_1.InternalServerErrorException(`Failed to run video generation workflow: ${error.message}`);
         }
     }
+    async runVideoGenerationForAssetId(assetId, positivePrompt, negativePrompt) {
+        this.logger.log(`Starting video generation for asset id: ${assetId}`);
+        const asset = await this.generatedAssetService.findOne(assetId);
+        if (!asset || !asset.assets?.firstFrame?.url) {
+            throw new common_1.NotFoundException(`Asset with id ${assetId} not found or does not have a valid image url.`);
+        }
+        const imageUrl = asset.assets.firstFrame.url;
+        const response = await this.httpService.axiosRef.get(imageUrl, { responseType: 'arraybuffer' });
+        const imageBuffer = Buffer.from(response.data, 'binary');
+        const videoRequest = {
+            workflowPath: path.resolve('./comfy_workflows/video-15_seconds.json'),
+            inputImage: imageBuffer,
+            positivePrompt: positivePrompt || 'Faces are smiling',
+            negativePrompt: negativePrompt || 'low quality, blurry, static',
+        };
+        return this.runVideoGenerationFromWorkflow(videoRequest, assetId);
+    }
 };
 exports.ComfyVideoService = ComfyVideoService;
 exports.ComfyVideoService = ComfyVideoService = ComfyVideoService_1 = __decorate([

package/index.d.ts

@@ -16,3 +16,5 @@ export * from './services/generated-asset.service';
 export * from './models/gemini-models';
 export * from './models/adapter.models';
 export * from './models/google-voices';
+export * from './models/generated-asset.entity';
+export * from './models/generated-asset.model';

package/index.js

@@ -32,4 +32,6 @@ __exportStar(require("./services/generated-asset.service"), exports);
 __exportStar(require("./models/gemini-models"), exports);
 __exportStar(require("./models/adapter.models"), exports);
 __exportStar(require("./models/google-voices"), exports);
+__exportStar(require("./models/generated-asset.entity"), exports);
+__exportStar(require("./models/generated-asset.model"), exports);
 //# sourceMappingURL=index.js.map

package/package.json

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