让 PyMOL 听懂人话：Agent 自动安装 PyMolAI，并接入免费的 NVIDIA NIM + Kimi K2.6

让 PyMOL 听懂人话：Agent 自动安装 PyMolAI，并接入免费的 NVIDIA NIM + Kimi K2.6

目录

1. 前言
2. 为什么要用 PyMolAI
3. 为什么选择 NVIDIA NIM 免费 API
4. 为什么选择 Kimi K2.6
5. 安装前准备
6. Agent 自动安装提示词
7. 说明
8. 在 PyMolAI 中切换模型
9. 常见问题
10. 总结

1. 前言

PyMOL 是结构生物学中非常常用的分子可视化工具，但很多操作都需要记忆命令。比如给不同链设置不同颜色、显示 cartoon、突出活性位点、调整视角、导出图片等。

最近我尝试使用 ravishar313/PyMolAI，让 PyMOL 支持 AI 对话操作，并进一步接入 NVIDIA NIM 免费 API。实际测试中，moonshotai/kimi-k2.6表现非常好，速度快，理解强，比较适合 PyMolAI 这种“自然语言 -> PyMOL 命令 -> 执行”的工作流。

【PyMolAI 启动后的界面截图】

2. 为什么要用 PyMolAI

PyMolAI 的核心价值是：用自然语言控制 PyMOL。

例如，我们可以直接输入：

下载 1UBQ，并用 cartoon 模式显示

或者：

给每个亚基设置不同颜色

AI 会自动理解任务，并尝试调用 PyMOL 命令完成操作。

相比手动输入命令，PyMolAI 的优势是：

减少查命令时间 适合快速生成结构图 适合教学和演示 可以连续执行多步 PyMOL 操作 对不熟悉 PyMOL 命令的新手更友好

3. 为什么选择 NVIDIA NIM 免费 API

PyMolAI 默认接入 OpenRouter，下图中的都是收费模型。OpenRouter也有免费模型，但是免费的api key有次数限制，从未充值，或累计购买不足 10 美元 credits的，每天最多 50 次免费模型请求。

NVIDIA NIM API申请网站

https://build.nvidia.com/

NVIDIA NIM 提供了免费使用1年的api key，超过100个模型供选择，截至2026 年 6 月，NVIDIA NIM 仍然允许开发者免费申请 API Key。

如果你有购买的其他的api key也可以，在配置软件的ai部分时，在提示词中说清楚是哪家的api，agent会自行解决

4. 为什么选择 Kimi K2.6

我测试了多个模型后，NVIDIA NIM 中的：

moonshotai/kimi-k2.6

表现最好。

它的优势主要体现在：

响应速度快 理解 PyMOL 操作意图比较准确 多步任务执行更稳定 工具调用表现较好 执行成功后更容易给出总结

当然，我并没有测试太多模型，大家也可以自行测试哪个更好用一些。

5. 安装前准备

1. NVIDIA NIM API key
2. PyMolAI的包，从github下载PyMolAI的压缩包，或者让agent自己下载
3. Agent可以在vs code中安装codex、coplit、Cline等插件，小米的mimo注册就能领10元体验，足够用于安装PyMolAI了。
   vs code插件：Codex – OpenAI’s coding agent
   GitHub Copilot Chat（vs code内置）
   Xiaomi MiMo for Copilot Chat

6. Agent 自动安装提示词

1. PyMolAI安装：下面是我给 Agent 使用的提示词，目标是让它自动处理安装，而不是只给说明。

你是一个 Windows PowerShell 自动安装 agent。请自动帮我安装和配置 PyMolAI，不要只给说明，要实际执行命令、检查结果、处理常见错误。 项目地址： https://github.com/ravishar313/PyMolAI 任务： 1. 在 Windows 上自动安装 PyMolAI到D盘。 2. 检查本机是否安装了合适版本的conda和python，如果没有提示用户安装或用户授权后自动下载安装到D盘。 3. 如果用户没有提供本地PyMolAI的git包，需要从GitHub下载。 4. 如果Git没安装，提示用户需要安装 Git for Windows。使用 conda 创建环境，环境名为 pymolai，在pymolai环境中安装。 5. 安装 PyMolAI 所需依赖，不要安装VS Studio，太占硬盘空间了，用conda中的cmake编译。 6. 如果cmake编译失败，安装预编译的 PyMOL，用 PyPI 预编译 wheel，无需 C++ 编译器。 7. 最终能通过 PowerShell 运行： conda activate pymolai pymol 请自动执行检查、安装、报错处理、验证，并输出最终启动命令。

2. PyMolAI配置：pymol成功运行之后，再对其进行配置以使用NVIDIA NIM，以下是给 Agent 使用的提示词

