@entro314labs/ai-changelog-generator 3.0.5 → 3.2.0

package/src/infrastructure/providers/implementations/vertex.js

@@ -1,32 +1,35 @@
-import { GoogleGenAI } from '@google/genai';
-import { GoogleAuth } from 'google-auth-library';
-import { BaseProvider } from '../core/base-provider.js';
-import { applyMixins } from '../utils/base-provider-helpers.js';
-import { buildClientOptions } from '../utils/provider-utils.js';
+import process from 'node:process'
+
+import { GoogleGenAI } from '@google/genai'
+import { GoogleAuth } from 'google-auth-library'
+
+import { BaseProvider } from '../core/base-provider.js'
+import { applyMixins } from '../utils/base-provider-helpers.js'
+import { buildClientOptions } from '../utils/provider-utils.js'
 
 // Cache for model instances to avoid recreating them (with size limit to prevent memory leaks)
-const modelCache = new Map();
-const MAX_MODEL_CACHE_SIZE = 50;
+const modelCache = new Map()
+const MAX_MODEL_CACHE_SIZE = 50
 
 function addToModelCache(key, value) {
   if (modelCache.size >= MAX_MODEL_CACHE_SIZE) {
     // Remove oldest entries (first added)
-    const firstKey = modelCache.keys().next().value;
-    modelCache.delete(firstKey);
+    const firstKey = modelCache.keys().next().value
+    modelCache.delete(firstKey)
   }
-  modelCache.set(key, value);
+  modelCache.set(key, value)
 }
 
 class VertexAIProvider extends BaseProvider {
   constructor(config) {
-    super(config);
-    this.client = null;
-    this.generativeModel = null;
-    this.auth = null;
-    this.authClient = null;
-    
+    super(config)
+    this.client = null
+    this.generativeModel = null
+    this.auth = null
+    this.authClient = null
+
     if (this.isAvailable()) {
-      this.initializeClient();
+      this.initializeClient()
     }
   }
 
@@ -34,159 +37,164 @@ class VertexAIProvider extends BaseProvider {
     try {
       const clientOptions = buildClientOptions(this.getProviderConfig(), {
         location: 'us-central1',
-        apiVersion: 'v1'
-      });
-      
+        apiVersion: 'v1',
+      })
+
       // Initialize Google Auth if needed
       if (clientOptions.keyFile || clientOptions.credentials) {
         this.auth = new GoogleAuth({
           scopes: ['https://www.googleapis.com/auth/cloud-platform'],
           keyFile: clientOptions.keyFile,
-          credentials: clientOptions.credentials ? JSON.parse(clientOptions.credentials) : undefined
-        });
-        this.authClient = await this.auth.getClient();
+          credentials: clientOptions.credentials
+            ? JSON.parse(clientOptions.credentials)
+            : undefined,
+        })
+        this.authClient = await this.auth.getClient()
       }
-      
+
       // Initialize the GoogleGenAI client with Vertex AI configuration
       const vertexOptions = {
         vertexai: true,
         project: clientOptions.projectId,
         location: clientOptions.location,
-        apiVersion: clientOptions.apiVersion
-      };
-      
+        apiVersion: clientOptions.apiVersion,
+      }
+
       if (clientOptions.apiEndpoint) {
-        vertexOptions.apiEndpoint = clientOptions.apiEndpoint;
+        vertexOptions.apiEndpoint = clientOptions.apiEndpoint
       }
-      
+
       if (this.authClient) {
-        const accessToken = await this.authClient.getAccessToken();
-        vertexOptions.apiKey = accessToken.token;
+        const accessToken = await this.authClient.getAccessToken()
+        vertexOptions.apiKey = accessToken.token
       }
-      
-      this.client = new GoogleGenAI(vertexOptions);
 
