pyproject.toml

# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

[build-system]
requires = ["maturin>=1.12.6,<2.0"]
build-backend = "maturin"

[project]
name = "nemo-relay"
dynamic = ["version"]
description = "Python bindings for the NeMo Relay agent runtime."
readme = "python/nemo_relay/README.md"
requires-python = ">=3.11"
license = "Apache-2.0"
authors = [
    { name = "NVIDIA Corporation & Affiliates" },
]
keywords = [
    "agents",
    "ai",
    "llm",
    "middleware",
    "nemo-relay",
    "observability",
    "runtime",
    "tools",
]
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Developers",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
    "Programming Language :: Rust",
]

[project.urls]
Documentation = "https://nvidia-nemo-relay.docs.buildwithfern.com/nemo-relay"
Homepage = "https://github.com/NVIDIA/NeMo-Relay"
Issues = "https://github.com/NVIDIA/NeMo-Relay/issues"
Repository = "https://github.com/NVIDIA/NeMo-Relay"

[dependency-groups]
dev = [
    # Transitive dependency of optional integrations; keep it at a patched release.
    "anyio>=4.14.2",
    "beautifulsoup4>=4.14,<5",
    "ipython~=8.20",
    "maturin[zig]>=1.0,<2.0; sys_platform == 'linux'",
    "maturin>=1.0,<2.0; sys_platform != 'linux'",
    "pip-licenses>=5,<6",
    "pre-commit~=4.0",
    "pylic>=5.0.1",
    "ruff~=0.11",
    # Transitive dependency of beautifulsoup4; keep it at a patched release.
    "soupsieve>=2.9.0",
    "ty>=0.0.1a7",
    "uv~=0.11.0",
]

test = [
    # De Morgan form: exclude only when the platform is both Windows and ARM64.
    "grpcio>=1.81.1,<2; sys_platform != 'win32' or (platform_machine != 'ARM64' and platform_machine != 'arm64' and platform_machine != 'aarch64')",
    "nemo-relay-plugin; sys_platform != 'win32' or (platform_machine != 'ARM64' and platform_machine != 'arm64' and platform_machine != 'aarch64')",
    "opentelemetry-proto>=1.39,<2",
    "pydantic>=2",
    "pytest>=8",
    "pytest-asyncio>=0.26",
    "pytest-cov~=7.0",
]

[project.optional-dependencies]
cli = [
    "nemo-relay-cli-bin==0.10.0",
]

langchain = [
    "langchain>=1.3.15,<2.0.0",
    "langchain-anthropic>=1.4.8,<2.0.0",
    "langchain-core",
    "langgraph>=1.2.9,<2.0.0",
    "langgraph-checkpoint>=4.1.1",
    "langsmith>=0.9.4",
]

langgraph = [
    "nemo-relay[langchain]",
    "langgraph>=1.2.9,<2.0.0",
]

deepagents = [
    "nemo-relay[langgraph]",
    "cryptography>=50.0.0",
    "deepagents>=0.7.4,<0.8.0",
    "pyasn1>=0.6.4",
]

langchain-nvidia = [
    "nemo-relay[langchain]",
    "langchain-nvidia-ai-endpoints>=1.4.1,<2.0.0",
    "aiohttp>=3.14.3",
    "idna>=3.19",
]

[tool.maturin]
manifest-path = "crates/python/Cargo.toml"
python-source = "python"
module-name = "nemo_relay._native"
features = ["pyo3/extension-module"]
include = [{ path = "LICENSE", format = ["sdist", "wheel"] }]

[tool.maturin.sbom]
rust = true
auditwheel = true

[tool.uv]
default-groups = ["dev", "test"]
package = true

[tool.uv.sources]
nemo-relay-cli-bin = { path = "python/cli-bin" }
nemo-relay-plugin = { workspace = true }

[tool.uv.workspace]
members = ["python/plugin"]

[tool.pytest.ini_options]
testpaths = ["python/tests"]
asyncio_mode = "auto"
pythonpath = ["python"]

[tool.coverage.run]
# Exclude integration tests from coverage, since we don't run these by default
omit = [
    "python/nemo_relay/integrations/*",
    "python/plugin/src/nemo_relay_plugin/_proto/*",
]

[tool.ty.analysis]
# nemo_relay._native is a compiled Rust extension (built by maturin) that only
# exists after `uv sync` / `pip install -e .`. Suppress unresolved-import for it.
# LangChain, LangGraph, and Deep Agents are optional integration dependencies
# which aren't installed by default.
allowed-unresolved-imports = ["deepagents.**", "grpc", "langchain.**", "langchain_*.**", "langgraph.**", "nemo_relay._native", "pytest"]

[[tool.ty.overrides]]
include = ["python/tests/**"]

[tool.ty.overrides.rules]
# Managed execution may return any valid JSON subtype after intercepts run.
# Tests that exercise a known object result assert its runtime shape directly.
invalid-argument-type = "ignore"
not-subscriptable = "ignore"
unsupported-operator = "ignore"

[tool.ruff]
line-length = 120
target-version = "py311"
extend-exclude = ["python/plugin/src/nemo_relay_plugin/_proto"]

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

[tool.ruff.lint]
select = ["E", "F", "W", "I"]

[tool.ruff.lint.isort]
known-first-party = ["nemo_relay", "nemo_relay_plugin"]

[tool.pylic]
safe_licenses = [
    # Apache
    "Apache-2.0",
    "Apache-2.0 AND MIT",
    "Apache 2.0",
    "Apache License 2.0",
    "Apache Software License",
    # BSD
    "BSD License",
    "BSD-3-Clause",
    "3-Clause BSD License",
    "BSD-2-Clause",
    # ISC
    "ISC License (ISCL)",
    # MIT
    "MIT License",
    "MIT",
    # PSFL
    "Python Software Foundation License",
    "PSF-2.0",
]
unsafe_packages = [
    "certifi", # Mozilla Public License 2.0 (MPL 2.0)
    "matplotlib-inline", # unknown derived
]