Pi 集成架构

本文档描述了 OpenClaw 如何集成 pi-coding-agent 及其兄弟包（pi-ai、pi-agent-core、pi-tui）来驱动 AI agent 能力。

概述

OpenClaw 使用 Pi SDK 将 AI 编程 agent 嵌入到消息网关架构中。它不是把 Pi 作为子进程启动或使用 RPC 模式，而是直接导入并通过 createAgentSession() 实例化 Pi 的 AgentSession。这种嵌入式方案带来了以下优势：

  • 对会话生命周期和事件处理的完全控制
  • 自定义工具注入（消息、沙箱、渠道专属操作）
  • 按渠道/上下文定制系统提示词
  • 支持分支/压缩的会话持久化
  • 多账号认证配置轮换与故障转移
  • 与提供商无关的模型切换

包依赖

{
  "@mariozechner/pi-agent-core": "0.49.3",
  "@mariozechner/pi-ai": "0.49.3",
  "@mariozechner/pi-coding-agent": "0.49.3",
  "@mariozechner/pi-tui": "0.49.3"
}
包用途
pi-ai核心 LLM 抽象：Model、streamSimple、消息类型、提供商 API
pi-agent-coreAgent 循环、工具执行、AgentMessage 类型
pi-coding-agent高层 SDK：createAgentSession、SessionManager、AuthStorage、ModelRegistry、内置工具
pi-tui终端 UI 组件（用于 OpenClaw 的本地 TUI 模式）

文件结构

