abshkbh/arrakis

A fully customizable and self-hosted sandboxing solution for AI agent code execution and computer use. It features out-of-the-box support for backtracking, a simple REST API and Python SDK, automatic port forwarding, and secure MicroVM isolation. Perfect for safely running, testing, and backtracking multi-step agent workflows.

GAI 中文摘要

Arrakis 是一个基于 Go 语言构建的开源沙盒解决方案，旨在为 AI Agent 提供安全、可完全自定义且可自托管的代码执行与计算机操作环境。该项目通过微型虚拟机（MicroVM）技术实现环境隔离，有效解决 AI 生成代码可能带来的系统安全风险，同时支持复杂任务的快照与回溯功能。

该平台提供开箱即用的快照与恢复功能，允许 AI Agent 在执行多步工作流时保存并随时回溯至历史状态。通过内置的 REST API、Python SDK 及 MCP 服务器，用户可以便捷地远程管理沙盒环境、上传文件及执行代码。系统具备自动端口转发能力，能够让用户或 Agent 无缝访问沙盒内的 GUI 界面或浏览器环境。基于 MicroVM 的设计确保了每个沙盒实例在隔离的 Linux 环境下运行，提供了极高的安全性保障。

该项目适用于需要运行 AI Agent、开发自动化编码工作流或进行计算机使用（Computer Use）实验的开发者及研究人员，特别适合处理涉及复杂试错逻辑的任务。

⭐

790

Stars

🔱

Forks

👁

Watchers

📋

Issues

GoAGPL-3.0创建于 2024/8/4更新于昨天

在 GitHub 上查看

README

由 Gemini 翻译整理

Arrakis

简介

AI Agent 可能会生成恶意代码或存在 Bug 的代码，从而攻击其运行的主机系统。

许多 Agent 具有复杂的多步计划来达成目标，并能从回溯到中间状态的能力中获益。

Arrakis 提供了一个安全、高度可定制且自托管的解决方案，用于创建和管理代码执行与计算机操作（Computer Use）的沙盒。它通过**快照与恢复（snapshot-and-restore）**功能实现了对回溯的原生支持。

安全设计：每个沙盒都运行在 MicroVM 中。
功能完备：每个沙盒内部运行 Ubuntu，并在启动时自带代码执行服务和 VNC 服务器。
丰富的接口：通过 REST API、Python SDK py-arrakis 以及 MCP 服务器，客户端（无论是人类还是 AI Agent）都可以编程方式创建沙盒、上传文件并在沙盒内执行代码。
自动端口转发：自动设置并管理从自托管公共服务器到内部沙盒的端口转发，即客户端无需额外配置即可轻松访问沙盒 GUI（包括用于计算机操作的 Chrome 浏览器）。
原生支持回溯：AI Agent 在执行过程中可以对沙盒进行快照，随后通过恢复快照回溯到之前的精确状态。这意味着沙盒内产生的任何进程、修改的文件等都将完全还原。这对于基于蒙特卡洛树搜索（MCTS）的 Agent 或解释复杂的 Agent 执行流程非常有用。

演示

观看 Claude 如何通过 MCP 使用 Arrakis 实时编写 Google Docs 的克隆版本。它甚至会对沙盒进行快照以检查进度。

安装设置

先决条件

cloud-hypervisor 仅支持在 Linux 机器上使用 /dev/kvm 进行虚拟化。因此，我们仅支持 Linux 系统。
通过运行以下命令检查主机是否启用了虚拟化：
```
stat /dev/kvm
```

GCP 设置

按照 GCP Setup 中的说明在 GCE VM 上设置 Arrakis。

使用预构建版本快速安装

您可以利用我们的 setup.sh 脚本和预构建二进制文件轻松安装 Arrakis：

curl -sSL https://raw.githubusercontent.com/abshkbh/arrakis/main/setup/setup.sh | bash
ls arrakis-prebuilt

运行 arrakis-restserver

现在我们拥有了一个包含所有二进制文件和已拉取镜像的文件夹。我们始终需要先运行 arrakis-restserver：
```
cd arrakis-prebuilt
sudo ./arrakis-restserver
```

使用 CLI 或 py-arrakis

您可以使用 CLI 或 py-arrakis 来创建和管理虚拟机：
```
cd arrakis-prebuilt
./arrakis-client start -n agent-sandbox
```

快速开始

SDK

Arrakis 附带了一个 Python SDK py-arrakis，让您可以无缝创建、管理和与虚拟机进行交互。

安装 SDK：
```
pip install py-arrakis
```
按照 Usage 中的说明在 Linux 机器上运行 arrakis-restserver，或者从官方发布页面下载预构建二进制文件。
使用 py-arrakis 与 arrakis-restserver 进行交互。

运行不可信代码：

# 将其替换为 `arrakis-restserver` 运行的 ip:port
sandbox_manager = SandboxManager('http://127.0.0.1:7000')

# 启动一个新沙盒
with sb as sandbox_manager.start_sandbox('agent-sandbox'):
  sb.run_cmd('echo hello world')

# 上下文退出时，沙盒 `sb` 会自动销毁

快照与恢复沙盒：

# 启动沙盒并向文件写入一些数据
sandbox_name = 'agent-sandbox'
sandbox = sandbox_manager.start_sandbox(sandbox_name)
sandbox.run_cmd("echo 'test data before snapshot' > /tmp/testfile")
snapshot_id = sandbox.snapshot("initial-state")
sandbox.run_cmd("echo 'test data after snapshot' > /tmp/testfile")

# 恢复到快照状态
sandbox.restore(snapshot_id)