pyproject.toml

# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES.
# All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

[build-system]
build-backend = "setuptools.build_meta"
requires = ["setuptools >= 64", "setuptools-scm>=8"]

[tool.setuptools.packages.find]
where = ["src"]
include = ["aiq", "nat.*"]

[tool.setuptools_scm]
git_describe_command = "git describe --long --first-parent"

[project]
name = "nvidia-nat"
dynamic = ["version"]
dependencies = [
  # Keep package version constraints as open as possible to avoid conflicts with other packages. Always define a minimum
  # version when adding a new package. If unsure, default to using `~=` instead of `==`. When using `~=`, use 2 digits
  # of precision in the version specifier. For example, use `~=1.2` instead of `~=1.2.3` and `~=0.1.3` instead of
  # `~=0.1.3.5`.
  # Keep sorted!!!
  "aioboto3>=11.0.0",
  "authlib>=1.6.5,<2.0.0",
  "click~=8.1",
  "colorama>=0.4.6,<1.0.0",
  "datasets~=4.4", # workaround for uv's solver choosing different versions of datasets based on sys_platform
  "expandvars~=1.0",
  "fastapi~=0.119",
  "httpx~=0.27",
  "jinja2~=3.1",
  "jsonpath-ng~=1.7",
  "nest-asyncio2==1.7.1",
  "networkx~=3.4",
  # langchain-aws pins numpy to 1.* for Python <3.12 -- remove once package is updated
  "numpy~=1.26; python_version < '3.12'",
  "numpy~=2.3; python_version >= '3.12'",
  "openinference-semantic-conventions>=0.1.14,<1.0.0",
  "openpyxl~=3.1",
  "optuna~=4.4",
  "pandas~=2.2",
  "pip>=24.3.1",
  "pkce==1.0.3",
  "pkginfo~=1.12",
  "platformdirs~=4.3",
  "plotly~=6.0",
  "pydantic~=2.11",
  "pymilvus~=2.6",
  "python-dotenv>=1.1.1,<2.0.0",
  "PyYAML~=6.0",
  "ragas>=0.2.14,<1.0.0", # If you update ragas be sure to update github links in docs/source/workflows/evaluate.md as well
  "rich>=14.0.0,<15.0.0",  # Upgraded from ~=13.9 to support strands-agents-tools (requires >=14.0.0,<15.0.0)
  "tabulate~=0.9",
  "uvicorn[standard]~=0.38",
  "wikipedia~=1.4",
  # transitive dependencies
  "urllib3>=2.6.3,<3.0.0",
]
requires-python = ">=3.11,<3.14"
description = "NVIDIA NeMo Agent toolkit"
readme = "src/nat/meta/pypi.md"
license = { text = "Apache-2.0" }
keywords = ["ai", "rag", "agents"]
classifiers = [
  "Programming Language :: Python",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
]
authors = [{ name = "NVIDIA Corporation" }]
maintainers = [{ name = "NVIDIA Corporation" }]

[project.urls]
documentation = "https://docs.nvidia.com/nemo/agent-toolkit/latest/"
source = "https://github.com/NVIDIA/NeMo-Agent-Toolkit"

