pyproject.toml

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "zerofuse"
version = "0.1.0"
description = "Automated, capability-preserving abliteration for open-weight language models."
readme = "README.md"
requires-python = ">=3.11"
license = { file = "LICENSE" }
keywords = ["llm", "abliteration", "refusal-direction", "transformers", "alignment"]
authors = [{ name = "ZeroFuse Authors" }]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Environment :: GPU",
    "Intended Audience :: Science/Research",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

# Runtime dependencies. Versions are lower bounds known to work; pin in a lockfile
# for reproducible runs.
dependencies = [
    "torch>=2.2",
    "transformers>=4.44",
    "accelerate>=0.33",
    "datasets>=2.20",
    "optuna>=3.6",
    "numpy>=1.26",
    "rich>=13.7",
]

[project.optional-dependencies]
# Optional 4-bit loading to fit large models on smaller accelerators.
quant = ["bitsandbytes>=0.43"]
# Agent integration: the Model Context Protocol server (Claude Desktop, Codex, …).
mcp = ["mcp>=1.2.0"]
# Developer tooling.
dev = ["pytest>=8.0", "ruff>=0.5"]

[project.scripts]
zerofuse = "zerofuse.cli:main"
zerofuse-mcp = "zerofuse.mcp_server:main"

[project.urls]
Documentation = "./docs/index.html"

[tool.hatch.build.targets.wheel]
packages = ["src/zerofuse"]

[tool.ruff]
line-length = 100
target-version = "py311"

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