"""Document store protocol and backends for AI pipeline flows."""

from ._models import MAX_PROVENANCE_GRAPH_NODES, DocumentNode, build_provenance_graph, walk_provenance
from ._summary import SummaryGenerator, SummaryUpdateFn
from .factory import create_document_store
from .protocol import DocumentStore, get_document_store, set_document_store

__all__ = [
    "MAX_PROVENANCE_GRAPH_NODES",
    "DocumentNode",
    "DocumentStore",
    "SummaryGenerator",
    "SummaryUpdateFn",
    "build_provenance_graph",
    "create_document_store",
    "get_document_store",
    "set_document_store",
    "walk_provenance",
]