src/agents/
├── pi-embedded-runner.ts          # 从 pi-embedded-runner/ 重新导出
├── pi-embedded-runner/
│   ├── run.ts                     # 主入口：runEmbeddedPiAgent()
│   ├── run/
│   │   ├── attempt.ts             # 单次尝试逻辑与会话设置
│   │   ├── params.ts              # RunEmbeddedPiAgentParams 类型
│   │   ├── payloads.ts            # 从运行结果构建响应载荷
│   │   ├── images.ts              # 视觉模型图片注入
│   │   └── types.ts               # EmbeddedRunAttemptResult
│   ├── abort.ts                   # Abort 错误检测
│   ├── cache-ttl.ts               # 缓存 TTL 跟踪，用于上下文裁剪
│   ├── compact.ts                 # 手动/自动压缩逻辑
│   ├── extensions.ts              # 加载嵌入运行的 Pi 扩展
│   ├── extra-params.ts            # 提供商专属的流参数
│   ├── google.ts                  # Google/Gemini 轮次排序修正
│   ├── history.ts                 # 历史记录限制（私聊 vs 群聊）
│   ├── lanes.ts                   # 会话/全局命令通道
│   ├── logger.ts                  # 子系统日志器
│   ├── model.ts                   # 通过 ModelRegistry 解析模型
│   ├── runs.ts                    # 活跃运行跟踪、中止、队列
│   ├── sandbox-info.ts            # 系统提示词的沙箱信息
│   ├── session-manager-cache.ts   # SessionManager 实例缓存
│   ├── session-manager-init.ts    # 会话文件初始化
│   ├── system-prompt.ts           # 系统提示词构建器
│   ├── tool-split.ts              # 将工具分为 builtIn 和 custom
│   ├── types.ts                   # EmbeddedPiAgentMeta, EmbeddedPiRunResult
│   └── utils.ts                   # ThinkLevel 映射、错误描述
├── pi-embedded-subscribe.ts       # 会话事件订阅/分发
├── pi-embedded-subscribe.types.ts # SubscribeEmbeddedPiSessionParams
├── pi-embedded-subscribe.handlers.ts # 事件处理器工厂
├── pi-embedded-subscribe.handlers.lifecycle.ts
├── pi-embedded-subscribe.handlers.types.ts
├── pi-embedded-block-chunker.ts   # 流式分块回复
├── pi-embedded-messaging.ts       # 消息工具发送跟踪
├── pi-embedded-helpers.ts         # 错误分类、轮次验证
├── pi-embedded-helpers/           # 辅助模块
├── pi-embedded-utils.ts           # 格式化工具
├── pi-tools.ts                    # createOpenClawCodingTools()
├── pi-tools.abort.ts              # 工具的 AbortSignal 包装
├── pi-tools.policy.ts             # 工具白名单/黑名单策略
├── pi-tools.read.ts               # Read 工具定制
├── pi-tools.schema.ts             # 工具 schema 规范化
├── pi-tools.types.ts              # AnyAgentTool 类型别名
├── pi-tool-definition-adapter.ts  # AgentTool -> ToolDefinition 适配器
├── pi-settings.ts                 # 设置覆盖
├── pi-extensions/                 # 自定义 Pi 扩展
│   ├── compaction-safeguard.ts    # 压缩保护扩展
│   ├── compaction-safeguard-runtime.ts
│   ├── context-pruning.ts         # 基于缓存 TTL 的上下文裁剪扩展
│   └── context-pruning/
├── model-auth.ts                  # 认证配置解析
├── auth-profiles.ts               # 配置存储、冷却、故障转移
├── model-selection.ts             # 默认模型解析
├── models-config.ts               # models.json 生成
├── model-catalog.ts               # 模型目录缓存
├── context-window-guard.ts        # 上下文窗口校验
├── failover-error.ts              # FailoverError 类
├── defaults.ts                    # DEFAULT_PROVIDER, DEFAULT_MODEL
├── system-prompt.ts               # buildAgentSystemPrompt()
├── system-prompt-params.ts        # 系统提示词参数解析
├── system-prompt-report.ts        # 调试报告生成
├── tool-summaries.ts              # 工具描述摘要
├── tool-policy.ts                 # 工具策略解析
├── transcript-policy.ts           # 对话记录验证策略
├── skills.ts                      # 技能快照/提示词构建
├── skills/                        # 技能子系统
├── sandbox.ts                     # 沙箱上下文解析
├── sandbox/                       # 沙箱子系统
├── channel-tools.ts               # 渠道专属工具注入
├── openclaw-tools.ts              # OpenClaw 专属工具
├── bash-tools.ts                  # exec/process 工具
├── apply-patch.ts                 # apply_patch 工具（OpenAI）
├── tools/                         # 各工具实现
│   ├── browser-tool.ts
│   ├── canvas-tool.ts
│   ├── cron-tool.ts
│   ├── discord-actions*.ts
│   ├── gateway-tool.ts
│   ├── image-tool.ts
│   ├── message-tool.ts
│   ├── nodes-tool.ts
│   ├── session*.ts
│   ├── slack-actions.ts
│   ├── telegram-actions.ts
│   ├── web-*.ts
│   └── whatsapp-actions.ts
└── ...

核心集成流程

1. 运行嵌入式 Agent

主入口是 pi-embedded-runner/run.ts 中的 runEmbeddedPiAgent()：

import { runEmbeddedPiAgent } from "./agents/pi-embedded-runner.js";

const result = await runEmbeddedPiAgent({
  sessionId: "user-123",
  sessionKey: "main:whatsapp:+1234567890",
  sessionFile: "/path/to/session.jsonl",
  workspaceDir: "/path/to/workspace",
  config: openclawConfig,
  prompt: "Hello, how are you?",
  provider: "anthropic",
  model: "claude-sonnet-4-20250514",
  timeoutMs: 120_000,
  runId: "run-abc",
  onBlockReply: async (payload) => {
    await sendToChannel(payload.text, payload.mediaUrls);
  },
});

2. 创建会话

在 runEmbeddedAttempt()（由 runEmbeddedPiAgent() 调用）内部使用 Pi SDK：

import {
  createAgentSession,
  DefaultResourceLoader,
  SessionManager,
  SettingsManager,
} from "@mariozechner/pi-coding-agent";

const resourceLoader = new DefaultResourceLoader({
  cwd: resolvedWorkspace,
  agentDir,
  settingsManager,
  additionalExtensionPaths,
});
await resourceLoader.reload();

