[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "apollo-plugin"
version = "0.1.0"
description = "Apollo training plugin for vLLM"
requires-python = ">=3.10"
dependencies = [
    "torch",
    "aiohttp",
    "safetensors",
    "pyzmq",
]

[project.optional-dependencies]
dev = ["pytest"]

[project.entry-points."vllm.general_plugins"]
apollo = "apollo_plugin:register"

[project.scripts]
apollo-checkpoint = "apollo_plugin.checkpoint_sync:main"
apollo-worker = "apollo_plugin.training_worker:main"

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