Pi 集成架构
本文档介绍 OpenClaw 如何与 pi-coding-agent 及其同级包(pi-ai、pi-agent-core、pi-tui)集成,以驱动其 AI 智能体能力。
概述
OpenClaw 使用 pi SDK,将一个 AI 编码智能体嵌入到其消息 Gateway 网关架构中。OpenClaw 不会将 pi 作为子进程启动,也不会使用 RPC 模式,而是通过createAgentSession() 直接导入并实例化 pi 的 AgentSession。这种内嵌方式提供了:
- 对会话生命周期和事件处理的完全控制
- 自定义工具注入(消息、沙箱、渠道特定操作)
- 按渠道/上下文定制系统提示词
- 支持分支/压缩的会话持久化
- 支持故障切换的多账户凭证配置轮换
- 与提供商无关的模型切换
包依赖
| Package | 用途 |
|---|---|
pi-ai | 核心 LLM 抽象:Model、streamSimple、消息类型、提供商 API |
pi-agent-core | 智能体循环、工具执行、AgentMessage 类型 |
pi-coding-agent | 高层 SDK:createAgentSession、SessionManager、AuthStorage、ModelRegistry、内置工具 |
pi-tui | 终端 UI 组件(用于 OpenClaw 的本地 TUI 模式) |
文件结构
src/agents/tools 下,例如:
- Discord 插件操作运行时文件
- Slack 插件操作运行时文件
- Telegram 插件操作运行时文件
- WhatsApp 插件操作运行时文件
核心集成流程
1. 运行内嵌智能体
主入口是pi-embedded-runner/run.ts 中的 runEmbeddedPiAgent():
2. 创建会话
在runEmbeddedAttempt()(由 runEmbeddedPiAgent() 调用)内部,使用了 pi SDK:
3. 事件订阅
subscribeEmbeddedPiSession() 会订阅 pi 的 AgentSession 事件:
message_start/message_end/message_update(流式文本/思考)tool_execution_start/tool_execution_update/tool_execution_endturn_start/turn_endagent_start/agent_endcompaction_start/compaction_end
4. 提示输入
完成设置后,会向会话发送提示:images 将其传递给当前轮次。它不会重新扫描更早的历史轮次来重新注入图像负载。
工具架构
工具流水线
- 基础工具:pi 的
codingTools(read、bash、edit、write) - 自定义替换:OpenClaw 将 bash 替换为
exec/process,并为沙箱定制 read/edit/write - OpenClaw 工具:消息、浏览器、画布、会话、cron、Gateway 网关等
- 渠道工具:Discord/Telegram/Slack/WhatsApp 特定操作工具
- 策略过滤:按配置、提供商、智能体、群组、沙箱策略过滤工具
- Schema 规范化:为兼容 Gemini/OpenAI 的特殊行为而清理 schema
- AbortSignal 包装:对工具进行包装,以遵守 abort signal
工具定义适配器
pi-agent-core 的AgentTool 与 pi-coding-agent 的 ToolDefinition 具有不同的 execute 签名。pi-tool-definition-adapter.ts 中的适配器负责桥接二者:
工具拆分策略
splitSdkTools() 通过 customTools 传递全部工具:
系统提示词构建
系统提示词在buildAgentSystemPrompt()(system-prompt.ts)中构建。它会组装一个完整提示词,其中包含 Tooling、工具调用风格、安全护栏、OpenClaw CLI 参考、Skills、文档、工作区、沙箱、消息、回复标签、语气、静默回复、心跳、运行时元数据,以及在启用时包含的 Memory 和 Reactions,还包括可选的上下文文件和额外系统提示词内容。用于子智能体的最小提示词模式会对这些部分进行裁剪。
系统提示词会在会话创建后通过 applySystemPromptOverrideToSession() 应用:
会话管理
会话文件
会话以 JSONL 文件存储,并具有树状结构(通过 id/parentId 链接)。pi 的SessionManager 负责持久化:
guardSessionManager() 对其进行包装,以确保工具结果安全。
会话缓存
session-manager-cache.ts 会缓存 SessionManager 实例,以避免重复解析文件:
历史记录限制
limitHistoryTurns() 会根据渠道类型(私信 与 群组)裁剪对话历史。
压缩
当上下文溢出时会触发自动压缩。常见的溢出特征包括request_too_large、context length exceeded、input exceeds the maximum number of tokens、input token count exceeds the maximum number of input tokens、input is too long for the model,以及 ollama error: context length exceeded。compactEmbeddedPiSessionDirect() 负责处理手动
压缩:
认证与模型解析
凭证配置
OpenClaw 维护一个凭证配置存储,为每个提供商保存多个 API 密钥:模型解析
故障切换
当配置了回退时,FailoverError 会触发模型回退:
Pi 扩展
OpenClaw 会加载自定义 Pi 扩展,以实现专门行为:压缩保护
src/agents/pi-hooks/compaction-safeguard.ts 为压缩增加护栏,包括自适应 token 预算,以及工具失败和文件操作摘要:
上下文裁剪
src/agents/pi-hooks/context-pruning.ts 实现了基于缓存 TTL 的上下文裁剪:
流式传输与分块回复
分块切块
EmbeddedBlockChunker 负责将流式文本整理为离散的回复块:
Thinking/Final 标签剥离
流式输出会经过处理,以剥离<think>/<thinking> 块并提取 <final> 内容:
回复指令
会解析并提取[[media:url]]、[[voice]]、[[reply:id]] 等回复指令:
错误处理
错误分类
pi-embedded-helpers.ts 会对错误进行分类,以便采取适当处理:
Thinking level 回退
如果某个 thinking level 不受支持,则会回退:沙箱集成
启用沙箱模式时,工具和路径都会受到约束:提供商特定处理
Anthropic
- 拒绝魔术字符串清理
- 连续角色的轮次校验
- 严格的上游 Pi 工具参数校验
Google/Gemini
- 插件自有的工具 schema 清理
OpenAI
- 面向 Codex 模型的
apply_patch工具 - thinking level 降级处理
TUI 集成
OpenClaw 还提供本地 TUI 模式,可直接使用 pi-tui 组件:与 Pi CLI 的关键差异
| 方面 | Pi CLI | OpenClaw 内嵌模式 |
|---|---|---|
| 调用方式 | pi 命令 / RPC | 通过 createAgentSession() 使用 SDK |
| 工具 | 默认编码工具 | 自定义 OpenClaw 工具套件 |
| 系统提示词 | AGENTS.md + prompts | 按渠道/上下文动态生成 |
| 会话存储 | ~/.pi/agent/sessions/ | ~/.openclaw/agents/<agentId>/sessions/(或 $OPENCLAW_STATE_DIR/agents/<agentId>/sessions/) |
| 认证 | 单一凭证 | 支持轮换的多配置 |
| 扩展 | 从磁盘加载 | 编程方式 + 磁盘路径 |
| 事件处理 | TUI 渲染 | 基于回调(onBlockReply 等) |
未来考虑
潜在需要重构的领域:- 工具签名对齐:当前需要在 pi-agent-core 与 pi-coding-agent 签名之间做适配
- Session manager 包装:
guardSessionManager增加了安全性,但也提高了复杂度 - 扩展加载:可以更直接地使用 pi 的
ResourceLoader - 流式处理器复杂度:
subscribeEmbeddedPiSession已变得较大 - 提供商特殊行为:存在许多提供商特定代码路径,而 pi 未来可能可以统一处理这些问题
测试
Pi 集成覆盖以下测试套件:src/agents/pi-*.test.tssrc/agents/pi-auth-json.test.tssrc/agents/pi-embedded-*.test.tssrc/agents/pi-embedded-helpers*.test.tssrc/agents/pi-embedded-runner*.test.tssrc/agents/pi-embedded-runner/**/*.test.tssrc/agents/pi-embedded-subscribe*.test.tssrc/agents/pi-tools*.test.tssrc/agents/pi-tool-definition-adapter*.test.tssrc/agents/pi-settings.test.tssrc/agents/pi-hooks/**/*.test.ts
src/agents/pi-embedded-runner-extraparams.live.test.ts(启用OPENCLAW_LIVE_TEST=1)