AX 是 Google 的开源智能体编排运行时,官方将其描述为高吞吐、声明式的编排器,用于在集群中运行数十亿个自主智能体工作负载。它构建在 Agent Substrate 之上,借助其提供沙箱化执行能力,设计目标为单集群运行数十亿任务。若使用过 Kubernetes,AX 的体验会较为相似。
用途方面,AX 面向智能体这一新型工作负载:它们既不是无状态微服务,也不是运行到完成的批处理任务,而是会累积状态、需要严格隔离、调用模型 API 与工具服务器,且若无人监控可能陷入循环烧钱。AX 提供三个小型原语以声明式方式处理这些需求。所有内容均以 ax.io/v1alpha1 清单表达,并通过单条命令应用。使用流程大致为:声明任务后应用它,观察其启动,并从智能体的视角进行查看。
主要能力包括:声明式定义智能体任务(含工作区与模型规格);将任务以沙箱化 actor 形式调度到 Agent Substrate;支持为任务设置 CPU/内存限制;提供挂起(suspend)与恢复(resume);支持通过 ax ssh 进行调试;命令行接口采用类似 kubectl 的形态,包含 apply、get、describe、watch、delete 以及若干智能体专用动词,并可与 kubectx 配合使用。ax 通过 gRPC 与控制平面通信,且会跟随当前活跃的 Kubernetes 上下文,在后台解析并隧道连接到该集群的控制平面。
使用方式上,部署前需满足若干前置条件:一个已安装 Agent Substrate 的 Kubernetes 集群;安装 ko(可通过 brew install ko)以及集群可拉取的容器镜像仓库。需先按 Substrate README 的说明安装 Agent Substrate,它会部署在 ate-system 命名空间,并在 api.ate-system.svc.cluster.local:443 暴露 Control API,AX 默认在此查找。安装 CLI 后,ax 二进制会放入 $(go env GOPATH)/bin,需确保该目录在 PATH 中。随后部署控制平面:该步骤会先部署 Redis,再用 ko 构建并部署控制平面镜像,所有内容落在 ax-system 命名空间。最后可运行第一个任务;若想端到端查看完整生命周期,可运行 ./demo.sh,它会应用自定义工作区、等待就绪、通过 ax ssh 执行命令并挂起任务。
适用场景为在 Kubernetes 集群中规模化运行自主智能体工作负载。限制包括:项目仍处于重度开发阶段,核心概念、协议与规范正在积极完善,在稳定版本发布前很可能引入重大破坏性变更;运行 AX 前必须先让 Agent Substrate 在集群中运行。项目采用 Apache License 2.0 许可,详细内容见仓库 LICENSE 文件。