@lichen-ai/lichen-sdk 0.1.0

package/dist/ui/branching.d.ts

@@ -0,0 +1,35 @@
+import type { ThreadState } from '../schema.js';
+interface Node<StateType = any> {
+    type: 'node';
+    value: ThreadState<StateType>;
+    path: string[];
+}
+interface Fork<StateType = any> {
+    type: 'fork';
+    items: Array<Sequence<StateType>>;
+}
+export interface Sequence<StateType = any> {
+    type: 'sequence';
+    items: Array<Node<StateType> | Fork<StateType>>;
+}
+export declare function getBranchSequence<StateType extends Record<string, unknown>>(history: ThreadState<StateType>[]): {
+    rootSequence: Sequence<any>;
+    paths: string[][];
+};
+export declare function getBranchView<StateType extends Record<string, unknown>>(sequence: Sequence<StateType>, paths: string[][], branch: string): {
+    history: ThreadState<StateType>[];
+    branchByCheckpoint: Record<string, {
+        branch: string | undefined;
+        branchOptions: string[] | undefined;
+    }>;
+};
+export declare function getBranchContext<StateType extends Record<string, unknown>>(branch: string, history: ThreadState<StateType>[] | undefined): {
+    branchTree: Sequence<any>;
+    flatHistory: ThreadState<any>[];
+    branchByCheckpoint: Record<string, {
+        branch: string | undefined;
+        branchOptions: string[] | undefined;
+    }>;
+    threadHead: ThreadState<any> | undefined;
+};
+export {};

package/dist/ui/branching.js

