@dataclouder/nest-vertex 0.0.4 → 0.0.7

package/controllers/gemini-chat.controller.js

@@ -0,0 +1,51 @@
+"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 __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+var GeminiChatController_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GeminiChatController = void 0;
+const common_1 = require("@nestjs/common");
+const gemini_chat_service_1 = require("../services/gemini-chat.service");
+const swagger_1 = require("@nestjs/swagger");
+let GeminiChatController = GeminiChatController_1 = class GeminiChatController {
+    geminiChatService;
+    logger = new common_1.Logger(GeminiChatController_1.name);
+    constructor(geminiChatService) {
+        this.geminiChatService = geminiChatService;
+    }
+    async generateText(messages) {
+        this.logger.log('Received request for text generation');
+        try {
+            const result = await this.geminiChatService.chat(messages);
+            return result;
+        }
+        catch (error) {
+            this.logger.error(`Error generating text: ${error.message}`, error.stack);
+            throw error;
+        }
+    }
+};
+exports.GeminiChatController = GeminiChatController;
+__decorate([
+    (0, common_1.Post)('generate-text'),
+    __param(0, (0, common_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Array]),
+    __metadata("design:returntype", Promise)
+], GeminiChatController.prototype, "generateText", null);
+exports.GeminiChatController = GeminiChatController = GeminiChatController_1 = __decorate([
+    (0, swagger_1.ApiTags)('Vertex Gemini'),
+    (0, common_1.Controller)('api/vertex/gemini'),
+    __metadata("design:paramtypes", [gemini_chat_service_1.GeminiChatService])
+], GeminiChatController);
+//# sourceMappingURL=gemini-chat.controller.js.map

package/controllers/nest-tts.controller.js

@@ -0,0 +1,112 @@
+"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 __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NestTtsController = void 0;
+const common_1 = require("@nestjs/common");
+const swagger_1 = require("@nestjs/swagger");
+const nest_tts_service_1 = require("../services/nest-tts.service");
+let NestTtsController = class NestTtsController {
+    nestTtsService;
+    logger = new common_1.Logger('VertexTtsController');
+    constructor(nestTtsService) {
+        this.nestTtsService = nestTtsService;
+    }
+    ping() {
+        this.logger.log('Received ping request');
+        return this.nestTtsService.getHello();
+    }
+    async synthesizeSpeech(body, res) {
+        console.log('Received synthesize request', body);
+        this.logger.log(`Received synthesize request for text: "${body.text.substring(0, 50)}..."`);
+        try {
+            const audioBuffer = await this.nestTtsService.synthesizeSpeech(body);
+            if (!audioBuffer) {
+                this.logger.error('Synthesis resulted in null buffer');
+                throw new common_1.NotFoundException('Could not generate audio for the given input.');
+            }
+            res.header('Content-Length', audioBuffer.length.toString());
+            this.logger.log(`Successfully synthesized audio (${audioBuffer.length} bytes)`);
+            return audioBuffer;
+        }
+        catch (error) {
+            this.logger.error(`Synthesis failed: ${error.message}`, error.stack);
+            throw new common_1.InternalServerErrorException(`Speech synthesis failed: ${error.message}`);
+        }
+    }
+    async listVoices(languageCode) {
+        this.logger.log(`Received request to list voices${languageCode ? ` for language ${languageCode}` : ''}`);
+        try {
+            const voices = await this.nestTtsService.listVoices(languageCode);
+            this.logger.log(`Returning ${voices.length} voice${voices.length === 1 ? '' : 's'}.`);
+            return voices;
+        }
+        catch (error) {
+            this.logger.error(`Failed to list voices: ${error.message}`, error.stack);
+            throw new common_1.InternalServerErrorException(`Failed to list voices: ${error.message}`);
+        }
+    }
+};
+exports.NestTtsController = NestTtsController;
+__decorate([
+    (0, common_1.Get)('ping'),
+    (0, swagger_1.ApiOperation)({ summary: 'Ping the TTS service' }),
+    (0, swagger_1.ApiResponse)({ status: 200, description: 'Service is available' }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", String)
+], NestTtsController.prototype, "ping", null);
+__decorate([
+    (0, common_1.Post)('synthesize'),
+    (0, common_1.HttpCode)(common_1.HttpStatus.OK),
+    (0, swagger_1.ApiOperation)({ summary: 'Synthesize speech from text' }),
+    (0, swagger_1.ApiBody)({
+        description: 'Text and configuration for speech synthesis',
+        schema: {
+            type: 'object',
+            properties: {
+                text: { type: 'string', example: 'Hello world' },
+                languageCode: { type: 'string', example: 'en-US', description: 'Optional language code (default: en-US)' },
+                ssmlGender: { type: 'string', example: 'NEUTRAL', description: 'Optional SSML gender (default: NEUTRAL)' },
+                voiceName: { type: 'string', example: 'en-US-Neural2-F', description: 'Optional specific voice name' },
+                audioEncoding: { type: 'string', example: 'MP3', description: 'Optional audio encoding (default: MP3)' },
+            },
+            required: ['text'],
+        },
+    }),
+    (0, swagger_1.ApiResponse)({ status: 200, description: 'Returns the synthesized audio file (e.g., MP3)' }),
+    (0, swagger_1.ApiResponse)({ status: 500, description: 'Internal server error during synthesis' }),
+    (0, common_1.Header)('Content-Type', 'audio/mpeg'),
+    __param(0, (0, common_1.Body)()),
+    __param(1, (0, common_1.Res)({ passthrough: true })),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, Object]),
+    __metadata("design:returntype", Promise)
+], NestTtsController.prototype, "synthesizeSpeech", null);
+__decorate([
+    (0, common_1.Get)('voices'),
+    (0, swagger_1.ApiOperation)({ summary: 'List available TTS voices, optionally filtered by language' }),
+    (0, swagger_1.ApiQuery)({ name: 'languageCode', required: false, type: String, description: 'Optional language code (e.g., en-US) to filter voices' }),
+    (0, swagger_1.ApiResponse)({ status: 200, description: 'Returns a list of available voices', type: [Object] }),
+    (0, swagger_1.ApiResponse)({ status: 500, description: 'Internal server error while fetching voices' }),
+    __param(0, (0, common_1.Query)('languageCode')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [String]),
+    __metadata("design:returntype", Promise)
+], NestTtsController.prototype, "listVoices", null);
+exports.NestTtsController = NestTtsController = __decorate([
+    (0, swagger_1.ApiTags)('Vertex TTS'),
+    (0, common_1.Controller)('api/vertex/tts'),
+    __metadata("design:paramtypes", [nest_tts_service_1.NestTtsService])
+], NestTtsController);
+//# sourceMappingURL=nest-tts.controller.js.map

