@dataclouder/nest-vertex 0.0.46 → 0.0.47

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

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

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

@@ -2,15 +2,15 @@ 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';
+import { StorageFactory } from '@dataclouder/nest-storage';
 export declare class ComfySDKService {
     private readonly generatedAssetService;
     private readonly httpService;
-    private readonly cloudStorageService;
+    private readonly storageFactory;
     private readonly logger;
     private outputDirectory;
     private api;
-    constructor(generatedAssetService: GeneratedAssetService, httpService: HttpService, cloudStorageService: CloudStorageService);
+    constructor(generatedAssetService: GeneratedAssetService, httpService: HttpService, storageFactory: StorageFactory);
     private init;
     testMethod(): Promise<{
         status: string;

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

@@ -54,14 +54,14 @@ const nest_storage_1 = require("@dataclouder/nest-storage");
 let ComfySDKService = ComfySDKService_1 = class ComfySDKService {
     generatedAssetService;
     httpService;
-    cloudStorageService;
+    storageFactory;
     logger = new common_1.Logger(ComfySDKService_1.name);
     outputDirectory = './comfyui_generated_images';
     api;
-    constructor(generatedAssetService, httpService, cloudStorageService) {
+    constructor(generatedAssetService, httpService, storageFactory) {
         this.generatedAssetService = generatedAssetService;
         this.httpService = httpService;
-        this.cloudStorageService = cloudStorageService;
+        this.storageFactory = storageFactory;
         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);
     }
@@ -119,15 +119,12 @@ let ComfySDKService = ComfySDKService_1 = class ComfySDKService {
         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,
-        };
+        const videoRequest = { ...genAsset.request };
+        videoRequest.inputImage = imageBuffer;
+        videoRequest.positivePrompt = genAsset.prompt || genAsset?.description || 'Random movement for video';
+        videoRequest.negativePrompt = 'low quality, blurry, static';
+        videoRequest.workflowPath = path.join(process.cwd(), 'comfy_workflows', 'video-15_seconds.json');
+        console.log(' -> USANDO el viejo confiable flujo: garantizar que exite: video-15_seconds.json');
         try {
             const result = await this.runVideoGenerationFromWorkflow(videoRequest, assetId);
             this.logger.log(`Video generation for asset ${assetId} completed successfully.`, result);
@@ -183,9 +180,13 @@ let ComfySDKService = ComfySDKService_1 = class ComfySDKService {
                     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');
+                        const storageFileName = videoRequest?.fileName || `${assetId}.mp4`;
+                        const storageFinalPath = `${storageFolder}/${storageFileName}`;
+                        console.log(' -> 📼 Saving video to storage path:', storageFinalPath);
+                        const storageService = this.storageFactory.getStorageService('CLOUDFLARE');
+                        console.log(' -> USING STORAGE SERVICE:', storageService);
+                        const options = { fileName: storageFinalPath, fileBuffer: videoBuffer, contentType: 'video/mp4' };
+                        const uploadResult = await storageService.uploadPublicFile(options);
                         this.logger.log(`Successfully uploaded video to: ${uploadResult.url}`);
                         await this.generatedAssetService.partialUpdate(assetId, {
                             url: uploadResult.url,
@@ -221,6 +222,6 @@ 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])
+        nest_storage_1.StorageFactory])
 ], ComfySDKService);
 //# sourceMappingURL=comfy-sdk.service.js.map

package/config/gemini.config.d.ts

@@ -0,0 +1 @@
+export declare const GEMINI_API_KEY = "{\n  \"type\": \"service_account\",\n  \"project_id\": \"dataclouder-dev\",\n  \"private_key_id\": \"044453d30d7f99ec3b2231a6e293f003f7dc6e14\",\n  \"private_key\": \"-----BEGIN PRIVATE KEY-----\\nMIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQCn+rv6kYKpuXtI\\noCe/3pcacKejTfYFAcd63JYGsqXe/iNa1ou4gESG5xVNQ20/0nB0Ubb+x4KpWWNv\\n7Y/LlNN+7V4eTk+fORxt4iZ3iQXBCQfQKMBPEvr/T1h1vviKmiknrFdKEByvXo1Y\\nfLkFtyAUg8B/mSecsQDFlPnqfQnUPARFVtajvy8S8RlRleshoJbaMX0YcLDO6lCa\\np6wx/BoEBdICg9XpNiYm2edvoGwgWQU8PMZmNtHGX6WVLWHlvP9w1k6wPEKBQTry\\nnpFMDVkll63XGgFBiFKO7KKvxVcyrKRafbBXRXHrafDRLljkeltFNRGYw83Xht6Y\\nFLaNsubZAgMBAAECggEAAmYVWd9K1EFksi1uT+I9+mfOLwnxl9H2Dyo3/r29mmT6\\n7ITIj0COqfYCEv3B4UHQLwsNL895zVEMV1h3P5cfG46Ic+qWnr2WKzgJ6FZVtZNb\\nUBTK5T4uRSfVCa5p1nHb4VgH0ZQQEs2I+3DhhARA0iYtAS122DkbW5b0aC2zrpsM\\nW88P5obIg4eWCYTbGnesYLLbJ9ZqpduD6IoSKme+RKNP6dSsH2v62tFmK64067jV\\nMB2LziNGguCJKUe3+/hIhopzxab7+wgxD3Jd58vCDsEUTwjrLfx700rmDDk5mjLG\\nAQCSzeIF15QGvqUbRYMmwfywXUfqUT3ABj1v4TEI4QKBgQDhBAJeRvJtg8RhiQbb\\nJfFOZ61K63d8NHu+a1C/qlVytDEEO5G7BEPrcbHk6fM46xl4wDpGQ7n0kY7ZmH78\\n8eZNkfXgMLvpNzKpj4oTCqAcY4abHuYA0IMA56hp+Sgn+pos7drdVFXCELD6Mys9\\nnh0rP1D/Kiuv2htmHkshchd84QKBgQC/HCZsgfN8O9HrE+oVd9Sp5aW13eaBkG/v\\nNUKxuaGdMGulxFLZQMJrugt2T4yNdjaUq625fmja1ujtIsU7JtlltBLZOAC9ZphF\\nPIfdgX48pvLdJIgOjgoxD4bfVav4Flum16wBskE+PSowdV0XEiOIi0GyofGBAMlT\\nNWDrnThw+QKBgQCcRLaIVulCGUKnIgy64WzmMKZu43yFq8VMhyb7FQZmtLFd6Sja\\nIqowtzL+cWNS4iM8cvOFnL9mFDiK8Zcvf6ebs4R4gB/gZU8v5tnTQ4vSa2QuEpVV\\nd3mjhT626ixzJ10vWrinAUzVi9X8zNB46HgFcuGDc/wHEof4sY0n2x5owQKBgQCY\\nXr1T7hhPiPJJV0tfancoLmz8n7tUufBE7FYMlVCBn0V0mR7lEHjyoxbYcXxCnWnP\\nB0IvjtXTBV4ZE4vlKYhTSkts3PXaDJ6DNUuuE7yuZv1jy+iekikWDKcgEZJW7pyM\\nXZBKmfDiyqmEkU5zQocKKt1qqyt+Cz55FSn9MRD7IQKBgDeUMSwAwQgtRu4RHNCJ\\n5IBaEfEX4fVIRscvFWKxUyhvUb5hpGj1asO0SoUboBw1wGjW/HcKZ7Gw/v+8103v\\n7y1fFqQxMY7eWrLegLtywAfnQW+Uyvaha6n85Ekpuk6X6mSd/IFhe4TPknCbb4+d\\n+NwWPEalsS86ekmB4TeZLQ+H\\n-----END PRIVATE KEY-----\\n\",\n  \"client_email\": \"app-keys-dataclouder-dev@dataclouder-dev.iam.gserviceaccount.com\",\n  \"client_id\": \"107525823981418207310\",\n  \"auth_uri\": \"https://accounts.google.com/o/oauth2/auth\",\n  \"token_uri\": \"https://oauth2.googleapis.com/token\",\n  \"auth_provider_x509_cert_url\": \"https://www.googleapis.com/oauth2/v1/certs\",\n  \"client_x509_cert_url\": \"https://www.googleapis.com/robot/v1/metadata/x509/app-keys-dataclouder-dev%40dataclouder-dev.iam.gserviceaccount.com\",\n  \"universe_domain\": \"googleapis.com\"\n}";

package/config/gemini.config.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GEMINI_API_KEY = void 0;
+exports.GEMINI_API_KEY = `{
+  "type": "service_account",
+  "project_id": "dataclouder-dev",
+  "private_key_id": "044453d30d7f99ec3b2231a6e293f003f7dc6e14",
+  "private_key": "-----BEGIN PRIVATE KEY-----\\nMIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQCn+rv6kYKpuXtI\\noCe/3pcacKejTfYFAcd63JYGsqXe/iNa1ou4gESG5xVNQ20/0nB0Ubb+x4KpWWNv\\n7Y/LlNN+7V4eTk+fORxt4iZ3iQXBCQfQKMBPEvr/T1h1vviKmiknrFdKEByvXo1Y\\nfLkFtyAUg8B/mSecsQDFlPnqfQnUPARFVtajvy8S8RlRleshoJbaMX0YcLDO6lCa\\np6wx/BoEBdICg9XpNiYm2edvoGwgWQU8PMZmNtHGX6WVLWHlvP9w1k6wPEKBQTry\\nnpFMDVkll63XGgFBiFKO7KKvxVcyrKRafbBXRXHrafDRLljkeltFNRGYw83Xht6Y\\nFLaNsubZAgMBAAECggEAAmYVWd9K1EFksi1uT+I9+mfOLwnxl9H2Dyo3/r29mmT6\\n7ITIj0COqfYCEv3B4UHQLwsNL895zVEMV1h3P5cfG46Ic+qWnr2WKzgJ6FZVtZNb\\nUBTK5T4uRSfVCa5p1nHb4VgH0ZQQEs2I+3DhhARA0iYtAS122DkbW5b0aC2zrpsM\\nW88P5obIg4eWCYTbGnesYLLbJ9ZqpduD6IoSKme+RKNP6dSsH2v62tFmK64067jV\\nMB2LziNGguCJKUe3+/hIhopzxab7+wgxD3Jd58vCDsEUTwjrLfx700rmDDk5mjLG\\nAQCSzeIF15QGvqUbRYMmwfywXUfqUT3ABj1v4TEI4QKBgQDhBAJeRvJtg8RhiQbb\\nJfFOZ61K63d8NHu+a1C/qlVytDEEO5G7BEPrcbHk6fM46xl4wDpGQ7n0kY7ZmH78\\n8eZNkfXgMLvpNzKpj4oTCqAcY4abHuYA0IMA56hp+Sgn+pos7drdVFXCELD6Mys9\\nnh0rP1D/Kiuv2htmHkshchd84QKBgQC/HCZsgfN8O9HrE+oVd9Sp5aW13eaBkG/v\\nNUKxuaGdMGulxFLZQMJrugt2T4yNdjaUq625fmja1ujtIsU7JtlltBLZOAC9ZphF\\nPIfdgX48pvLdJIgOjgoxD4bfVav4Flum16wBskE+PSowdV0XEiOIi0GyofGBAMlT\\nNWDrnThw+QKBgQCcRLaIVulCGUKnIgy64WzmMKZu43yFq8VMhyb7FQZmtLFd6Sja\\nIqowtzL+cWNS4iM8cvOFnL9mFDiK8Zcvf6ebs4R4gB/gZU8v5tnTQ4vSa2QuEpVV\\nd3mjhT626ixzJ10vWrinAUzVi9X8zNB46HgFcuGDc/wHEof4sY0n2x5owQKBgQCY\\nXr1T7hhPiPJJV0tfancoLmz8n7tUufBE7FYMlVCBn0V0mR7lEHjyoxbYcXxCnWnP\\nB0IvjtXTBV4ZE4vlKYhTSkts3PXaDJ6DNUuuE7yuZv1jy+iekikWDKcgEZJW7pyM\\nXZBKmfDiyqmEkU5zQocKKt1qqyt+Cz55FSn9MRD7IQKBgDeUMSwAwQgtRu4RHNCJ\\n5IBaEfEX4fVIRscvFWKxUyhvUb5hpGj1asO0SoUboBw1wGjW/HcKZ7Gw/v+8103v\\n7y1fFqQxMY7eWrLegLtywAfnQW+Uyvaha6n85Ekpuk6X6mSd/IFhe4TPknCbb4+d\\n+NwWPEalsS86ekmB4TeZLQ+H\\n-----END PRIVATE KEY-----\\n",
+  "client_email": "app-keys-dataclouder-dev@dataclouder-dev.iam.gserviceaccount.com",
+  "client_id": "107525823981418207310",
+  "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+  "token_uri": "https://oauth2.googleapis.com/token",
+  "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
+  "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/app-keys-dataclouder-dev%40dataclouder-dev.iam.gserviceaccount.com",
+  "universe_domain": "googleapis.com"
+}`;
+//# sourceMappingURL=gemini.config.js.map

package/controllers/vertex-gemini-chat.controller.d.ts

@@ -1,8 +1,20 @@
 import { GeminiChatService } from '../services/vertex-gemini-chat.service';
 import { GenerateTextDto } from '../dto/generate-text.dto';
+import { ChatMessageDict, IAIModel } from '../models/adapter.models';
+export declare class ConversationDTO {
+    id?: string;
+    entityId?: string;
+    type?: string;
+    createdAt?: Date;
+    messages: Array<ChatMessageDict>;
+    transcription?: boolean;
+    model?: IAIModel;
+    returnJson?: boolean;
+}
 export declare class GeminiChatController {
     private readonly geminiChatService;
     private readonly logger;
     constructor(geminiChatService: GeminiChatService);
-    generateText(generateTextDto: GenerateTextDto): Promise<import("..").ChatMessageDict>;
+    generateText(generateTextDto: GenerateTextDto): Promise<ChatMessageDict>;
+    continueConversation(conversation: ConversationDTO): Promise<ChatMessageDict>;
 }

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

@@ -13,12 +13,25 @@ var __param = (this && this.__param) || function (paramIndex, decorator) {
 };
 var GeminiChatController_1;
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.GeminiChatController = void 0;
+exports.GeminiChatController = exports.ConversationDTO = void 0;
 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");
+const gemini_models_1 = require("../models/gemini-models");
+const adapter_models_1 = require("../models/adapter.models");
+class ConversationDTO {
+    id;
+    entityId;
+    type;
+    createdAt;
+    messages;
+    transcription;
+    model;
+    returnJson;
+}
+exports.ConversationDTO = ConversationDTO;
 let GeminiChatController = GeminiChatController_1 = class GeminiChatController {
     geminiChatService;
     logger = new common_1.Logger(GeminiChatController_1.name);
@@ -36,6 +49,40 @@ let GeminiChatController = GeminiChatController_1 = class GeminiChatController {
             throw error;
         }
     }
+    async continueConversation(conversation) {
+        const startTime = Date.now();
+        if (!conversation?.model) {
+            conversation.model = { provider: 'google', modelName: gemini_models_1.GeminiModels.Gemini2_5Lite, id: 'no-id' };
+        }
+        if (conversation?.model?.quality) {
+            conversation.model.provider = 'google';
+            conversation.model.modelName = this.geminiChatService.getDefaultQualityModel(conversation.model.quality);
+        }
+        const returnJson = conversation.returnJson;
+        if (returnJson) {
+            const obj = await this.geminiChatService.chatAndExtractJson(conversation.messages, conversation.model.modelName);
+            const endTime = Date.now();
+            const processTime = (endTime - startTime) / 1000;
+            const metadata = {
+                type: 'json',
+                provider: conversation.model.provider,
+                model: conversation.model.modelName,
+                processTime,
+                ...obj?.metadata,
+            };
+            return { content: obj.json, role: adapter_models_1.ChatRole.Assistant, metadata };
+        }
+        else {
+            const response = await this.geminiChatService.chat(conversation.messages, conversation.model.modelName);
+            const endTime = Date.now();
+            const processTime = (endTime - startTime) / 1000;
+            return {
+                content: response.content,
+                role: response.role,
+                metadata: { provider: conversation.model.provider, model: conversation.model.modelName, processTime, tokens: response.metadata.tokens },
+            };
+        }
+    }
 };
 exports.GeminiChatController = GeminiChatController;
 __decorate([
@@ -45,6 +92,15 @@ __decorate([
     __metadata("design:paramtypes", [generate_text_dto_1.GenerateTextDto]),
     __metadata("design:returntype", Promise)
 ], GeminiChatController.prototype, "generateText", null);
+__decorate([
+    (0, common_1.Post)('/chat'),
+    (0, swagger_1.ApiOperation)({ summary: 'Continue the conversation' }),
+    (0, swagger_1.ApiResponse)({ status: 200, description: 'Return the conversation.' }),
+    __param(0, (0, common_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [ConversationDTO]),
+    __metadata("design:returntype", Promise)
+], GeminiChatController.prototype, "continueConversation", null);
 exports.GeminiChatController = GeminiChatController = GeminiChatController_1 = __decorate([
     (0, swagger_1.ApiTags)('LLM Gemini '),
     (0, common_1.Controller)('api/ai-services/gemini'),

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

@@ -0,0 +1,9 @@
+import { type FastifyReply } from 'fastify';
+import { VertexGeminiTtsService } from '../services/vertex-gemini-tts.service';
+import { type SynthesizeSpeechInput } from '../services/vertex-tts.service';
+export declare class VertexGeminiTtsController {
+    private readonly vertexGeminiTtsService;
+    private readonly logger;
+    constructor(vertexGeminiTtsService: VertexGeminiTtsService);
+    synthesizeSpeech(body: SynthesizeSpeechInput, res: FastifyReply): Promise<Buffer<ArrayBufferLike>>;
+}

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

@@ -0,0 +1,82 @@
+"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.VertexGeminiTtsController = void 0;
+const common_1 = require("@nestjs/common");
+const swagger_1 = require("@nestjs/swagger");
+const vertex_gemini_tts_service_1 = require("../services/vertex-gemini-tts.service");
+const nest_core_1 = require("@dataclouder/nest-core");
+let VertexGeminiTtsController = class VertexGeminiTtsController {
+    vertexGeminiTtsService;
+    logger = new common_1.Logger('VertexGeminiTtsController');
+    constructor(vertexGeminiTtsService) {
+        this.vertexGeminiTtsService = vertexGeminiTtsService;
+    }
+    async synthesizeSpeech(body, res) {
+        this.logger.log(`Received Gemini synthesize request for text: "${body.text.substring(0, 50)}..."`);
+        const audioBuffer = await this.vertexGeminiTtsService.synthesizeSpeech(body);
+        if (!audioBuffer) {
+            this.logger.error('Synthesis resulted in null buffer');
+            throw new nest_core_1.AppException({
+                error_message: '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;
+    }
+};
+exports.VertexGeminiTtsController = VertexGeminiTtsController;
+__decorate([
+    (0, common_1.Post)('synthesize'),
+    (0, swagger_1.ApiOperation)({ summary: 'Synthesize speech from text using Gemini' }),
+    (0, swagger_1.ApiBody)({
+        description: 'Text and configuration for speech synthesis with Gemini',
+        schema: {
+            type: 'object',
+            properties: {
+                text: { type: 'string', example: 'Hello, this is a test.' },
+                prompt: {
+                    type: 'string',
+                    example: 'Say it in a friendly tone.',
+                    description: 'Optional styling instructions.',
+                },
+                voiceName: {
+                    type: 'string',
+                    example: 'Charon',
+                    description: 'Optional specific voice name (default: Charon)',
+                },
+                languageCode: {
+                    type: 'string',
+                    example: 'en-US',
+                    description: 'Optional language code (default: en-US)',
+                },
+            },
+            required: ['text'],
+        },
+    }),
+    (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)
+], VertexGeminiTtsController.prototype, "synthesizeSpeech", null);
+exports.VertexGeminiTtsController = VertexGeminiTtsController = __decorate([
+    (0, swagger_1.ApiTags)('Vertex Gemini TTS'),
+    (0, common_1.Controller)('api/vertex-gemini/tts'),
+    (0, common_1.UseFilters)(nest_core_1.AllExceptionsHandler),
+    __metadata("design:paramtypes", [vertex_gemini_tts_service_1.VertexGeminiTtsService])
+], VertexGeminiTtsController);
+//# sourceMappingURL=vertex-gemini-tts.controller.js.map

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

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

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

@@ -18,16 +18,16 @@ const swagger_1 = require("@nestjs/swagger");
 const vertex_tts_service_1 = require("../services/vertex-tts.service");
 const nest_core_1 = require("@dataclouder/nest-core");
 const common_2 = require("@nestjs/common");
+const nest_auth_1 = require("@dataclouder/nest-auth");
 let VertexTtsAdapterController = class VertexTtsAdapterController {
     nestTtsService;
     logger = new common_1.Logger('VertexTtsAdapterController');
     constructor(nestTtsService) {
         this.nestTtsService = nestTtsService;
     }
-    async synthesizeSpeech(body, res) {
-        console.log('Received synthesize request', body);
+    async synthesizeSpeech(body, res, token) {
         this.logger.log(`Received synthesize request for text: "${body.text.substring(0, 50)}..."`);
-        const audioBuffer = await this.nestTtsService.synthesizeSpeech(body);
+        const audioBuffer = await this.nestTtsService.synthesizeSpeech(body, token);
         if (!audioBuffer) {
             this.logger.error('Synthesis resulted in null buffer');
             throw new nest_core_1.AppException({
@@ -82,8 +82,9 @@ __decorate([
     (0, common_1.Header)('Content-Type', 'audio/mpeg'),
     __param(0, (0, common_1.Body)()),
     __param(1, (0, common_1.Res)({ passthrough: true })),
+    __param(2, (0, nest_auth_1.DecodedToken)()),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [Object, Object]),
+    __metadata("design:paramtypes", [Object, Object, Object]),
     __metadata("design:returntype", Promise)
 ], VertexTtsAdapterController.prototype, "synthesizeSpeech", null);
 __decorate([
@@ -103,9 +104,10 @@ __decorate([
     __metadata("design:returntype", Promise)
 ], VertexTtsAdapterController.prototype, "listVoices", null);
 exports.VertexTtsAdapterController = VertexTtsAdapterController = __decorate([
+    (0, common_1.UseGuards)(nest_auth_1.AuthGuard),
     (0, swagger_1.ApiTags)('Vertex TTS Adapter'),
     (0, common_1.Controller)('api/vertex-adapter/tts'),
     (0, common_2.UseFilters)(nest_core_1.AllExceptionsHandler),
-    __metadata("design:paramtypes", [vertex_tts_service_1.NestTtsService])
+    __metadata("design:paramtypes", [vertex_tts_service_1.VertextTtsService])
 ], VertexTtsAdapterController);
 //# sourceMappingURL=vertex-tts-adapter.controller.js.map

package/drizzle/drizzle.module.d.ts

@@ -0,0 +1,2 @@
+export declare class DrizzleModule {
+}

package/drizzle/drizzle.module.js

@@ -0,0 +1,35 @@
+"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;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DrizzleModule = void 0;
+const common_1 = require("@nestjs/common");
+const drizzle_service_1 = require("./drizzle.service");
+const key_balancer_service_1 = require("./key-balancer.service");
+const key_balancer_controller_1 = require("./key-balancer.controller");
+const user_requests_service_1 = require("./user-requests.service");
+const user_requests_controller_1 = require("./user-requests.controller");
+const key_usage_service_1 = require("./key-usage.service");
+let DrizzleModule = class DrizzleModule {
+};
+exports.DrizzleModule = DrizzleModule;
+exports.DrizzleModule = DrizzleModule = __decorate([
+    (0, common_1.Module)({
+        controllers: [key_balancer_controller_1.KeyBalancerController, user_requests_controller_1.UserRequestsController],
+        providers: [
+            {
+                provide: drizzle_service_1.DRIZZLE_SERVICE,
+                useClass: drizzle_service_1.DrizzleService,
+            },
+            key_balancer_service_1.KeyBalancerService,
+            user_requests_service_1.UserRequestsService,
+            key_usage_service_1.KeyUsageService,
+        ],
+        exports: [key_balancer_service_1.KeyBalancerService, user_requests_service_1.UserRequestsService, key_usage_service_1.KeyUsageService],
+    })
+], DrizzleModule);
+//# sourceMappingURL=drizzle.module.js.map

package/drizzle/drizzle.service.d.ts

@@ -0,0 +1,10 @@
+import { Pool } from 'pg';
+import * as schema from './schema';
+export declare const DRIZZLE_SERVICE = "DRIZZLE_SERVICE";
+export declare class DrizzleService {
+    private _drizzle;
+    constructor();
+    get db(): import("drizzle-orm/node-postgres").NodePgDatabase<typeof schema> & {
+        $client: Pool;
+    };
+}

package/drizzle/drizzle.service.js

@@ -0,0 +1,68 @@
+"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);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DrizzleService = exports.DRIZZLE_SERVICE = void 0;
+const common_1 = require("@nestjs/common");
+const node_postgres_1 = require("drizzle-orm/node-postgres");
+const pg_1 = require("pg");
+const schema = __importStar(require("./schema"));
+exports.DRIZZLE_SERVICE = 'DRIZZLE_SERVICE';
+let DrizzleService = class DrizzleService {
+    _drizzle;
+    constructor() {
+        const pool = new pg_1.Pool({
+            connectionString: process.env.DATABASE_URL,
+        });
+        this._drizzle = (0, node_postgres_1.drizzle)(pool, { schema });
+    }
+    get db() {
+        return this._drizzle;
+    }
+};
+exports.DrizzleService = DrizzleService;
+exports.DrizzleService = DrizzleService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [])
+], DrizzleService);
+//# sourceMappingURL=drizzle.service.js.map

package/drizzle/dto/create-key-balancer.dto.d.ts

@@ -0,0 +1,9 @@
+export declare class CreateKeyBalancerDto {
+    name: string;
+    email: string;
+    disabled?: boolean;
+    description?: string;
+    key: string;
+    type: string;
+    provider: string;
+}

package/drizzle/dto/create-key-balancer.dto.js

@@ -0,0 +1,54 @@
+"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.CreateKeyBalancerDto = void 0;
+const class_validator_1 = require("class-validator");
+class CreateKeyBalancerDto {
+    name;
+    email;
+    disabled;
+    description;
+    key;
+    type;
+    provider;
+}
+exports.CreateKeyBalancerDto = CreateKeyBalancerDto;
+__decorate([
+    (0, class_validator_1.IsString)(),
+    __metadata("design:type", String)
+], CreateKeyBalancerDto.prototype, "name", void 0);
+__decorate([
+    (0, class_validator_1.IsEmail)(),
+    __metadata("design:type", String)
+], CreateKeyBalancerDto.prototype, "email", void 0);
+__decorate([
+    (0, class_validator_1.IsBoolean)(),
+    (0, class_validator_1.IsOptional)(),
+    __metadata("design:type", Boolean)
+], CreateKeyBalancerDto.prototype, "disabled", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsOptional)(),
+    __metadata("design:type", String)
+], CreateKeyBalancerDto.prototype, "description", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    __metadata("design:type", String)
+], CreateKeyBalancerDto.prototype, "key", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    __metadata("design:type", String)
+], CreateKeyBalancerDto.prototype, "type", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    __metadata("design:type", String)
+], CreateKeyBalancerDto.prototype, "provider", void 0);
+//# sourceMappingURL=create-key-balancer.dto.js.map

package/drizzle/dto/create-key-usage.dto.d.ts

@@ -0,0 +1,6 @@
+export declare class CreateKeyUsageDto {
+    id?: string;
+    key: string;
+    charCount?: number;
+    totalRequest?: number;
+}

package/drizzle/dto/create-key-usage.dto.js

@@ -0,0 +1,11 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CreateKeyUsageDto = void 0;
+class CreateKeyUsageDto {
+    id;
+    key;
+    charCount;
+    totalRequest;
+}
+exports.CreateKeyUsageDto = CreateKeyUsageDto;
+//# sourceMappingURL=create-key-usage.dto.js.map

package/drizzle/dto/create-user-request.dto.d.ts

@@ -0,0 +1,7 @@
+export declare class CreateUserRequestDto {
+    userId: string;
+    email: string;
+    idKey?: string;
+    amountUsed: number;
+    serviceType: string;
+}

package/drizzle/dto/create-user-request.dto.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);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CreateUserRequestDto = void 0;
+const class_validator_1 = require("class-validator");
+class CreateUserRequestDto {
+    userId;
+    email;
+    idKey;
+    amountUsed;
+    serviceType;
+}
+exports.CreateUserRequestDto = CreateUserRequestDto;
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsNotEmpty)(),
+    __metadata("design:type", String)
+], CreateUserRequestDto.prototype, "userId", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsNotEmpty)(),
+    __metadata("design:type", String)
+], CreateUserRequestDto.prototype, "email", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsOptional)(),
+    __metadata("design:type", String)
+], CreateUserRequestDto.prototype, "idKey", void 0);
+__decorate([
+    (0, class_validator_1.IsInt)(),
+    (0, class_validator_1.IsNotEmpty)(),
+    __metadata("design:type", Number)
+], CreateUserRequestDto.prototype, "amountUsed", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsNotEmpty)(),
+    __metadata("design:type", String)
+], CreateUserRequestDto.prototype, "serviceType", void 0);
+//# sourceMappingURL=create-user-request.dto.js.map