-      const modelConfig = this.getProviderModelConfig();
-      this.generativeModel = this.getModelInstance(modelConfig.standardModel);
-      
-      return true;
+      this.client = new GoogleGenAI(vertexOptions)
+
+      const modelConfig = this.getProviderModelConfig()
+      this.generativeModel = this.getModelInstance(modelConfig.standardModel)
+
+      return true
     } catch (error) {
-      console.error('Failed to initialize Vertex AI client:', error);
-      this.client = null;
-      this.generativeModel = null;
-      return false;
+      console.error('Failed to initialize Vertex AI client:', error)
+      this.client = null
+      this.generativeModel = null
+      return false
     }
   }
 
   getModelInstance(modelName, options = {}) {
     if (!this.client) {
-      throw new Error('Vertex AI client not initialized');
+      throw new Error('Vertex AI client not initialized')
     }
-    
+
     // Create a cache key based on model name and options
-    const cacheKey = `${modelName}-${JSON.stringify(options)}`;
-    
+    const cacheKey = `${modelName}-${JSON.stringify(options)}`
+
     // Check if we already have this model instance cached
     if (modelCache.has(cacheKey)) {
-      return modelCache.get(cacheKey);
+      return modelCache.get(cacheKey)
     }
-    
+
     // Default generation config
     const generationConfig = {
       temperature: options.temperature || this.config.VERTEX_TEMPERATURE || 0.7,
       topP: options.top_p || this.config.VERTEX_TOP_P || 0.95,
       topK: options.top_k || this.config.VERTEX_TOP_K || 40,
       maxOutputTokens: options.max_tokens || this.config.VERTEX_MAX_TOKENS || 8192,
-    };
-    
+    }
+
     // Add stop sequences if provided
     if (options.stop && options.stop.length > 0) {
-      generationConfig.stopSequences = options.stop;
+      generationConfig.stopSequences = options.stop
     }
-    
+
     // Create model instance
     const modelInstance = this.client.getGenerativeModel({
       model: modelName,
       generationConfig,
-      safetySettings: this.getSafetySettings()
-    });
-    
+      safetySettings: this.getSafetySettings(),
+    })
+
     // Cache the model instance with size limit
-    addToModelCache(cacheKey, modelInstance);
-    
-    return modelInstance;
+    addToModelCache(cacheKey, modelInstance)
+
+    return modelInstance
   }
-  
+
   getSafetySettings() {
     // Configure safety settings based on environment variables or defaults
     const safetySettings = [
       {
         category: 'HARM_CATEGORY_HATE_SPEECH',
-        threshold: this.config.VERTEX_SAFETY_HATE_SPEECH || 'BLOCK_MEDIUM_AND_ABOVE'
+        threshold: this.config.VERTEX_SAFETY_HATE_SPEECH || 'BLOCK_MEDIUM_AND_ABOVE',
       },
       {
         category: 'HARM_CATEGORY_DANGEROUS_CONTENT',
-        threshold: this.config.VERTEX_SAFETY_DANGEROUS || 'BLOCK_MEDIUM_AND_ABOVE'
+        threshold: this.config.VERTEX_SAFETY_DANGEROUS || 'BLOCK_MEDIUM_AND_ABOVE',
       },
       {
         category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
-        threshold: this.config.VERTEX_SAFETY_SEXUALLY_EXPLICIT || 'BLOCK_MEDIUM_AND_ABOVE'
+        threshold: this.config.VERTEX_SAFETY_SEXUALLY_EXPLICIT || 'BLOCK_MEDIUM_AND_ABOVE',
       },
       {
         category: 'HARM_CATEGORY_HARASSMENT',
-        threshold: this.config.VERTEX_SAFETY_HARASSMENT || 'BLOCK_MEDIUM_AND_ABOVE'
-      }
-    ];
-    
-    return safetySettings;
+        threshold: this.config.VERTEX_SAFETY_HARASSMENT || 'BLOCK_MEDIUM_AND_ABOVE',
+      },
+    ]
+
+    return safetySettings
   }
 
   getName() {
-    return 'vertex';
+    return 'vertex'
   }
 
   isAvailable() {
     // Check if we have the required configuration for Vertex AI
-    return !!(this.config.VERTEX_PROJECT_ID && 
-             (this.config.VERTEX_KEY_FILE || this.config.VERTEX_CREDENTIALS || 
-              process.env.GOOGLE_APPLICATION_CREDENTIALS));
+    return !!(
+      this.config.VERTEX_PROJECT_ID &&
+      (this.config.VERTEX_KEY_FILE ||
+        this.config.VERTEX_CREDENTIALS ||
+        process.env.GOOGLE_APPLICATION_CREDENTIALS)
+    )
   }
 
   async generateCompletion(messages, options = {}) {
     if (!this.isAvailable()) {
       return this.handleProviderError(
-        new Error('Vertex AI provider is not configured'), 
+        new Error('Vertex AI provider is not configured'),
         'generate_completion'
-      );
+      )
     }
