summaryrefslogtreecommitdiff
path: root/hag/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'hag/__init__.py')
-rw-r--r--hag/__init__.py21
1 files changed, 21 insertions, 0 deletions
diff --git a/hag/__init__.py b/hag/__init__.py
new file mode 100644
index 0000000..18496e9
--- /dev/null
+++ b/hag/__init__.py
@@ -0,0 +1,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",
+]