@claritylabs/cl-sdk 0.7.5 → 0.8.1

package/dist/index.mjs

@@ -1525,19 +1525,28 @@ async function runExtractor(params) {
     maxTokens = 4096,
     providerOptions
   } = params;
-  const pagesPdf = await extractPageRange(pdfBase64, startPage, endPage);
-  const fullPrompt = convertPdfToImages ? `${prompt}
-
-[Document pages ${startPage}-${endPage} are provided as images above.]` : `${prompt}
+  const extractorProviderOptions = { ...providerOptions };
+  let fullPrompt;
+  if (convertPdfToImages) {
+    const images = await convertPdfToImages(pdfBase64, startPage, endPage);
+    extractorProviderOptions.images = images;
+    fullPrompt = `${prompt}
+
+[Document pages ${startPage}-${endPage} are provided as images.]`;
+  } else {
+    const pagesPdf = await extractPageRange(pdfBase64, startPage, endPage);
+    extractorProviderOptions.pdfBase64 = pagesPdf;
+    fullPrompt = `${prompt}
 
-[Document pages ${startPage}-${endPage} are provided as a PDF file above.]`;
+[Document pages ${startPage}-${endPage} are provided as a PDF file.]`;
+  }
   const strictSchema = toStrictSchema(schema);
   const result = await withRetry(
     () => generateObject({
       prompt: fullPrompt,
       schema: strictSchema,
       maxTokens,
-      providerOptions
+      providerOptions: extractorProviderOptions
     })
   );
   return {
@@ -3357,7 +3366,7 @@ function createExtractor(config) {
           prompt: buildClassifyPrompt(),
           schema: ClassifyResultSchema,
           maxTokens: 512,
-          providerOptions
+          providerOptions: { ...providerOptions, pdfBase64 }
         },
         {
           fallback: { documentType: "policy", policyTypes: ["other"], confidence: 0 },
@@ -3401,7 +3410,7 @@ function createExtractor(config) {
           prompt: buildPlanPrompt(templateHints),
           schema: ExtractionPlanSchema,
           maxTokens: 2048,
-          providerOptions
+          providerOptions: { ...providerOptions, pdfBase64 }
         },
         {
           fallback: {