pyproject.toml

[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",
  "datasets~=4.0", # workaround for uv's solver choosing different versions of datasets based on sys_platform
  "expandvars~=1.0",
  "fastapi~=0.119.0",
  "httpx~=0.27",
  "jinja2~=3.1",
  "jsonpath-ng~=1.7",
  "mcp~=1.13",
  "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",
  "openpyxl~=3.1",
  "optuna~=4.4.0",
  "pip>=24.3.1",
  "pkce==1.0.3",
  "pkginfo~=1.12",
  "platformdirs~=4.3",
  "pydantic~=2.11",
  "pymilvus~=2.5",
  "python-dotenv~=1.1.1",
  "PyYAML~=6.0",
  "ragas~=0.2.14", # 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",
]
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"]
crewai = ["nvidia-nat-crewai"]
data-flywheel = ["nvidia-nat-data-flywheel"]
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"]
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_haystack_deep_research_agent",
  "nat_alert_triage_agent",
  "nat_automated_description_generation",
  "nat_currency_agent_a2a",
  "nat_dpo_tic_tac_toe",
  "nat_email_phishing_analyzer",
  "nat_multi_frameworks",
  "nat_plot_charts",
  "nat_por_to_jiratickets",
  "nat_profiler_agent",
  "nat_react_benchmark_agent",
  "nat_router_agent",
  "nat_rl_with_openpipe_art",
  "nat_semantic_kernel_demo",
  "nat_sequential_executor",
  "nat_service_account_auth_mcp",
  "nat_simple_auth",
  "nat_simple_auth_mcp",
  "nat_per_user_workflow",
  "nat_simple_web_query",
  "nat_simple_web_query_eval",
  "nat_simple_calculator",
  "nat_math_assistant_a2a",
  "nat_simple_calculator_custom_routes",
  "nat_simple_calculator_eval",
  "nat_simple_calculator_mcp",
  "nat_simple_calculator_observability",
  "nat_simple_calculator_hitl",
  "nat_simple_rag",
  "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~=2025.11",
  "distributed~=2025.11",
  "sqlalchemy[asyncio]~=2.0",
]

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

[tool.uv]
managed = true
config-settings = { editable_mode = "compat" }
override-dependencies = [
  # TODO: remove this when ragaai-catalsty supports tenacity version ~=9.0
  # 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"
]


[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-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_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_haystack_deep_research_agent = { path = "examples/frameworks/haystack_deep_research_agent", editable = true }
nat_alert_triage_agent = { path = "examples/advanced_agents/alert_triage_agent", editable = true }
nat_automated_description_generation = { path = "examples/custom_functions/automated_description_generation", editable = true }
nat_email_phishing_analyzer = { path = "examples/evaluation_and_profiling/email_phishing_analyzer", editable = true }
nat_multi_frameworks = { path = "examples/frameworks/multi_frameworks", 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_router_agent = { path = "examples/control_flow/router_agent", editable = true }
nat_rl_with_openpipe_art = { path = "examples/finetuning/rl_with_openpipe_art", 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_per_user_workflow = { path = "examples/front_ends/per_user_workflow", editable = true }
nat_simple_calculator = { path = "examples/getting_started/simple_calculator", editable = true }
nat_math_assistant_a2a = { path = "examples/A2A/math_assistant_a2a", 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_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_httpserver==1.1.*",
  "pytest-asyncio==0.24.*",
  "pytest-cov~=6.1",
  "pytest~=8.3",
  "python-docx~=1.1.0",
  "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",
  "tomlkit~=0.13.2",
  "twine~=6.0",
  "vale~=3.12",
  "yapf==0.43.*",
]

[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.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"


[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",
  "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"