@heripo/pdf-parser 0.1.32 → 0.1.33

package/dist/index.cjs

@@ -172,6 +172,8 @@ function detectProvider(model) {
   if (providerId.includes("google")) return "google";
   if (providerId.includes("anthropic")) return "anthropic";
   if (providerId.includes("together")) return "togetherai";
+  if (providerId.includes("ollama")) return "ollama";
+  if (providerId.includes("lmstudio")) return "lmstudio";
   return "unknown";
 }
 var LLMCaller = class {
@@ -259,14 +261,18 @@ var LLMCaller = class {
    *
    * Strategy per provider:
    * - OpenAI / Anthropic / Google Gemini: `Output.object()` with schema retry
-   * - Together AI / unknown: forced tool call pattern
+   * - Together AI / Ollama / unknown: forced tool call pattern
+   *
+   * Ollama uses tool-call because MLX backends ignore the `format` grammar
+   * that `Output.object()` relies on, and gguf's grammar is unstable once
+   * thinking is disabled. Applies to both text and vision paths.
    *
    * Retries up to MAX_STRUCTURED_OUTPUT_RETRIES times on NoObjectGeneratedError
    * (schema mismatch), re-throwing the last error if all attempts fail.
    */
   static async generateStructuredOutput(model, schema, promptParams) {
     const providerType = detectProvider(model);
-    if (providerType === "togetherai" || providerType === "unknown") {
+    if (providerType === "togetherai" || providerType === "ollama" || providerType === "unknown") {
       return this.generateViaToolCall(model, schema, promptParams);
     }
     let lastError;
@@ -1368,6 +1374,7 @@ var PdfLanguageDetector = class {
         }
       ],
       primaryModel: options.model,
+      fallbackModel: options.fallbackModel,
       maxRetries: options.maxRetries ?? DEFAULT_MAX_RETRIES,
       temperature: options.temperature ?? 0,
       abortSignal: options.abortSignal,
@@ -3320,6 +3327,7 @@ var ReviewAssistancePageGate = class {
         }
       ],
       primaryModel: model,
+      fallbackModel: options.fallbackModel,
       maxRetries: options.maxRetries ?? REVIEW_ASSISTANCE_PAGE_GATE_MAX_RETRIES,
       temperature: options.temperature ?? REVIEW_ASSISTANCE_PAGE_GATE_TEMPERATURE,
       abortSignal: options.abortSignal,
@@ -3591,6 +3599,7 @@ var PostDoclingPageProcessor = class {
           }
         ],
         primaryModel: model,
+        fallbackModel: options?.fallbackModel,
         maxRetries: options?.maxRetries ?? DEFAULT_MAX_RETRIES2,
         temperature: options?.temperature ?? DEFAULT_TEMPERATURE,
         abortSignal: options?.abortSignal,
@@ -3636,6 +3645,7 @@ var PostDoclingPageProcessor = class {
         image,
         gateOptions.model,
         {
+          fallbackModel: gateOptions.fallback,
           maxRetries: gateOptions.maxRetries,
           temperature: gateOptions.temperature,
           abortSignal: options?.abortSignal,
@@ -8881,6 +8891,7 @@ var ReviewAssistanceRunner = class {
             image,
             options.pageGateModel,
             {
+              fallbackModel: options.pageGateFallback,
               maxRetries: options.pageGateMaxRetries,
               temperature: options.pageGateTemperature,
               abortSignal: options.abortSignal,
@@ -9159,7 +9170,7 @@ var ReviewAssistanceRunner = class {
   async reviewWorkItem(context, workItem, image, pageCount, modelResolver, options) {
     const startedAtMs = Date.now();
     const startedAt = new Date(startedAtMs).toISOString();
-    const model = modelResolver(workItem.task);
+    const { model, fallback } = modelResolver(workItem.task);
     const modelId = this.getModelId(model);
     let attempts = 0;
     let validationFeedback = [];
@@ -9200,6 +9211,7 @@ var ReviewAssistanceRunner = class {
               }
             ],
             primaryModel: model,
+            fallbackModel: fallback,
             maxRetries: this.getTaskMaxRetries(workItem.task, options),
             temperature: options.temperature,
             abortSignal: options.abortSignal,
@@ -9986,7 +9998,7 @@ var ReviewAssistanceRunner = class {
    */
   async callMergeArbiter(candidates, context, image, modelResolver, options) {
     const arbiterTask = this.resolveMergeTaskDefinition(candidates);
-    const model = modelResolver(arbiterTask);
+    const { model, fallback } = modelResolver(arbiterTask);
     const conflictRefs = [
       ...new Set(
         candidates.flatMap(
@@ -10025,6 +10037,7 @@ var ReviewAssistanceRunner = class {
           }
         ],
         primaryModel: model,
+        fallbackModel: fallback,
         maxRetries: REVIEW_ASSISTANCE_MERGE_MAX_RETRIES,
         temperature: options.temperature,
         abortSignal: options.abortSignal,
@@ -10577,6 +10590,7 @@ var PostDoclingCorrectionPipeline = class {
         outputDir,
         correction.models.textCorrection,
         {
+          fallbackModel: correction.models.textCorrectionFallback,
           concurrency: correction.concurrency.pages,
           maxRetries: correction.maxRetries.textCorrection,
           temperature: correction.temperature,
@@ -10587,6 +10601,7 @@ var PostDoclingCorrectionPipeline = class {
           pageTexts,
           reviewAssistanceGate: {
             model: correction.models.pageGate,
+            fallback: correction.models.pageGateFallback,
             maxRetries: correction.maxRetries.pageGate,
             temperature: correction.temperature,
             outputLanguage: correction.outputLanguage
@@ -10602,6 +10617,7 @@ var PostDoclingCorrectionPipeline = class {
           pdfPath,
           pageTexts,
           pageGateModel: correction.models.pageGate,
+          pageGateFallback: correction.models.pageGateFallback,
           pageGateMaxRetries: correction.maxRetries.pageGate,
           pageGateTemperature: correction.temperature,
           pageConcurrency: correction.concurrency.pages,
@@ -10627,9 +10643,15 @@ var PostDoclingCorrectionPipeline = class {
   buildReviewAssistanceModelResolver(correction) {
     return (task) => {
       if (task.id === "tables") {
-        return correction.models.tableCorrection ?? correction.models.reviewAssistanceTasks?.tables ?? correction.models.reviewAssistance;
+        return {
+          model: correction.models.tableCorrection ?? correction.models.reviewAssistanceTasks?.tables ?? correction.models.reviewAssistance,
+          fallback: correction.models.tableCorrectionFallback ?? correction.models.reviewAssistanceTasksFallback?.tables ?? correction.models.reviewAssistanceFallback
+        };
       }
-      return correction.models.reviewAssistanceTasks?.[task.id] ?? correction.models.reviewAssistance;
+      return {
+        model: correction.models.reviewAssistanceTasks?.[task.id] ?? correction.models.reviewAssistance,
+        fallback: correction.models.reviewAssistanceTasksFallback?.[task.id] ?? correction.models.reviewAssistanceFallback
+      };
     };
   }
   async extractPageTexts(pdfPath, outputDir) {
@@ -10777,6 +10799,7 @@ var PDFConverter = class {
     try {
       const result = await detector.detect(pdfPath, languageDetectionDir, {
         model: options.languageDetectionModel,
+        fallbackModel: options.languageDetectionFallbackModel,
         aggregator: options.aggregator,
         abortSignal
       });