const { session } = await createAgentSession({
  cwd: resolvedWorkspace,
  agentDir,
  authStorage: params.authStorage,
  modelRegistry: params.modelRegistry,
  model: params.model,
  thinkingLevel: mapThinkingLevel(params.thinkLevel),
  tools: builtInTools,
  customTools: allCustomTools,
  sessionManager,
  settingsManager,
  resourceLoader,
});

applySystemPromptOverrideToSession(session, systemPromptOverride);

3. 事件订阅

subscribeEmbeddedPiSession() 订阅 Pi 的 AgentSession 事件：

const subscription = subscribeEmbeddedPiSession({
  session: activeSession,
  runId: params.runId,
  verboseLevel: params.verboseLevel,
  reasoningMode: params.reasoningLevel,
  toolResultFormat: params.toolResultFormat,
  onToolResult: params.onToolResult,
  onReasoningStream: params.onReasoningStream,
  onBlockReply: params.onBlockReply,
  onPartialReply: params.onPartialReply,
  onAgentEvent: params.onAgentEvent,
});

处理的事件包括：

  • message_start / message_end / message_update（流式文本/思考）
  • tool_execution_start / tool_execution_update / tool_execution_end
  • turn_start / turn_end
  • agent_start / agent_end
  • auto_compaction_start / auto_compaction_end

4. 发送提示

设置完成后，向会话发送提示：

await session.prompt(effectivePrompt, { images: imageResult.images });

SDK 会处理完整的 agent 循环：发送到 LLM、执行工具调用、流式输出响应。

图片注入仅作用于当前提示：OpenClaw 从当前提示中加载图片引用，通过 images 参数传入该轮次。它不会重新扫描历史轮次来重新注入图片载荷。

工具架构

工具流水线

  1. 基础工具：Pi 的 codingTools（read、bash、edit、write）
  2. 自定义替换：OpenClaw 用 exec/process 替换 bash，为沙箱定制 read/edit/write
  3. OpenClaw 工具：消息、浏览器、canvas、会话、定时任务、gateway 等
  4. 渠道工具：Discord/Telegram/Slack/WhatsApp 专属操作工具
  5. 策略过滤：按配置、提供商、agent、群组、沙箱策略过滤工具
  6. Schema 规范化：为 Gemini/OpenAI 的特殊要求清理 schema
  7. AbortSignal 包装：工具被包装以支持中止信号

工具定义适配器

pi-agent-core 的 AgentTool 与 pi-coding-agent 的 ToolDefinition 有不同的 execute 签名。pi-tool-definition-adapter.ts 中的适配器负责桥接：

export function toToolDefinitions(tools: AnyAgentTool[]): ToolDefinition[] {
  return tools.map((tool) => ({
    name: tool.name,
    label: tool.label ?? name,
    description: tool.description ?? "",
    parameters: tool.parameters,
    execute: async (toolCallId, params, onUpdate, _ctx, signal) => {
      // pi-coding-agent 的签名与 pi-agent-core 不同
      return await tool.execute(toolCallId, params, signal, onUpdate);
    },
  }));
}

工具拆分策略

splitSdkTools() 将所有工具都通过 customTools 传入：

export function splitSdkTools(options: { tools: AnyAgentTool[]; sandboxEnabled: boolean }) {
  return {
    builtInTools: [], // 留空。我们覆盖所有内置工具
    customTools: toToolDefinitions(options.tools),
  };
}

这样可以确保 OpenClaw 的策略过滤、沙箱集成和扩展工具集在所有提供商之间保持一致。

系统提示词构建

系统提示词在 buildAgentSystemPrompt()（system-prompt.ts）中构建。它组装一个完整的提示词，包含工具、工具调用风格、安全护栏、OpenClaw CLI 参考、技能、文档、工作区、沙箱、消息、回复标签、语音、静默回复、心跳、运行时元数据等部分，启用时还包括记忆和反应，以及可选的上下文文件和额外系统提示词内容。子 agent 使用的最小提示模式会裁剪部分内容。

提示词在会话创建后通过 applySystemPromptOverrideToSession() 应用：

