USDcodeNIM_MCP

NVIDIA NIM（NVIDIA智能模型）与Cursor IDE的集成，特别配置用于USD（通用场景描述）代码生成和验证。

概述

此仓库提供了工具和配置，通过模型上下文协议（MCP），将NVIDIA的专用USD代码模型集成到Cursor IDE中。它支持：

• 使用NVIDIA的专用USD代码模型进行USD代码生成
• 带有结构化反馈的自动代码验证
• 针对USD的最佳实践和建议
• 对生成的代码进行双重检查以查找错误和改进

快速开始

1. 从build.nvidia.com/nvidia/usdcode获取您的API密钥

2. 设置环境变量：

   $env:NIM_API_KEY = "your_api_key_here"  # Windows PowerShell
   

   set NIM_API_KEY=your_api_key_here  # Windows CMD
   

   export NIM_API_KEY=your_api_key_here  # Linux/Mac
   

3. 安装依赖项：

   pip install -r scripts/requirements_nim.txt
   

4. 测试集成：

   python scripts/nim_mcp_server.py test
   

仓库结构

USDcodeNIM_MCP/
├── scripts/
│   ├── nim_mcp_server.py      # 用于Cursor集成的MCP服务器包装器
│   ├── nim_direct_client.py    # 独立使用的直接客户端
│   ├── validate_with_nim.py   # 用于验证USD Python文件的脚本
│   ├── requirements_nim.txt    # Python依赖项
│   └── README_NIM.md          # 快速参考指南
├── config/
│   └── cursor_mcp_config.json  # Cursor MCP配置模板
├── docs/
│   └── NVIDIA_NIM_Integration_Guide.md  # 完整设置文档
└── README.md                  # 此文件

功能

1. MCP服务器集成

通过Model Context Protocol实现与Cursor IDE的完全集成，提供USD代码生成和验证工具。

2. 直接客户端

独立的Python客户端，可用于脚本、CI/CD流水线或其他工具。

3. 代码验证

自动验证USD Python代码，并提供结构化的反馈（错误、警告、建议）。

4. 代码生成

使用NVIDIA的专用模型从自然语言提示生成USD代码。

使用方法

在Cursor中使用（推荐）✨

配置完成后，可以直接在Cursor的聊天框中使用NVIDIA NIM：

生成USD代码：

生成一个创建红色立方体网格的USD Python脚本

验证USD代码：

使用NVIDIA NIM验证这个USD脚本

（打开一个USD文件时）

询问USD问题：

创建USD舞台的最佳方式是什么？

独立验证

python scripts/validate_with_nim.py 路径/to/usd_script.py
python scripts/validate_with_nim.py 路径/to/usd_script.py --context "资产验证脚本"

代码生成

python scripts/nim_direct_client.py "创建一个带有立方体网格的USD舞台"

程序化使用

from scripts.nim_direct_client import NIMDirectClient
import asyncio

async def main():
    client = NIMDirectClient()
    
    # 生成代码
    code = await client.generate_code("创建一个USD舞台")
    print(code)
    
    # 验证代码
    result = await client.validate_code(code)
    print(result)
    
    await client.close()

asyncio.run(main())

Cursor MCP配置

要与Cursor IDE集成：

1. 从build.nvidia.com/nvidia/usdcode获取您的API密钥
2. 编辑您的MCP配置：C:\Users\<用户名>\.cursor\mcp.json
3. 从config/cursor_mcp_config.json添加NVIDIA NIM服务器配置
4. 更新配置中的路径以匹配您的仓库位置
5. 在env.NIM_API_KEY字段中设置您的API密钥
6. 设置模型名称：nvidia/usdcode-llama-3.1-70b-instruct
7. 完全重启Cursor

快速设置：

{
  "mcpServers": {
    "nvidia-nim": {
      "command": "python",
      "args": ["C:\\路径\\到\\USDcodeNIM_MCP\\scripts\\nim_mcp_server.py"],
      "env": {
        "NIM_API_KEY": "your_api_key_here",
        "NIM_MODEL": "nvidia/usdcode-llama-3.1-70b-instruct"
      }
    }
  }
}

参见docs/NVIDIA_NIM_Integration_Guide.md获取详细的设置说明。 参见QUICK_REFERENCE.md获取快速参考指南。

要求

• Python 3.10+
• httpx库（通过pip install -r scripts/requirements_nim.txt安装）
• 从build.nvidia.com获取的NVIDIA NIM API密钥
• Cursor IDE（用于MCP集成）

文档

• 快速参考：QUICK_REFERENCE.md（从此处开始！）
• 脚本指南：scripts/README_NIM.md
• 完整指南：docs/NVIDIA_NIM_Integration_Guide.md
• 配置：config/cursor_m_配置.json

资源

• NVIDIA NIM USD代码模型
• NVIDIA NIM文档
• MCP协议规范
• USD Python API文档

许可

该项目按原样提供，用于与NVIDIA NIM服务集成。

---

版本：1.0.0
最后更新：2025-11-21