请帮我修改当前 conda 环境中 PyMolAI 实际加载的后端代码，使其支持 NVIDIA NIM 的 OpenAI-compatible API。 要求： 1. 不要修改 API key，不要把 API key 写进文件。 2. 先运行以下命令定位实际加载文件： python -c "import pymol.ai.runtime as r; print(r.__file__)" python -c "import pymol.ai.openrouter_client as c; print(c.__file__)" 3. 修改前先备份 runtime.py 和 openrouter_client.py。 4. 在 runtime.py 中增加 PYMOL_AI_BACKEND=openai 后端。 5. 当 PYMOL_AI_BACKEND=openai 时，使用 OpenAI-compatible /v1/chat/completions，而不是 Claude SDK。 6. 支持以下环境变量： PYMOL_AI_BACKEND=openai PYMOL_AI_OPENAI_BASE_URL=https://integrate.api.nvidia.com/v1 NVIDIA_API_KEY NIM_API_KEY PYMOL_AI_OPENAI_API_KEY PYMOL_AI_DEFAULT_MODEL PYMOL_AI_MAX_TOKENS PYMOL_AI_MAX_STEPS 7. 保留原来的 OpenRouter / Claude SDK 后端，默认行为不要破坏。 8. 给 OpenAI-compatible 后端注册两个核心工具： run_pymol_command capture_viewer_snapshot 9. 让模型通过 tool calling 调用 PyMOL 命令，并把工具结果回传给模型继续生成最终回答。 10. 在 openrouter_client.py 中，让 parallel_tool_calls 参数可以通过环境变量关闭： PYMOL_AI_SEND_PARALLEL_TOOL_CALLS=0 11. 修改完成后运行语法检查： python -m py_compile runtime.py openrouter_client.py 12. 检查一下修改后的pymol能否打开，如果报错要自动解决。

3. 创建运行脚本：当pymolai能成功运行后，还需创建一键运行脚本，避免每次输入api key，给AI的提示词

请帮我为 Windows 创建一个调用power shell的“一键启动 PyMOL AI.bat”脚本。 1. 询问用户脚本保存位置 2. 询问用户储存NVIDIA NIM API KEY文件的路径 3. 有些电脑通过 .bat 双击启动 PowerShell 时，不会自动加载 Conda 初始化配置。可以将 conda activate pymolai 替换为 Conda 的绝对路径调用。 4. 默认环境变量 $env:PYMOL_AI_BACKEND = "openai" $env:PYMOL_AI_OPENAI_BASE_URL = "https://integrate.api.nvidia.com/v1" $env:PYMOL_AI_DEFAULT_MODEL = "moonshotai/kimi-k2.6" $env:PYMOL_AI_MAX_TOKENS = "4000" $env:PYMOL_AI_MAX_STEPS = "6" $env:PYMOL_AI_SCREENSHOT_VALIDATE_REQUIRED = "0" $env:PYMOL_AI_SEND_PARALLEL_TOOL_CALLS = "0" 5. 在脚本中用中文注明各参数的意义 6. 完成脚本编写后，运行脚本，并测试让pymolai执行“下载 1UBQ，并用 cartoon 模式显示”，自行处理报错。

7. 说明

PYMOL_AI_BACKEND=openai：使用 OpenAI-compatible 后端 PYMOL_AI_OPENAI_BASE_URL：NVIDIA NIM API 地址 NVIDIA_API_KEY：NVIDIA NIM API key PYMOL_AI_DEFAULT_MODEL：默认模型 PYMOL_AI_MAX_TOKENS：控制 token 上限 PYMOL_AI_MAX_STEPS：控制工具调用轮数 PYMOL_AI_SCREENSHOT_VALIDATE_REQUIRED=0：避免反复截图验证

手动测试，进入 PyMolAI 后对话框测试：

下载 1UBQ，并用 cartoon 模式显示

8. 在 PyMolAI 中切换模型

启动后，可以在 PyMolAI 对话框中输入：

/ai model moonshotai/kimi-k2.6

备用模型可以试：

/ai model nvidia/llama-3.3-nemotron-super-49b-v1.5

9. 常见问题

9.1 ERROR 404

在pymol中，新建pymol窗口之后，可能会报错：404 page not found（使用一键运行脚本，基本不出现）
解决方法是重新设置一下模型，在对话框输入

/ai model moonshotai/kimi-k2.6

9.2 AI 执行成功但不知道结束

有时 PyMOL 命令已经成功执行，但 AI 没有生成最终总结。一般没有影响，再问一下：“你是否完成任务”就行。

I completed the loop but did not receive a final answer from the model.

9.3 模型容易达到 max turn limit

我测试时遇到：

OpenAI-compatible backend reached max turn limit before producing a final answer.

这个报错表示：模型已经连续进行了多轮工具调用，但在达到 PYMOL_AI_MAX_STEPS 限制前，没有返回最终的文字总结。

解决方法，增加最大steps，或者在对话框中输入“继续任务”。

$env:PYMOL_AI_MAX_STEPS = "12"

10. 总结

本文使用的组合是：

PyMolAI：让 PyMOL 支持自然语言操作 Agent：自动处理安装和配置 NVIDIA NIM：提供免费 API moonshotai/kimi-k2.6：作为主力模型

实际测试下来，Kimi K2.6 在 PyMolAI 中表现很好，速度快、理解强、工具调用稳定，适合用来完成 PyMOL 中的分子可视化操作。