Pi 集成架构
本文档介绍 OpenClaw 如何与 pi-coding-agent 及其同级软件包(pi-ai、pi-agent-core、pi-tui)集成,以提供其 AI 智能体能力。
概览
OpenClaw 使用 pi SDK 将 AI 编码智能体嵌入到其消息 Gateway 网关架构中。OpenClaw 不会将 pi 作为子进程启动,也不会使用 RPC 模式,而是通过createAgentSession() 直接导入并实例化 pi 的 AgentSession。这种内嵌方式提供了:
- 对会话生命周期和事件处理的完全控制
- 自定义工具注入(消息、沙箱、渠道特定操作)
- 按渠道 / 上下文定制系统提示词
- 支持分支 / 压缩的会话持久化
- 带故障转移的多账户凭证配置轮换
- 与提供商无关的模型切换
软件包依赖
| 软件包 | 用途 |
|---|---|
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 下,例如:
extensions/discord/src/actions/runtime*.tsextensions/slack/src/action-runtime.tsextensions/telegram/src/action-runtime.tsextensions/whatsapp/src/action-runtime.ts
核心集成流程
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_endauto_compaction_start/auto_compaction_end
4. 发送提示
完成设置后,会向会话发送提示:images 将其传入该轮。它不会重新扫描较早的历史轮次来重新注入图像负载。
工具架构
工具流水线
- 基础工具:pi 的
codingTools(read、bash、edit、write) - 自定义替换:OpenClaw 用
exec/process替换 bash,并为沙箱定制 read / edit / write - OpenClaw 工具:消息、浏览器、画布、会话、cron、Gateway 网关 等
- 渠道工具:Discord / Telegram / Slack / WhatsApp 特定操作工具
- 策略过滤:按配置、提供商、智能体、群组、沙箱策略过滤工具
- 模式归一化:清理模式以适配 Gemini / OpenAI 的特殊行为
- AbortSignal 包装:包装工具以遵循中止信号
工具定义适配器
pi-agent-core 的AgentTool 与 pi-coding-agent 的 ToolDefinition 在 execute 签名上不同。pi-tool-definition-adapter.ts 中的适配器用于桥接这一差异:
工具拆分策略
splitSdkTools() 会通过 customTools 传入所有工具:
系统提示词构建
系统提示词在buildAgentSystemPrompt()(system-prompt.ts)中构建。它会组装完整提示词,包含工具、工具调用风格、安全护栏、OpenClaw CLI 参考、Skills、文档、工作区、沙箱、消息、回复标签、语音、静默回复、心跳、运行时元数据等部分,并在启用时包含 Memory 和 Reactions,以及可选的上下文文件和额外系统提示词内容。为子智能体使用的最小提示词模式会对各部分进行裁剪。
系统提示词会在会话创建后通过 applySystemPromptOverrideToSession() 应用:
会话管理
会话文件
会话是具有树状结构(通过 id / parentId 关联)的 JSONL 文件。Pi 的SessionManager 负责持久化:
guardSessionManager() 对其进行包装,以保证工具结果安全。
会话缓存
session-manager-cache.ts 会缓存 SessionManager 实例,以避免重复解析文件:
历史限制
limitHistoryTurns() 会根据渠道类型(私信 与群组)裁剪对话历史。
压缩
上下文溢出时会触发自动压缩。compactEmbeddedPiSessionDirect() 负责手动压缩:
身份验证与模型解析
凭证配置
OpenClaw 维护一个凭证配置存储,为每个提供商保存多个 API 密钥:模型解析
故障转移
配置了回退时,FailoverError 会触发模型故障转移:
Pi 扩展
OpenClaw 会加载自定义的 pi 扩展,以实现专门行为:压缩保护
src/agents/pi-hooks/compaction-safeguard.ts 会为压缩添加护栏,包括自适应 token 预算,以及工具失败和文件操作摘要:
上下文裁剪
src/agents/pi-hooks/context-pruning.ts 实现了基于 cache-TTL 的上下文裁剪:
流式传输与分块回复
分块处理
EmbeddedBlockChunker 负责将流式文本管理为离散的回复块:
思考 / 最终标签剥离
流式输出会经过处理,以去除<think> / <thinking> 块并提取 <final> 内容:
回复指令
会解析并提取诸如[[media:url]]、[[voice]]、[[reply:id]] 之类的回复指令:
错误处理
错误分类
pi-embedded-helpers.ts 会对错误进行分类,以便进行适当处理:
思考级别回退
如果某个思考级别不受支持,它会回退:沙箱集成
启用沙箱模式时,工具和路径都会受到约束:提供商特定处理
Anthropic
- 清理拒绝魔法字符串
- 针对连续角色的轮次验证
- Claude Code 参数兼容性
Google / Gemini
- 轮次顺序修复(
applyGoogleTurnOrderingFix) - 工具模式净化(
sanitizeToolsForGoogle) - 会话历史净化(
sanitizeSessionHistory)
OpenAI
- 面向 Codex 模型的
apply_patch工具 - 思考级别降级处理
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 的签名之间进行适配
- 会话管理器包装:
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)