hag/__init__.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

"""HAG: Hopfield-Augmented Generation."""

from hag.config import (
    EncoderConfig,
    GeneratorConfig,
    HopfieldConfig,
    MemoryBankConfig,
    PipelineConfig,
)
from hag.datatypes import HopfieldResult, PipelineResult, RetrievalResult

__all__ = [
    "HopfieldConfig",
    "MemoryBankConfig",
    "EncoderConfig",
    "GeneratorConfig",
    "PipelineConfig",
    "HopfieldResult",
    "RetrievalResult",
    "PipelineResult",
]