package/controllers/nest-vertex.controller.js

@@ -0,0 +1,88 @@
+"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 __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+var NestVertexController_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NestVertexController = void 0;
+const common_1 = require("@nestjs/common");
+const nest_vertex_service_1 = require("../services/nest-vertex.service");
+const generate_video_dto_1 = require("../dto/generate-video.dto");
+const generate_image_dto_1 = require("../dto/generate-image.dto");
+const swagger_1 = require("@nestjs/swagger");
+let NestVertexController = NestVertexController_1 = class NestVertexController {
+    nestVertexService;
+    logger = new common_1.Logger(NestVertexController_1.name);
+    constructor(nestVertexService) {
+        this.nestVertexService = nestVertexService;
+    }
+    async generateVideo(generateVideoDto) {
+        this.logger.log(`Received request to generate video: ${JSON.stringify(generateVideoDto)}`);
+        try {
+            const operation = await this.nestVertexService.startVideoGeneration(generateVideoDto);
+            return { operationName: operation.name };
+        }
+        catch (error) {
+            this.logger.error('Error in generateVideo endpoint:', error);
+            throw error;
+        }
+    }
+    async getVideoStatus(operationName) {
+        this.logger.log(`Received request to check status for operation: ${operationName}`);
+        try {
+            return await this.nestVertexService.checkVideoOperationStatus(operationName);
+        }
+        catch (error) {
+            this.logger.error(`Error checking status for operation ${operationName}:`, error);
+            throw error;
+        }
+    }
+    async generateImage(generateImageDto) {
+        this.logger.log(`Received request to generate image: ${JSON.stringify(generateImageDto)}`);
+        try {
+            const response = await this.nestVertexService.startImageGeneration(generateImageDto);
+            return response;
+        }
+        catch (error) {
+            this.logger.error('Error in generateImage endpoint:', error);
+            throw error;
+        }
+    }
+};
+exports.NestVertexController = NestVertexController;
+__decorate([
+    (0, common_1.Post)('generate-video'),
+    __param(0, (0, common_1.Body)(new common_1.ValidationPipe({ transform: true, whitelist: true }))),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [generate_video_dto_1.GenerateVideoDto]),
+    __metadata("design:returntype", Promise)
+], NestVertexController.prototype, "generateVideo", null);
+__decorate([
+    (0, common_1.Get)('video-status/:operationName'),
+    __param(0, (0, common_1.Param)('operationName')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [String]),
+    __metadata("design:returntype", Promise)
+], NestVertexController.prototype, "getVideoStatus", null);
+__decorate([
+    (0, common_1.Post)('generate-image'),
+    __param(0, (0, common_1.Body)(new common_1.ValidationPipe({ transform: true, whitelist: true }))),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [generate_image_dto_1.GenerateImageDto]),
+    __metadata("design:returntype", Promise)
+], NestVertexController.prototype, "generateImage", null);
+exports.NestVertexController = NestVertexController = NestVertexController_1 = __decorate([
+    (0, swagger_1.ApiTags)('Vertex Image/video'),
+    (0, common_1.Controller)('api/vertex'),
+    __metadata("design:paramtypes", [nest_vertex_service_1.NestVertexService])
+], NestVertexController);
+//# sourceMappingURL=nest-vertex.controller.js.map

package/dto/generate-image.dto.js

