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

[project]
name = "base-model-trpgner"
version = "0.1.6"
description = "HydroRoll TRPG NER 模型 - 桌上角色扮演游戏日志命名实体识别"
authors = [{ name = "HsiangNianian", email = "leader@hydroroll.team" }]
readme = "README.md"
requires-python = ">=3.10"
license = { text = "AFL-3.0" }
keywords = [
    "hydroroll",
    "trpg",
    "nlp",
    "ner",
    "chinese",
    "onnx",
    "robot framework",
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Developers",
    "License :: OSI Approved :: Academic Free License (AFL)",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Text Processing :: Linguistic",
]

dependencies = [
    "numpy>=1.24.0",
    "onnxruntime>=1.23.2",
    "transformers>=4.57.3",
]

[project.optional-dependencies]
train = [
    "torch>=2.9.1",
    "datasets>=2.18.0",
    "accelerate>=0.27.0",
    "tqdm>=4.66.0",
    "onnxscript>=0.5.7",
]
dev = [
    "base-model-trpgner[train]",
    "pytest>=8.0.0",
    "black>=24.0.0",
    "ruff>=0.1.0",
]
webui = ["base-model-trpgner[train]", "gradio>=6.2.0", "scikit-learn>=1.4.0"]
all = ["base-model-trpgner[train,webui,dev]"]

[project.urls]
Homepage = "https://ailab.hydroroll.team/"
Repository = "https://github.com/HydroRoll-Team/base-model"
Documentation = "https://ailab.hydroroll.team/"
Issues = "https://github.com/HydroRoll-Team/base-model/issues"

[tool.hatch.build.targets.wheel]
packages = ["src/base_model_trpgner"]
artifacts = ["src/base_model_trpgner/**/*.py"]

[tool.hatch.build.targets.sdist]
include = ["/src", "/README.md", "/COPYING"]

[tool.ruff]
exclude = [
    ".bzr",
    ".direnv",
    ".eggs",
    ".git",
    ".git-rewrite",
    ".hg",
    ".ipynb_checkpoints",
    ".mypy_cache",
    ".nox",
    ".pants.d",
    ".pyenv",
    ".pytest_cache",
    ".pytype",
    ".ruff_cache",
    ".svn",
    ".tox",
    ".venv",
    ".vscode",
    "__pypackages__",
    "_build",
    "buck-out",
    "build",
    "dist",
    "node_modules",
    "site-packages",
    "venv",
]
line-length = 100
indent-width = 4
target-version = "py312"

[tool.ruff.lint]
select = ["E4", "E7", "E9", "F", "I", "N", "W"]
ignore = ["E501"]
fixable = ["ALL"]
unfixable = []

[tool.ruff.format]
quote-style = "double"
indent-style = "space"
skip-magic-trailing-comma = false
line-ending = "auto"

[tool.black]
line-length = 100
target-version = ["py312"]

[tool.uv]
dev-dependencies = []