@dataclouder/nest-vertex 0.0.52 → 0.0.53

package/package.json

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

package/services/google-genai.service.js

@@ -28,14 +28,14 @@ let GoogleGenaiService = GoogleGenaiService_1 = class GoogleGenaiService {
     }
     async getGoogleGenAIClient(request) {
         this.logger.warn('⚖️ getGoogleGenAIClient() Requesting key from Balancer 🗝️ ');
-        if (process.env.KEY_BALANCER_HOST && request.aiType) {
+        if (process.env.KEY_BALANCER_HOST) {
             const balancedKey = await this.keyBalancerClient.getBestKey({
                 provider: 'google',
                 service: request.model,
                 tierType: request.keyTierType,
             }, null);
             if (balancedKey?.key) {
-                this.logger.debug(`⚖️ Using balanced key: ${balancedKey.name} ${balancedKey.id}`);
+                this.logger.debug(`⚖️ Using balanced key: ${balancedKey.id})  ${balancedKey.name} - ${balancedKey.key} `);
                 return { client: new genai_1.GoogleGenAI({ apiKey: balancedKey.key }), balancedKey };
             }
         }

package/services/vertex-gemini-chat.service.js

@@ -105,6 +105,7 @@ let GeminiChatService = GeminiChatService_1 = class GeminiChatService {
             };
         }
         catch (error) {
+            this.logger.error(`First Error Gemini Gemini {${balancedKey?.key} chat.sendMessage failed: ${error.message}`);
             if (error instanceof nest_core_1.AppException) {
                 throw error;
             }