-    
+
     // Initialize client if not already done
     if (!this.client) {
-      await this.initializeClient();
+      await this.initializeClient()
       if (!this.client) {
         return this.handleProviderError(
-          new Error('Failed to initialize Vertex AI client'), 
+          new Error('Failed to initialize Vertex AI client'),
           'generate_completion'
-        );
+        )
       }
     }
 
     try {
-      const modelConfig = this.getProviderModelConfig();
-      const modelName = options.model || modelConfig.standardModel;
-      
+      const modelConfig = this.getProviderModelConfig()
+      const modelName = options.model || modelConfig.standardModel
+
       // Get or create model instance with the specified options
-      const model = this.getModelInstance(modelName, options);
+      const model = this.getModelInstance(modelName, options)
 
       // Convert messages to Vertex AI format
-      const formattedMessages = this.formatMessages(messages);
+      const formattedMessages = await this.formatMessages(messages)
 
       // Add function calling if provided and the model supports it
-      let tools = null;
+      let tools = null
       if (options.tools && Array.isArray(options.tools) && options.tools.length > 0) {
-        const capabilities = this.getCapabilities(modelName);
+        const capabilities = this.getCapabilities(modelName)
         if (capabilities.function_calling) {
-          tools = this.formatTools(options.tools);
+          tools = this.formatTools(options.tools)
         }
       }
 
@@ -195,206 +203,228 @@ class VertexAIProvider extends BaseProvider {
         // For streaming, we'll use the generateContentStream method
         const streamResult = await model.generateContentStream({
           contents: formattedMessages,
-          tools: tools
-        });
-        
-        let fullContent = '';
+          tools,
+        })
+
+        let fullContent = ''
         for await (const chunk of streamResult.stream) {
-          const chunkContent = chunk.text();
-          fullContent += chunkContent;
+          const chunkContent = chunk.text()
+          fullContent += chunkContent
           options.onUpdate({
             content: chunkContent,
-            done: false
-          });
+            done: false,
+          })
         }
-        
+
         // Signal completion
         options.onUpdate({
           content: '',
-          done: true
-        });
-        
+          done: true,
+        })
+
         // Return the full result
         return {
           content: fullContent,
-          model: modelName
-        };
-      } else {
-        // Non-streaming request
-        const result = await model.generateContent({
-          contents: formattedMessages,
-          tools: tools
-        });
-        
-        // Extract the response text
-        const responseText = result.response?.text() || '';
-        
-        // Handle function calls if present
-        let functionCalls = null;
-        if (result.response?.functionCalls && result.response.functionCalls.length > 0) {
-          functionCalls = result.response.functionCalls.map(call => ({
-            name: call.name,
-            arguments: JSON.parse(call.args)
-          }));
-        }
-        
-        return {
-          content: responseText,
           model: modelName,
-          function_call: functionCalls ? functionCalls[0] : undefined,
-          function_calls: functionCalls
-        };
+        }
+      }
+      // Non-streaming request
+      const result = await model.generateContent({
+        contents: formattedMessages,
+        tools,
+      })
+
+      // Extract the response text
+      const responseText = result.response?.text() || ''
+
+      // Handle function calls if present
+      let functionCalls = null
+      if (result.response?.functionCalls && result.response.functionCalls.length > 0) {
+        functionCalls = result.response.functionCalls.map((call) => ({
+          name: call.name,
+          arguments: JSON.parse(call.args),
+        }))
+      }
+
+      return {
+        content: responseText,
+        model: modelName,
+        function_call: functionCalls ? functionCalls[0] : undefined,
+        function_calls: functionCalls,
       }
     } catch (error) {
       // Handle rate limiting with exponential backoff
-      if (error.message && (error.message.includes('quota') || error.message.includes('rate') || error.message.includes('limit'))) {
-        const retryCount = options.retryCount || 0;
+      if (
+        error.message &&
+        (error.message.includes('quota') ||
+          error.message.includes('rate') ||
+          error.message.includes('limit'))
+      ) {
+        const retryCount = options.retryCount || 0
         if (retryCount < 3) {
-          const delay = Math.pow(2, retryCount) * 1000; // Exponential backoff: 1s, 2s, 4s
-          console.warn(`Rate limit hit, retrying in ${delay}ms...`);
-          
+          const delay = 2 ** retryCount * 1000 // Exponential backoff: 1s, 2s, 4s
+          console.warn(`Rate limit hit, retrying in ${delay}ms...`)
+
           return new Promise((resolve) => {
             setTimeout(async () => {
-              const retryOptions = { ...options, retryCount: retryCount + 1 };
-              const result = await this.generateCompletion(messages, retryOptions);
-              resolve(result);
-            }, delay);
-          });
+              const retryOptions = { ...options, retryCount: retryCount + 1 }
+              const result = await this.generateCompletion(messages, retryOptions)
+              resolve(result)
+            }, delay)
+          })
         }
       }
