EnrichmentConfig

Configuration for how enrichment results are applied.

Usage

Source

EnrichmentConfig()

Controls how many related documents to include in context, and which enrichment features to enable.

Parameters

add_related_docs: int = 3

Number of most-related documents to include as context when building prompts.

add_entity_context: bool = True

Whether to include entity descriptions in prompt context.

add_temporal_context: bool = False

Whether to include temporal annotations in context.

Examples

import talk_box as tb

config = tb.EnrichmentConfig(
    add_related_docs=3,
    add_entity_context=True,
)