🏆 黑客松项目 - 基于NVIDIA官方NeMo Agent Toolkit构建的智能对话机器人,展示AI Agent的强大功能
本项目是为推广NVIDIA NeMo Agent Toolkit而开发的AI对话机器人示例,完全基于NVIDIA官方技术栈构建。系统集成了实时网络搜索、时间查询等功能,支持用户自定义OpenAI兼容的API接口,是学习和体验AI Agent技术的完美起点。
- 🤖 官方架构: 100%使用NVIDIA官方NeMo Agent Toolkit
- 🌐 实时搜索: 集成Tavily API,支持实时网络搜索
- ⏰ 时间查询: 获取当前日期和时间信息
- 🔧 灵活配置: 支持任何OpenAI兼容的API接口
- 🎨 现代界面: 官方UI,支持实时对话和流式响应
- 🚀 一键部署: 跨平台安装脚本,支持Windows/Linux/macOS
- 框架: Next.js 14 + TypeScript
- UI库: 官方NeMo-Agent-Toolkit-UI
- 特性: 实时聊天、主题切换、历史记录
- 核心: NVIDIA NeMo Agent Toolkit (AIQ)
- 工作流: React Agent
- 工具: Tavily搜索、时间查询
- 默认: Qwen模型
- 兼容: 任何OpenAI格式的API
- 自定义: 用户可配置API密钥、模型名称、base_url
- Python: 3.12+
- Node.js: 18+
- Git: 最新版本
- 操作系统: Windows 10+/macOS 10.15+/Ubuntu 20.04+
git clone https://github.com/HeKun-NVIDIA/hackathon_aiqtoolkit.git
cd hackathon_aiqtoolkit
安装完成后,您需要配置以下API密钥:
在install.sh
文件中185行左右,将Your API Key替换成你自己的Tavily API Key 来保证搜索功能正常
# 设置环境变量
export TAVILY_API_KEY=Your API Key
获取Tavily API密钥:
- 访问 Tavily官网
- 注册账户并获取免费API密钥
- 将密钥添加到环境变量中
编辑 install.sh
文件中154行左右,将Your API Key替换成你自己的Bailian API Key:
llms:
# 默认使用Bailian API (用户可修改)
default_llm:
_type: openai
model_name: "qwen-plus"
api_key: "Your API Key"
base_url: "https://dashscope.aliyuncs.com/compatible-mode/v1"
temperature: 0.7
max_tokens: 2048
支持的API提供商:
- 阿里云百炼平台Qwen系列:
https://bailian.console.aliyun.com/?tab=model#/model-market
- 其他: 任何OpenAI兼容的API
# 运行安装脚本
chmod +x install.sh
./install.sh
# 运行安装脚本
install.bat
# 启动服务
cd NeMo-Agent-Toolkit
./start.sh
# 停止服务
./stop.sh
- 前端界面: http://localhost:3000
- API文档: http://localhost:8001/docs
- 健康检查: http://localhost:8001/health
用户: 北京今天的天气怎么样,气温是多少?
AI: 今天北京天气晴朗,气温在18℃至31℃之间,当前温度约30℃,西南风3级,相对湿度43%,空气质量良好,体感温度舒适。白天最高气温可达31℃,夜间最低18℃,全天无降水,紫外线较强,建议做好防晒措施。
用户: 现在几点了?
AI: 现在是晚上11点17分。
用户: 帮我介绍一下NVIDIA Agent Intelligence Toolkit
AI: [搜索并介绍NVIDIA AIQ工具包的详细信息]
nvidia-nemo-agent-toolkit-hackathon/
├── configs/ # 配置文件
│ └── hackathon_config.yml # 主配置文件
├── external/ # 外部模块
│ └── aiqtoolkit-opensource-ui/ # 官方UI
├── docs/ # 文档和截图
│ └── ui_screenshot.png # 界面截图
├── src/ # 源代码
├── install.sh # Linux/macOS安装脚本
├── install.bat # Windows安装脚本
├── start.sh # 启动脚本
├── stop.sh # 停止脚本
└── README.md # 说明文档
在配置文件中添加新工具:
functions:
your_custom_tool:
_type: your_tool_type
description: "工具描述"
# 其他配置参数
workflow:
_type: react_agent
tool_names:
- internet_search
- current_datetime
- your_custom_tool
llm_name: default_llm
verbose: true
# 启用详细日志
aiq serve --config_file configs/hackathon_config.yml --verbose
# 检查端口占用
netstat -tlnp | grep :8001
# 使用不同端口
aiq serve --port 8002
- 检查
configs/hackathon_config.yml
中的API密钥配置 - 确认环境变量
TAVILY_API_KEY
已正确设置 - 验证API密钥的有效性和权限
# 清理缓存重新安装
uv cache clean
uv pip install -e . --force-reinstall
- 检查后端是否正常启动(访问 http://localhost:8001/health)
- 确认端口配置正确
- 检查防火墙设置
# 查看后端日志
tail -f logs/aiq.log
# 查看前端日志
cd external/aiqtoolkit-opensource-ui
npm run dev -- --verbose
本项目专为推广NVIDIA NeMo Agent Toolkit技术而开发,旨在:
- 🎯 展示AI Agent能力: 通过实际应用展示NVIDIA NeMo Agent Toolkit的强大功能
- 🚀 降低学习门槛: 提供完整的示例代码和详细文档,帮助开发者快速上手
- 🌟 促进技术交流: 为AI Agent技术爱好者提供学习和交流的平台
- 💡 激发创新思维: 鼓励开发者基于此项目创建更多创新应用
- ✅ 完全官方架构: 严格遵循NVIDIA官方技术规范
- ✅ 生产级质量: 包含完整的错误处理、日志记录和监控
- ✅ 易于扩展: 模块化设计,支持快速添加新功能
- ✅ 跨平台支持: 一套代码,多平台运行
🎯 让我们一起探索AI Agent的无限可能!
本项目展示了NVIDIA NeMo Agent Toolkit在实际应用中的强大能力,为AI Agent技术的普及和发展贡献力量。无论您是AI初学者还是资深开发者,都能从这个项目中获得有价值的学习体验。