const systemPromptOverride = createSystemPromptOverride(appendPrompt);
applySystemPromptOverrideToSession(session, systemPromptOverride);

会话管理

会话文件

会话是带树形结构的 JSONL 文件（通过 id/parentId 关联）。Pi 的 SessionManager 负责持久化：

const sessionManager = SessionManager.open(params.sessionFile);

OpenClaw 使用 guardSessionManager() 进行包装，增加工具结果的安全保护。

会话缓存

session-manager-cache.ts 缓存 SessionManager 实例，避免重复解析文件：

await prewarmSessionFile(params.sessionFile);
sessionManager = SessionManager.open(params.sessionFile);
trackSessionManagerAccess(params.sessionFile);

历史记录限制

limitHistoryTurns() 根据渠道类型（私聊 vs 群聊）裁剪对话历史。

压缩

上下文溢出时会触发自动压缩。compactEmbeddedPiSessionDirect() 处理手动压缩：

const compactResult = await compactEmbeddedPiSessionDirect({
  sessionId, sessionFile, provider, model, ...
});

认证与模型解析

认证配置

OpenClaw 维护一个认证配置存储，每个提供商可以有多个 API 密钥：

const authStore = ensureAuthProfileStore(agentDir, { allowKeychainPrompt: false });
const profileOrder = resolveAuthProfileOrder({ cfg, store: authStore, provider, preferredProfile });

失败时配置会自动轮换，并带有冷却跟踪：

await markAuthProfileFailure({ store, profileId, reason, cfg, agentDir });
const rotated = await advanceAuthProfile();

模型解析

import { resolveModel } from "./pi-embedded-runner/model.js";

const { model, error, authStorage, modelRegistry } = resolveModel(
  provider,
  modelId,
  agentDir,
  config,
);

// 使用 Pi 的 ModelRegistry 和 AuthStorage
authStorage.setRuntimeApiKey(model.provider, apiKeyInfo.apiKey);

故障转移

FailoverError 在配置了回退方案时触发模型切换：

if (fallbackConfigured && isFailoverErrorMessage(errorText)) {
  throw new FailoverError(errorText, {
    reason: promptFailoverReason ?? "unknown",
    provider,
    model: modelId,
    profileId,
    status: resolveFailoverStatus(promptFailoverReason),
  });
}

Pi 扩展

OpenClaw 加载自定义 Pi 扩展来实现特殊行为：

压缩保护

src/agents/pi-extensions/compaction-safeguard.ts 为压缩添加护栏，包括自适应 token 预算以及工具失败和文件操作摘要：

if (resolveCompactionMode(params.cfg) === "safeguard") {
  setCompactionSafeguardRuntime(params.sessionManager, { maxHistoryShare });
  paths.push(resolvePiExtensionPath("compaction-safeguard"));
}

上下文裁剪

src/agents/pi-extensions/context-pruning.ts 实现了基于缓存 TTL 的上下文裁剪：

if (cfg?.agents?.defaults?.contextPruning?.mode === "cache-ttl") {
  setContextPruningRuntime(params.sessionManager, {
    settings,
    contextWindowTokens,
    isToolPrunable,
    lastCacheTouchAt,
  });
  paths.push(resolvePiExtensionPath("context-pruning"));
}

流式输出与分块回复

分块处理

EmbeddedBlockChunker 管理将流式文本拆分成离散的回复块：

const blockChunker = blockChunking ? new EmbeddedBlockChunker(blockChunking) : null;

思考/最终标签剥离

流式输出会被处理以剥离 <think>/<thinking> 块并提取 <final> 内容：

const stripBlockTags = (text: string, state: { thinking: boolean; final: boolean }) => {
  // 剥离 <think>...</think> 内容
  // 如果开启了 enforceFinalTag，只返回 <final>...</final> 内容
};

回复指令

回复指令如 [[media:url]]、[[voice]]、[[reply:id]] 会被解析并提取：

const { text: cleanedText, mediaUrls, audioAsVoice, replyToId } = consumeReplyDirectives(chunk);