[project.optional-dependencies]
# Optional dependencies are things that users would want to install with NAT. i.e. `uv pip install "nvidia-nat[langchain]"`
# Keep sorted!!!
all = ["nvidia-nat-all"] # meta-package
a2a = ["nvidia-nat-a2a"]
adk = ["nvidia-nat-adk"]
agno = ["nvidia-nat-agno"]
autogen = ["nvidia-nat-autogen"]
crewai = ["nvidia-nat-crewai"]
data-flywheel = ["nvidia-nat-data-flywheel"]
huggingface = [
  "transformers[torch,accelerate]~=4.57",
]
ingestion = ["nvidia-nat-ingestion"] # meta-package
langchain = ["nvidia-nat-langchain"]
llama-index = ["nvidia-nat-llama-index"]
mcp = ["nvidia-nat-mcp"]
mem0ai = ["nvidia-nat-mem0ai"]
opentelemetry = ["nvidia-nat-opentelemetry"]
phoenix = ["nvidia-nat-phoenix"]
pii-defense = ["presidio-analyzer", "presidio-anonymizer"]
profiling = ["nvidia-nat-profiling"] # meta-package
ragaai = ["nvidia-nat-ragaai"]
mysql = ["nvidia-nat-mysql"]
redis = ["nvidia-nat-redis"]
s3 = ["nvidia-nat-s3"]
semantic-kernel = ["nvidia-nat-semantic-kernel"]
strands = ["nvidia-nat-strands"]
telemetry = [
  "nvidia-nat-data-flywheel",
  "nvidia-nat-opentelemetry",
  "nvidia-nat-phoenix",
  "nvidia-nat-ragaai",
  "nvidia-nat-weave",
]
test = ["nvidia-nat-test"]
vanna = ["nvidia-nat-vanna"]
weave = ["nvidia-nat-weave"]
zep-cloud = ["nvidia-nat-zep-cloud"]
openpipe-art = ["nvidia-nat-openpipe-art"]
nemo-customizer = ["nvidia-nat-nemo-customizer"]

examples = [
  "nat_adk_demo",
  "nat_agno_personal_finance",
  "nat_alert_triage_agent",
  "nat_autogen_demo",
  "nat_automated_description_generation",
  "nat_currency_agent_a2a",
  "nat_dpo_tic_tac_toe",
  "nat_email_phishing_analyzer",
  "nat_haystack_deep_research_agent",
  "nat_kaggle_mcp",
  "nat_math_assistant_a2a",
  "nat_math_assistant_a2a_protected",
  "nat_multi_frameworks",
  "nat_per_user_workflow",
  "nat_plot_charts",
  "nat_por_to_jiratickets",
  "nat_profiler_agent",
  "nat_react_benchmark_agent",
  "nat_retail_agent",
  "nat_rl_with_openpipe_art",
  "nat_router_agent",
  "nat_semantic_kernel_demo",
  "nat_sequential_executor",
  "nat_service_account_auth_mcp",
  "nat_simple_auth",
  "nat_simple_auth_mcp",
  "nat_simple_calculator",
  "nat_simple_calculator_custom_routes",
  "nat_simple_calculator_eval",
  "nat_simple_calculator_hitl",
  "nat_simple_calculator_mcp",
  "nat_simple_calculator_mcp_protected",
  "nat_simple_calculator_observability",
  "nat_simple_rag",
  "nat_simple_web_query",
  "nat_simple_web_query_eval",
  "nat_strands_demo",
  "nat_swe_bench",
  "nat_user_report",
]

# Optional dependency needed when use_gunicorn is set to true
gunicorn = [
  "gunicorn~=23.0",
]

# Optional dependency needed when async endpoints are needed
async_endpoints = [
  "aiosqlite~=0.21",
  "dask~=2026.1",
  "distributed~=2026.1",
  "sqlalchemy[asyncio]~=2.0",
]

# specific versions of shared packages within optional subpackages
nvidia_haystack = [
  "nvidia-haystack~=0.3",
]
litellm = [
  "litellm==1.74.9"
]
openai = [
  "openai~=1.106"
]

[tool.uv]
managed = true
config-settings = { editable_mode = "compat" }

# Only use override-dependencies as a temporary workaround across package version resolution.
#
# We currently have all packages part of the same workspace, which means a single lockfile is in
# use across all packages. However, subsets of packages currently conflict:
#
# nvidia-nat-openpipe-art <> nvidia-nat-crewai**
# - these packages conflict due to a litellm conflict
#
# nvidia-nat-ragaai <> nvidia-nat-strands**
# - these packages conflict due to a tenacity conflict
#
# All conflicts have specific resolutions below, namely conforming to the package which is kept
# in nvidia-nat-all. Packages denoted with ** are those which are preferred as part of an "all"
# distribution (thus included in both nvidia-nat-all and nvidia-nat[all])
#
# Long-term solutions include better package isolation and testing, which is on the roadmap.
#
override-dependencies = [
  # Override tenacity version to resolve conflict between ragaai-catalyst ("tenacity==8.3.0")
  #   and strands-agents-tools ("tenacity>=9.1.2")
  # tenacity 9.x maintains API compatibility with 8.x except for the statistics attribute (breaking change in 9.0.0)
  # See: https://github.com/jd/tenacity/releases
  "tenacity>=9.1.2,<10.0.0",

  # Override dependencies that nvidia-nat-openpipe-art would otherwise regress.
  # This keeps status-quo for other packages.
  "litellm==1.74.9",
  "openai>=1.106,<2.dev0",

  # Override pdfminer.six version (transitive dependency of polyfill-weave)
  "pdfminer.six>=20251229"
]

