USD代码生成与验证工具

NVIDIA NIM集成工具,提供USD代码生成、自动化验证及最佳实践建议,适用于3D场景描述开发。

  • by jph2
  • Python
  • 本地部署
  • 6 星标
  • 更新于
在 GitHub 查看 jph2/USDcodeNIM_MCP

使用说明


arys_schema_version:'1.2' id:b1c4c143-c78e-405f-9e99-40eec1d23dc2 标题:USDcodeNIM_MCP 类型:技术 状态:活动 信任级别:2 创建时间:2026-02-17T09:42:16Z 最后修改:'2026-02-17T09:42:16Z'

版本: 1.0.0 | 日期: 16.02.2026 | 时间: 12:00 | Globalid:20260216_1200_USDCode NIM_MCP_batch

标签块: #framework_integration#stage#v_model#aas_integration#cursor#vscode#validation#specializes#mcp_pprotocol#usd_core#opencode#openusd#semantic_governance#export#workflow_automation#ai_coding_agents#quality_assurance#deterministic_workflows#best_pactics

美元代码NIM_MCP

用于Cursor IDE的NVIDIA NIM(NVIDIA智能模型)集成,专门配置用于USD(通用场景描述)代码生成和验证。

概述

此存储库提供工具和配置,通过模型上下文协议(MCP)将NVIDIA的专用USD代码模型集成到Cursor IDE中。它能够:

  • 美元代码生成 使用NVIDIA的专用美元代码模型
  • 自动代码验证 结构化反馈
  • 美元具体建议 最佳实践
  • 复核 为错误和改进生成的代码

快速开始

  1. 获取API密钥

    1. 从 build.nvidia.com/nvidia/usdcode

    导航到NVIDIA NIM平台,在型号目录中搜索“USD代码”或“usdcode”。一旦您找到美元代码模型,请登录或创建一个帐户来访问您的API密钥。API密钥将在验证后显示在模型的页面上。

    1. 设置环境变量:
    $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
    
    1. 安装依赖项:
    pip install -r scripts/requirements_nim.txt
    
    1. 测试集成:
    python scripts/nim_mcp_server.py test
    

2.配置光标

打开光标并单击 齿轮图标 在右上角访问设置。选择 光标设置,然后导航到 工具和MCP。单击 添加新的MCP服务器 并将以下JSON配置粘贴到文件中:

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

确保更新路径以匹配您的存储库位置并替换 your_api_key_here 使用您的实际API密钥。

  1. 重新启动游标 完全关闭并重新打开游标。

  2. 验证连接 设置→ 工具和MCP→ Check nvidia-nim 是 绿色 ✅

💡 文档层次结构:


仓库结构

USDcodeNIM_MCP/
├── scripts/
│   ├── nim_mcp_server.py      # MCP server wrapper for Cursor integration
│   ├── nim_direct_client.py    # Direct client for standalone use
│   ├── validate_with_nim.py   # Validation script for USD Python files
│   ├── requirements_nim.txt    # Python dependencies
│   └── README_NIM.md          # Quick reference guide
├── config/
│   └── cursor_mcp_config.json  # Cursor MCP configuration template
├── docs/
│   └── NVIDIA_NIM_Integration_Guide.md  # Complete setup documentation
└── README.md                  # This file

特性

1.MCP服务器集成

通过模型上下文协议实现完整的Cursor IDE集成,提供USD代码生成和验证工具。

2.直接客户

用于脚本、CI/CD管道或其他工具的独立Python客户端。

3.代码验证

通过结构化反馈(错误、警告、建议)自动验证USD Python代码。

4.代码生成

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

用法

在游标中使用(推荐)✨

配置后,直接在Cursor的聊天中使用NVIDIA NIM:

生成美元代码:

Generate a USD Python script that creates a red cube mesh

验证美元代码:

Validate this USD script using NVIDIA NIM

(打开美元文件)

询问美元问题:

What's the best way to create a USD stage?

独立验证

python scripts/validate_with_nim.py path/to/usd_script.py
python scripts/validate_with_nim.py path/to/usd_script.py --context "Asset validation script"

代码生成

python scripts/nim_direct_client.py "Create a USD stage with a cube mesh"

程序化使用

from scripts.nim_direct_client import NIMDirectClient
import asyncio

async def main():
    client = NIMDirectClient()
    
    # Generate code
    code = await client.generate_code("Create a USD stage")
    print(code)
    
    # Validate code
    result = await client.validate_code(code)
    print(result)
    
    await client.close()

asyncio.run(main())

光标MCP配置

要与Cursor IDE集成,请执行以下操作:

  1. 获取API密钥 从 build.nvidia.com/nvidia/usdcode
  2. 编辑您的MCP配置: C:\Users\<username>\.cursor\mcp.json
  3. 添加NVIDIA NIM服务器 配置从 config/cursor_mcp_config.json
  4. 更新路径 在配置中匹配您的存储库位置
  5. 设置API密钥 在 env.NIM_API_KEY 领域
  6. 设置型号名称: nvidia/usdcode-llama-3.1-70b-instruct
  7. 重新启动游标 完全

快速设置:

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

需要更多帮助?

需求

  • Python 3.10+
  • httpx 库(通过安装 pip install -r scripts/requirements_nim.txt)
  • 来自的NVIDIA NIM API密钥 build.nvidia.com
  • 游标IDE(用于MCP集成)

文档

文档层次结构:

  1. README.md (你在这里)→ 快速入门基本设置说明
  2. 快速引用.md → 更多深度、故障排除、示例和常用命令
  3. 完整指南 → 全面的文档,详细描述了所有内容

其他资源:

资源

许可证

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


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