pyproject.toml

[project]
name = "lightspeed-stack"
dynamic = ["version"]
description = "LLM tooling stack"
authors = []
maintainers = [
    {email = "tisnik@centrum.cz", name = "Pavel Tišnovský"},
]
classifiers = [
    "Development Status :: 4 - Beta",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Software Development",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3 :: Only",
    "License :: OSI Approved :: Apache Software License",
]
keywords = [
    "LLM",
    "RAG"
]
requires-python = ">=3.12,<3.14"
readme = "README.md"
license = {file = "LICENSE"}
dependencies = [
    # Used for HTTP service base
    "fastapi>=0.115.12",
    "uvicorn>=0.34.3",
    # Used by authentication/k8s integration
    "kubernetes>=30.1.0",
    # Used to call Llama Stack APIs
    "llama-stack==0.2.22",
    "llama-stack-client==0.2.22",
    # Used by Logger
    "rich>=14.0.0",
    # Used by JWK token auth handler
    "cachetools>=6.1.0",
    # Used for metric exporter
    "prometheus-client>=0.22.1",
    # Used for Bearer token extraction
    "starlette>=0.47.1",
    # Used by JWK token auth handler
    "aiohttp>=3.12.14",
    "authlib>=1.6.0",
    # OpenAPI exporter
    "email-validator>=2.2.0",
    "openai>=1.99.9",
    # Used by database interface
    "sqlalchemy>=2.0.42",
    # Used by Llama Stack version checker
    "semver<4.0.0",
    # Used by authorization resolvers
    "jsonpath-ng>=1.6.1",
    "psycopg2-binary>=2.9.10",
    "litellm>=1.75.5.post1",
]


[tool.pyright]
exclude = [
  # TODO(lucasagomes): This module was copied from road-core
  # service/ols/src/auth/k8s.py and currently has 58 Pyright issues. It
  # might need to be rewritten down the line.
  "src/authentication/k8s.py",
]
extraPaths = ["./src"]

[tool.pdm]
distribution = true

[tool.pdm.version]
source = "file"
path = "src/version.py"

[tool.behave]
paths = ["tests/e2e/features"]

[project.scripts]
lightspeed-stack = "lightspeed_stack:main"

[project.urls]
Homepage = "https://github.com/lightspeed-core/lightspeed-stack"
Issues = "https://github.com/lightspeed-core/lightspeed-stack/issues"

# PyTorch has multiple wheel variants for different backends - cpu, gpu, etc.
# By default on pypi.org is the gpu variant. Forces uv to use the cpu variant.
[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true
[tool.uv.sources]
torch = [{ index = "pytorch-cpu" }]

[dependency-groups]
dev = [
    "black>=25.1.0",
    "pytest>=8.3.2",
    "pytest-cov>=5.0.0",
    "pytest-mock>=3.14.0",
    "pytest-asyncio>=1.0.0",
    "pyright>=1.1.401",
    "pylint>=3.3.7",
    "pydocstyle>=6.3.0",
    "mypy>=1.16.0",
    "types-PyYAML>=6.0.2",
    "types-requests>=2.28.0",
    "ruff>=0.11.13",
    "aiosqlite",
    "behave>=1.3.0",
    "types-cachetools>=6.1.0.20250717",
    "build>=1.2.2.post1",
    "twine>=6.1.0",
    "openapi-to-md>=0.1.0b2",
    "pytest-subtests>=0.14.2",
    "bandit>=1.8.6",
]
llslibdev = [
    # To check llama-stack API provider dependecies:
    #
    #    $ uv run llama stack list-providers
    #
    # API agents: inline::meta-reference
    "matplotlib>=3.10.0",
    "pillow>=11.1.0",
    "pandas>=2.2.3",
    "scikit-learn>=1.5.2",
    "psycopg2-binary>=2.9.10",
    # API eval: inline::meta-reference
    "tree_sitter>=0.24.0",
    "pythainlp>=3.0.10",
    "langdetect>=1.0.9",
    "emoji>=2.1.0",
    "nltk>=3.8.1",
    # API inference: inline::sentence-transformers
    "sentence-transformers>=5.0.0",
    # API vector_io: inline::faiss
    "faiss-cpu>=1.11.0",
    # API scoring: inline::basic
    "requests>=2.32.4",
    # API datasetio: inline::localfs
    "aiosqlite>=0.21.0",
    # API datasetio: remote::huggingface
    "datasets>=3.6.0",
    # API telemetry: inline::meta-reference
    "opentelemetry-sdk>=1.34.1",
    "opentelemetry-exporter-otlp>=1.34.1",
    # API tool_runtime: inline::rag-runtime
    "transformers>=4.34.0",
    "numpy==2.2.6",
    # API tool_runtime: remote::model-context-protocol
    "mcp>=1.9.4",
    # API post_training: inline::huggingface
    "torch==2.7.1",
    "trl>=0.18.2",
    "peft>=0.15.2",
    # Other
    "autoevals>=0.0.129",
    "fire>=0.7.0",
    "opentelemetry-instrumentation>=0.55b0",
    "blobfile>=3.0.0",
    "psutil>=7.0.0",
    # API inference: remote::ollama
    "ollama>=0.4.7",
    "h11>=0.16.0",
]

build = [
    "build>=1.2.2.post1",
    "twine>=6.1.0",
]

[tool.pytest.ini_options]
asyncio_mode = "auto"
pythonpath = [
    "src"
]
addopts = [
    "--import-mode=importlib",
]

[tool.pylint.main]
source-roots = "src"
ignore = ["query.py"]

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

[tool.pylint."MESSAGES CONTROL"]
disable = ["R0801"]

[tool.ruff]
[tool.ruff.lint.flake8-tidy-imports]
banned-api = { "unittest" = { msg = "use pytest instead of unittest" }, "unittest.mock" = { msg = "use pytest-mock instead of unittest.mock" } }

[tool.mypy]
explicit_package_bases = true
disallow_untyped_calls = true
disallow_untyped_defs = true
disallow_incomplete_defs = true
ignore_missing_imports = true
disable_error_code = "attr-defined"