@dataclouder/nest-vertex 0.0.44 → 0.0.45

package/package.json

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

package/services/key-balancer-api.service.js

@@ -29,7 +29,7 @@ let KeyBalancerApiService = KeyBalancerApiService_1 = class KeyBalancerApiServic
         try {
             const url = `${keyBalancerHost}/api/key-balancer/redis/key`;
             const { data } = await (0, rxjs_1.firstValueFrom)(this.httpService.post(url, request));
-            this.logger.verbose(`Request to: ${url} for key :${request.provider} getting key: ${data.name}`);
+            this.logger.verbose(`Requested to: ${url} for key :${request.provider} type: ${request.aiType} model: ${request.model} tier: ${request.tier} getting key: ${data.name}`);
             return data;
         }
         catch (error) {

package/services/vertex-image.service.js

@@ -34,7 +34,7 @@ let ImageVertexService = ImageVertexService_1 = class ImageVertexService {
     async listModels() {
         return gemini_models_1.GeminiImageModelsList;
     }
-    async _getGenAiClient(model, tierType = null) {
+    async _getGenAiClient(model, tierType = key_balancer_models_1.TierType.TIER_1) {
         let apiKey;
         let keyId;
         let availableKey = {};
@@ -43,12 +43,13 @@ let ImageVertexService = ImageVertexService_1 = class ImageVertexService {
             keyId = 'local';
         }
         else {
-            availableKey = (await this.keyBalancerApiService.postQueryKey({
+            this.logger.log(`Vertex Image"Getting key for model: ${model} and tier: ${tierType}`);
+            availableKey = await this.keyBalancerApiService.postQueryKey({
                 provider: 'google',
                 model: model,
-                tier: key_balancer_models_1.TierType.TIER_1,
+                tier: tierType,
                 aiType: key_balancer_models_1.ModelType.IMAGE,
-            }));
+            });
             if (availableKey === null) {
                 this.logger.error('KEY BALANCER IS DOWN: :::: No available API key from key balancer.');
             }

package/services/vertex-veo-genai.service.js

@@ -152,14 +152,25 @@ let VertexVeoGenaiService = class VertexVeoGenaiService {
         });
         this.logger.log(` ❗️ Bug de GEN AI API, No FUNCIONA ASYNC para descargar video, esperando 3.5 segundos a que descarge.....`);
         await new Promise(resolve => setTimeout(resolve, 3500));
-        const videoBuffer = await fs.readFile(filename);
-        console.log('Subiendo video a storage', filename, process.env.STORAGE_BUCKET);
-        const uploadResult = await this.cloudStorageService.uploadFileAndMakePublic(process.env.STORAGE_BUCKET, `generated-videos/${filename}`, videoBuffer, 'video/mp4');
-        this.logger.log(`Generated video uploaded to ${uploadResult.url}`);
-        return {
-            url: uploadResult.url,
-            result: uploadResult,
-        };
+        try {
+            const videoBuffer = await fs.readFile(filename);
+            console.log('Subiendo video a storage', filename, process.env.STORAGE_BUCKET);
+            const uploadResult = await this.cloudStorageService.uploadFileAndMakePublic(process.env.STORAGE_BUCKET, `generated-videos/${filename}`, videoBuffer, 'video/mp4');
+            this.logger.log(`Generated video uploaded to ${uploadResult.url}`);
+            return {
+                url: uploadResult.url,
+                result: uploadResult,
+            };
+        }
+        finally {
+            try {
+                await fs.unlink(filename);
+                this.logger.log(`Successfully deleted temporary video file: ${filename}`);
+            }
+            catch (error) {
+                this.logger.error(`Failed to delete temporary video file: ${filename}`, error);
+            }
+        }
     }
 };
 exports.VertexVeoGenaiService = VertexVeoGenaiService;