@@ -0,0 +1,143 @@
+export function getBranchSequence(history) {
+    const nodeIds = new Set();
+    const childrenMap = {};
+    // Short circuit if there's only a singular one state
+    if (history.length <= 1) {
+        return {
+            rootSequence: {
+                type: 'sequence',
+                items: history.map((value) => ({ type: 'node', value, path: [] }))
+            },
+            paths: []
+        };
+    }
+    // First pass - collect nodes for each checkpoint
+    history.forEach((state) => {
+        const checkpointId = state.parent_checkpoint?.checkpoint_id ?? '$';
+        childrenMap[checkpointId] ??= [];
+        childrenMap[checkpointId].push(state);
+        if (state.checkpoint?.checkpoint_id != null) {
+            nodeIds.add(state.checkpoint.checkpoint_id);
+        }
+    });
+    // If dealing with partial history, take the branch
+    // with the latest checkpoint and mark it as the root.
+    const maxId = (...ids) => ids
+        .filter((i) => i != null)
+        .sort((a, b) => a.localeCompare(b))
+        .at(-1);
+    const lastOrphanedNode = childrenMap.$ == null
+        ? Object.keys(childrenMap)
+            .filter((parentId) => !nodeIds.has(parentId))
+            .map((parentId) => {
+            const queue = [parentId];
+            const seen = new Set();
+            let lastId = parentId;
+            while (queue.length > 0) {
+                const current = queue.shift();
+                if (seen.has(current))
+                    continue;
+                seen.add(current);
+                const children = (childrenMap[current] ?? []).flatMap((i) => i.checkpoint?.checkpoint_id ?? []);
+                lastId = maxId(lastId, ...children);
+                queue.push(...children);
+            }
+            return { parentId, lastId };
+        })
+            .sort((a, b) => a.lastId.localeCompare(b.lastId))
+            .at(-1)?.parentId
+        : undefined;
+    if (lastOrphanedNode != null)
+        childrenMap.$ = childrenMap[lastOrphanedNode];
+    const rootSequence = { type: 'sequence', items: [] };
+    const queue = [{ id: '$', sequence: rootSequence, path: [] }];
+    const paths = [];
+    const visited = new Set();
+    while (queue.length > 0) {
+        const task = queue.shift();
+        if (visited.has(task.id))
+            continue;
+        visited.add(task.id);
+        const children = childrenMap[task.id];
+        if (children == null || children.length === 0)
+            continue;
+        // If we've encountered a fork (2+ children), push the fork
+        // to the sequence and add a new sequence for each child
+        let fork;
+        if (children.length > 1) {
+            fork = { type: 'fork', items: [] };
+            task.sequence.items.push(fork);
+        }
+        for (const value of children) {
+            const id = value.checkpoint?.checkpoint_id;
+            if (id == null)
+                continue;
+            let { sequence } = task;
+            let { path } = task;
+            if (fork != null) {
+                sequence = { type: 'sequence', items: [] };
+                fork.items.unshift(sequence);
+                path = path.slice();
+                path.push(id);
+                paths.push(path);
+            }
+            sequence.items.push({ type: 'node', value, path });
+            queue.push({ id, sequence, path });
+        }
+    }
+    return { rootSequence, paths };
+}
+const PATH_SEP = '>';
+const ROOT_ID = '$';
+// Get flat view
+export function getBranchView(sequence, paths, branch) {
+    const path = branch.split(PATH_SEP);
+    const pathMap = {};
+    for (const path of paths) {
+        const parent = path.at(-2) ?? ROOT_ID;
+        pathMap[parent] ??= [];
+        pathMap[parent].unshift(path);
+    }
+    const history = [];
+    const branchByCheckpoint = {};
+    const forkStack = path.slice();
+    const queue = [...sequence.items];
+    while (queue.length > 0) {
+        const item = queue.shift();
+        if (item.type === 'node') {
+            history.push(item.value);
+            const checkpointId = item.value.checkpoint?.checkpoint_id;
+            if (checkpointId == null)
+                continue;
+            branchByCheckpoint[checkpointId] = {
+                branch: item.path.join(PATH_SEP),
+                branchOptions: (item.path.length > 0 ? (pathMap[item.path.at(-2) ?? ROOT_ID] ?? []) : []).map((p) => p.join(PATH_SEP))
+            };
+        }
+        if (item.type === 'fork') {
+            const forkId = forkStack.shift();
+            const index = forkId != null
+                ? item.items.findIndex((value) => {
+                    const firstItem = value.items.at(0);
+                    if (!firstItem || firstItem.type !== 'node')
+                        return false;
+                    return firstItem.value.checkpoint?.checkpoint_id === forkId;
+                })
+                : -1;
+            const nextItems = item.items.at(index)?.items ?? [];
+            queue.push(...nextItems);
+        }
+    }
+    return { history, branchByCheckpoint };
+}
+export function getBranchContext(branch, history) {
+    const { rootSequence: branchTree, paths } = getBranchSequence(history ?? []);
+    const { history: flatHistory, branchByCheckpoint } = getBranchView(branchTree, paths, branch);
+    return {
+        branchTree,
+        flatHistory,
+        branchByCheckpoint,
+        threadHead: flatHistory.at(-1)
+    };
+}
+//# sourceMappingURL=branching.js.map