[tool.uv.sources]
# Workspace members
nvidia-nat-all = { workspace = true }
nvidia-nat-a2a = { workspace = true }
nvidia-nat-agno = { workspace = true }
nvidia-nat-adk = { workspace = true }
nvidia-nat-autogen = { workspace = true }
nvidia-nat-crewai = { workspace = true }
nvidia-nat-data-flywheel = { workspace = true }
nvidia-nat-ingestion = { workspace = true }
nvidia-nat-langchain = { workspace = true }
nvidia-nat-llama-index = { workspace = true }
nvidia-nat-mcp = { workspace = true }
nvidia-nat-mem0ai = { workspace = true }
nvidia-nat-mysql = { workspace = true }
nvidia-nat-opentelemetry = { workspace = true }
nvidia-nat-phoenix = { workspace = true }
nvidia-nat-profiling = { workspace = true }
nvidia-nat-ragaai = { workspace = true }
nvidia-nat-redis = { workspace = true }
nvidia-nat-s3 = { workspace = true }
nvidia-nat-semantic-kernel = { workspace = true }
nvidia-nat-strands = { workspace = true }
nvidia-nat-test = { workspace = true }
nvidia-nat-vanna = { workspace = true }
nvidia-nat-weave = { workspace = true }
nvidia-nat-zep-cloud = { workspace = true }
nvidia-nat-openpipe-art = { workspace = true }
nvidia-nat-nemo-customizer = { workspace = true }

