[build-system]
requires = ["setuptools>=68.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "hag"
version = "0.1.0"
requires-python = ">=3.10"
description = "Hopfield-Augmented Generation: RAG with iterative Modern Hopfield Network retrieval"

dependencies = [
    "torch>=2.0",
    "numpy>=1.24",
    "faiss-cpu>=1.7",
    "transformers>=4.36",
    "datasets>=2.14",
    "pyyaml>=6.0",
    "tqdm>=4.65",
]

[project.optional-dependencies]
dev = [
    "pytest>=7.0",
    "pytest-cov>=4.0",
]
eval = [
    "scikit-learn>=1.3",
    "umap-learn>=0.5",
    "matplotlib>=3.7",
]

[tool.setuptools.packages.find]
include = ["hag*"]

[tool.pytest.ini_options]
testpaths = ["tests"]