package/dist/ui/branching.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"branching.js","sourceRoot":"","sources":["../../src/ui/branching.ts"],"names":[],"mappings":"AAiCA,MAAM,UAAU,iBAAiB,CAA4C,OAAiC;IAC5G,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAA;IACjC,MAAM,WAAW,GAA6C,EAAE,CAAA;IAEhE,qDAAqD;IACrD,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;QACxB,OAAO;YACL,YAAY,EAAE;gBACZ,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;aACrC;YAC/B,KAAK,EAAE,EAAE;SACV,CAAA;IACH,CAAC;IAED,iDAAiD;IACjD,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;QACxB,MAAM,YAAY,GAAG,KAAK,CAAC,iBAAiB,EAAE,aAAa,IAAI,GAAG,CAAA;QAClE,WAAW,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;QAChC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAErC,IAAI,KAAK,CAAC,UAAU,EAAE,aAAa,IAAI,IAAI,EAAE,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,CAAA;QAC7C,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,mDAAmD;IACnD,sDAAsD;IACtD,MAAM,KAAK,GAAG,CAAC,GAAG,GAAsB,EAAE,EAAE,CAC1C,GAAG;SACA,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC;SACxB,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;SAClC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAA;IAEZ,MAAM,gBAAgB,GACpB,WAAW,CAAC,CAAC,IAAI,IAAI;QACnB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC;aACrB,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;aAC5C,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YAChB,MAAM,KAAK,GAAa,CAAC,QAAQ,CAAC,CAAA;YAClC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAA;YAE9B,IAAI,MAAM,GAAG,QAAQ,CAAA;YAErB,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,EAAG,CAAA;gBAE9B,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;oBAAE,SAAQ;gBAC/B,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;gBAEjB,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,EAAE,aAAa,IAAI,EAAE,CAAC,CAAA;gBAE/F,MAAM,GAAG,KAAK,CAAC,MAAM,EAAE,GAAG,QAAQ,CAAC,CAAA;gBACnC,KAAK,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAA;YACzB,CAAC;YAED,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;QAC7B,CAAC,CAAC;aACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;aAChD,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ;QACrB,CAAC,CAAC,SAAS,CAAA;IAEf,IAAI,gBAAgB,IAAI,IAAI;QAAE,WAAW,CAAC,CAAC,GAAG,WAAW,CAAC,gBAAgB,CAAC,CAAA;IAI3E,MAAM,YAAY,GAAa,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,CAAA;IAC9D,MAAM,KAAK,GAAW,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAA;IAErE,MAAM,KAAK,GAAe,EAAE,CAAA;IAE5B,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAA;IACjC,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAG,CAAA;QAC3B,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAAE,SAAQ;QAClC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAEpB,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QACrC,IAAI,QAAQ,IAAI,IAAI,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,SAAQ;QAEvD,2DAA2D;QAC3D,wDAAwD;QACxD,IAAI,IAAsB,CAAA;QAC1B,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAA;YAClC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAChC,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,EAAE,GAAG,KAAK,CAAC,UAAU,EAAE,aAAa,CAAA;YAC1C,IAAI,EAAE,IAAI,IAAI;gBAAE,SAAQ;YAExB,IAAI,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAA;YACvB,IAAI,EAAE,IAAI,EAAE,GAAG,IAAI,CAAA;YACnB,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;gBACjB,QAAQ,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,CAAA;gBAC1C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;gBAE5B,IAAI,GAAG,IAAI,CAAC,KAAK,EAAE,CAAA;gBACnB,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;gBACb,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAClB,CAAC;YAED,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;YAClD,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAA;QACpC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,CAAA;AAChC,CAAC;AAED,MAAM,QAAQ,GAAG,GAAG,CAAA;AACpB,MAAM,OAAO,GAAG,GAAG,CAAA;AAEnB,gBAAgB;AAChB,MAAM,UAAU,aAAa,CAC3B,QAA6B,EAC7B,KAAiB,EACjB,MAAc;IAEd,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;IACnC,MAAM,OAAO,GAA+B,EAAE,CAAA;IAE9C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,OAAO,CAAA;QACrC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;QACtB,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;IAC/B,CAAC;IAED,MAAM,OAAO,GAA6B,EAAE,CAAA;IAC5C,MAAM,kBAAkB,GAAwF,EAAE,CAAA;IAElH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,EAAE,CAAA;IAC9B,MAAM,KAAK,GAA0C,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAA;IAExE,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAG,CAAA;QAE3B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACxB,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,aAAa,CAAA;YACzD,IAAI,YAAY,IAAI,IAAI;gBAAE,SAAQ;YAElC,kBAAkB,CAAC,YAAY,CAAC,GAAG;gBACjC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAChC,aAAa,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAClG,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CACjB;aACF,CAAA;QACH,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,EAAE,CAAA;YAChC,MAAM,KAAK,GACT,MAAM,IAAI,IAAI;gBACZ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE;oBAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;oBACnC,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,KAAK,MAAM;wBAAE,OAAO,KAAK,CAAA;oBACzD,OAAO,SAAS,CAAC,KAAK,CAAC,UAAU,EAAE,aAAa,KAAK,MAAM,CAAA;gBAC7D,CAAC,CAAC;gBACJ,CAAC,CAAC,CAAC,CAAC,CAAA;YAER,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,IAAI,EAAE,CAAA;YACnD,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAA;QAC1B,CAAC;IACH,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAA;AACxC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,OAA6C;IAE7C,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,iBAAiB,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;IAC5E,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,GAAG,aAAa,CAAC,UAAU,EAAE,KAAK,EAAE,MAAM,CAAC,CAAA;IAE7F,OAAO;QACL,UAAU;QACV,WAAW;QACX,kBAAkB;QAClB,UAAU,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;KAC/B,CAAA;AACH,CAAC"}