# All examples here
nat_adk_demo = { path = "examples/frameworks/adk_demo", editable = true }
nat_agno_personal_finance = { path = "examples/frameworks/agno_personal_finance", editable = true }
nat_alert_triage_agent = { path = "examples/advanced_agents/alert_triage_agent", editable = true }
nat_autogen_demo = { path = "examples/frameworks/nat_autogen_demo", editable = true }
nat_automated_description_generation = { path = "examples/custom_functions/automated_description_generation", editable = true }
nat_currency_agent_a2a = { path = "examples/A2A/currency_agent_a2a", editable = true }
nat_dpo_tic_tac_toe = { path = "examples/finetuning/dpo_tic_tac_toe", editable = true }
nat_email_phishing_analyzer = { path = "examples/evaluation_and_profiling/email_phishing_analyzer", editable = true }
nat_haystack_deep_research_agent = { path = "examples/frameworks/haystack_deep_research_agent", editable = true }
nat_kaggle_mcp = { path = "examples/MCP/kaggle_mcp", editable = true }
nat_math_assistant_a2a = { path = "examples/A2A/math_assistant_a2a", editable = true }
nat_math_assistant_a2a_protected = { path = "examples/A2A/math_assistant_a2a_protected", editable = true }
nat_multi_frameworks = { path = "examples/frameworks/multi_frameworks", editable = true }
nat_per_user_workflow = { path = "examples/front_ends/per_user_workflow", editable = true }
nat_plot_charts = { path = "examples/custom_functions/plot_charts", editable = true }
nat_por_to_jiratickets = { path = "examples/HITL/por_to_jiratickets", editable = true }
nat_profiler_agent = { path = "examples/advanced_agents/profiler_agent", editable = true }
nat_react_benchmark_agent = { path = "examples/dynamo_integration/react_benchmark_agent", editable = true }
nat_retail_agent = { path = "examples/safety_and_security/retail_agent", editable = true }
nat_rl_with_openpipe_art = { path = "examples/finetuning/rl_with_openpipe_art", editable = true }
nat_router_agent = { path = "examples/control_flow/router_agent", editable = true }
nat_semantic_kernel_demo = { path = "examples/frameworks/semantic_kernel_demo", editable = true }
nat_sequential_executor = { path = "examples/control_flow/sequential_executor", editable = true }
nat_service_account_auth_mcp = { path = "examples/MCP/service_account_auth_mcp", editable = true }
nat_simple_auth = { path = "examples/front_ends/simple_auth", editable = true }
nat_simple_auth_mcp = { path = "examples/MCP/simple_auth_mcp", editable = true }
nat_simple_calculator = { path = "examples/getting_started/simple_calculator", editable = true }
nat_simple_calculator_custom_routes = { path = "examples/front_ends/simple_calculator_custom_routes", editable = true }
nat_simple_calculator_eval = { path = "examples/evaluation_and_profiling/simple_calculator_eval", editable = true }
nat_simple_calculator_hitl = { path = "examples/HITL/simple_calculator_hitl", editable = true }
nat_simple_calculator_mcp = { path = "examples/MCP/simple_calculator_mcp", editable = true }
nat_simple_calculator_mcp_protected = { path = "examples/MCP/simple_calculator_mcp_protected", editable = true }
nat_simple_calculator_observability = { path = "examples/observability/simple_calculator_observability", editable = true }
nat_simple_rag = { path = "examples/RAG/simple_rag", editable = true }
nat_simple_web_query = { path = "examples/getting_started/simple_web_query", editable = true }
nat_simple_web_query_eval = { path = "examples/evaluation_and_profiling/simple_web_query_eval", editable = true }
nat_strands_demo = { path = "examples/frameworks/strands_demo", editable = true }
nat_swe_bench = { path = "examples/evaluation_and_profiling/swe_bench", editable = true }
nat_user_report = { path = "examples/object_store/user_report", editable = true }


[tool.uv.workspace]
members = ["packages/*"]
exclude = ["packages/compat"]

[dependency-groups]
# Dependency groups are only for developers to aid in managing dependencies local to a dev machine.
dev = [
  "asgi-lifespan~=2.1",
  "galileo~=1.27",
  "beautifulsoup4~=4.13",
  "httpx-sse~=0.4",
  "ipython~=8.31",
  "langsmith", # this is currently a transitive dependency of nat, not specifying a version to avoid conflicts
  "myst-parser~=4.0",
  "nbconvert", # Version determined by jupyter
  "nbsphinx~=0.9",
  "nvidia-nat_test",
  "nvidia-sphinx-theme>=0.0.9",
  "pre-commit>=4.0,<5.0",
  "pytest~=8.3",
  "pytest-asyncio==0.24.*",
  "pytest-cov~=6.1",
  "pytest_httpserver==1.1.*",
  "pytest-timeout~=2.4",
  "python-docx~=1.1",
  "ruff~=0.12",
  "setuptools >= 64",
  "setuptools_scm>=8",
  "sphinx~=8.2",
  "sphinx-autoapi>=3.6",
  "sphinx-copybutton>=0.5",
  "sphinx_design~=0.6",
  "sphinx-mermaid",
  "sphinx-reredirects~=1.1",
  "tomlkit>=0.13.2",
  "twine~=6.0",
  "vale~=3.12",
  "yapf==0.43.*",
  # transitive dependencies
  "werkzeug>=3.1.5",
]

[project.entry-points.'nat.components']
nat_agents = "nat.agent.register"
nat_authentication = "nat.authentication.register"
nat_control_flow = "nat.control_flow.register"
nat_embedders = "nat.embedder.register"
nat_evaluators = "nat.eval.register"
nat_middleware = "nat.middleware.register"
nat_test_time_compute = "nat.experimental.test_time_compute.register"
nat_llms = "nat.llm.register"
nat_object_stores = "nat.object_store.register"
nat_observability = "nat.observability.register"
nat_retrievers = "nat.retriever.register"
nat_tools = "nat.tool.register"