-      
+
       // If model not found, try with a fallback model
-      if (error.message && (error.message.includes('not found') || error.message.includes('invalid model'))) {
-        const fallbackModels = this.getSuggestedModels(modelName);
+      if (
+        error.message &&
+        (error.message.includes('not found') || error.message.includes('invalid model'))
+      ) {
+        const fallbackModels = this.getSuggestedModels(modelName)
         if (fallbackModels.length > 0 && !options.triedFallback) {
-          console.warn(`Model ${modelName} not found, trying fallback model: ${fallbackModels[0]}`);
-          return this.generateCompletion(messages, { 
-            ...options, 
+          console.warn(`Model ${modelName} not found, trying fallback model: ${fallbackModels[0]}`)
+          return this.generateCompletion(messages, {
+            ...options,
             model: fallbackModels[0],
-            triedFallback: true 
-          });
+            triedFallback: true,
+          })
         }
       }
-      
-      return this.handleProviderError(error, 'generate_completion', { model: options.model });
+
+      return this.handleProviderError(error, 'generate_completion', { model: options.model })
     }
   }
 
-  formatMessages(messages) {
+  async formatMessages(messages) {
     // Convert messages to Vertex AI format for the new SDK
-    const formattedMessages = [];
-    
+    const formattedMessages = []
+
     for (const message of messages) {
-      const role = message.role === 'assistant' ? 'model' : message.role;
-      
+      const role = message.role === 'assistant' ? 'model' : message.role
+
       // Handle different content formats
-      let parts = [];
-      
+      const parts = []
+
       // If content is a string, convert to text part
       if (typeof message.content === 'string') {
-        parts.push({ text: message.content });
+        parts.push({ text: message.content })
       }
       // If content is an array (multimodal), convert each part
       else if (Array.isArray(message.content)) {
         for (const part of message.content) {
           if (part.type === 'text') {
-            parts.push({ text: part.text });
-          }
-          else if (part.type === 'image_url') {
+            parts.push({ text: part.text })
+          } else if (part.type === 'image_url') {
             // Handle inline image data
             if (part.image_url.url.startsWith('data:image/')) {
-              const imageData = this.getImageData(part.image_url.url);
-              parts.push({ inlineData: { data: imageData, mimeType: 'image/jpeg' } });
+              const imageData = await this.getImageData(part.image_url.url)
+              parts.push({ inlineData: { data: imageData, mimeType: 'image/jpeg' } })
             }
-            // Handle remote image URL
+            // Handle remote image URL - for remote URLs, we now fetch and convert
             else {
-              parts.push({ fileData: { mimeType: 'image/jpeg', fileUri: part.image_url.url } });
+              try {
+                const imageData = await this.getImageData(part.image_url.url)
+                parts.push({ inlineData: { data: imageData, mimeType: 'image/jpeg' } })
+              } catch (_error) {
+                // Fallback to fileData if remote fetch fails
+                parts.push({ fileData: { mimeType: 'image/jpeg', fileUri: part.image_url.url } })
+              }
             }
           }
         }
       }
-      
+
       formattedMessages.push({
         role,
-        parts
-      });
+        parts,
+      })
     }
-    
-    return formattedMessages;
+
+    return formattedMessages
   }
 
