pyproject.toml

[project]
name = "alphatrion"
version = "0.2.0"
description = "⚒️ AlphaTrion is an open-source framework to help build GenAI applications, including experiment tracking, adaptive model routing, prompt optimization and performance evaluation."
license = {text = "Apache-2.0"}
readme = "README.md"
requires-python = ">=3.12"

dependencies = [
    "sqlalchemy (>=2.0.43,<3.0.0)",
    "python-dotenv (>=1.1.1,<2.0.0)",
    "oras (>=0.2.38,<0.3.0)",
    "psycopg2-binary (>=2.9.10,<3.0.0)",
    "pydantic (>=2.11.7,<3.0.0)",
    "traceloop-sdk (>=0.47.5,<1.0.0)",
    "openai (>=2.6.1,<3.0.0)",
    "alembic (>=1.17.1,<2.0.0)",
    "pg8000>=1.31.5",
    "uvicorn>=0.38.0",
    "strawberry-graphql>=0.285.0",
    "fastapi>=0.121.2",
    "rich>=14.2.0",
    "httpx>=0.28.1",
    "clickhouse-connect>=0.7.0",
    "faker>=37.12",
    "prometheus-client>=0.21.1",
    "python-jose[cryptography]>=3.3.0",
    "bcrypt>=4.0.0",
    "python-multipart>=0.0.6",
    "pyyaml>=6.0.3",
]

[project.optional-dependencies]
s3 = [
    "boto3>=1.35.0",
]

[dependency-groups]
dev = [
  {include-group = "lint"},
  {include-group = "test"}
]
lint = [
    "ruff>=0.12",
]
test = [
    "pytest>=8.4",
    "pytest-asyncio>=0.22",
    "pytest-timeout>=2.1",
    "nest-asyncio>=1.5.0",  # Required for DataLoader tests
    "moto>=5.0.0",  # For S3 backend testing
    "boto3>=1.35.0",  # Required by moto for S3 mocking
]

[project.scripts]
alphatrion = "alphatrion.server.cmd.main:main"

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

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

[tool.hatch.build.targets.wheel.force-include]
"config/modelspec.yaml" = "alphatrion/config/modelspec.yaml"

# Configuration for ruff linter and formatter

[tool.ruff]
line-length = 88
target-version = "py312"
lint.select = [
    "E",  # pycodestyle,
    "F",  # Pyflakes
    "UP",  # pyupgrade
    "B",  # flake8-bugbear
    "SIM",  # flake8-simplify
    "I",  # isort
    "PL",  # PyLint,
]
lint.ignore = [
    "E501",  # Line too long
    "PLR0913",  # Too many arguments in function definition
    "PLR0912",  # Too many branches
    "PLR0911",  # Too many return statements
    "SIM117", # Use a single `with` statement with multiple contexts instead of nested `with` statements
    "PLR0915", # Too many statements in function
    "PLC0415", # `import` should be at the top-level of a file
    "PLW0603", # Using the global statement
]
exclude = ["venv", "migrations", "__pycache__"]

[tool.ruff.format]
quote-style = "double"
indent-style = "space"

[tool.ruff.lint.per-file-ignores]
"tests/*" = ["PLR2004", "B017"]