错误处理

错误分类

pi-embedded-helpers.ts 对错误进行分类以便合理处理：

isContextOverflowError(errorText)     // 上下文过大
isCompactionFailureError(errorText)   // 压缩失败
isAuthAssistantError(lastAssistant)   // 认证失败
isRateLimitAssistantError(...)        // 被限流
isFailoverAssistantError(...)         // 应该故障转移
classifyFailoverReason(errorText)     // "auth" | "rate_limit" | "quota" | "timeout" | ...

思考级别回退

如果不支持某个思考级别，会自动回退：

const fallbackThinking = pickFallbackThinkingLevel({
  message: errorText,
  attempted: attemptedThinking,
});
if (fallbackThinking) {
  thinkLevel = fallbackThinking;
  continue;
}

沙箱集成

启用沙箱模式后，工具和路径会受到约束：

const sandbox = await resolveSandboxContext({
  config: params.config,
  sessionKey: sandboxSessionKey,
  workspaceDir: resolvedWorkspace,
});

if (sandboxRoot) {
  // 使用沙箱版的 read/edit/write 工具
  // exec 在容器中运行
  // 浏览器使用 bridge URL
}

提供商专属处理

Anthropic

  • 拒绝魔术字符串清理
  • 连续角色的轮次验证
  • Claude Code 参数兼容

Google/Gemini

  • 轮次排序修正（applyGoogleTurnOrderingFix）
  • 工具 schema 清理（sanitizeToolsForGoogle）
  • 会话历史清理（sanitizeSessionHistory）

OpenAI

  • Codex 模型的 apply_patch 工具
  • 思考级别降级处理

TUI 集成

OpenClaw 也有一个本地 TUI 模式，直接使用 pi-tui 组件：

// src/tui/tui.ts
import { ... } from "@mariozechner/pi-tui";

这提供了类似 Pi 原生模式的交互式终端体验。

与 Pi CLI 的主要区别

方面Pi CLIOpenClaw 嵌入式
调用方式pi 命令 / RPC通过 SDK 调用 createAgentSession()
工具默认编程工具自定义的 OpenClaw 工具套件
系统提示词AGENTS.md + prompts按渠道/上下文动态生成
会话存储~/.pi/agent/sessions/~/.openclaw/agents/<agentId>/sessions/（或 $OPENCLAW_STATE_DIR/agents/<agentId>/sessions/）
认证单一凭据多配置轮换
扩展从磁盘加载编程式 + 磁盘路径
事件处理TUI 渲染基于回调（onBlockReply 等）

未来考虑

可能重构的方向：

  1. 工具签名对齐：目前在 pi-agent-core 和 pi-coding-agent 签名之间做适配
  2. SessionManager 包装：guardSessionManager 增加了安全性但也增加了复杂度
  3. 扩展加载：可以更直接地使用 Pi 的 ResourceLoader
  4. 流式处理器复杂度：subscribeEmbeddedPiSession 已经变得很庞大
  5. 提供商特殊处理：很多提供商专属的代码路径，Pi 本身可能可以处理

测试

Pi 集成的测试覆盖了以下套件：

  • src/agents/pi-*.test.ts
  • src/agents/pi-auth-json.test.ts
  • src/agents/pi-embedded-*.test.ts
  • src/agents/pi-embedded-helpers*.test.ts
  • src/agents/pi-embedded-runner*.test.ts
  • src/agents/pi-embedded-runner/**/*.test.ts
  • src/agents/pi-embedded-subscribe*.test.ts
  • src/agents/pi-tools*.test.ts
  • src/agents/pi-tool-definition-adapter*.test.ts
  • src/agents/pi-settings.test.ts
  • src/agents/pi-extensions/**/*.test.ts

实时/可选测试：

  • src/agents/pi-embedded-runner-extraparams.live.test.ts（启用 OPENCLAW_LIVE_TEST=1）

运行命令详见 Pi 开发工作流。

arrow_back
上一页
Troubleshooting
下一页
Architecture
arrow_forward