自今年 8 月推出以来,Prime Agent 下载量已突破 30 万次,处理 Token 数超过 8 万亿。今日,经过彻底重构的 Rust 版 Prime Agent 正式发布,旨在提供更快速、简洁且可靠的体验。

在两周时间内,Prime Agent 协调由 2000 多个智能体组成的群体,在 10000 多个 Prime Sandboxes 中完成了从 TypeScript 到 Rust 的自我重写,消耗了来自 Prime Inference GLM-5.3 端点的超 2000 亿个 Token。此次重写不仅是对多智能体能力的压力测试,也验证了其支持大规模智能体集群、自主研究和强化学习的基础设施。

为何选择 Rust

TypeScript 虽有助于快速迭代,但其运行时类型检查缺失、垃圾回收开销以及单事件循环对 CPU 密集型任务的限制,成为性能瓶颈。Rust 的引入带来了以下优势:

  • 性能:原生代码无需垃圾回收,显著提升了内存效率和启动速度。
  • 并发:Rust 的 Send 和 Sync 特性确保了线程间数据移动和共享的安全性,支持守护进程同时处理模型输出流、工具调用及多客户端服务。
  • 编译时保证:所有权机制和严格的 Clippy lint 规则在运行前排除大量潜在 Bug,这对于由智能体编写大部分代码的场景尤为重要。

此外,重写带来了代码模块化、Windows 支持、会话崩溃隔离及更一致的守护进程协议等架构红利。

智能体如何自我重写

为实现最小化人类干预的自主重写,团队建立了严格的验证机制,确保功能对等性:

  • TUI 对等性:通过差分测试对比 TypeScript 和 Rust 二进制文件的终端渲染帧,覆盖启动、菜单、工具调用等用户流程。
  • Harness 对等性:确保两者从相同输入产生相同的会话转录本和模型请求。
  • 协议对等性:检查守护进程协议中的所有消息类型,确保 API 一致性。
  • 功能对等性:智能体逐个组件审计 TypeScript 产品,分类标记匹配状态。

重写工作由根智能体编排,任务被划分为拓扑顺序,并通过“规划者-实现者-审查者-验证者”的四智能体流程执行。审查或验证失败的任务将被退回,只有全部通过后才会合并代码。这种基于有限状态机的编排模式已被构建到 Prime Agent 中,以实现工作流程的可复用性。

架构重构与性能优化

Rust 移植推动了代码库的深度重构:

  • 模块化:代码拆分为九个 Crate,最大源文件从 TypeScript 的 15000 行降至 2500 行左右。
  • 隔离:每个会话在独立工作进程中运行,故障互不影响,且支持重启后重新附加。
  • 平台抽象:通过特定接口抽象传输和进程控制,简化了 Windows 等平台的适配。
  • 目录驱动模型:模型和插件列表移至运行时获取的目录,支持不发布新版本即可推送更新。

在功能对等实现后,团队利用基准测试框架进行“爬山法”性能优化。智能体在三天内进行了超过 144 次实验,合并了 69 次性能提升更改。结果显示,Rust 版本在多项指标上显著优于 TypeScript 版本:

  • 冷启动:输入就绪延迟从 736.1ms 降至 51.9ms,提速 14.18 倍。
  • 热启动:输入就绪延迟从 552.3ms 降至 41.4ms,提速 13.34 倍。
  • 内存占用:大会话内存从 1130.0MB 降至 237.3MB,减少 4.76 倍。
  • 安装大小:从 172.1MB 降至 59.6MB,减小 2.89 倍。

与 Claude Code、Codex CLI 等竞品相比,Prime Agent (Rust) 在首屏绘制、打字时间及内存占用上均表现出显著优势。

下一步计划

随着 Rust 端口完成,Prime Agent 将加速能力发展,并开放背后的多智能体工作流程。未来,Prime Agent 将更紧密融入 Prime Intellect 生态系统,支持云智能体集群、推理、追踪等全栈功能。

此次发布还包括原生支持 Windows (Beta) 及 Homebrew 安装。Prime Agent 保持开源,用户可通过以下命令安装:

curl -fsSL https://app.primeintellect.ai/prime-agent/install.sh | sh

Windows 用户可使用:

irm https://app.primeintellect.ai/prime-agent/install.ps1 | iex