@claritylabs/cl-sdk 3.1.17 → 3.1.19

package/dist/{index-CTxWtbqE.d.mts → index-q8LQXen9.d.mts}

@@ -71,6 +71,8 @@ interface ModelCallTrace {
     batchCount?: number;
     phase?: string;
     sourceBacked?: boolean;
+    coverageGroup?: string;
+    itemCount?: number;
 }
 /** Callback to generate text from a prompt. Provider-agnostic. */
 type GenerateText = (params: {

package/dist/{index-CTxWtbqE.d.ts → index-q8LQXen9.d.ts}

@@ -71,6 +71,8 @@ interface ModelCallTrace {
     batchCount?: number;
     phase?: string;
     sourceBacked?: boolean;
+    coverageGroup?: string;
+    itemCount?: number;
 }
 /** Callback to generate text from a prompt. Provider-agnostic. */
 type GenerateText = (params: {

package/dist/index.d.mts

@@ -1,10 +1,14 @@
-import { L as LogFn, M as ModelTaskKind, a as ModelBudgetResolution, b as ModelCallTrace, G as GenerateObject, T as TokenUsage, S as SourceSpanLocation, c as SourceSpan, D as DocumentSourceNode, d as SourceKind, e as SourceSpanUnit, f as SourceSpanTableLocation, g as SourceChunk, h as SourceSpanBBox, U as UnifiedQualityReport, B as BaseQualityIssue, Q as QualityGateStatus, I as InsuranceDocument, P as PdfInput, i as DocumentChunk, j as PolicyOperationalProfile, k as PerformanceReport, l as GenerateText, C as ConvertPdfToImagesFn, m as QualityGateMode, n as ModelCapabilities, o as ModelBudgetConstraint, A as AgentContext, p as DocumentStore, q as MemoryStore, r as Platform } from './index-CTxWtbqE.mjs';
-export { s as AcroFormMapping, t as AcroFormMappingSchema, u as AnswerParsingResult, v as AnswerParsingResultSchema, w as ApplicationClassifyResult, x as ApplicationClassifyResultSchema, y as ApplicationContextProposal, z as ApplicationContextProposalSchema, E as ApplicationEmailReviewSchema, F as ApplicationField, H as ApplicationFieldSchema, J as ApplicationListFilters, K as ApplicationNextQuestions, N as ApplicationPacket, O as ApplicationPacketAnswer, R as ApplicationPacketAnswerSchema, V as ApplicationPacketSchema, W as ApplicationPipelineConfig, X as ApplicationQualityArtifactSchema, Y as ApplicationQualityIssueSchema, Z as ApplicationQualityReportSchema, _ as ApplicationQualityRoundSchema, $ as ApplicationQuestionCondition, a0 as ApplicationQuestionConditionSchema, a1 as ApplicationQuestionGraph, a2 as ApplicationQuestionGraphSchema, a3 as ApplicationQuestionNode, a4 as ApplicationQuestionNodeSchema, a5 as ApplicationRepeat, a6 as ApplicationRepeatSchema, a7 as ApplicationState, a8 as ApplicationStateSchema, a9 as ApplicationStore, aa as ApplicationTemplate, ab as ApplicationTemplateListFilters, ac as ApplicationTemplateSchema, ad as ApplicationTemplateStore, ae as AutoFillMatch, af as AutoFillMatchSchema, ag as AutoFillResult, ah as AutoFillResultSchema, ai as AuxiliaryFact, aj as AuxiliaryFactSchema, ak as BackfillProvider, al as BuildApplicationPacketInput, am as BuildApplicationPacketResult, an as ChunkFilter, ao as CommunicationIntent, ap as CommunicationIntentSchema, aq as ContextProposalResult, ar as ConversationTurn, as as CoveredReason, at as CoveredReasonSchema, au as CreateApplicationRunInput, av as Definition, aw as DefinitionSchema, ax as DocumentAgentGuidance, ay as DocumentAgentGuidanceSchema, az as DocumentFilters, aA as DocumentMetadata, aB as DocumentMetadataSchema, aC as DocumentNode, aD as DocumentNodeSchema, aE as DocumentPageMapEntry, aF as DocumentPageMapEntrySchema, aG as DocumentSourceNodeKind, aH as DocumentSourceNodeKindSchema, aI as DocumentSourceNodeSchema, aJ as DocumentTableOfContentsEntry, aK as DocumentTableOfContentsEntrySchema, aL as EmbedText, aM as FieldExtractionResult, aN as FieldExtractionResultSchema, aO as FieldType, aP as FieldTypeSchema, aQ as FlatPdfPlacement, aR as FlatPdfPlacementSchema, aS as InsuranceDocumentSchema, aT as LookupFill, aU as LookupFillResult, aV as LookupFillResultSchema, aW as LookupFillSchema, aX as LookupRequest, aY as LookupRequestSchema, aZ as ModelCallReport, a_ as OperationalCoverageLine, a$ as OperationalCoverageLineSchema, b0 as OperationalCoverageTerm, b1 as OperationalCoverageTermSchema, b2 as OperationalEndorsementSupport, b3 as OperationalEndorsementSupportSchema, b4 as OperationalParty, b5 as OperationalPartySchema, b6 as PLATFORM_CONFIGS, b7 as ParsedAnswer, b8 as ParsedAnswerSchema, b9 as PlatformConfig, ba as PlatformSchema, bb as PolicyDocument, bc as PolicyDocumentSchema, bd as PolicyOperationalProfileSchema, be as PremiumLine, bf as PremiumLineSchema, bg as PriorAnswer, bh as ProcessApplicationInput, bi as ProcessApplicationResult, bj as ProcessReplyInput, bk as ProcessReplyResult, bl as QuestionBatchResult, bm as QuestionBatchResultSchema, bn as QuoteDocument, bo as QuoteDocumentSchema, bp as ReplyIntent, bq as ReplyIntentSchema, br as ResolveModelBudgetParams, bs as Section, bt as SectionSchema, bu as SourceBackedValue, bv as SourceBackedValueSchema, bw as SourceChunkSchema, bx as SourceKindSchema, by as SourceSpanBBoxSchema, bz as SourceSpanKind, bA as SourceSpanKindSchema, bB as SourceSpanLocationSchema, bC as SourceSpanRef, bD as SourceSpanRefSchema, bE as SourceSpanSchema, bF as SourceSpanTableLocationSchema, bG as SourceSpanUnitSchema, bH as Subjectivity, bI as SubjectivitySchema, bJ as Subsection, bK as SubsectionSchema, bL as UnderwritingCondition, bM as UnderwritingConditionSchema, bN as applyApplicationAnswers, bO as buildApplicationPacket, bP as buildQuestionGraphFromFields, bQ as createApplicationPipeline, bR as createApplicationRun, bS as extractQuestionGraphFromFields, bT as flattenQuestionGraph, bU as getActiveApplicationFields, bV as getNextApplicationQuestions, bW as normalizeApplicationQuestionGraph, bX as planNextApplicationQuestions, bY as proposeContextWrites, bZ as resolveModelBudget, b_ as validateApplicationPacket } from './index-CTxWtbqE.mjs';
+import { L as LogFn, M as ModelTaskKind, a as ModelBudgetResolution, b as ModelCallTrace, G as GenerateObject, T as TokenUsage, S as SourceSpanLocation, c as SourceSpan, D as DocumentSourceNode, d as SourceKind, e as SourceSpanUnit, f as SourceSpanTableLocation, g as SourceChunk, h as SourceSpanBBox, U as UnifiedQualityReport, B as BaseQualityIssue, Q as QualityGateStatus, I as InsuranceDocument, P as PdfInput, i as DocumentChunk, j as PolicyOperationalProfile, k as PerformanceReport, l as GenerateText, C as ConvertPdfToImagesFn, m as QualityGateMode, n as ModelCapabilities, o as ModelBudgetConstraint, A as AgentContext, p as DocumentStore, q as MemoryStore, r as Platform } from './index-q8LQXen9.mjs';
+export { s as AcroFormMapping, t as AcroFormMappingSchema, u as AnswerParsingResult, v as AnswerParsingResultSchema, w as ApplicationClassifyResult, x as ApplicationClassifyResultSchema, y as ApplicationContextProposal, z as ApplicationContextProposalSchema, E as ApplicationEmailReviewSchema, F as ApplicationField, H as ApplicationFieldSchema, J as ApplicationListFilters, K as ApplicationNextQuestions, N as ApplicationPacket, O as ApplicationPacketAnswer, R as ApplicationPacketAnswerSchema, V as ApplicationPacketSchema, W as ApplicationPipelineConfig, X as ApplicationQualityArtifactSchema, Y as ApplicationQualityIssueSchema, Z as ApplicationQualityReportSchema, _ as ApplicationQualityRoundSchema, $ as ApplicationQuestionCondition, a0 as ApplicationQuestionConditionSchema, a1 as ApplicationQuestionGraph, a2 as ApplicationQuestionGraphSchema, a3 as ApplicationQuestionNode, a4 as ApplicationQuestionNodeSchema, a5 as ApplicationRepeat, a6 as ApplicationRepeatSchema, a7 as ApplicationState, a8 as ApplicationStateSchema, a9 as ApplicationStore, aa as ApplicationTemplate, ab as ApplicationTemplateListFilters, ac as ApplicationTemplateSchema, ad as ApplicationTemplateStore, ae as AutoFillMatch, af as AutoFillMatchSchema, ag as AutoFillResult, ah as AutoFillResultSchema, ai as AuxiliaryFact, aj as AuxiliaryFactSchema, ak as BackfillProvider, al as BuildApplicationPacketInput, am as BuildApplicationPacketResult, an as ChunkFilter, ao as CommunicationIntent, ap as CommunicationIntentSchema, aq as ContextProposalResult, ar as ConversationTurn, as as CoveredReason, at as CoveredReasonSchema, au as CreateApplicationRunInput, av as Definition, aw as DefinitionSchema, ax as DocumentAgentGuidance, ay as DocumentAgentGuidanceSchema, az as DocumentFilters, aA as DocumentMetadata, aB as DocumentMetadataSchema, aC as DocumentNode, aD as DocumentNodeSchema, aE as DocumentPageMapEntry, aF as DocumentPageMapEntrySchema, aG as DocumentSourceNodeKind, aH as DocumentSourceNodeKindSchema, aI as DocumentSourceNodeSchema, aJ as DocumentTableOfContentsEntry, aK as DocumentTableOfContentsEntrySchema, aL as EmbedText, aM as FieldExtractionResult, aN as FieldExtractionResultSchema, aO as FieldType, aP as FieldTypeSchema, aQ as FlatPdfPlacement, aR as FlatPdfPlacementSchema, aS as InsuranceDocumentSchema, aT as LookupFill, aU as LookupFillResult, aV as LookupFillResultSchema, aW as LookupFillSchema, aX as LookupRequest, aY as LookupRequestSchema, aZ as ModelCallReport, a_ as OperationalCoverageLine, a$ as OperationalCoverageLineSchema, b0 as OperationalCoverageTerm, b1 as OperationalCoverageTermSchema, b2 as OperationalEndorsementSupport, b3 as OperationalEndorsementSupportSchema, b4 as OperationalParty, b5 as OperationalPartySchema, b6 as PLATFORM_CONFIGS, b7 as ParsedAnswer, b8 as ParsedAnswerSchema, b9 as PlatformConfig, ba as PlatformSchema, bb as PolicyDocument, bc as PolicyDocumentSchema, bd as PolicyOperationalProfileSchema, be as PremiumLine, bf as PremiumLineSchema, bg as PriorAnswer, bh as ProcessApplicationInput, bi as ProcessApplicationResult, bj as ProcessReplyInput, bk as ProcessReplyResult, bl as QuestionBatchResult, bm as QuestionBatchResultSchema, bn as QuoteDocument, bo as QuoteDocumentSchema, bp as ReplyIntent, bq as ReplyIntentSchema, br as ResolveModelBudgetParams, bs as Section, bt as SectionSchema, bu as SourceBackedValue, bv as SourceBackedValueSchema, bw as SourceChunkSchema, bx as SourceKindSchema, by as SourceSpanBBoxSchema, bz as SourceSpanKind, bA as SourceSpanKindSchema, bB as SourceSpanLocationSchema, bC as SourceSpanRef, bD as SourceSpanRefSchema, bE as SourceSpanSchema, bF as SourceSpanTableLocationSchema, bG as SourceSpanUnitSchema, bH as Subjectivity, bI as SubjectivitySchema, bJ as Subsection, bK as SubsectionSchema, bL as UnderwritingCondition, bM as UnderwritingConditionSchema, bN as applyApplicationAnswers, bO as buildApplicationPacket, bP as buildQuestionGraphFromFields, bQ as createApplicationPipeline, bR as createApplicationRun, bS as extractQuestionGraphFromFields, bT as flattenQuestionGraph, bU as getActiveApplicationFields, bV as getNextApplicationQuestions, bW as normalizeApplicationQuestionGraph, bX as planNextApplicationQuestions, bY as proposeContextWrites, bZ as resolveModelBudget, b_ as validateApplicationPacket } from './index-q8LQXen9.mjs';
 import * as zod from 'zod';
 import { ZodTypeAny, z, ZodSchema } from 'zod';
 import { PDFDocument } from 'pdf-lib';
 
-declare function withRetry<T>(fn: () => Promise<T>, log?: LogFn): Promise<T>;
+interface RetryOptions {
+    maxRetries?: number;
+    baseDelayMs?: number;
+}
+declare function withRetry<T>(fn: () => Promise<T>, log?: LogFn, options?: RetryOptions): Promise<T>;
 
 /**
  * Concurrency limiter — returns a function that wraps async tasks
@@ -31,6 +35,8 @@ interface SafeGenerateOptions<T> {
     onError?: (error: unknown, attempt: number) => void;
     /** Logger for pipeline status messages. */
     log?: LogFn;
+    /** Controls retryable provider-error backoff around the host callback. Use false when the host already owns fallback. */
+    retry?: RetryOptions | false;
 }
 interface SafeGenerateParams {
     prompt: string;
@@ -44,7 +50,7 @@ interface SafeGenerateParams {
 /**
  * Wraps a `generateObject` call with two layers of resilience:
  *
- * 1. Inner: `withRetry` handles 429 / rate-limit errors with exponential backoff
+ * 1. Inner: `withRetry` handles retryable provider errors with exponential backoff unless disabled.
  * 2. Outer: catches all other errors (schema validation, malformed JSON, transient API errors)
  *    and retries up to `maxRetries` times. If all retries fail, returns `fallback` (if provided)
  *    or re-throws.
@@ -8734,6 +8740,7 @@ interface ExtractorConfig {
     sourceStore?: SourceStore;
     qualityGate?: QualityGateMode;
     modelCapabilities?: ModelCapabilities;
+    modelCapabilitiesByTaskKind?: Partial<Record<ModelTaskKind, ModelCapabilities>>;
     modelBudgetConstraints?: Partial<Record<ModelTaskKind, ModelBudgetConstraint>>;
     /** Optional checkpoint persistence callback. */
     onCheckpointSave?: (checkpoint: PipelineCheckpoint<ExtractionState>) => Promise<void>;

package/dist/index.d.ts

@@ -1,10 +1,14 @@
-import { L as LogFn, M as ModelTaskKind, a as ModelBudgetResolution, b as ModelCallTrace, G as GenerateObject, T as TokenUsage, S as SourceSpanLocation, c as SourceSpan, D as DocumentSourceNode, d as SourceKind, e as SourceSpanUnit, f as SourceSpanTableLocation, g as SourceChunk, h as SourceSpanBBox, U as UnifiedQualityReport, B as BaseQualityIssue, Q as QualityGateStatus, I as InsuranceDocument, P as PdfInput, i as DocumentChunk, j as PolicyOperationalProfile, k as PerformanceReport, l as GenerateText, C as ConvertPdfToImagesFn, m as QualityGateMode, n as ModelCapabilities, o as ModelBudgetConstraint, A as AgentContext, p as DocumentStore, q as MemoryStore, r as Platform } from './index-CTxWtbqE.js';
-export { s as AcroFormMapping, t as AcroFormMappingSchema, u as AnswerParsingResult, v as AnswerParsingResultSchema, w as ApplicationClassifyResult, x as ApplicationClassifyResultSchema, y as ApplicationContextProposal, z as ApplicationContextProposalSchema, E as ApplicationEmailReviewSchema, F as ApplicationField, H as ApplicationFieldSchema, J as ApplicationListFilters, K as ApplicationNextQuestions, N as ApplicationPacket, O as ApplicationPacketAnswer, R as ApplicationPacketAnswerSchema, V as ApplicationPacketSchema, W as ApplicationPipelineConfig, X as ApplicationQualityArtifactSchema, Y as ApplicationQualityIssueSchema, Z as ApplicationQualityReportSchema, _ as ApplicationQualityRoundSchema, $ as ApplicationQuestionCondition, a0 as ApplicationQuestionConditionSchema, a1 as ApplicationQuestionGraph, a2 as ApplicationQuestionGraphSchema, a3 as ApplicationQuestionNode, a4 as ApplicationQuestionNodeSchema, a5 as ApplicationRepeat, a6 as ApplicationRepeatSchema, a7 as ApplicationState, a8 as ApplicationStateSchema, a9 as ApplicationStore, aa as ApplicationTemplate, ab as ApplicationTemplateListFilters, ac as ApplicationTemplateSchema, ad as ApplicationTemplateStore, ae as AutoFillMatch, af as AutoFillMatchSchema, ag as AutoFillResult, ah as AutoFillResultSchema, ai as AuxiliaryFact, aj as AuxiliaryFactSchema, ak as BackfillProvider, al as BuildApplicationPacketInput, am as BuildApplicationPacketResult, an as ChunkFilter, ao as CommunicationIntent, ap as CommunicationIntentSchema, aq as ContextProposalResult, ar as ConversationTurn, as as CoveredReason, at as CoveredReasonSchema, au as CreateApplicationRunInput, av as Definition, aw as DefinitionSchema, ax as DocumentAgentGuidance, ay as DocumentAgentGuidanceSchema, az as DocumentFilters, aA as DocumentMetadata, aB as DocumentMetadataSchema, aC as DocumentNode, aD as DocumentNodeSchema, aE as DocumentPageMapEntry, aF as DocumentPageMapEntrySchema, aG as DocumentSourceNodeKind, aH as DocumentSourceNodeKindSchema, aI as DocumentSourceNodeSchema, aJ as DocumentTableOfContentsEntry, aK as DocumentTableOfContentsEntrySchema, aL as EmbedText, aM as FieldExtractionResult, aN as FieldExtractionResultSchema, aO as FieldType, aP as FieldTypeSchema, aQ as FlatPdfPlacement, aR as FlatPdfPlacementSchema, aS as InsuranceDocumentSchema, aT as LookupFill, aU as LookupFillResult, aV as LookupFillResultSchema, aW as LookupFillSchema, aX as LookupRequest, aY as LookupRequestSchema, aZ as ModelCallReport, a_ as OperationalCoverageLine, a$ as OperationalCoverageLineSchema, b0 as OperationalCoverageTerm, b1 as OperationalCoverageTermSchema, b2 as OperationalEndorsementSupport, b3 as OperationalEndorsementSupportSchema, b4 as OperationalParty, b5 as OperationalPartySchema, b6 as PLATFORM_CONFIGS, b7 as ParsedAnswer, b8 as ParsedAnswerSchema, b9 as PlatformConfig, ba as PlatformSchema, bb as PolicyDocument, bc as PolicyDocumentSchema, bd as PolicyOperationalProfileSchema, be as PremiumLine, bf as PremiumLineSchema, bg as PriorAnswer, bh as ProcessApplicationInput, bi as ProcessApplicationResult, bj as ProcessReplyInput, bk as ProcessReplyResult, bl as QuestionBatchResult, bm as QuestionBatchResultSchema, bn as QuoteDocument, bo as QuoteDocumentSchema, bp as ReplyIntent, bq as ReplyIntentSchema, br as ResolveModelBudgetParams, bs as Section, bt as SectionSchema, bu as SourceBackedValue, bv as SourceBackedValueSchema, bw as SourceChunkSchema, bx as SourceKindSchema, by as SourceSpanBBoxSchema, bz as SourceSpanKind, bA as SourceSpanKindSchema, bB as SourceSpanLocationSchema, bC as SourceSpanRef, bD as SourceSpanRefSchema, bE as SourceSpanSchema, bF as SourceSpanTableLocationSchema, bG as SourceSpanUnitSchema, bH as Subjectivity, bI as SubjectivitySchema, bJ as Subsection, bK as SubsectionSchema, bL as UnderwritingCondition, bM as UnderwritingConditionSchema, bN as applyApplicationAnswers, bO as buildApplicationPacket, bP as buildQuestionGraphFromFields, bQ as createApplicationPipeline, bR as createApplicationRun, bS as extractQuestionGraphFromFields, bT as flattenQuestionGraph, bU as getActiveApplicationFields, bV as getNextApplicationQuestions, bW as normalizeApplicationQuestionGraph, bX as planNextApplicationQuestions, bY as proposeContextWrites, bZ as resolveModelBudget, b_ as validateApplicationPacket } from './index-CTxWtbqE.js';
+import { L as LogFn, M as ModelTaskKind, a as ModelBudgetResolution, b as ModelCallTrace, G as GenerateObject, T as TokenUsage, S as SourceSpanLocation, c as SourceSpan, D as DocumentSourceNode, d as SourceKind, e as SourceSpanUnit, f as SourceSpanTableLocation, g as SourceChunk, h as SourceSpanBBox, U as UnifiedQualityReport, B as BaseQualityIssue, Q as QualityGateStatus, I as InsuranceDocument, P as PdfInput, i as DocumentChunk, j as PolicyOperationalProfile, k as PerformanceReport, l as GenerateText, C as ConvertPdfToImagesFn, m as QualityGateMode, n as ModelCapabilities, o as ModelBudgetConstraint, A as AgentContext, p as DocumentStore, q as MemoryStore, r as Platform } from './index-q8LQXen9.js';
+export { s as AcroFormMapping, t as AcroFormMappingSchema, u as AnswerParsingResult, v as AnswerParsingResultSchema, w as ApplicationClassifyResult, x as ApplicationClassifyResultSchema, y as ApplicationContextProposal, z as ApplicationContextProposalSchema, E as ApplicationEmailReviewSchema, F as ApplicationField, H as ApplicationFieldSchema, J as ApplicationListFilters, K as ApplicationNextQuestions, N as ApplicationPacket, O as ApplicationPacketAnswer, R as ApplicationPacketAnswerSchema, V as ApplicationPacketSchema, W as ApplicationPipelineConfig, X as ApplicationQualityArtifactSchema, Y as ApplicationQualityIssueSchema, Z as ApplicationQualityReportSchema, _ as ApplicationQualityRoundSchema, $ as ApplicationQuestionCondition, a0 as ApplicationQuestionConditionSchema, a1 as ApplicationQuestionGraph, a2 as ApplicationQuestionGraphSchema, a3 as ApplicationQuestionNode, a4 as ApplicationQuestionNodeSchema, a5 as ApplicationRepeat, a6 as ApplicationRepeatSchema, a7 as ApplicationState, a8 as ApplicationStateSchema, a9 as ApplicationStore, aa as ApplicationTemplate, ab as ApplicationTemplateListFilters, ac as ApplicationTemplateSchema, ad as ApplicationTemplateStore, ae as AutoFillMatch, af as AutoFillMatchSchema, ag as AutoFillResult, ah as AutoFillResultSchema, ai as AuxiliaryFact, aj as AuxiliaryFactSchema, ak as BackfillProvider, al as BuildApplicationPacketInput, am as BuildApplicationPacketResult, an as ChunkFilter, ao as CommunicationIntent, ap as CommunicationIntentSchema, aq as ContextProposalResult, ar as ConversationTurn, as as CoveredReason, at as CoveredReasonSchema, au as CreateApplicationRunInput, av as Definition, aw as DefinitionSchema, ax as DocumentAgentGuidance, ay as DocumentAgentGuidanceSchema, az as DocumentFilters, aA as DocumentMetadata, aB as DocumentMetadataSchema, aC as DocumentNode, aD as DocumentNodeSchema, aE as DocumentPageMapEntry, aF as DocumentPageMapEntrySchema, aG as DocumentSourceNodeKind, aH as DocumentSourceNodeKindSchema, aI as DocumentSourceNodeSchema, aJ as DocumentTableOfContentsEntry, aK as DocumentTableOfContentsEntrySchema, aL as EmbedText, aM as FieldExtractionResult, aN as FieldExtractionResultSchema, aO as FieldType, aP as FieldTypeSchema, aQ as FlatPdfPlacement, aR as FlatPdfPlacementSchema, aS as InsuranceDocumentSchema, aT as LookupFill, aU as LookupFillResult, aV as LookupFillResultSchema, aW as LookupFillSchema, aX as LookupRequest, aY as LookupRequestSchema, aZ as ModelCallReport, a_ as OperationalCoverageLine, a$ as OperationalCoverageLineSchema, b0 as OperationalCoverageTerm, b1 as OperationalCoverageTermSchema, b2 as OperationalEndorsementSupport, b3 as OperationalEndorsementSupportSchema, b4 as OperationalParty, b5 as OperationalPartySchema, b6 as PLATFORM_CONFIGS, b7 as ParsedAnswer, b8 as ParsedAnswerSchema, b9 as PlatformConfig, ba as PlatformSchema, bb as PolicyDocument, bc as PolicyDocumentSchema, bd as PolicyOperationalProfileSchema, be as PremiumLine, bf as PremiumLineSchema, bg as PriorAnswer, bh as ProcessApplicationInput, bi as ProcessApplicationResult, bj as ProcessReplyInput, bk as ProcessReplyResult, bl as QuestionBatchResult, bm as QuestionBatchResultSchema, bn as QuoteDocument, bo as QuoteDocumentSchema, bp as ReplyIntent, bq as ReplyIntentSchema, br as ResolveModelBudgetParams, bs as Section, bt as SectionSchema, bu as SourceBackedValue, bv as SourceBackedValueSchema, bw as SourceChunkSchema, bx as SourceKindSchema, by as SourceSpanBBoxSchema, bz as SourceSpanKind, bA as SourceSpanKindSchema, bB as SourceSpanLocationSchema, bC as SourceSpanRef, bD as SourceSpanRefSchema, bE as SourceSpanSchema, bF as SourceSpanTableLocationSchema, bG as SourceSpanUnitSchema, bH as Subjectivity, bI as SubjectivitySchema, bJ as Subsection, bK as SubsectionSchema, bL as UnderwritingCondition, bM as UnderwritingConditionSchema, bN as applyApplicationAnswers, bO as buildApplicationPacket, bP as buildQuestionGraphFromFields, bQ as createApplicationPipeline, bR as createApplicationRun, bS as extractQuestionGraphFromFields, bT as flattenQuestionGraph, bU as getActiveApplicationFields, bV as getNextApplicationQuestions, bW as normalizeApplicationQuestionGraph, bX as planNextApplicationQuestions, bY as proposeContextWrites, bZ as resolveModelBudget, b_ as validateApplicationPacket } from './index-q8LQXen9.js';
 import * as zod from 'zod';
 import { ZodTypeAny, z, ZodSchema } from 'zod';
 import { PDFDocument } from 'pdf-lib';
 
-declare function withRetry<T>(fn: () => Promise<T>, log?: LogFn): Promise<T>;
+interface RetryOptions {
+    maxRetries?: number;
+    baseDelayMs?: number;
+}
+declare function withRetry<T>(fn: () => Promise<T>, log?: LogFn, options?: RetryOptions): Promise<T>;
 
 /**
  * Concurrency limiter — returns a function that wraps async tasks
@@ -31,6 +35,8 @@ interface SafeGenerateOptions<T> {
     onError?: (error: unknown, attempt: number) => void;
     /** Logger for pipeline status messages. */
     log?: LogFn;
+    /** Controls retryable provider-error backoff around the host callback. Use false when the host already owns fallback. */
+    retry?: RetryOptions | false;
 }
 interface SafeGenerateParams {
     prompt: string;
@@ -44,7 +50,7 @@ interface SafeGenerateParams {
 /**
  * Wraps a `generateObject` call with two layers of resilience:
  *
- * 1. Inner: `withRetry` handles 429 / rate-limit errors with exponential backoff
+ * 1. Inner: `withRetry` handles retryable provider errors with exponential backoff unless disabled.
  * 2. Outer: catches all other errors (schema validation, malformed JSON, transient API errors)
  *    and retries up to `maxRetries` times. If all retries fail, returns `fallback` (if provided)
  *    or re-throws.
@@ -8734,6 +8740,7 @@ interface ExtractorConfig {
     sourceStore?: SourceStore;
     qualityGate?: QualityGateMode;
     modelCapabilities?: ModelCapabilities;
+    modelCapabilitiesByTaskKind?: Partial<Record<ModelTaskKind, ModelCapabilities>>;
     modelBudgetConstraints?: Partial<Record<ModelTaskKind, ModelBudgetConstraint>>;
     /** Optional checkpoint persistence callback. */
     onCheckpointSave?: (checkpoint: PipelineCheckpoint<ExtractionState>) => Promise<void>;

package/dist/index.js

@@ -396,17 +396,19 @@ function isRetryableError(error) {
   }
   return false;
 }
-async function withRetry(fn, log) {
+async function withRetry(fn, log, options) {
+  const maxRetries = options?.maxRetries ?? MAX_RETRIES;
+  const baseDelayMs = options?.baseDelayMs ?? BASE_DELAY_MS;
   for (let attempt = 0; ; attempt++) {
     try {
       return await fn();
     } catch (error) {
-      if (!isRetryableError(error) || attempt >= MAX_RETRIES) {
+      if (!isRetryableError(error) || attempt >= maxRetries) {
         throw error;
       }
       const jitter = Math.random() * 1e3;
-      const delay = BASE_DELAY_MS * Math.pow(2, attempt) + jitter;
-      await log?.(`Retryable error, retrying in ${(delay / 1e3).toFixed(1)}s (attempt ${attempt + 1}/${MAX_RETRIES})...`);
+      const delay = baseDelayMs * Math.pow(2, attempt) + jitter;
+      await log?.(`Retryable error, retrying in ${(delay / 1e3).toFixed(1)}s (attempt ${attempt + 1}/${maxRetries})...`);
       await new Promise((resolve) => setTimeout(resolve, delay));
     }
   }
@@ -533,10 +535,8 @@ async function safeGenerateObject(generateObject, params, options) {
   const strictParams = { ...params, schema: toStrictSchema(params.schema) };
   for (let attempt = 0; attempt <= maxRetries; attempt++) {
     try {
-      const result = await withRetry(
-        () => generateObject(strictParams),
-        options?.log
-      );
+      const generate = () => generateObject(strictParams);
+      const result = options?.retry === false ? await generate() : await withRetry(generate, options?.log, options?.retry);
       return {
         ...result,
         object: params.schema.parse(sanitizeNulls(result.object))
@@ -9578,11 +9578,20 @@ function coverageTextForSelection(coverage) {
     ])
   ].filter(Boolean).join(" ");
 }
+function coverageCleanupEntries(profile, coverageIndexes) {
+  if (!coverageIndexes) {
+    return profile.coverages.map((coverage, coverageIndex) => ({ coverage, coverageIndex }));
+  }
+  return [...new Set(coverageIndexes)].sort((left, right) => left - right).map((coverageIndex) => {
+    const coverage = profile.coverages[coverageIndex];
+    return coverage ? { coverage, coverageIndex } : void 0;
+  }).filter((entry) => Boolean(entry));
+}
 function nodeTextMatchesCoverage(node, coverageTerms) {
   const text = nodeTextForSelection(node).toLowerCase();
   return coverageTerms.some((term) => term.length >= 5 && text.includes(term));
 }
-function selectCoverageCleanupNodes(sourceTree, profile) {
+function selectCoverageCleanupNodes(sourceTree, coverages) {
   const nodeById = new Map(sourceTree.map((node) => [node.id, node]));
   const childrenByParent = /* @__PURE__ */ new Map();
   for (const node of sourceTree) {
@@ -9600,12 +9609,12 @@ function selectCoverageCleanupNodes(sourceTree, profile) {
     const current = selected.get(node.id);
     if (!current || score > current.score) selected.set(node.id, { node, score });
   };
-  const sourceNodeIds = uniqueStrings(profile.coverages.flatMap((coverage) => [
+  const sourceNodeIds = uniqueStrings(coverages.flatMap((coverage) => [
     ...coverage.sourceNodeIds,
     ...coverage.limits.flatMap((term) => term.sourceNodeIds)
   ]));
   const coveragePages2 = /* @__PURE__ */ new Set();
-  const coverageTerms = uniqueStrings(profile.coverages.flatMap(
+  const coverageTerms = uniqueStrings(coverages.flatMap(
     (coverage) => coverageTextForSelection(coverage).toLowerCase().split(/[^a-z0-9$,.]+/i).filter((part) => part.length >= 5)
   ));
   for (const id of sourceNodeIds) {
@@ -9648,8 +9657,9 @@ function selectCoverageCleanupNodes(sourceTree, profile) {
     (left, right) => (left.node.pageStart ?? Number.MAX_SAFE_INTEGER) - (right.node.pageStart ?? Number.MAX_SAFE_INTEGER) || left.node.order - right.node.order
   ).map((entry) => entry.node);
 }
-function buildOperationalProfileCleanupPrompt(sourceTree, profile) {
-  const nodes = selectCoverageCleanupNodes(sourceTree, profile).map((node) => compactNode(
+function buildOperationalProfileCleanupPrompt(sourceTree, profile, options = {}) {
+  const coverageEntries = coverageCleanupEntries(profile, options.coverageIndexes);
+  const nodes = selectCoverageCleanupNodes(sourceTree, coverageEntries.map((entry) => entry.coverage)).map((node) => compactNode(
     node,
     node.kind === "page" || node.kind === "page_group" ? 260 : node.kind === "table_row" || node.kind === "table_cell" ? 520 : 360
   ));
@@ -9657,9 +9667,12 @@ function buildOperationalProfileCleanupPrompt(sourceTree, profile) {
     documentType: profile.documentType,
     policyTypes: profile.policyTypes,
     coverageTypes: profile.coverageTypes,
-    coverages: profile.coverages.map(compactCoverageForCleanup)
+    coverages: coverageEntries.map(({ coverage, coverageIndex }) => compactCoverageForCleanup(coverage, coverageIndex))
   };
   return `Review and clean a source-backed operational profile projection for an insurance policy.
+${options.label ? `
+Coverage group: ${options.label}
+` : ""}
 
 Task:
 - Inspect the candidate coverage projection against the source nodes.
@@ -11744,7 +11757,8 @@ async function runVisualTableRepair(params) {
   if (candidates.length === 0) return { sourceTree: params.sourceTree, warnings: [] };
   let sourceTree = params.sourceTree;
   const warnings = [];
-  for (const [index, candidate] of candidates.entries()) {
+  const repairResults = await Promise.all(candidates.map(async (candidate, index) => {
+    const label = `source_tree_visual_table_repair_p${candidate.page}`;
     try {
       const budget = params.resolveBudget("extraction_source_tree", 1800);
       const maxTokens = Math.min(budget.maxTokens, 2400);
@@ -11758,7 +11772,7 @@ async function runVisualTableRepair(params) {
           taskKind: "extraction_source_tree",
           budgetDiagnostics: { ...budget, maxTokens },
           trace: {
-            label: `source_tree_visual_table_repair_p${candidate.page}`,
+            label,
             startPage: candidate.page,
             endPage: candidate.page,
             batchIndex: index + 1,
@@ -11768,23 +11782,43 @@ async function runVisualTableRepair(params) {
         },
         {
           fallback: { tables: [], warnings: [] },
-          log: params.log
+          maxRetries: 0,
+          log: params.log,
+          retry: false
         }
       );
-      params.trackUsage(response.usage, {
-        taskKind: "extraction_source_tree",
-        label: `source_tree_visual_table_repair_p${candidate.page}`,
+      return {
+        index,
+        candidate,
+        label,
         maxTokens,
-        durationMs: Date.now() - startedAt
-      });
-      const repair = response.object;
-      sourceTree = applyVisualTableRepair(sourceTree, repair);
-      warnings.push(...repair.warnings.map(
-        (warning) => `Visual table repair warning on page ${candidate.page}: ${warning}`
-      ));
+        durationMs: Date.now() - startedAt,
+        usage: response.usage,
+        repair: response.object
+      };
     } catch (error) {
-      warnings.push(`Visual table repair skipped on page ${candidate.page}; parsed table kept (${error instanceof Error ? error.message : String(error)})`);
+      return {
+        index,
+        candidate,
+        error: error instanceof Error ? error.message : String(error)
+      };
     }
+  }));
+  for (const result of repairResults.sort((left, right) => left.index - right.index)) {
+    if ("error" in result) {
+      warnings.push(`Visual table repair skipped on page ${result.candidate.page}; parsed table kept (${result.error})`);
+      continue;
+    }
+    params.trackUsage(result.usage, {
+      taskKind: "extraction_source_tree",
+      label: result.label,
+      maxTokens: result.maxTokens,
+      durationMs: result.durationMs
+    });
+    sourceTree = applyVisualTableRepair(sourceTree, result.repair);
+    warnings.push(...result.repair.warnings.map(
+      (warning) => `Visual table repair warning on page ${result.candidate.page}: ${warning}`
+    ));
   }
   return { sourceTree, warnings };
 }
@@ -12037,9 +12071,9 @@ async function runSourceTreeExtraction(params) {
   };
   if (shouldRunSourceTreeOrganizer(sourceTree, sourceSpans)) {
     try {
-      const organizations = [];
       const batches = organizationBatches(sourceTree);
-      for (const [batchIndex, batch] of batches.entries()) {
+      const organizationResults = await Promise.all(batches.map(async (batch, batchIndex) => {
+        const label = batches.length > 1 ? `source_tree_organizer_${batchIndex + 1}` : "source_tree_organizer";
         const budget = params.resolveBudget("extraction_source_tree", 4096);
         const startedAt = Date.now();
         const response = await safeGenerateObject(
@@ -12049,20 +12083,39 @@ async function runSourceTreeExtraction(params) {
             schema: SourceTreeOrganizationSchema,
             maxTokens: budget.maxTokens,
             taskKind: "extraction_source_tree",
-            budgetDiagnostics: budget
+            budgetDiagnostics: budget,
+            trace: {
+              label,
+              batchIndex: batchIndex + 1,
+              batchCount: batches.length,
+              sourceBacked: true
+            }
           },
           {
             fallback: { labels: [], groups: [] },
-            log: params.log
+            maxRetries: 0,
+            log: params.log,
+            retry: false
           }
         );
-        localTrack(response.usage, {
+        return {
+          batchIndex,
+          label,
+          budget,
+          durationMs: Date.now() - startedAt,
+          usage: response.usage,
+          organization: response.object
+        };
+      }));
+      const organizations = [];
+      for (const result of organizationResults.sort((left, right) => left.batchIndex - right.batchIndex)) {
+        localTrack(result.usage, {
           taskKind: "extraction_source_tree",
-          label: batches.length > 1 ? `source_tree_organizer_${batchIndex + 1}` : "source_tree_organizer",
-          maxTokens: budget.maxTokens,
-          durationMs: Date.now() - startedAt
+          label: result.label,
+          maxTokens: result.budget.maxTokens,
+          durationMs: result.durationMs
         });
-        organizations.push(response.object);
+        organizations.push(result.organization);
       }
       sourceTree = applyOrganization(sourceTree, mergeOrganizationResults(organizations));
     } catch (error) {
@@ -12087,7 +12140,9 @@ async function runSourceTreeExtraction(params) {
         },
         {
           fallback: { labels: [], groups: [] },
-          log: params.log
+          maxRetries: 0,
+          log: params.log,
+          retry: false
         }
       );
       localTrack(response.usage, {
@@ -12129,7 +12184,9 @@ async function runSourceTreeExtraction(params) {
       },
       {
         fallback: emptyProfile,
-        log: params.log
+        maxRetries: 0,
+        log: params.log,
+        retry: false
       }
     );
     localTrack(response.usage, {
@@ -12162,16 +12219,23 @@ async function runSourceTreeExtraction(params) {
           taskKind: "extraction_coverage_cleanup",
           budgetDiagnostics: budget,
           providerOptions: params.providerOptions,
-          trace: { phase: "coverage_cleanup", sourceBacked: true }
+          trace: {
+            phase: "coverage_cleanup",
+            label: "Coverage cleanup",
+            itemCount: operationalProfile.coverages.length,
+            sourceBacked: true
+          }
         },
         {
           fallback: { coverageDecisions: [], warnings: [] },
-          log: params.log
+          maxRetries: 0,
+          log: params.log,
+          retry: false
         }
       );
       localTrack(response.usage, {
         taskKind: "extraction_coverage_cleanup",
-        label: "operational_profile_cleanup",
+        label: "coverage_cleanup",
         maxTokens: budget.maxTokens,
         durationMs: Date.now() - startedAt
       });
@@ -12231,6 +12295,7 @@ function createExtractor(config) {
     sourceStore,
     qualityGate = "warn",
     modelCapabilities,
+    modelCapabilitiesByTaskKind,
     modelBudgetConstraints,
     onCheckpointSave
   } = config;
@@ -12247,10 +12312,11 @@ function createExtractor(config) {
   };
   let activeProviderOptions = providerOptions;
   function resolveBudget(taskKind, hintTokens) {
+    const taskModelCapabilities = modelCapabilitiesByTaskKind?.[taskKind] ?? modelCapabilities;
     return resolveModelBudget({
       taskKind,
       hintTokens,
-      modelCapabilities,
+      modelCapabilities: taskModelCapabilities,
       constraint: modelBudgetConstraints?.[taskKind]
     });
   }
@@ -12573,7 +12639,9 @@ ${sourceText}`;
           },
           {
             fallback: { forms: [] },
+            maxRetries: 0,
             log,
+            retry: false,
             onError: (err, attempt) => log?.(`Form inventory attempt ${attempt + 1} failed: ${err instanceof Error ? err.message : String(err)}`)
           }
         );