@@ -0,0 +1,51 @@
+"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.GenerateImageDto = void 0;
+const class_validator_1 = require("class-validator");
+class GenerateImageDto {
+    prompt;
+    numberOfImages = 1;
+    aspectRatio = '1:1';
+    negativePrompt;
+    seed;
+}
+exports.GenerateImageDto = GenerateImageDto;
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsNotEmpty)(),
+    __metadata("design:type", String)
+], GenerateImageDto.prototype, "prompt", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsInt)(),
+    (0, class_validator_1.Min)(1),
+    (0, class_validator_1.Max)(8),
+    __metadata("design:type", Number)
+], GenerateImageDto.prototype, "numberOfImages", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsEnum)(['1:1', '16:9', '9:16', '3:4', '4:3'], { message: 'aspectRatio must be one of "1:1", "16:9", "9:16", "3:4", "4:3"' }),
+    __metadata("design:type", String)
+], GenerateImageDto.prototype, "aspectRatio", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsString)(),
+    __metadata("design:type", String)
+], GenerateImageDto.prototype, "negativePrompt", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsInt)(),
+    (0, class_validator_1.Min)(0),
+    (0, class_validator_1.Max)(4294967295),
+    __metadata("design:type", Number)
+], GenerateImageDto.prototype, "seed", void 0);
+//# sourceMappingURL=generate-image.dto.js.map

package/dto/generate-video.dto.js

@@ -0,0 +1,112 @@
+"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.GenerateVideoDto = void 0;
+const class_validator_1 = require("class-validator");
+const class_transformer_1 = require("class-transformer");
+class ImageInputDto {
+    bytesBase64Encoded;
+    gcsUri;
+    mimeType;
+}
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsBase64)(),
+    (0, class_validator_1.ValidateIf)((o) => !o.gcsUri),
+    (0, class_validator_1.IsNotEmpty)({ message: 'Either bytesBase64Encoded or gcsUri must be provided in image object' }),
+    __metadata("design:type", String)
+], ImageInputDto.prototype, "bytesBase64Encoded", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.Matches)(/^gs:\/\/.+/, { message: 'gcsUri must be a valid Cloud Storage URI (gs://...)' }),
+    (0, class_validator_1.ValidateIf)((o) => !o.bytesBase64Encoded),
+    (0, class_validator_1.IsNotEmpty)({ message: 'Either bytesBase64Encoded or gcsUri must be provided in image object' }),
+    __metadata("design:type", String)
+], ImageInputDto.prototype, "gcsUri", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsNotEmpty)(),
+    __metadata("design:type", String)
+], ImageInputDto.prototype, "mimeType", void 0);
+class GenerateVideoDto {
+    prompt;
+    image;
+    durationSeconds = 8;
+    aspectRatio = '16:9';
+    negativePrompt;
+    personGeneration = 'allow_adult';
+    sampleCount = 1;
+    seed;
+    storageUri;
+    enhancePrompt = true;
+}
+exports.GenerateVideoDto = GenerateVideoDto;
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.ValidateIf)((o) => !o.image),
+    (0, class_validator_1.IsNotEmpty)({ message: 'prompt is required if image is not provided' }),
+    __metadata("design:type", String)
+], GenerateVideoDto.prototype, "prompt", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.ValidateNested)(),
+    (0, class_transformer_1.Type)(() => ImageInputDto),
+    __metadata("design:type", ImageInputDto)
+], GenerateVideoDto.prototype, "image", void 0);
+__decorate([
+    (0, class_validator_1.IsInt)(),
+    (0, class_validator_1.Min)(5),
+    (0, class_validator_1.Max)(8),
+    __metadata("design:type", Number)
+], GenerateVideoDto.prototype, "durationSeconds", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsEnum)(['16:9', '9:16'], { message: 'aspectRatio must be either "16:9" or "9:16"' }),
+    __metadata("design:type", String)
+], GenerateVideoDto.prototype, "aspectRatio", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsString)(),
+    __metadata("design:type", String)
+], GenerateVideoDto.prototype, "negativePrompt", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsEnum)(['allow_adult', 'disallow'], { message: 'personGeneration must be either "allow_adult" or "disallow"' }),
+    __metadata("design:type", String)
+], GenerateVideoDto.prototype, "personGeneration", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsInt)(),
+    (0, class_validator_1.Min)(1),
+    (0, class_validator_1.Max)(4),
+    __metadata("design:type", Number)
+], GenerateVideoDto.prototype, "sampleCount", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsInt)(),
+    (0, class_validator_1.Min)(0),
+    (0, class_validator_1.Max)(4294967295),
+    __metadata("design:type", Number)
+], GenerateVideoDto.prototype, "seed", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.Matches)(/^gs:\/\/.+\/.+/, { message: 'storageUri must be a valid Cloud Storage URI (gs://BUCKET_NAME/SUBDIRECTORY)' }),
+    __metadata("design:type", String)
+], GenerateVideoDto.prototype, "storageUri", void 0);
+__decorate([
+    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsBoolean)(),
+    __metadata("design:type", Boolean)
+], GenerateVideoDto.prototype, "enhancePrompt", void 0);
+//# sourceMappingURL=generate-video.dto.js.map