pyproject.toml

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

[project]
name = "cricket-captain"
version = "0.1.0"
description = "CricketCaptain-LLM: Strategic Coherence Training via Tool-Aware RL"
requires-python = ">=3.10"
dependencies = [
    "openenv-core[core]>=0.2.2",
    "numpy>=1.24.0",
    "pyyaml>=6.0.1",
    "gradio>=4.0.0",
    "openai>=1.0.0",
]

# Training extras — these are the versions that actually reconcile in 2026:
#   transformers 5.6.2 ─┐
#   trl 1.2.0           ├─ TRL multi-turn environment_factory needs transformers >=5.2,
#   vllm 0.19.1         ┘  vLLM 0.19+ supports transformers 5, vLLM 0.18 does not.
# Earlier we tried vllm 0.11.x — it pinned transformers <5 and broke environment_factory.
# mergekit removed: pinned pydantic <2.11 which conflicts with openenv-core 0.2.3 (>=2.11.7).
# Not used by training anyway.
[project.optional-dependencies]
train = [
    "torch==2.10.0",
    "transformers==5.6.2",
    "trl==1.2.0",
    "vllm==0.19.1",
    "peft>=0.13.0,<0.20.0",
    "accelerate>=1.0.0",
    "datasets>=4.0.0",
    "bitsandbytes>=0.43.0",
    "wandb>=0.16",
    # flash-attn is optional — vLLM has its own attention backends; uncomment if you want it:
    # "flash-attn>=2.5.0",
]
eval = [
    "matplotlib>=3.8.0",
    "seaborn>=0.13.0",
    "pandas>=2.1.0",
]

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

[tool.setuptools]
include-package-data = true
packages = ["cricket_captain", "cricket_captain.server"]
package-dir = { "cricket_captain" = ".", "cricket_captain.server" = "server" }