NeMo-Relay/pyproject.toml at main · NVIDIA/NeMo-Relay · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

[build-system]
requires = ["maturin>=1.0,<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 = [
    "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",
    "ruff~=0.11",
    "ty>=0.0.1a7",
    "uv~=0.10.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]
langchain = [
    "langchain>=1.3.11,<2.0.0",
    "langchain-core",
    "langgraph", # allow any compatible version
    "langgraph-checkpoint>=4.1.1",
    "langsmith>=0.9.4",
]

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

deepagents = [
    "nemo-relay[langgraph]",
    "deepagents>=0.5.3,<0.6.0",
    "langchain-anthropic>=1.4.8,<2.0.0",
]

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

[tool.maturin]
manifest-path = "crates/python/Cargo.toml"
python-source = "python"
module-name = "nemo_relay._native"
features = ["pyo3/extension-module"]

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

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

[tool.uv.sources]
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.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"]