NVIDIA/NeMo-Agent-Toolkit

The NVIDIA NeMo Agent toolkit is an open-source library for efficiently connecting and optimizing teams of AI agents.

NVIDIA NeMo Agent Toolkit – 简介

NeMo Agent Toolkit (NAT) 是 NVIDIA 推出的开源 Python 库,帮助您构建、运行和改进 AI 智能体——使用大型语言模型(LLM)进行推理、调用工具和执行操作的软件。该工具包位于您选择的 LLM(例如 NVIDIA Nemotron-3.5)和您已使用的智能体框架(LangChain、LlamaIndex、CrewAI、Semantic Kernel 等)之间。它增加了一层检测、可观测性和优化,使智能体能够以企业级规模部署,并具有更好的性能和更简单的调试。


核心能力(根据 README 描述)

领域 工具包提供的内容
智能体构建 针对主流智能体库的框架无关适配器;可复用组件;内置 UI 用于与智能体聊天。
检测 令牌级性能分析、运行时遥测、OpenTelemetry 追踪,以及 LangSmith 集成,实现端到端可观测性。
优化 离线评估工具、超参数和提示词优化器、基于强化学习的微调,以及 Agent Performance Primitives(并行执行、推测分支、优先级路由)。
运行时智能 实验性的 Dynamo 集成,可推断请求延迟敏感性并动态调整缓存/路由提示。
协议支持 实现模型上下文协议(MCP)以将工具/智能体作为服务提供,以及用于安全分布式团队的智能体间(A2A)协议。
可扩展性 公共插件 API,使第三方包(例如 Redis、Tavily、Memanto)无需修改框架特定代码即可接入。

典型工作流(Hello-World 示例)

  1. 定义 YAML 工作流,列出工具(例如 Wikipedia 搜索)、要使用的 LLM(通过 NVIDIA NIM 的 Nemotron 模型)以及智能体类型(react_agent)。
  2. 使用 nat CLI 运行:
    nat run --config_file workflow.yml --input "List five subspecies of Aardvarks"
    
  3. 工具包处理:
    • 调用 LLM,
    • 调用 Wikipedia 工具,
    • 分析执行过程,
    • 返回最终答案。

适用人群

  • 机器学习工程师,希望为 LangChain 风格的智能体添加生产就绪的可观测性和性能调节功能。
  • 数据科学家,希望系统性地评估和调整提示词,或针对特定智能体任务微调 LLM。
  • 企业,需要安全、经过认证的 A2A 通信和兼容 MCP 的内部工具服务。

安装与入口点

  • 从 PyPI 安装核心包:pip install nvidia-nat。
  • 可选扩展包引入框架特定集成,例如 pip install "nvidia-nat[langchain]"。
  • 主要命令行工具:nat(运行工作流、提供 MCP 服务器、评估、配置遥测等)。

生态系统与社区

  • 官方文档:https://docs.nvidia.com/nemo/agent-toolkit/latest/
  • 开源插件(Redis、Tavily 等)位于独立仓库,但遵循相同的公共 API。
  • 遥测是可选加入的,并且刻意保持最小化以尊重隐私。
  • 路线图包括更多 RL 自动化、KV 缓存优化和更广泛的语言支持。

简而言之: NeMo Agent Toolkit 是一个真正的、面向生产的 SDK,用于跨多种框架构建、观察和优化基于 LLM 的 AI 智能体,具有 NVIDIA 特定的性能增强,并强调可扩展性和可观测性。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目