pyproject.toml

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

[project]
name = "undertrial-ai"
version = "1.0.0"
description = "OpenEnv-compliant RL environment for Indian bail decision support"
readme = "README.md"
requires-python = ">=3.10"
license = {text = "MIT"}
keywords = ["openenv", "rl", "legal-ai", "bail", "india", "grpo"]
classifiers = [
    "Programming Language :: Python :: 3",
    "License :: OSI Approved :: MIT License",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
    "pydantic>=2.6.0",
    "openenv-core>=0.1.0",
]

[project.optional-dependencies]
server = [
    "fastapi>=0.110.0",
    "uvicorn[standard]>=0.27.0",
    "websockets>=12.0",
]
train = [
    "unsloth",
    "trl>=0.8.0",
    "torch>=2.1.0",
    "datasets>=2.18.0",
    "transformers>=4.40.0",
    "matplotlib>=3.7.0",
    "wandb",
]

[project.scripts]
server = "undertrial_ai.server.app:main"

[tool.setuptools.packages.find]
where = ["."]
include = ["undertrial_ai*"]