package/dist/ui/branching.mjs

@@ -0,0 +1,143 @@
+export function getBranchSequence(history) {
+    const nodeIds = new Set();
+    const childrenMap = {};
+    // Short circuit if there's only a singular one state
+    if (history.length <= 1) {
+        return {
+            rootSequence: {
+                type: 'sequence',
+                items: history.map((value) => ({ type: 'node', value, path: [] }))
+            },
+            paths: []
+        };
+    }
+    // First pass - collect nodes for each checkpoint
+    history.forEach((state) => {
+        const checkpointId = state.parent_checkpoint?.checkpoint_id ?? '$';
+        childrenMap[checkpointId] ??= [];
+        childrenMap[checkpointId].push(state);
+        if (state.checkpoint?.checkpoint_id != null) {
+            nodeIds.add(state.checkpoint.checkpoint_id);
+        }
+    });
+    // If dealing with partial history, take the branch
+    // with the latest checkpoint and mark it as the root.
+    const maxId = (...ids) => ids
+        .filter((i) => i != null)
+        .sort((a, b) => a.localeCompare(b))
+        .at(-1);
+    const lastOrphanedNode = childrenMap.$ == null
+        ? Object.keys(childrenMap)
+            .filter((parentId) => !nodeIds.has(parentId))
+            .map((parentId) => {
+            const queue = [parentId];
+            const seen = new Set();
+            let lastId = parentId;
+            while (queue.length > 0) {
+                const current = queue.shift();
+                if (seen.has(current))
+                    continue;
+                seen.add(current);
+                const children = (childrenMap[current] ?? []).flatMap((i) => i.checkpoint?.checkpoint_id ?? []);
+                lastId = maxId(lastId, ...children);
+                queue.push(...children);
+            }
+            return { parentId, lastId };
+        })
+            .sort((a, b) => a.lastId.localeCompare(b.lastId))
+            .at(-1)?.parentId
+        : undefined;
+    if (lastOrphanedNode != null)
+        childrenMap.$ = childrenMap[lastOrphanedNode];
+    const rootSequence = { type: 'sequence', items: [] };
+    const queue = [{ id: '$', sequence: rootSequence, path: [] }];
+    const paths = [];
+    const visited = new Set();
+    while (queue.length > 0) {
+        const task = queue.shift();
+        if (visited.has(task.id))
+            continue;
+        visited.add(task.id);
+        const children = childrenMap[task.id];
+        if (children == null || children.length === 0)
+            continue;
+        // If we've encountered a fork (2+ children), push the fork
+        // to the sequence and add a new sequence for each child
+        let fork;
+        if (children.length > 1) {
+            fork = { type: 'fork', items: [] };
+            task.sequence.items.push(fork);
+        }
+        for (const value of children) {
+            const id = value.checkpoint?.checkpoint_id;
+            if (id == null)
+                continue;
+            let { sequence } = task;
+            let { path } = task;
+            if (fork != null) {
+                sequence = { type: 'sequence', items: [] };
+                fork.items.unshift(sequence);
+                path = path.slice();
+                path.push(id);
+                paths.push(path);
+            }
+            sequence.items.push({ type: 'node', value, path });
+            queue.push({ id, sequence, path });
+        }
+    }
+    return { rootSequence, paths };
+}
+const PATH_SEP = '>';
+const ROOT_ID = '$';
+// Get flat view
+export function getBranchView(sequence, paths, branch) {
+    const path = branch.split(PATH_SEP);
+    const pathMap = {};
+    for (const path of paths) {
+        const parent = path.at(-2) ?? ROOT_ID;
+        pathMap[parent] ??= [];
+        pathMap[parent].unshift(path);
+    }
+    const history = [];
+    const branchByCheckpoint = {};
+    const forkStack = path.slice();
+    const queue = [...sequence.items];
+    while (queue.length > 0) {
+        const item = queue.shift();
+        if (item.type === 'node') {
+            history.push(item.value);
+            const checkpointId = item.value.checkpoint?.checkpoint_id;
+            if (checkpointId == null)
+                continue;
+            branchByCheckpoint[checkpointId] = {
+                branch: item.path.join(PATH_SEP),
+                branchOptions: (item.path.length > 0 ? (pathMap[item.path.at(-2) ?? ROOT_ID] ?? []) : []).map((p) => p.join(PATH_SEP))
+            };
+        }
+        if (item.type === 'fork') {
+            const forkId = forkStack.shift();
+            const index = forkId != null
+                ? item.items.findIndex((value) => {
+                    const firstItem = value.items.at(0);
+                    if (!firstItem || firstItem.type !== 'node')
+                        return false;
+                    return firstItem.value.checkpoint?.checkpoint_id === forkId;
+                })
+                : -1;
+            const nextItems = item.items.at(index)?.items ?? [];
+            queue.push(...nextItems);
+        }
+    }
+    return { history, branchByCheckpoint };
+}
+export function getBranchContext(branch, history) {
+    const { rootSequence: branchTree, paths } = getBranchSequence(history ?? []);
+    const { history: flatHistory, branchByCheckpoint } = getBranchView(branchTree, paths, branch);
+    return {
+        branchTree,
+        flatHistory,
+        branchByCheckpoint,
+        threadHead: flatHistory.at(-1)
+    };
+}
+//# sourceMappingURL=branching.mjs.map

