pyproject.toml

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

[project]
name = "adagraphrag"
version = "0.1.0"
description = "Adaptive Hierarchical Graph-RAG with Incremental Construction and Self-Verified Retrieval"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "MIT" }
authors = [
    { name = "Maryam Zaheer" }
]
keywords = [
    "retrieval-augmented-generation",
    "knowledge-graph",
    "large-language-models",
    "information-retrieval",
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

dependencies = [
    "pydantic>=2.6.0",
    "pydantic-settings>=2.2.0",
    "PyYAML>=6.0.1",
    "python-dotenv>=1.0.1",
    "structlog>=24.1.0",
    "tenacity>=8.2.3",
    "openai>=1.30.0",
    "qdrant-client>=1.9.0",
    "neo4j>=5.19.0",
    "networkx>=3.2",
    "python-igraph>=0.11",
    "leidenalg>=0.10",
    "typer>=0.12.0",
    "fastapi>=0.110.0",
    "uvicorn[standard]>=0.29.0",
    "streamlit>=1.33.0",
    "matplotlib>=3.8.0",
]

# NOTE: no `typer`-based CLI ships yet (see README) -- indexing/querying
# currently go through scripts/build_graph.py and the Streamlit UI.

[project.optional-dependencies]
dev = [
    "pytest>=8.0.0",
    "pytest-cov>=4.1.0",
    "pytest-mock>=3.12.0",
    "black>=24.1.0",
    "ruff>=0.2.0",
    "mypy>=1.8.0",
    "types-PyYAML>=6.0.12",
]
local-llm = [
    "torch>=2.2.0",
    "transformers>=4.39.0",
    "sentence-transformers>=2.6.0",
    "bitsandbytes>=0.43.0",
]

[tool.setuptools.packages.find]
where = ["src"]

[tool.black]
line-length = 100
target-version = ["py310", "py311"]

[tool.ruff]
line-length = 100
target-version = "py310"
src = ["src", "tests"]

[tool.ruff.lint]
select = ["E", "F", "W", "I", "N", "UP", "B", "C4", "SIM"]
ignore = [
    "E501",
    # FastAPI's Depends(...) and Typer's Option()/Argument() are designed
    # to be used exactly as default argument values -- both frameworks
    # inspect the call at import time to build their dependency-injection
    # / CLI-parsing metadata. This is the correct, idiomatic usage, not
    # the mutable-default-argument bug this rule otherwise catches.
    "B008",
]

[tool.mypy]
python_version = "3.12"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = true
ignore_missing_imports = true
mypy_path = "src"

[tool.pytest.ini_options]
minversion = "8.0"
testpaths = ["tests"]
pythonpath = ["src"]
addopts = "-ra -q --cov=adagraphrag --cov-report=term-missing"