RPC Mode(RPC 模式)

本页面是 Pi 官方文档 的中文翻译。仅供学习参考。

RPC 模式通过 stdin/stdout 上的 JSON 协议实现 Agent 的无界面操作,用于嵌入其他应用、IDE 或自定义 UI。

Node.js/TypeScript 用户注意:如果你正在构建 Node.js 应用,建议直接使用 @earendil-works/pi-coding-agent 中的 AgentSession,而非生成子进程。参见 src/core/agent-session.ts 了解 API。对于基于子进程的 TypeScript 客户端,参见 src/modes/rpc/rpc-client.ts

启动 RPC 模式

pi --mode rpc [options]

常用选项:

  • --provider <name>: 设置 LLM Provider(anthropic、openai、google 等)
  • --model <pattern>: 模型模式或 ID(支持 provider/id 和可选的 :<thinking>
  • --no-session: 禁用会话持久化
  • --session-dir <path>: 自定义会话存储目录

协议概述

  • Commands(命令):发送到 stdin 的 JSON 对象,每行一个
  • Responses(响应):带 type: "response" 的 JSON 对象,指示命令成功/失败
  • Events(事件):流式输出到 stdout 的 Agent 事件(JSON 行)

所有命令支持可选的 id 字段用于请求/响应关联。如果提供,相应的响应将包含相同的 id

帧协议

RPC 模式使用严格的 JSONL 语义,仅使用 LF(\n)作为记录分隔符。

这对客户端很重要:

  • 仅以 \n 分割记录
  • 接受可选的 \r\n 输入,通过去除尾随的 \r 处理
  • 不要使用将 Unicode 分隔符视为换行的通用行读取器

特别是,Node 的 readline 不符合 RPC 模式的协议规范,因为它也会在 U+2028U+2029 处分割,而这些在 JSON 字符串中是有效的。

命令

Prompting

prompt

向 Agent 发送用户 Prompt。命令响应在 prompt 被接受、排队或处理后发出。事件在接受后继续异步流式传输。

{"id": "req-1", "type": "prompt", "message": "Hello, world!"}

带图像:

{"type": "prompt", "message": "What's in this image?", "images": [{"type": "image", "data": "base64-encoded-data", "mimeType": "image/png"}]}

流式传输期间:如果 Agent 已在流式传输,必须指定 streamingBehavior 来队列消息:

{"type": "prompt", "message": "New instruction", "streamingBehavior": "steer"}
  • "steer":在 Agent 运行时队列消息。在当前助手回合完成工具调用后、下一次 LLM 调用前发送。
  • "followUp":等待 Agent 完成。仅在 Agent 停止时发送消息。

如果 Agent 正在流式传输且未指定 streamingBehavior,命令返回错误。

扩展命令:如果消息是扩展命令(例如 /mycommand),即使在流式传输期间也会立即执行。扩展命令通过 pi.sendMessage() 管理自己的 LLM 交互。

输入扩展:Skill 命令(/skill:name)和 prompt 模板(/template)在发送/队列前被扩展。

响应:

{"id": "req-1", "type": "response", "command": "prompt", "success": true}

success: true 表示 prompt 被接受、排队或立即处理。success: false 表示 prompt 在接受前被拒绝。接受后的失败通过正常的事件和消息流报告,而不是针对同一请求 ID 的第二个 response

images 字段可选。每个图像使用 ImageContent 格式:{"type": "image", "data": "base64-encoded-data", "mimeType": "image/png"}

steer

在 Agent 运行时队列一条 steering 消息。在当前助手回合完成工具调用后、下一次 LLM 调用前发送。Skill 命令和 prompt 模板会被扩展。不允许扩展命令(请改用 prompt)。

{"type": "steer", "message": "Stop and do this instead"}

带图像:

{"type": "steer", "message": "Look at this instead", "images": [{"type": "image", "data": "base64-encoded-data", "mimeType": "image/png"}]}

images 字段可选。每个图像使用 ImageContent 格式(与 prompt 相同)。

响应:

{"type": "response", "command": "steer", "success": true}

参见 set_steering_mode 了解如何控制 steering 消息的处理方式。

follow_up

队列一条 follow-up 消息,在 Agent 完成后处理。仅在 Agent 没有更多工具调用或 steering 消息时发送。Skill 命令和 prompt 模板会被扩展。不允许扩展命令(请改用 prompt)。

{"type": "follow_up", "message": "After you're done, also do this"}

带图像:

{"type": "follow_up", "message": "Also check this image", "images": [{"type": "image", "data": "base64-encoded-data", "mimeType": "image/png"}]}

images 字段可选。每个图像使用 ImageContent 格式(与 prompt 相同)。

响应:

{"type": "response", "command": "follow_up", "success": true}

参见 set_follow_up_mode 了解如何控制 follow-up 消息的处理方式。

abort

中断当前 Agent 操作。

{"type": "abort"}

响应:

{"type": "response", "command": "abort", "success": true}

new_session

开始新会话。可以被 session_before_switch 扩展事件处理器取消。

{"type": "new_session"}

带可选的父会话跟踪:

{"type": "new_session", "parentSession": "/path/to/parent-session.jsonl"}

响应:

{"type": "response", "command": "new_session", "success": true, "data": {"cancelled": false}}

如果扩展取消了:

{"type": "response", "command": "new_session", "success": true, "data": {"cancelled": true}}

状态

get_state

获取当前会话状态。

{"type": "get_state"}

响应:

{
  "type": "response",
  "command": "get_state",
  "success": true,
  "data": {
    "model": {...},
    "thinkingLevel": "medium",
    "isStreaming": false,
    "isCompacting": false,
    "steeringMode": "all",
    "followUpMode": "one-at-a-time",
    "sessionFile": "/path/to/session.jsonl",
    "sessionId": "abc123",
    "sessionName": "my-feature-work",
    "autoCompactionEnabled": true,
    "messageCount": 5,
    "pendingMessageCount": 0
  }
}

model 字段是完整的 Model 对象或 nullsessionName 字段是通过 set_session_name 设置的显示名称,未设置时省略。

get_messages

获取对话中的所有消息。

{"type": "get_messages"}

响应:

{
  "type": "response",
  "command": "get_messages",
  "success": true,
  "data": {"messages": [...]}
}

消息是 AgentMessage 对象(参见 消息类型)。

模型

set_model

切换到指定模型。

{"type": "set_model", "provider": "anthropic", "modelId": "claude-sonnet-4-20250514"}

响应包含完整的 Model 对象:

{
  "type": "response",
  "command": "set_model",
  "success": true,
  "data": {...}
}

cycle_model

循环到下一个可用模型。如果只有一个模型可用,返回 null

{"type": "cycle_model"}

响应:

{
  "type": "response",
  "command": "cycle_model",
  "success": true,
  "data": {
    "model": {...},
    "thinkingLevel": "medium",
    "isScoped": false
  }
}

model 字段是完整的 Model 对象。

get_available_models

列出所有已配置的模型。

{"type": "get_available_models"}

响应包含完整的 Model 对象数组:

{
  "type": "response",
  "command": "get_available_models",
  "success": true,
  "data": {
    "models": [...]
  }
}

思考

set_thinking_level

为支持的模型设置推理/思考级别。

{"type": "set_thinking_level", "level": "high"}

级别:"off""minimal""low""medium""high""xhigh"

注意:"xhigh" 仅 OpenAI codex-max 模型支持。

响应:

{"type": "response", "command": "set_thinking_level", "success": true}

cycle_thinking_level

循环可用的思考级别。如果模型不支持思考,返回 null

{"type": "cycle_thinking_level"}

响应:

{
  "type": "response",
  "command": "cycle_thinking_level",
  "success": true,
  "data": {"level": "high"}
}

队列模式

set_steering_mode

控制 steering 消息(来自 steer)的发送方式。

{"type": "set_steering_mode", "mode": "one-at-a-time"}

模式:

  • "all":在当前助手回合完成工具调用后发送所有 steering 消息
  • "one-at-a-time":每个已完成的助手回合发送一条 steering 消息(默认)

响应:

{"type": "response", "command": "set_steering_mode", "success": true}

set_follow_up_mode

控制 follow-up 消息(来自 follow_up)的发送方式。

{"type": "set_follow_up_mode", "mode": "one-at-a-time"}

模式:

  • "all":Agent 完成时发送所有 follow-up 消息
  • "one-at-a-time":每次 Agent 完成时发送一条 follow-up 消息(默认)

响应:

{"type": "response", "command": "set_follow_up_mode", "success": true}

压缩

compact

手动压缩对话上下文以减少 Token 使用量。

{"type": "compact"}

带自定义指令:

{"type": "compact", "customInstructions": "Focus on code changes"}

响应:

{
  "type": "response",
  "command": "compact",
  "success": true,
  "data": {
    "summary": "Summary of conversation...",
    "firstKeptEntryId": "abc123",
    "tokensBefore": 150000,
    "details": {}
  }
}

set_auto_compaction

启用或禁用上下文接近满时的自动压缩。

{"type": "set_auto_compaction", "enabled": true}

响应:

{"type": "response", "command": "set_auto_compaction", "success": true}

重试

set_auto_retry

启用或禁用临时错误(过载、速率限制、5xx)的自动重试。

{"type": "set_auto_retry", "enabled": true}

响应:

{"type": "response", "command": "set_auto_retry", "success": true}

abort_retry

中止正在进行的重试(取消延迟并停止重试)。

{"type": "abort_retry"}

响应:

{"type": "response", "command": "abort_retry", "success": true}

Bash

bash

执行 Shell 命令并将输出添加到对话上下文。

{"type": "bash", "command": "ls -la"}

响应:

{
  "type": "response",
  "command": "bash",
  "success": true,
  "data": {
    "output": "total 48\ndrwxr-xr-x ...",
    "exitCode": 0,
    "cancelled": false,
    "truncated": false
  }
}

如果输出被截断,包含 fullOutputPath

{
  "type": "response",
  "command": "bash",
  "success": true,
  "data": {
    "output": "truncated output...",
    "exitCode": 0,
    "cancelled": false,
    "truncated": true,
    "fullOutputPath": "/tmp/pi-bash-abc123.log"
  }
}

bash 结果如何到达 LLM:

bash 命令立即执行并返回 BashResult。内部会创建一个 BashExecutionMessage 并存储在 Agent 的消息状态中。此消息不会发出事件。

当下一个 prompt 命令发送时,所有消息(包括 BashExecutionMessage)在发送到 LLM 之前会被转换。BashExecutionMessage 会转换为包含以下格式的 UserMessage

Ran `ls -la`
```
total 48
drwxr-xr-x ...
```

这意味着:

  1. Bash 输出会在下一次 prompt 时包含在 LLM 上下文中,而非立即
  2. 可以在 prompt 之前执行多个 bash 命令;所有输出都会被包含
  3. BashExecutionMessage 本身不会发出事件

abort_bash

中止正在运行的 bash 命令。

{"type": "abort_bash"}

响应:

{"type": "response", "command": "abort_bash", "success": true}

会话

get_session_stats

获取 Token 使用量、成本统计和当前上下文窗口使用情况。

{"type": "get_session_stats"}

响应:

{
  "type": "response",
  "command": "get_session_stats",
  "success": true,
  "data": {
    "sessionFile": "/path/to/session.jsonl",
    "sessionId": "abc123",
    "userMessages": 5,
    "assistantMessages": 5,
    "toolCalls": 12,
    "toolResults": 12,
    "totalMessages": 22,
    "tokens": {
      "input": 50000,
      "output": 10000,
      "cacheRead": 40000,
      "cacheWrite": 5000,
      "total": 105000
    },
    "cost": 0.45,
    "contextUsage": {
      "tokens": 60000,
      "contextWindow": 200000,
      "percent": 30
    }
  }
}

tokens 包含当前会话状态的助手使用量总计。contextUsage 包含用于压缩和底部显示的实际当前上下文窗口估计值。

当没有模型或上下文窗口可用时,contextUsage 被省略。在压缩后直到新的压缩后助手响应提供有效的使用量数据前,contextUsage.tokenscontextUsage.percentnull

export_html

将会话导出为 HTML 文件。

{"type": "export_html"}

带自定义路径:

{"type": "export_html", "outputPath": "/tmp/session.html"}

响应:

{
  "type": "response",
  "command": "export_html",
  "success": true,
  "data": {"path": "/tmp/session.html"}
}

switch_session

加载不同的会话文件。可以被 session_before_switch 扩展事件处理器取消。

{"type": "switch_session", "sessionPath": "/path/to/session.jsonl"}

响应:

{"type": "response", "command": "switch_session", "success": true, "data": {"cancelled": false}}

如果扩展取消了切换:

{"type": "response", "command": "switch_session", "success": true, "data": {"cancelled": true}}

fork

从当前分支上的先前用户消息创建新分叉。可以被 session_before_fork 扩展事件处理器取消。返回被分叉消息的文本。

{"type": "fork", "entryId": "abc123"}

响应:

{
  "type": "response",
  "command": "fork",
  "success": true,
  "data": {"text": "The original prompt text...", "cancelled": false}
}

如果扩展取消了分叉:

{
  "type": "response",
  "command": "fork",
  "success": true,
  "data": {"text": "The original prompt text...", "cancelled": true}
}

clone

将当前活跃分支复制到新会话的当前位置。可以被 session_before_fork 扩展事件处理器取消。

{"type": "clone"}

响应:

{
  "type": "response",
  "command": "clone",
  "success": true,
  "data": {"cancelled": false}
}

如果扩展取消了克隆:

{
  "type": "response",
  "command": "clone",
  "success": true,
  "data": {"cancelled": true}
}

get_fork_messages

获取可用于分叉的用户消息。

{"type": "get_fork_messages"}

响应:

{
  "type": "response",
  "command": "get_fork_messages",
  "success": true,
  "data": {
    "messages": [
      {"entryId": "abc123", "text": "First prompt..."},
      {"entryId": "def456", "text": "Second prompt..."}
    ]
  }
}

get_last_assistant_text

获取最后一条助手消息的文本内容。

{"type": "get_last_assistant_text"}

响应:

{
  "type": "response",
  "command": "get_last_assistant_text",
  "success": true,
  "data": {"text": "The assistant's response..."}
}

如果没有助手消息,返回 {"text": null}

set_session_name

为当前会话设置显示名称。该名称会出现在会话列表中,有助于识别会话。

{"type": "set_session_name", "name": "my-feature-work"}

响应:

{
  "type": "response",
  "command": "set_session_name",
  "success": true
}

当前会话名称可通过 get_statesessionName 字段获取。

命令

get_commands

获取可用命令(扩展命令、prompt 模板和 Skills)。可以通过在 prompt 命令前加 / 来调用它们。

{"type": "get_commands"}

响应:

{
  "type": "response",
  "command": "get_commands",
  "success": true,
  "data": {
    "commands": [
      {"name": "session-name", "description": "Set or clear session name", "source": "extension", "path": "/home/user/.pi/agent/extensions/session.ts"},
      {"name": "fix-tests", "description": "Fix failing tests", "source": "prompt", "location": "project", "path": "/home/user/myproject/.pi/agent/prompts/fix-tests.md"},
      {"name": "skill:brave-search", "description": "Web search via Brave API", "source": "skill", "location": "user", "path": "/home/user/.pi/agent/skills/brave-search/SKILL.md"}
    ]
  }
}

每个命令包含:

  • name:命令名称(使用 /name 调用)
  • description:人类可读的描述(扩展命令可选)
  • source:命令类型:
    • "extension":通过扩展中的 pi.registerCommand() 注册
    • "prompt":从 prompt 模板 .md 文件加载
    • "skill":从 skill 目录加载(名称前缀为 skill:
  • location:加载位置(扩展命令可无):
    • "user":用户级别(~/.pi/agent/
    • "project":项目级别(./.pi/agent/
    • "path":通过 CLI 或设置的显式路径
  • path:命令源的绝对文件路径(可选)

注意:内置 TUI 命令(/settings/hotkeys 等)不包含在内。它们仅在交互模式中处理,如果通过 prompt 发送不会执行。

事件

事件作为 JSON 行流式输出到 stdout。事件包含 id 字段(只有响应包含)。

事件类型

事件描述
agent_startAgent 开始处理
agent_endAgent 完成(包含所有生成的消息)
turn_start新回合开始
turn_end回合完成(包含助手消息和工具结果)
message_start消息开始
message_update流式更新(文本/思考/工具调用增量)
message_end消息完成
tool_execution_start工具开始执行
tool_execution_update工具执行进度(流式输出)
tool_execution_end工具完成
queue_update待处理的 steering/follow-up 队列变化
compaction_start压缩开始
compaction_end压缩完成
auto_retry_start自动重试开始(临时错误后)
auto_retry_end自动重试完成(成功或最终失败)
extension_error扩展抛出错误

agent_start

Agent 开始处理 prompt 时发出。

{"type": "agent_start"}

agent_end

Agent 完成时发出。包含此次运行中生成的所有消息。

{
  "type": "agent_end",
  "messages": [...]
}

turn_start / turn_end

一个回合包括一次助手响应以及由此产生的任何工具调用和结果。

{"type": "turn_start"}
{
  "type": "turn_end",
  "message": {...},
  "toolResults": [...]
}

message_start / message_end

消息开始和完成时发出。message 字段包含 AgentMessage

{"type": "message_start", "message": {...}}
{"type": "message_end", "message": {...}}

message_update(流式)

在助手消息流式传输期间发出。包含部分消息和流式增量事件。

{
  "type": "message_update",
  "message": {...},
  "assistantMessageEvent": {
    "type": "text_delta",
    "contentIndex": 0,
    "delta": "Hello ",
    "partial": {...}
  }
}

assistantMessageEvent 字段包含以下增量类型之一:

类型描述
start消息生成开始
text_start文本内容块开始
text_delta文本内容片段
text_end文本内容块结束
thinking_start思考块开始
thinking_delta思考内容片段
thinking_end思考块结束
toolcall_start工具调用开始
toolcall_delta工具调用参数片段
toolcall_end工具调用结束(包含完整 toolCall 对象)
done消息完成(reason:"stop""length""toolUse"
error发生错误(reason:"aborted""error"

文本响应流式示例:

{"type":"message_update","message":{...},"assistantMessageEvent":{"type":"text_start","contentIndex":0,"partial":{...}}}
{"type":"message_update","message":{...},"assistantMessageEvent":{"type":"text_delta","contentIndex":0,"delta":"Hello","partial":{...}}}
{"type":"message_update","message":{...},"assistantMessageEvent":{"type":"text_delta","contentIndex":0,"delta":" world","partial":{...}}}
{"type":"message_update","message":{...},"assistantMessageEvent":{"type":"text_end","contentIndex":0,"content":"Hello world","partial":{...}}}

tool_execution_start / tool_execution_update / tool_execution_end

工具开始、流式进度和完成执行时发出。

{
  "type": "tool_execution_start",
  "toolCallId": "call_abc123",
  "toolName": "bash",
  "args": {"command": "ls -la"}
}

执行期间,tool_execution_update 事件流式传输部分结果(例如 bash 输出到达时):

{
  "type": "tool_execution_update",
  "toolCallId": "call_abc123",
  "toolName": "bash",
  "args": {"command": "ls -la"},
  "partialResult": {
    "content": [{"type": "text", "text": "partial output so far..."}],
    "details": {"truncation": null, "fullOutputPath": null}
  }
}

完成时:

{
  "type": "tool_execution_end",
  "toolCallId": "call_abc123",
  "toolName": "bash",
  "result": {
    "content": [{"type": "text", "text": "total 48\n..."}],
    "details": {...}
  },
  "isError": false
}

使用 toolCallId 关联事件。tool_execution_update 中的 partialResult 包含到目前为止的累积输出(不仅仅是增量),允许客户端在每次更新时直接替换显示。

queue_update

每当待处理的 steering 或 follow-up 队列发生变化时发出。

{
  "type": "queue_update",
  "steering": ["Focus on error handling"],
  "followUp": ["After that, summarize the result"]
}

compaction_start / compaction_end

压缩运行时发出,无论手动还是自动。

{"type": "compaction_start", "reason": "threshold"}

reason 字段为 "manual""threshold""overflow"

{
  "type": "compaction_end",
  "reason": "threshold",
  "result": {
    "summary": "Summary of conversation...",
    "firstKeptEntryId": "abc123",
    "tokensBefore": 150000,
    "details": {}
  },
  "aborted": false,
  "willRetry": false
}

如果 reason"overflow" 且压缩成功,则 willRetrytrue,Agent 将自动重试 prompt。

如果压缩被中止,resultnullabortedtrue

如果压缩失败(例如 API 配额超限),resultnullabortedfalseerrorMessage 包含错误描述。

auto_retry_start / auto_retry_end

在临时错误(过载、速率限制、5xx)后触发自动重试时发出。

{
  "type": "auto_retry_start",
  "attempt": 1,
  "maxAttempts": 3,
  "delayMs": 2000,
  "errorMessage": "529 {\"type\":\"error\",\"error\":{\"type\":\"overloaded_error\",\"message\":\"Overloaded\"}}"
}
{
  "type": "auto_retry_end",
  "success": true,
  "attempt": 2
}

最终失败(超过最大重试次数):

{
  "type": "auto_retry_end",
  "success": false,
  "attempt": 3,
  "finalError": "529 overloaded_error: Overloaded"
}

extension_error

扩展抛出错误时发出。

{
  "type": "extension_error",
  "extensionPath": "/path/to/extension.ts",
  "event": "tool_call",
  "error": "Error message..."
}

扩展 UI 协议

扩展可以通过 ctx.ui.select()ctx.ui.confirm() 等方法请求用户交互。在 RPC 模式中,这些被转换为在基本命令/事件流之上的请求/响应子协议。

扩展 UI 方法分为两类:

  • 对话框方法selectconfirminputeditor):在 stdout 上发出 extension_ui_request,并阻塞直到客户端在 stdin 上发回带有匹配 idextension_ui_response
  • 即发即弃方法notifysetStatussetWidgetsetTitleset_editor_text):在 stdout 上发出 extension_ui_request,但不期望响应。客户端可以显示信息或忽略。

如果对话框方法包含 timeout 字段,Agent 端会在超时过期时自动解析为默认值。客户端无需跟踪超时。

某些 ExtensionUIContext 方法在 RPC 模式下不受支持或功能降级,因为它们需要直接 TUI 访问:

  • custom() 返回 undefined
  • setWorkingMessage()setWorkingIndicator()setFooter()setHeader()setEditorComponent()setToolsExpanded() 为空操作
  • getEditorText() 返回 ""
  • getToolsExpanded() 返回 false
  • pasteToEditor() 委托给 setEditorText()(无粘贴/折叠处理)
  • getAllThemes() 返回 []
  • getTheme() 返回 undefined
  • setTheme() 返回 { success: false, error: "..." }

注意:ctx.hasUI 在 RPC 模式下为 true,因为对话框和即发即弃方法通过扩展 UI 子协议是功能性的。

扩展 UI 请求(stdout)

所有请求都有 type: "extension_ui_request"、唯一的 idmethod 字段。

select

提示用户从列表中选择。带 timeout 字段的对话框方法包含以毫秒为单位的超时;如果客户端未及时响应,Agent 自动解析为 undefined

{
  "type": "extension_ui_request",
  "id": "uuid-1",
  "method": "select",
  "title": "Allow dangerous command?",
  "options": ["Allow", "Block"],
  "timeout": 10000
}

预期响应:包含 value(所选选项字符串)或 cancelled: trueextension_ui_response

confirm

提示用户进行是/否确认。

{
  "type": "extension_ui_request",
  "id": "uuid-2",
  "method": "confirm",
  "title": "Clear session?",
  "message": "All messages will be lost.",
  "timeout": 5000
}

预期响应:包含 confirmed: true/falsecancelled: trueextension_ui_response

input

提示用户输入自由文本。

{
  "type": "extension_ui_request",
  "id": "uuid-3",
  "method": "input",
  "title": "Enter a value",
  "placeholder": "type something..."
}

预期响应:包含 value(输入的文本)或 cancelled: trueextension_ui_response

editor

打开多行文本编辑器,带有可选预填内容。

{
  "type": "extension_ui_request",
  "id": "uuid-4",
  "method": "editor",
  "title": "Edit some text",
  "prefill": "Line 1\nLine 2\nLine 3"
}

预期响应:包含 value(编辑后的文本)或 cancelled: trueextension_ui_response

notify

显示通知。即发即弃,不期望响应。

{
  "type": "extension_ui_request",
  "id": "uuid-5",
  "method": "notify",
  "message": "Command blocked by user",
  "notifyType": "warning"
}

notifyType 字段为 "info""warning""error"。省略时默认为 "info"

setStatus

在底部/状态栏中设置或清除状态条目。即发即弃。

{
  "type": "extension_ui_request",
  "id": "uuid-6",
  "method": "setStatus",
  "statusKey": "my-ext",
  "statusText": "Turn 3 running..."
}

发送 statusText: undefined(或省略)来清除该键的状态条目。

setWidget

在编辑器上方或下方设置或清除小部件(文本行块)。即发即弃。

{
  "type": "extension_ui_request",
  "id": "uuid-7",
  "method": "setWidget",
  "widgetKey": "my-ext",
  "widgetLines": ["--- My Widget ---", "Line 1", "Line 2"],
  "widgetPlacement": "aboveEditor"
}

发送 widgetLines: undefined(或省略)来清除小部件。widgetPlacement 字段为 "aboveEditor"(默认)或 "belowEditor"。RPC 模式仅支持字符串数组;组件工厂被忽略。

setTitle

设置终端窗口/标签标题。即发即弃。

{
  "type": "extension_ui_request",
  "id": "uuid-8",
  "method": "setTitle",
  "title": "pi - my project"
}

set_editor_text

设置输入编辑器中的文本。即发即弃。

{
  "type": "extension_ui_request",
  "id": "uuid-9",
  "method": "set_editor_text",
  "text": "prefilled text for the user"
}

扩展 UI 响应(stdin)

仅对话框方法需要响应(selectconfirminputeditor)。id 必须与请求匹配。

值响应(select、input、editor)

{"type": "extension_ui_response", "id": "uuid-1", "value": "Allow"}

确认响应(confirm)

{"type": "extension_ui_response", "id": "uuid-2", "confirmed": true}

取消响应(任意对话框)

关闭任意对话框方法。扩展接收 undefined(对于 select/input/editor)或 false(对于 confirm)。

{"type": "extension_ui_response", "id": "uuid-3", "cancelled": true}

错误处理

失败的命令返回包含 success: false 的响应:

{
  "type": "response",
  "command": "set_model",
  "success": false,
  "error": "Model not found: invalid/model"
}

解析错误:

{
  "type": "response",
  "command": "parse",
  "success": false,
  "error": "Failed to parse command: Unexpected token..."
}

类型

源文件:

Model

{
  "id": "claude-sonnet-4-20250514",
  "name": "Claude Sonnet 4",
  "api": "anthropic-messages",
  "provider": "anthropic",
  "baseUrl": "https://api.anthropic.com",
  "reasoning": true,
  "input": ["text", "image"],
  "contextWindow": 200000,
  "maxTokens": 16384,
  "cost": {
    "input": 3.0,
    "output": 15.0,
    "cacheRead": 0.3,
    "cacheWrite": 3.75
  }
}

UserMessage

{
  "role": "user",
  "content": "Hello!",
  "timestamp": 1733234567890,
  "attachments": []
}

content 字段可以是字符串或 TextContent/ImageContent 块的数组。

AssistantMessage

{
  "role": "assistant",
  "content": [
    {"type": "text", "text": "Hello! How can I help?"},
    {"type": "thinking", "thinking": "User is greeting me..."},
    {"type": "toolCall", "id": "call_123", "name": "bash", "arguments": {"command": "ls"}}
  ],
  "api": "anthropic-messages",
  "provider": "anthropic",
  "model": "claude-sonnet-4-20250514",
  "usage": {
    "input": 100,
    "output": 50,
    "cacheRead": 0,
    "cacheWrite": 0,
    "cost": {"input": 0.0003, "output": 0.00075, "cacheRead": 0, "cacheWrite": 0, "total": 0.00105}
  },
  "stopReason": "stop",
  "timestamp": 1733234567890
}

停止原因:"stop""length""toolUse""error""aborted"

ToolResultMessage

{
  "role": "toolResult",
  "toolCallId": "call_123",
  "toolName": "bash",
  "content": [{"type": "text", "text": "total 48\ndrwxr-xr-x ..."}],
  "isError": false,
  "timestamp": 1733234567890
}

BashExecutionMessage

bash RPC 命令创建(非 LLM 工具调用):

{
  "role": "bashExecution",
  "command": "ls -la",
  "output": "total 48\ndrwxr-xr-x ...",
  "exitCode": 0,
  "cancelled": false,
  "truncated": false,
  "fullOutputPath": null,
  "timestamp": 1733234567890
}

Attachment

{
  "id": "img1",
  "type": "image",
  "fileName": "photo.jpg",
  "mimeType": "image/jpeg",
  "size": 102400,
  "content": "base64-encoded-data...",
  "extractedText": null,
  "preview": null
}

示例:基本客户端(Python)

import subprocess
import json

proc = subprocess.Popen(
    ["pi", "--mode", "rpc", "--no-session"],
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    text=True
)

def send(cmd):
    proc.stdin.write(json.dumps(cmd) + "\n")
    proc.stdin.flush()

def read_events():
    for line in proc.stdout:
        yield json.loads(line)

# 发送 prompt
send({"type": "prompt", "message": "Hello!"})

# 处理事件
for event in read_events():
    if event.get("type") == "message_update":
        delta = event.get("assistantMessageEvent", {})
        if delta.get("type") == "text_delta":
            print(delta["delta"], end="", flush=True)

    if event.get("type") == "agent_end":
        print()
        break

示例:交互式客户端(Node.js)

参见 test/rpc-example.ts 获取完整的交互式示例,或 src/modes/rpc/rpc-client.ts 获取带类型定义的客户端实现。

关于处理扩展 UI 协议的完整示例,参见 examples/rpc-extension-ui.ts,它与 examples/extensions/rpc-demo.ts 扩展配合使用。

const { spawn } = require("child_process");
const { StringDecoder } = require("string_decoder");

const agent = spawn("pi", ["--mode", "rpc", "--no-session"]);

function attachJsonlReader(stream, onLine) {
    const decoder = new StringDecoder("utf8");
    let buffer = "";

    stream.on("data", (chunk) => {
        buffer += typeof chunk === "string" ? chunk : decoder.write(chunk);

        while (true) {
            const newlineIndex = buffer.indexOf("\n");
            if (newlineIndex === -1) break;

            let line = buffer.slice(0, newlineIndex);
            buffer = buffer.slice(newlineIndex + 1);
            if (line.endsWith("\r")) line = line.slice(0, -1);
            onLine(line);
        }
    });

    stream.on("end", () => {
        buffer += decoder.end();
        if (buffer.length > 0) {
            onLine(buffer.endsWith("\r") ? buffer.slice(0, -1) : buffer);
        }
    });
}

attachJsonlReader(agent.stdout, (line) => {
    const event = JSON.parse(line);

    if (event.type === "message_update") {
        const { assistantMessageEvent } = event;
        if (assistantMessageEvent.type === "text_delta") {
            process.stdout.write(assistantMessageEvent.delta);
        }
    }
});

// 发送 prompt
agent.stdin.write(JSON.stringify({ type: "prompt", message: "Hello" }) + "\n");

// 在 Ctrl+C 时中止
process.on("SIGINT", () => {
    agent.stdin.write(JSON.stringify({ type: "abort" }) + "\n");
});

法律声明:本页面是 pi.dev 官方文档的中文翻译版本,仅供学习参考。本网站与 pi.dev 及 Earendil Inc. 无任何法律关系。