@contentgrowth/llm-service 1.4.1 → 1.4.3

package/dist/index.cjs

@@ -1744,14 +1744,18 @@ var LLMService = class {
     if (this.providerCache.has(cacheKey)) {
       return this.providerCache.get(cacheKey);
     }
-    const config = await ConfigManager.getConfig(tenantId, this.env);
-    if (!config.apiKey && config.provider !== "vertex") {
+    const configOptions = {};
+    if (this._ptalkTaskType) {
+      configOptions.capability = this._ptalkTaskType;
+    }
+    const config = await ConfigManager.getConfig(tenantId, this.env, configOptions);
+    if (!config.apiKey && config.provider !== "vertex" && config.provider !== "google") {
       throw new LLMServiceException(`LLM service is not configured for ${config.provider}. Missing API Key.`, 500);
     }
     let provider;
     if (config.provider === "openai") {
       provider = new OpenAIProvider(config);
-    } else if (config.provider === "gemini" || config.provider === "vertex") {
+    } else if (config.provider === "gemini" || config.provider === "vertex" || config.provider === "google") {
       provider = new GoogleProvider(config);
     } else if (config.provider === "anthropic") {
       provider = new AnthropicProvider(config);
@@ -1931,21 +1935,16 @@ var LLMService = class {
   }
   /**
    * Generate an image
-   * Falls back to system keys if tenant doesn't have image capability enabled
+   * 
+   * Note: The caller (backend) is responsible for capability checks and routing.
+   * This method simply delegates to the configured provider.
+   * 
    * @param {string} prompt - Text description of the image
    * @param {string} tenantId - Tenant identifier
    * @param {string} systemPrompt - System instructions for generation
-   * @param {Object} options - Generation options (aspectRatio, images, etc.)
+   * @param {Object} options - Generation options (aspectRatio, images, model, etc.)
    */
   async imageGeneration(prompt, tenantId, systemPrompt, options = {}) {
-    var _a;
-    if (tenantId) {
-      const config = await ConfigManager.getConfig(tenantId, this.env);
-      if (config.isTenantOwned && !((_a = config.capabilities) == null ? void 0 : _a.image)) {
-        console.log(`[LLMService] Tenant ${tenantId} BYOK doesn't have image capability. Using system keys.`);
-        tenantId = null;
-      }
-    }
     const provider = await this._getProvider(tenantId);
     return provider.imageGeneration(prompt, systemPrompt, options);
   }