@contentgrowth/llm-service 1.0.4 → 1.0.5

package/dist/index.cjs

@@ -1070,6 +1070,65 @@ ${prompt}` : prompt;
       throw error;
     }
   }
+  /**
+   * Extract structured data from a file (PDF, Image, etc.) using Gemini Multimodal capabilities.
+   * @param {Buffer|string} fileData - Base64 string or Buffer of the file
+   * @param {string} mimeType - Mime type (e.g., 'application/pdf', 'image/png')
+   * @param {string} prompt - Extraction prompt
+   * @param {Object} schema - JSON schema for the output
+   * @param {Object} options - Additional options
+   */
+  async extractWithLLM(fileData, mimeType, prompt, schema = null, options = {}) {
+    var _a, _b, _c, _d;
+    const tier = options.tier || "default";
+    const modelName = this.models[tier] || this.defaultModel;
+    const maxTokens = options.maxTokens || this.config.maxTokens;
+    const temperature = options.temperature !== void 0 ? options.temperature : this.config.temperature;
+    const generationConfig = this._buildGenerationConfig(
+      { ...options, responseFormat: schema ? { type: "json_schema", schema } : "json" },
+      maxTokens,
+      temperature
+    );
+    const parts = [
+      { text: prompt }
+    ];
+    let base64Data = fileData;
+    if (typeof fileData !== "string") {
+      try {
+        base64Data = Buffer.from(fileData).toString("base64");
+      } catch (e) {
+      }
+    }
+    parts.push({
+      inlineData: {
+        data: base64Data,
+        mimeType
+      }
+    });
+    const requestOptions = {
+      model: modelName,
+      contents: [{
+        role: "user",
+        parts
+      }],
+      config: generationConfig
+    };
+    try {
+      const response = await this.client.models.generateContent(requestOptions);
+      const candidate = (_a = response.candidates) == null ? void 0 : _a[0];
+      if (!candidate) {
+        throw new LLMServiceException("No candidates returned from model during extraction", 500);
+      }
+      const textContent = ((_d = (_c = (_b = candidate.content) == null ? void 0 : _b.parts) == null ? void 0 : _c[0]) == null ? void 0 : _d.text) || "";
+      if (this._shouldAutoParse(options)) {
+        return this._safeJsonParse(textContent);
+      }
+      return textContent;
+    } catch (error) {
+      console.error(`[GeminiProvider] extractWithLLM failed (API Key: ${this._getMaskedApiKey()}):`, error);
+      throw error;
+    }
+  }
 };
 
 // src/llm-service.js
@@ -1297,6 +1356,22 @@ var LLMService = class {
     const provider = await this._getProvider(tenantId);
     return provider.getDeepResearchStatus(operationId);
   }
+  /**
+   * Extract structured data from a file (Multimodal)
+   * @param {Buffer|string} fileData - File content (base64 or buffer)
+   * @param {string} mimeType - File mime type (e.g. 'application/pdf')
+   * @param {string} prompt - Extraction instructions
+   * @param {string} tenantId - Tenant ID
+   * @param {Object} schema - Optional JSON schema
+   * @param {Object} options - Options
+   */
+  async extractWithLLM(fileData, mimeType, prompt, tenantId, schema = null, options = {}) {
+    const provider = await this._getProvider(tenantId);
+    if (!provider.extractWithLLM) {
+      throw new LLMServiceException(`Provider ${provider.config.provider} does not support extractWithLLM (Multimodal extraction)`, 400);
+    }
+    return provider.extractWithLLM(fileData, mimeType, prompt, schema, options);
+  }
 };
 var LLMServiceException = class extends Error {
   constructor(message, statusCode = 500, details = null) {