7.8 KiB
s07: Session Tree — 历史不是一条线，是一棵树
走错了能回头，回头还能换条路。 Pi 边界：会话历史边界 —— 历史存成一棵树，一轮输入只取当前这条路径。
问题
到 s06 为止，消息都存在一个数组里，只能一条线往后加：A → B → C。
但现实里经常想这样：走到 B，发现不太对，想退回 A 换个方向再试一次。数组做不到——它只会往后追加，没有"回到某个点、从那里分叉"。
s07 要让历史能分叉。
解决方案
给每条消息记一个 parentId，指向它的上一条。这样历史不再是一条线，而是一棵树：同一个节点可以长出多条分支。
user(方案A)
/ \
asst(A的回答) asst(改走方案B)
currentPath 从当前位置一路回溯到根，得到当前这条线的消息序列——provider 拿到的还是线性的消息，对外完全没变。moveTo 切换当前位置，就能走到另一条分支。
注意：s07 不是让 provider 理解一棵树。树只存在于 core 内部，用来支持回到历史点再分叉；provider 仍然只看当前路径上的线性 messages。
[U1 升级]
AgentState.messages（数组）升级为SessionTree。这是受控升级：数组没法表达分叉，所以是替换。但currentPath()仍产出线性AgentMessage[]，ProviderInput 的构造方式一字不变——升级藏在 core 内部，不漏到外面。
工作原理
先定义节点。 一个节点就是一条消息，外加它在树里的位置。
export type SessionEntry = {
id: string;
parentId: string | null;
message: AgentMessage;
};
SessionTree 做三件事。 追加、切换位置、读当前路径。
export class SessionTree {
private entries = new Map<string, SessionEntry>();
private activeLeafId: string | null = null;
private counter = 0; // 实例级：每个树独立计数
append(message: AgentMessage): SessionEntry {
const entry = { id: `e${++this.counter}`, parentId: this.activeLeafId, message };
this.entries.set(entry.id, entry);
this.activeLeafId = entry.id;
return entry;
}
moveTo(entryId: string): void { /* 切换当前位置 */ }
currentPath(): AgentMessage[] {
// 从 activeLeaf 一路回溯到根，反转，得到当前这条线
}
}
append 总是接在当前位置后面；moveTo 把当前位置挪到任意已有节点（分叉的起点）；currentPath 回溯出当前这条线。切位置不会删掉旧节点——它们还在树里，只是不在当前路径上。
id 计数器是实例级的（this.counter），不是全局变量。这样多个 SessionTree 互不干扰，也不会因为新建一个树就接着旧树的编号往后数。
对外不变。 createTurnSnapshot 和 buildProviderInputFromSnapshot 现在从 state.session.currentPath() 取消息，但产出的还是线性的 ProviderMessage[]——provider 这边感觉不到 core 内部已经从数组换成了树。
这一节真正建立的是会话历史边界：历史在 core 内部是一棵树，但对一轮输入来说，它永远是"当前这条路径"的线性投影。后面 s08 会往输入里加项目资料，但"历史 = 当前路径"这条规矩，从这里立起来。
试一下
运行：
npm run s07
输出类似：
s07: Session Tree
[路径：方案 A]
user: 方案 A
assistant: A 的回答
[路径：方案 B]
user: 方案 A
assistant: 改走方案 B
[所有节点]
e1 parent=null user: 方案 A
e2 parent=e1 assistant: A 的回答
e3 parent=e1 assistant: 改走方案 B
观察重点：e2 和 e3 的 parent 都是 e1——从同一个节点分叉出两条路；切到方案 B 后，[路径：方案 B] 只含 e1 和 e3，不含 e2。
接入主线
s07 在 s06 上累积。相对 s06 的变更：
| 组件 | s06 | s07 |
|---|---|---|
AgentState |
{ messages: AgentMessage[] } |
{ session: SessionTree }（U1 升级） |
| 新增类型 | — | SessionEntry / SessionTree |
| 消息写入 | state.messages.push(...) |
state.session.append(...) |
| 消息读取 | state.messages |
state.session.currentPath() |
createTurnSnapshot / buildProviderInputFromSnapshot |
从 state.messages 取 |
从 state.session.currentPath() 取 |
ProviderInput 构造 |
— | 不变（currentPath 产出线性消息） |
焊接点：消息读写全改为走 state.session；但 currentPath() 产出线性 AgentMessage[]，所以 ProviderInput / TurnSnapshot 的构造逻辑一字未动。U1 升级藏在 core 内部。
接下来
现在一轮输入里有：当前路径上的历史、工具说明、模型名。
下一节会再往输入里加一样东西——项目本身的资料（比如一份说明文档、一个可复用的提示词）。
进入下一节：s08。
Pi 源码溯源：持久化的 SessionTree 和 11 种 entry
教学版的 SessionTree 在内存里、只有 message entry。Pi 的 session 是持久化的树，节点有 11 种类型，远不止消息。
源码在哪
packages/agent/src/harness/types.ts:334— SessionTreeEntry 联合类型packages/agent/src/harness/types.ts:409— 11 种 entrypackages/agent/src/harness/session/session.ts:82— Session 实现packages/agent/src/harness/session/session.ts:246—moveTo（分支）
不只是消息：11 种 entry
教学版的 SessionEntry 只有 { id, parentId, message }。Pi 的 entry 是个大联合（types.ts:409）：
type SessionTreeEntry =
| MessageEntry // 消息（教学版唯一有的）
| ThinkingLevelChangeEntry // 改了推理强度
| ModelChangeEntry // 换了模型
| ActiveToolsChangeEntry // 启用/禁用了工具
| CompactionEntry // 做了上下文压缩（s08 方向）
| BranchSummaryEntry // 分支摘要
| CustomEntry / CustomMessageEntry // 自定义内容
| LabelEntry // 给某个节点打标签
| SessionInfoEntry // 会话元信息
| LeafEntry; // 当前活跃叶子
历史不只记"说了什么"，还记"中途换了什么"——换模型、换工具、压缩上下文都是树上的节点。这样回到任何一个历史点，能完整还原当时的配置。
parentId + moveTo = 真分叉
每个 entry 都有 parentId（types.ts:337），moveTo(entryId)（session.ts:246）切换当前位置：
async appendMessage(message) {
return this.appendTypedEntry({
type: "message", id: ...,
parentId: await this.storage.getLeafId(), // 挂在当前叶子下
timestamp: ..., message,
});
}
新节点总挂在当前叶子下；moveTo 把叶子指针挪到任意历史节点，再 append 就长出一条新分支。和教学版的 SessionTree 一模一样的心智，但 Pi 的分支还能带 BranchSummaryEntry 记录"为什么岔出去"。
持久化，不是内存
教学版的 SessionTree 在内存里、进程退出就没了。Pi 的 session 走 storage（session.ts），落盘持久化——关掉重开能恢复，能跨会话。LeafEntry 专门跟踪"当前在哪条分支"，持久化后重启能接上。
边界
moveTo 一个不存在的 id 抛 SessionError（教学版也抛错，一致）。分支不会删旧节点——它们留在树里，只是不在当前路径上。
一句话
教学版的 SessionTree 立的是"历史是树、能分叉、一轮输入取当前路径"。Pi 把它坐实成持久化的树 + 11 种 entry（消息/模型变更/工具变更/压缩/分支摘要…），parentId + moveTo 实现分叉。教学版只留 MessageEntry 和分支骨架。