pyproject.toml

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

[project]
name = "tropt"
version = "0.1.1"
description = "A toolbox for optimizing discrete text triggers."
readme = "README.md"
requires-python = ">=3.10"
license = {text = "MIT"}
authors = [
  {name = "Matan Ben-Tov"}
]
keywords = [
  "adversarial-attacks",
  "jailbreak",
  "red-teaming",
  "trigger-optimization",
  "discrete-optimization",
  "prompt-optimization",
  "llm",
  "nlp",
  "robustness",
  "ai-safety",
]
classifiers = [
  "License :: OSI Approved :: MIT License",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

# Core dependencies - minimal for basic HuggingFace model optimization
dependencies = [
  # Core ML/numeric libraries
  "torch>=2.4.0",
  "numpy>=2.0.0",
  "jaxtyping",
  # HuggingFace models (core use case)
  "transformers==5.8.1",
  "accelerate>=1.0.0",
  "sentence-transformers>=5.5.0",
  # Data validation
  "pydantic>=2.0.0",
  # Configuration & utilities
  "tqdm",
  "tenacity",
  "pandas>=2.3.3",
  "datasets>=4.8.3",
  # Used directly by tropt/utils/refusal_dir.py (refusal-direction extraction).
  "requests",
  "scikit-learn",

  # Config [deprecated for now]
  # "hydra-core>=1.3.0",
  # "omegaconf",
]

[project.urls]
Homepage = "https://tropt.dev"
Repository = "https://github.com/matanbt/TROPT"
Issues = "https://github.com/matanbt/TROPT/issues"

[project.optional-dependencies]
# API model integrations (install separately as needed)
openai = [
  "openai>=1.0.0",
  "tiktoken>=0.5.0",
]

google = [
  "google-genai>=1.0.0",
]

voyage = [
  "voyageai>=0.3.0",
]

litellm = [
  "litellm>=1.0.0",
]

# vision-related integration (e.g., diffusion models)
vision = [
  "diffusers>=0.27.0",
]

# Experiment tracking and visualization
tracking = [
  "wandb>=0.25.1",
  "livelossplot>=0.5.0",
  "trackio>=0.3.0",
]

# Jupyter notebook support
notebooks = [
  "ipython>=8.0.0",
  "jupyter>=1.0.0",
]

# All optional features (convenience installer)
all = [
  "tropt[openai,google,voyage,litellm,vision,tracking,notebooks]",
]

# Development tools
dev = [
  # Testing
  "pytest>=7.0.0",
  "pytest-cov>=4.0.0",

  # Linting & formatting
  "ruff>=0.1.0",
  "ty>=0.0.24",
  "pre-commit>=3.0.0",

  # All optional features for comprehensive testing
  "tropt[all]",
  "litellm[proxy]",

  # Analysis & visualization
  "matplotlib>=3.7.0",
  "seaborn>=0.13.0",

  # Documentation
  "sphinx>=5.0.0",
  "sphinx_rtd_theme>=1.0.0",
  "pydata_sphinx_theme>=0.13.0",
  "sphinx-autodoc-typehints>=1.20.0",
  "sphinx-copybutton>=0.5.0",
  "sphinx-codeautolink>=0.15.0",
  "sphinx-design>=0.5.0",
  "sphinx-autobuild>=2024.0.0",
  "myst-parser>=0.18.0",
  # SEO: auto-generates sitemap.xml + Open Graph / Twitter card meta tags.
  "sphinx-sitemap>=2.5.0",
  "sphinxext-opengraph>=0.9.0",
]

[[tool.uv.index]]
name = "pytorch-cu126"
url = "https://download.pytorch.org/whl/cu126"
explicit = true

[tool.uv.sources]
torch = [{ index = "pytorch-cu126" }]

[tool.hatch.build.targets.wheel]
packages = ["tropt"]

# Explicit allowlist so the sdist ships only the package + metadata.
[tool.hatch.build.targets.sdist]
include = [
  "/tropt",
  "/README.md",
  "/LICENSE",
  "/pyproject.toml",
]

[tool.ruff]
include = ["tropt/**/*.py", "tests/**/*.py"]
extend-exclude = ["tests/_legacy"]

[tool.ruff.lint]
# Enable Pyflakes, pycodestyle, and isort rules
select = ["E", "F", "W", "I"]

ignore = [
    "E501",  # line length (currently ignored, TODO: re-format long lines)
    "F722",  # Forward annotation false positive from jaxtyping
    "F821",  # Undefined name false positive from jaxtyping
]

[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["F401", "I001"]  # custom import order in init files 

[tool.ty.src]
include = ["tropt/"]

[tool.ty.rules]
unresolved-attribute = "ignore"   # HF/ST models expose attrs dynamically (e.g. .generate, .encode, .config)
invalid-argument-type = "ignore"  # jaxtyping annotations (Float[Tensor, ...]) confuse ty's type narrowing
call-non-callable = "ignore"      # ST's _first_module(), .encode(), HF's .generate() seen as Tensor|Module union
invalid-method-override = "ignore"  # model/loss subclasses intentionally specialize abstract signatures

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py"]
python_classes = ["Test*"]
python_functions = ["test_*"]
norecursedirs = ["_legacy"]