package/dist/ui/branching.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"branching.mjs","sourceRoot":"","sources":["../../src/ui/branching.ts"],"names":[],"mappings":"AAiCA,MAAM,UAAU,iBAAiB,CAA4C,OAAiC;IAC5G,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAA;IACjC,MAAM,WAAW,GAA6C,EAAE,CAAA;IAEhE,qDAAqD;IACrD,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;QACxB,OAAO;YACL,YAAY,EAAE;gBACZ,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;aACrC;YAC/B,KAAK,EAAE,EAAE;SACV,CAAA;IACH,CAAC;IAED,iDAAiD;IACjD,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;QACxB,MAAM,YAAY,GAAG,KAAK,CAAC,iBAAiB,EAAE,aAAa,IAAI,GAAG,CAAA;QAClE,WAAW,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;QAChC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAErC,IAAI,KAAK,CAAC,UAAU,EAAE,aAAa,IAAI,IAAI,EAAE,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,CAAA;QAC7C,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,mDAAmD;IACnD,sDAAsD;IACtD,MAAM,KAAK,GAAG,CAAC,GAAG,GAAsB,EAAE,EAAE,CAC1C,GAAG;SACA,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC;SACxB,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;SAClC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAA;IAEZ,MAAM,gBAAgB,GACpB,WAAW,CAAC,CAAC,IAAI,IAAI;QACnB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC;aACrB,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;aAC5C,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YAChB,MAAM,KAAK,GAAa,CAAC,QAAQ,CAAC,CAAA;YAClC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAA;YAE9B,IAAI,MAAM,GAAG,QAAQ,CAAA;YAErB,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,EAAG,CAAA;gBAE9B,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;oBAAE,SAAQ;gBAC/B,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;gBAEjB,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,EAAE,aAAa,IAAI,EAAE,CAAC,CAAA;gBAE/F,MAAM,GAAG,KAAK,CAAC,MAAM,EAAE,GAAG,QAAQ,CAAC,CAAA;gBACnC,KAAK,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAA;YACzB,CAAC;YAED,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;QAC7B,CAAC,CAAC;aACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;aAChD,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ;QACrB,CAAC,CAAC,SAAS,CAAA;IAEf,IAAI,gBAAgB,IAAI,IAAI;QAAE,WAAW,CAAC,CAAC,GAAG,WAAW,CAAC,gBAAgB,CAAC,CAAA;IAI3E,MAAM,YAAY,GAAa,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,CAAA;IAC9D,MAAM,KAAK,GAAW,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAA;IAErE,MAAM,KAAK,GAAe,EAAE,CAAA;IAE5B,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAA;IACjC,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAG,CAAA;QAC3B,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAAE,SAAQ;QAClC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAEpB,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QACrC,IAAI,QAAQ,IAAI,IAAI,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,SAAQ;QAEvD,2DAA2D;QAC3D,wDAAwD;QACxD,IAAI,IAAsB,CAAA;QAC1B,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAA;YAClC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAChC,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,EAAE,GAAG,KAAK,CAAC,UAAU,EAAE,aAAa,CAAA;YAC1C,IAAI,EAAE,IAAI,IAAI;gBAAE,SAAQ;YAExB,IAAI,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAA;YACvB,IAAI,EAAE,IAAI,EAAE,GAAG,IAAI,CAAA;YACnB,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;gBACjB,QAAQ,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,CAAA;gBAC1C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;gBAE5B,IAAI,GAAG,IAAI,CAAC,KAAK,EAAE,CAAA;gBACnB,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;gBACb,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAClB,CAAC;YAED,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;YAClD,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAA;QACpC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,CAAA;AAChC,CAAC;AAED,MAAM,QAAQ,GAAG,GAAG,CAAA;AACpB,MAAM,OAAO,GAAG,GAAG,CAAA;AAEnB,gBAAgB;AAChB,MAAM,UAAU,aAAa,CAC3B,QAA6B,EAC7B,KAAiB,EACjB,MAAc;IAEd,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;IACnC,MAAM,OAAO,GAA+B,EAAE,CAAA;IAE9C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,OAAO,CAAA;QACrC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;QACtB,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;IAC/B,CAAC;IAED,MAAM,OAAO,GAA6B,EAAE,CAAA;IAC5C,MAAM,kBAAkB,GAAwF,EAAE,CAAA;IAElH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,EAAE,CAAA;IAC9B,MAAM,KAAK,GAA0C,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAA;IAExE,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAG,CAAA;QAE3B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACxB,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,aAAa,CAAA;YACzD,IAAI,YAAY,IAAI,IAAI;gBAAE,SAAQ;YAElC,kBAAkB,CAAC,YAAY,CAAC,GAAG;gBACjC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAChC,aAAa,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAClG,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CACjB;aACF,CAAA;QACH,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,EAAE,CAAA;YAChC,MAAM,KAAK,GACT,MAAM,IAAI,IAAI;gBACZ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE;oBAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;oBACnC,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,KAAK,MAAM;wBAAE,OAAO,KAAK,CAAA;oBACzD,OAAO,SAAS,CAAC,KAAK,CAAC,UAAU,EAAE,aAAa,KAAK,MAAM,CAAA;gBAC7D,CAAC,CAAC;gBACJ,CAAC,CAAC,CAAC,CAAC,CAAA;YAER,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,IAAI,EAAE,CAAA;YACnD,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAA;QAC1B,CAAC;IACH,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAA;AACxC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,OAA6C;IAE7C,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,iBAAiB,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;IAC5E,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,GAAG,aAAa,CAAC,UAAU,EAAE,KAAK,EAAE,MAAM,CAAC,CAAA;IAE7F,OAAO;QACL,UAAU;QACV,WAAW;QACX,kBAAkB;QAClB,UAAU,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;KAC/B,CAAA;AACH,CAAC"}

package/dist/ui/branching.test.d.ts

@@ -0,0 +1 @@
+export {};