AgentEnv:開源的大規模智能體環境運行平台
AgentEnv(AENV)是一個基於Firecracker微虛擬機和OCI鏡像的開源平台,用於大規模運行智能體環境,為Kimi K3的強化學習訓練提供支持。它具備快照啓動(<50ms)、增量快照、內存氣球、高I/O性能等特性,並提供了CLI和E2B兼容API。
AgentEnv(簡稱AENV)是一個開源的智能體環境運行平台,專門為大規模運行智能體環境而設計,為Kimi K3的強化學習訓練提供動力。該平台基於Firecracker微虛擬機技術,結合OCI兼容的容器鏡像,通過overlaybd按需加載,實現了跨物理機的大規模環境部署。本地磁盤作為有界緩存,保留熱數據並自動淘汰冷數據,使得鏡像總大小可以超過磁盤容量,同時保持整個集羣的快速啓動,無需預熱每台主機。
在性能方面,AENV的獨特之處在於其快照機制。基於快照的環境啓動或恢復時間低於50毫秒,暫停時間低於100毫秒。空閒環境可以迅速釋放CPU和內存資源,待新任務到來時再快速恢復,從而顯著降低空閒成本。平台支持對內存和文件系統進行增量快照,即使在高磁盤修改量下,完成時間也低於100毫秒。正在運行的環境可以分叉(fork)成多個獨立的沙箱,用於並行智能體工作流。快照可持久化到S3兼容的對象存儲或共享分佈式文件系統,防止數據丟失。
在I/O性能方面,AENV通過ublk提供高性能I/O,同時共享主機頁面緩存,用於存儲和內存快照數據。內存氣球技術將可回收的客户機內存歸還給主機,隨着環境運行時間增長和差異變大,依然維持高密度和高性能。
平台對硬件和軟件有特定要求:需要Linux內核6.8及以上版本,並且擁有/dev/kvm訪問權限以運行Firecracker微VM。快速部署方面,提供了Ubuntu 24.04的一鍵安裝腳本和Docker兩種方式。安裝腳本會同時安裝服務器和aenv CLI,並將服務器作為systemd服務啓動。Docker方式則通過docker-setup.sh腳本準備環境,然後拉取並運行服務器容器。
使用aenv CLI可以進行模板管理、沙箱生命週期管理(啓動、暫停、恢復、刪除)以及超時設置等操作。例如,通過“aenv pull ubuntu:22.04 --name ubuntu”拉取模板,然後用“aenv start ubuntu”啓動一個沙箱並附加交互式shell。CLI支持表格和JSON兩種輸出格式。
AENV還提供了與E2B兼容的HTTP API,只需將E2B_API_URL指向AENV服務器,即可使用標準的E2B Python或TypeScript SDK,無需修改代碼。在部署方面,支持Docker Compose和Kubernetes集羣部署,以及從源代碼構建。
該項目採用MIT許可證,目前在GitHub上已獲得超過1100顆星和92次分叉,社區活躍。