onf-v3/src/onf/graph/__init__.py

"""Demonstration-graph subsystem: shared schema/constants and the query-dependent GNN retriever.

    schema             — relation vocabulary + on-disk artifact layout (import constants from here)
    QueryEncoder        — pools a query history window into one embedding e_Q
    GraphRetrieverNet   — the NBFNet/GFM-RAG-style retriever: e_Q + graph -> p(v|Q)

See ``onf.graph.gnn`` for the full method (WHAT/WHY, checkpoint format, inductivity guarantee).
"""

from onf.graph.gnn import GraphRetrieverNet, QueryEncoder

__all__ = ["GraphRetrieverNet", "QueryEncoder"]