SuperAPI AI 工具集

setup-rfd4_proteina-remote-tool

智能体与模型

查看项目

更新于 10-27 06:20

技能介绍

用途

RFD4-Proteina 是一个远程工具提供方(provider)的部署技能,用于在 Linux + CUDA GPU 主机上安装、启动、验证与排查 RFD4-Proteina 蛋白质设计工具,并可选地通过 ToolUniverse Connect 中继对外分享。已在 NVIDIA GB10(aarch64、cc 12.1、128 GB 统一内存)、驱动 580.173.02 上通过该提供方自身路径验证。

主要能力

  • 按远程工具标准建立两个隔离环境:MCP 服务器运行于 ToolUniverse 环境,模型运行于由固定 lock 文件构建的独立环境,两者不可互相导入。
  • 提供 tu remote setup 分步执行:preflight、environment、weights、smoke;smoke 步骤加载模型、发送请求、断言产出 *_generated.cif,并把实测峰值 VRAM 记录到 ~/.cache/tooluniverse/measured/rfd4_proteina.json。
  • 提供契约检查脚本 scripts/remote_validation/setup_skill_preflight.py,支持 --check-provider-env、--live、--check-connect-prereqs 等选项。
  • 支持本地 MCP 端点 http://127.0.0.1:8062/mcp,工具名为 rfd4_proteina_design。
  • 支持通过 ToolUniverse Connect 中继分享(tu remote share),默认使用浏览器设备授权,无需复制粘贴密钥。

使用方式

  1. 前置条件:ToolUniverse 仓库根目录、Linux、Python 3.12.3、CUDA GPU;provider 数据、权重、缓存与凭据不得放入 Git;绑定 loopback,非 loopback 绑定需 TOOLUNIVERSE_API_TOKEN。
  2. 先运行标准库契约检查,再创建 .venvs/rfd4_proteina 虚拟环境并安装 -e . 与 fastmcp。
  3. 在 TOOLUNIVERSE_RFD4_PROTEINA_PYTHON 指定路径创建空解释器,由 tu remote setup --only environment 从 requirements.lock 以 --no-deps 填充。
  4. 设置 TOOLUNIVERSE_RFD4_PROTEINA_PYTHON,可选设置 RFD4_PROTEINA_HF_HOME(默认 ~/.cache/tooluniverse/rfd4_proteina-hf),然后执行 tu remote setup rfd4_proteina。
  5. 本地启动:python -m tooluniverse.remote.rfd4_proteina.rfd4_proteina_tool,用 fastmcp Client 列出工具确认包含 rfd4_proteina_design。
  6. 分享:tu remote login 后每次执行 tu remote share rfd4_proteina;可用 --name、--workers 覆盖默认值;tu remote check、tu remote run 分别用于就绪检查与仅本地前台运行。
  7. Connect 中继需安装固定 SHA-256 的 tuplatform-connect wheel,并用 tu doctor、tu serve 验证。

典型请求

操作 rfd4_proteina_design,参数包括 input_cif(带注释的 mmCIF,注释即设计规范)、num_replicates、seed。成功时返回 n_designs、含 design_id、链序列与 n_atoms 的 designs 数组、耗时、peak_reserved_gib、weights_held_by;return_structures 为 true 时才返回设计 mmCIF 文本。证据测量所用的输入位于 src/tooluniverse/remote/rfd4_proteina/smoke/monomer.cif。

适用场景与限制

适用于在单张 CUDA 卡上私有部署并验证 RFD4-Proteina 推理服务。限制:需 CUDA GPU,不面向 CPU;建议单 worker,第二个在途请求会在锁上排队;manifest 的 VRAM 数据基于 100 残基测得,上游设计上限 1024 tokens,更长输入未实测;refolding 指标不计算;仅验证了无条件单体模式,六种条件化设计形态未运行;设计质量未验证,无条件输入产生带电重复螺旋;公开出版、跨用户隔离、并发、恢复与长输入均未测试,应保持私有部署;该实现未经过 Platform 矩阵验证。