-  getImageData(imageUrl) {
+  async getImageData(imageUrl) {
     // For base64 data URLs
     if (imageUrl.startsWith('data:')) {
-      return imageUrl.split(',')[1];
+      return imageUrl.split(',')[1]
+    }
+
+    // For regular URLs, fetch the image and convert to base64
+    try {
+      const response = await fetch(imageUrl)
+      if (!response.ok) {
+        throw new Error(`Failed to fetch image: ${response.statusText}`)
+      }
+
+      const arrayBuffer = await response.arrayBuffer()
+      const buffer = Buffer.from(arrayBuffer)
+      return buffer.toString('base64')
+    } catch (error) {
+      throw new Error(`Remote image fetching failed: ${error.message}`)
     }
-    
-    // For regular URLs, we'd need to fetch the image and convert to base64
-    // Remote image fetching requires additional implementation
-    throw new Error('Remote image fetching not implemented');
   }
 
   formatTools(tools) {
     // Convert OpenAI-style function definitions to Google GenAI format
-    const functionDeclarations = [];
-    
+    const functionDeclarations = []
+
     for (const tool of tools) {
       if (tool.type === 'function' && tool.function) {
         functionDeclarations.push({
           name: tool.function.name,
           description: tool.function.description || '',
-          parameters: tool.function.parameters || {}
-        });
+          parameters: tool.function.parameters || {},
+        })
       } else if (Array.isArray(tool.functions)) {
         // Handle legacy format
         for (const fn of tool.functions) {
           functionDeclarations.push({
             name: fn.name,
             description: fn.description || '',
-            parameters: fn.parameters || {}
-          });
+            parameters: fn.parameters || {},
+          })
         }
       }
     }
-    
-    return { functionDeclarations };
+
+    return { functionDeclarations }
   }
 
   estimateTokenCount(text) {
     // Simple estimation: ~4 characters per token
-    return Math.ceil(text.length / 4);
+    return Math.ceil(text.length / 4)
   }
 
   // Vertex AI-specific helper methods
   getRetryDelay(attempt) {
-    return Math.pow(2, attempt) * 1000; // Exponential backoff
+    return 2 ** attempt * 1000 // Exponential backoff
   }
 
   // Vertex AI-specific method for testing model availability
   async testModel(modelName) {
     try {
-      const model = this.getModelInstance(modelName, { maxOutputTokens: 10 });
-      
+      const model = this.getModelInstance(modelName, { maxOutputTokens: 10 })
+
       await model.generateContent({
         contents: [{ role: 'user', parts: [{ text: 'Test' }] }],
         generationConfig: {
           maxOutputTokens: 10,
-          temperature: 0
-        }
-      });
-      
+          temperature: 0,
+        },
+      })
+
       return {
         success: true,
         model: modelName,
         project: this.config.VERTEX_PROJECT_ID,
-        location: this.config.VERTEX_LOCATION || 'us-central1'
-      };
+        location: this.config.VERTEX_LOCATION || 'us-central1',
+      }
     } catch (error) {
       return {
         success: false,
         error: error.message,
-        model: modelName
-      };
+        model: modelName,
+      }
     }
   }
 
@@ -406,9 +436,9 @@ class VertexAIProvider extends BaseProvider {
         contextWindow: 1048576,
         maxOutput: 8192,
         inputCost: 0.00000125,
-        outputCost: 0.00000500,
+        outputCost: 0.000005,
         features: ['text', 'vision', 'tools', 'reasoning'],
-        description: 'Latest Gemini model via Vertex AI'
+        description: 'Latest Gemini model via Vertex AI',
       },
       {
         id: 'gemini-1.5-pro',
@@ -416,9 +446,9 @@ class VertexAIProvider extends BaseProvider {
         contextWindow: 2097152,
         maxOutput: 8192,
         inputCost: 0.00000125,
-        outputCost: 0.00000500,
+        outputCost: 0.000005,
         features: ['text', 'vision', 'tools'],
-        description: 'Production-ready Gemini model via Vertex AI'
+        description: 'Production-ready Gemini model via Vertex AI',
       },
       {
         id: 'gemini-1.5-flash',
@@ -426,13 +456,13 @@ class VertexAIProvider extends BaseProvider {
         contextWindow: 1048576,
         maxOutput: 8192,
         inputCost: 0.00000075,
-        outputCost: 0.00000300,
+        outputCost: 0.000003,
         features: ['text', 'vision', 'tools'],
-        description: 'Fast Gemini model via Vertex AI'
-      }
-    ];
+        description: 'Fast Gemini model via Vertex AI',
+      },
+    ]
   }
 }
 
 // Apply mixins to add standard provider functionality
-export default applyMixins(VertexAIProvider, 'vertex');
+export default applyMixins(VertexAIProvider, 'vertex')