[project.entry-points.'nat.front_ends']
nat_front_ends = "nat.front_ends.register"


[project.entry-points.'nat.registry_handlers']
nat_registry_handlers = "nat.registry_handlers.register"


[project.entry-points.'nat.cli']
configure = "nat.cli.commands.configure.configure:configure_command"
eval = "nat.cli.commands.evaluate:eval_command"
finetune = "nat.cli.commands.finetune:finetune_command"
info = "nat.cli.commands.info.info:info_command"
object-store = "nat.cli.commands.object_store.object_store:object_store_command"
optimize = "nat.cli.commands.optimize:optimizer_command"
red-team = "nat.cli.commands.red_teaming.red_teaming:red_team_command"
registry = "nat.cli.commands.registry.registry:registry_command"
sizing = "nat.cli.commands.sizing.sizing:sizing"
start = "nat.cli.commands.start:start_command"
uninstall = "nat.cli.commands.uninstall:uninstall_command"
validate = "nat.cli.commands.validate:validate_command"
workflow = "nat.cli.commands.workflow.workflow:workflow_command"


[project.scripts]
aiq = "nat.cli.main:run_cli_aiq_compat"
nat = "nat.cli.main:run_cli"

[tool.setuptools]
include-package-data = true

# List any markers that users would reasonably want to filter by.
# These show up when querying `pytest --markers`
[tool.pytest.ini_options]
markers = [
  "benchmark: Benchmarks",
  "integration: Integration tests which do not use mocks and may require external services",
  "requires_cache: Tests that require a reverse-proxy cache to be running",
  "slow: Slow tests",
]
filterwarnings = [
  # Work-around for a warning emitted by strands https://github.com/strands-agents/sdk-python/issues/1236
  "ignore:^These events have been moved to production.*strands:DeprecationWarning",
]
testpaths = ["tests", "examples/**/tests", "packages/**/tests"]
asyncio_mode = "auto"
asyncio_default_fixture_loop_scope = "session"

# Global timeout of 5 minutes per test to catch hanging tests.
# Individual tests can override with @pytest.mark.timeout(seconds) or disable with @pytest.mark.timeout(0)
timeout = 300


[tool.ruff]
# Set the maximum line length to 120.
line-length = 120
target-version = "py311"

[tool.ruff.lint]

select = [
  # The E, F, W categories are flake8 rules.
  "E",
  "F",
  "W",
  # Enable isort support
  "I",
  # Enable partial pylint support
  "PL",
  # Enable pyupgrade support
  "UP"
]

extend-ignore = [
  "PLC0415", # Allow lazy imports inside functions
  # Disable all the "too-*" checks, as they are too strict for our codebase
  "PLR0904", # too-many-public-methods
  "PLR0911", # too-many-return-statements
  "PLR0912", # too-many-branches
  "PLR0913", # too-many-arguments
  "PLR0914", # too-many-locals
  "PLR0915", # too-many-statements
  "PLR0917", # too-many-positional-arguments
  "PLR1702", # too-many-nested-blocks
  "PLR2004", # Magic value used in comparison this happens a lot within unittests
  "PLW0108", # unnecessary-lambda
  "PLW2901", # allow re-assigning a loop variable
]

[tool.ruff.lint.isort]
combine-as-imports = true
default-section = "third-party"
force-single-line = true
known-first-party = ["aiq", "nat", "nat_*", "_utils"]
known-third-party = [
  "agno",
  "autogen",
  "crewai",
  "google",
  "langchain",
  "llama_index",
  "mem0ai",
  "redis",
  "semantic_kernel",
  "zep_cloud",
]
order-by-type = true

# ===== yapf Config =====
[tool.yapf]
based_on_style = "pep8"
column_limit = 120
split_all_top_level_comma_separated_values = true
join_multiple_lines = true
indent_dictionary_value = true


# ===== pyright Config =====
[tool.pyright]
deprecateTypingAliases = true
reportInvalidTypeForm = "warning"