getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-15 14:03:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix branch summarization abort handling and tree navigation

Browse source 
- Check response.stopReason instead of catching errors for abort detection
- Return result object from _generateBranchSummary instead of throwing
- Fix summary attachment: attach to navigation target position, not old branch
- Support root-level summaries (parentId=null) in branchWithSummary
- Remove setTimeout hack for re-showing tree selector on abort
This commit is contained in:
Mario Zechner 2025-12-29 20:15:19 +01:00
parent 9dac0a1423
commit 01dae9ebcc
7 changed files with 55 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

14
packages/coding-agent/test/session-manager/tree-traversal.test.ts
View file

@ -129,16 +129,16 @@ describe("SessionManager append and tree traversal", () => {
it("leaf pointer advances after each append", () => {
const session = SessionManager.inMemory();
expect(session.getLeafUuid()).toBe("");
expect(session.getLeafId()).toBe("");
const id1 = session.appendMessage(userMsg("1"));
expect(session.getLeafUuid()).toBe(id1);
expect(session.getLeafId()).toBe(id1);
const id2 = session.appendMessage(assistantMsg("2"));
expect(session.getLeafUuid()).toBe(id2);
expect(session.getLeafId()).toBe(id2);
const id3 = session.appendThinkingLevelChange("high");
expect(session.getLeafUuid()).toBe(id3);
expect(session.getLeafId()).toBe(id3);
});
});
@ -303,10 +303,10 @@ describe("SessionManager append and tree traversal", () => {
const _id2 = session.appendMessage(assistantMsg("2"));
const id3 = session.appendMessage(userMsg("3"));
expect(session.getLeafUuid()).toBe(id3);
expect(session.getLeafId()).toBe(id3);
session.branch(id1);
expect(session.getLeafUuid()).toBe(id1);
expect(session.getLeafId()).toBe(id1);
});
it("throws for non-existent entry", () => {
@ -341,7 +341,7 @@ describe("SessionManager append and tree traversal", () => {
const summaryId = session.branchWithSummary(id1, "Summary of abandoned work");
expect(session.getLeafUuid()).toBe(summaryId);
expect(session.getLeafId()).toBe(summaryId);
const entries = session.getEntries();
const summaryEntry = entries.find((e) => e.type === "branch_summary");