Source code for lalandre_rag.response.source_builder

"""Build final source document lists from enriched context references."""

from typing import Any, Dict, List

from .builder import build_source_document




def build_sources(
    *,
    refs: List[Dict[str, Any]],
    include_relations: bool,
    include_subjects: bool,
    include_full_content: bool,
) -> List[Dict[str, Any]]:
    """Build final source documents from context refs."""
    sources: List[Dict[str, Any]] = []
    for item in refs:
        doc = item["doc"]
        content_used = str(item["content_used"])
        content_truncated = bool(item["content_truncated"])
        source_id = str(item["source_id"])
        sources.append(
            build_source_document(
                doc,
                include_relations=include_relations,
                include_subjects=include_subjects,
                include_full_content=include_full_content,
                include_content_preview=True,
                content_used=content_used,
                content_truncated=content_truncated,
                source_id=source_id,
            )
        )
    return sources