mirror of
https://github.com/badlogic/pi-mono.git
synced 2026-11-05 23:37:25 +00:00
0c5b7ee393
- Switching branches (navigate, select) no longer waits for the main conversation: a branch left while running keeps working in the background, read-only with Stop allowed. - Branch rows in the sidebar show the working dot; background branches' answers count as unread. - Fence refusals name a running background branch instead of a subagent #N. - Subagent reports go to the conversation that sent the message; team send to "parent" goes to the conversation that last spawned or messaged the subagent.
83 lines
3.7 KiB
TypeScript
83 lines
3.7 KiB
TypeScript
// `/tree` (`docs/tree.md`): the session's branches as one tree of messages, switching the branch the session continues
|
|
// in, and the branches' labels. Each branch is a top-level Durable conversation forked from another at an entry.
|
|
import { arr, int, nul, obj, type Static, str, union } from "@earendil-works/pi-schema";
|
|
import { evolve, literal, method, param, remote, sessionHost } from "@p2/app/v1";
|
|
|
|
/** How much of a message a node previews, and the longest branch label. */
|
|
export const PREVIEW_LENGTH = 200;
|
|
|
|
/** A tree node's fields in `SessionTree` v1. */
|
|
const NODE_FIELDS = {
|
|
/** The entry's ID (Session-global, so unique across branches). */
|
|
entry: int({ minimum: 1 }),
|
|
/** The node before it on its branch, or null for the first message. */
|
|
parent: union([int({ minimum: 1 }), nul()]),
|
|
role: union([literal("user"), literal("assistant")]),
|
|
/** The message's text, at most `PREVIEW_LENGTH` characters. */
|
|
text: str(),
|
|
};
|
|
|
|
/** One message of the tree: a user message or an assistant answer with text. */
|
|
export const TreeNodeSchema = obj({
|
|
...NODE_FIELDS,
|
|
/** The top-level conversation storing it: the branch whose label it shows under, the first of its own nodes. */
|
|
conversation: int({ minimum: 1 }),
|
|
});
|
|
export type TreeNode = Static<typeof TreeNodeSchema>;
|
|
|
|
/** A branch's label; null shows the session title (conversation 1's default, `MainDoc`). */
|
|
export const BranchLabelSchema = union([str(), nul()]);
|
|
|
|
/** The tree's fields in `SessionTree` v1. */
|
|
const TREE_FIELDS = {
|
|
/** The current main conversation. */
|
|
main: int({ minimum: 1 }),
|
|
/** The current branch's last node: where the next message continues; null before the first message. */
|
|
current: union([int({ minimum: 1 }), nul()]),
|
|
};
|
|
|
|
export const TreeSchema = obj({
|
|
/** In entry ID order, so a parent always comes before its children. */
|
|
nodes: arr(TreeNodeSchema),
|
|
/** The top-level conversations and their labels, in creation order. */
|
|
branches: arr(obj({ conversation: int({ minimum: 1 }), label: BranchLabelSchema })),
|
|
...TREE_FIELDS,
|
|
});
|
|
export type Tree = Static<typeof TreeSchema>;
|
|
|
|
/** The session's branches, for the owner (worker): the base of the history, before branch labels. */
|
|
const SessionTreeV1 = remote("p2.tree", {
|
|
version: 1,
|
|
params: { session: param.str() },
|
|
hostAddress: sessionHost,
|
|
variants: {
|
|
owner: {
|
|
methods: {
|
|
tree: method(undefined, obj({ nodes: arr(obj(NODE_FIELDS)), ...TREE_FIELDS })),
|
|
/**
|
|
* Continues the session at `entry` (pi's `/tree`): after it for an answer, before it for a user message, whose
|
|
* text comes back for the editor (null otherwise). Switches to the branch ending there, else forks a new one
|
|
* (a new empty conversation before the first message), labelled with the chosen message's start. A run of
|
|
* the conversation left goes on in the background. Returns the new main conversation.
|
|
*/
|
|
navigate: method(
|
|
obj({ entry: int({ minimum: 1 }) }),
|
|
obj({ conversation: int({ minimum: 1 }), text: union([str(), nul()]) }),
|
|
),
|
|
/** Continues the session at the end of the branch `conversation` (a sidebar row); a run goes on. */
|
|
select: method(obj({ conversation: int({ minimum: 1 }) })),
|
|
},
|
|
},
|
|
},
|
|
});
|
|
|
|
/**
|
|
* The session's branches, for the owner (worker). 2: branch labels (`Tree.branches`, `TreeNode.conversation`) and
|
|
* `rename`: renames the branch `conversation`, 1 to `PREVIEW_LENGTH` characters after trimming; null or empty resets
|
|
* conversation 1 to the session title, other branches need a label. The provider still serves 1 (its `accepts`).
|
|
*/
|
|
export const SessionTree = evolve(SessionTreeV1, (e) =>
|
|
e
|
|
.result("tree", TreeSchema)
|
|
.addMethod("owner", "rename", obj({ conversation: int({ minimum: 1 }), label: BranchLabelSchema }), nul()),
|
|
);
|