@slopus/beer 0.1.1 → 0.1.2

package/dist/modules/tree/treeSearchRun.js

@@ -0,0 +1,178 @@
+import { mkdir } from "node:fs/promises";
+import path from "node:path";
+import { text } from "@text";
+import { generateText } from "@/modules/ai/generateText.js";
+import { treeChildrenParse } from "@/modules/tree/treeChildrenParse.js";
+import { treeChildrenRead } from "@/modules/tree/treeChildrenRead.js";
+import { treeChildrenWrite } from "@/modules/tree/treeChildrenWrite.js";
+import { treeInferenceProgressRun } from "@/modules/tree/treeInferenceProgressRun.js";
+import { treeLeafPick } from "@/modules/tree/treeLeafPick.js";
+import { treeNodeExpand } from "@/modules/tree/treeNodeExpand.js";
+import { treeNodeRead } from "@/modules/tree/treeNodeRead.js";
+import { treeNodeWrite } from "@/modules/tree/treeNodeWrite.js";
+import { treeStateLeaves } from "@/modules/tree/treeStateLeaves.js";
+import { treeStateRead } from "@/modules/tree/treeStateRead.js";
+/**
+ * Runs persisted two-phase tree search: root walk, then LLM-guided leaf expansion.
+ * Expects: config.documentPrompt/childrenPrompt are deterministic and rootDir is writable.
+ */
+export async function treeSearchRun(ctx, config) {
+    const concurrency = Math.max(1, config.concurrency ?? 4);
+    const maxDepth = Math.max(0, config.maxDepth ?? 4);
+    await mkdir(config.rootDir, { recursive: true });
+    let totalExpanded = 0;
+    let totalSkipped = 0;
+    const rootChildren = await treeSearchRunRootChildrenResolve(ctx, config);
+    for (const child of rootChildren) {
+        await treeSearchRunEnsureNode(path.join(config.rootDir, child.slug), {
+            title: child.title,
+            slug: child.slug,
+            status: "unexpanded"
+        });
+    }
+    const rootNodes = await treeStateRead(config.rootDir);
+    await treeSearchRunMapLimit(rootNodes, concurrency, async (node) => {
+        if (node.depth >= maxDepth) {
+            await treeSearchRunDepthLimitMark(node);
+            return;
+        }
+        const expanded = await treeNodeExpand(ctx, node, config);
+        if (expanded.skipped) {
+            totalSkipped += 1;
+            return;
+        }
+        totalExpanded += 1;
+    });
+    await treeSearchRunDispatchGuided(ctx, config, concurrency, maxDepth, {
+        onExpanded: () => {
+            totalExpanded += 1;
+        },
+        onSkipped: () => {
+            totalSkipped += 1;
+        }
+    });
+    const finalState = await treeStateRead(config.rootDir);
+    const totalLeaves = treeSearchRunLeafCount(finalState);
+    return {
+        totalExpanded,
+        totalLeaves,
+        totalSkipped
+    };
+}
+async function treeSearchRunRootChildrenResolve(ctx, config) {
+    const cached = await treeChildrenRead(config.rootDir);
+    if (cached !== null) {
+        return cached;
+    }
+    const result = await treeInferenceProgressRun(ctx, text.tree_search_root_expanding, async (onEvent) => {
+        return generateText(ctx, config.rootPrompt, {
+            modelSelectionMode: config.modelSelectionMode,
+            onEvent,
+            verify: ({ text }) => {
+                treeChildrenParse(text);
+            }
+        });
+    });
+    const parsed = treeChildrenParse(result.text);
+    await treeChildrenWrite(config.rootDir, parsed);
+    return parsed;
+}
+async function treeSearchRunDispatchGuided(ctx, config, concurrency, maxDepth, callbacks) {
+    const inFlight = new Set();
+    while (true) {
+        let pickerExhausted = false;
+        while (inFlight.size < concurrency) {
+            const picked = await treeLeafPick(ctx, config);
+            if (!picked) {
+                pickerExhausted = true;
+                break;
+            }
+            if (picked.depth >= maxDepth) {
+                await treeSearchRunDepthLimitMark(picked);
+                continue;
+            }
+            const current = await treeNodeRead(picked.dirPath);
+            if (current?.status === "in-progress") {
+                continue;
+            }
+            await treeNodeWrite(picked.dirPath, {
+                title: current?.title ?? picked.title,
+                slug: current?.slug ?? picked.slug,
+                sessionId: current?.sessionId,
+                status: "in-progress"
+            });
+            const job = (async () => {
+                const expanded = await treeNodeExpand(ctx, { ...picked, status: "in-progress" }, config);
+                if (expanded.skipped) {
+                    callbacks.onSkipped();
+                    return;
+                }
+                callbacks.onExpanded();
+            })();
+            inFlight.add(job);
+            job.finally(() => {
+                inFlight.delete(job);
+            });
+        }
+        if (inFlight.size === 0) {
+            if (pickerExhausted) {
+                if ((await treeSearchRunUnexpandedLeafCount(config.rootDir)) > 0) {
+                    await ctx.progress(text.tree_search_picker_unresolved, async () => undefined);
+                }
+                return;
+            }
+        }
+        await Promise.race(inFlight);
+    }
+}
+async function treeSearchRunEnsureNode(nodeDirPath, node) {
+    const existing = await treeNodeRead(nodeDirPath);
+    if (existing) {
+        return;
+    }
+    await treeNodeWrite(nodeDirPath, {
+        title: node.title,
+        slug: node.slug,
+        sessionId: node.sessionId,
+        status: node.status
+    });
+}
+async function treeSearchRunDepthLimitMark(node) {
+    const current = await treeNodeRead(node.dirPath);
+    await treeChildrenWrite(node.dirPath, []);
+    await treeNodeWrite(node.dirPath, {
+        title: current?.title ?? node.title,
+        slug: current?.slug ?? node.slug,
+        sessionId: current?.sessionId,
+        status: "leaf"
+    });
+}
+async function treeSearchRunMapLimit(entries, concurrency, run) {
+    const active = new Set();
+    for (const [index, entry] of entries.entries()) {
+        const task = run(entry, index);
+        active.add(task);
+        task.finally(() => {
+            active.delete(task);
+        });
+        if (active.size >= concurrency) {
+            await Promise.race(active);
+        }
+    }
+    await Promise.all(active);
+}
+function treeSearchRunLeafCount(nodes) {
+    let count = 0;
+    for (const node of nodes) {
+        if (node.status === "leaf") {
+            count += 1;
+        }
+        count += treeSearchRunLeafCount(node.children);
+    }
+    return count;
+}
+async function treeSearchRunUnexpandedLeafCount(rootDir) {
+    const state = await treeStateRead(rootDir);
+    return treeStateLeaves(state).length;
+}
+//# sourceMappingURL=treeSearchRun.js.map

package/dist/modules/tree/treeSearchRun.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeSearchRun.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeSearchRun.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACzC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,OAAO,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qCAAqC,CAAC;AACxE,OAAO,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,qCAAqC,CAAC;AACxE,OAAO,EAAE,wBAAwB,EAAE,MAAM,4CAA4C,CAAC;AACtF,OAAO,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAGhE;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,GAAY,EAAE,MAAwB;IACtE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;IACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC;IAEnD,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAEjD,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,YAAY,GAAG,CAAC,CAAC;IAErB,MAAM,YAAY,GAAG,MAAM,gCAAgC,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IACzE,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;QAC/B,MAAM,uBAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE;YACjE,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,MAAM,EAAE,YAAY;SACvB,CAAC,CAAC;IACP,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACtD,MAAM,qBAAqB,CAAC,SAAS,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;QAC/D,IAAI,IAAI,CAAC,KAAK,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,2BAA2B,CAAC,IAAI,CAAC,CAAC;YACxC,OAAO;QACX,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QACzD,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACnB,YAAY,IAAI,CAAC,CAAC;YAClB,OAAO;QACX,CAAC;QACD,aAAa,IAAI,CAAC,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,MAAM,2BAA2B,CAAC,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE,QAAQ,EAAE;QAClE,UAAU,EAAE,GAAG,EAAE;YACb,aAAa,IAAI,CAAC,CAAC;QACvB,CAAC;QACD,SAAS,EAAE,GAAG,EAAE;YACZ,YAAY,IAAI,CAAC,CAAC;QACtB,CAAC;KACJ,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACvD,MAAM,WAAW,GAAG,sBAAsB,CAAC,UAAU,CAAC,CAAC;IACvD,OAAO;QACH,aAAa;QACb,WAAW;QACX,YAAY;KACf,CAAC;AACN,CAAC;AAED,KAAK,UAAU,gCAAgC,CAAC,GAAY,EAAE,MAAwB;IAClF,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACtD,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;QAClB,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,GAAG,EAAE,IAAI,CAAC,0BAA2B,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QACnG,OAAO,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,UAAU,EAAE;YACxC,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;YAC7C,OAAO;YACP,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE;gBACjB,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;SACJ,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAChD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,KAAK,UAAU,2BAA2B,CACtC,GAAY,EACZ,MAAwB,EACxB,WAAmB,EACnB,QAAgB,EAChB,SAA4D;IAE5D,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;IAE1C,OAAO,IAAI,EAAE,CAAC;QACV,IAAI,eAAe,GAAG,KAAK,CAAC;QAC5B,OAAO,QAAQ,CAAC,IAAI,GAAG,WAAW,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAC/C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACV,eAAe,GAAG,IAAI,CAAC;gBACvB,MAAM;YACV,CAAC;YAED,IAAI,MAAM,CAAC,KAAK,IAAI,QAAQ,EAAE,CAAC;gBAC3B,MAAM,2BAA2B,CAAC,MAAM,CAAC,CAAC;gBAC1C,SAAS;YACb,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACnD,IAAI,OAAO,EAAE,MAAM,KAAK,aAAa,EAAE,CAAC;gBACpC,SAAS;YACb,CAAC;YAED,MAAM,aAAa,CAAC,MAAM,CAAC,OAAO,EAAE;gBAChC,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,MAAM,CAAC,KAAK;gBACrC,IAAI,EAAE,OAAO,EAAE,IAAI,IAAI,MAAM,CAAC,IAAI;gBAClC,SAAS,EAAE,OAAO,EAAE,SAAS;gBAC7B,MAAM,EAAE,aAAa;aACxB,CAAC,CAAC;YAEH,MAAM,GAAG,GAAG,CAAC,KAAK,IAAI,EAAE;gBACpB,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,GAAG,EAAE,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,MAAM,CAAC,CAAC;gBACzF,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;oBACnB,SAAS,CAAC,SAAS,EAAE,CAAC;oBACtB,OAAO;gBACX,CAAC;gBACD,SAAS,CAAC,UAAU,EAAE,CAAC;YAC3B,CAAC,CAAC,EAAE,CAAC;YAEL,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAClB,GAAG,CAAC,OAAO,CAAC,GAAG,EAAE;gBACb,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC,CAAC,CAAC;QACP,CAAC;QAED,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,IAAI,eAAe,EAAE,CAAC;gBAClB,IAAI,CAAC,MAAM,gCAAgC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC/D,MAAM,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,6BAA8B,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAAC;gBACnF,CAAC;gBACD,OAAO;YACX,CAAC;QACL,CAAC;QAED,MAAM,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACjC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,uBAAuB,CAClC,WAAmB,EACnB,IAAyE;IAEzE,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,CAAC;IACjD,IAAI,QAAQ,EAAE,CAAC;QACX,OAAO;IACX,CAAC;IAED,MAAM,aAAa,CAAC,WAAW,EAAE;QAC7B,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,MAAM,EAAE,IAAI,CAAC,MAAM;KACtB,CAAC,CAAC;AACP,CAAC;AAED,KAAK,UAAU,2BAA2B,CAAC,IAAc;IACrD,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjD,MAAM,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1C,MAAM,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE;QAC9B,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,KAAK;QACnC,IAAI,EAAE,OAAO,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI;QAChC,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,MAAM,EAAE,MAAM;KACjB,CAAC,CAAC;AACP,CAAC;AAED,KAAK,UAAU,qBAAqB,CAChC,OAAqB,EACrB,WAAmB,EACnB,GAA+C;IAE/C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAiB,CAAC;IAExC,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;QAC7C,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACjB,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE;YACd,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,IAAI,IAAI,WAAW,EAAE,CAAC;YAC7B,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC;IACL,CAAC;IAED,MAAM,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,sBAAsB,CAAC,KAA0B;IACtD,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YACzB,KAAK,IAAI,CAAC,CAAC;QACf,CAAC;QACD,KAAK,IAAI,sBAAsB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACnD,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,gCAAgC,CAAC,OAAe;IAC3D,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,OAAO,CAAC,CAAC;IAC3C,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;AACzC,CAAC"}

package/dist/modules/tree/treeSearchRun.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=treeSearchRun.spec.d.ts.map

package/dist/modules/tree/treeSearchRun.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeSearchRun.spec.d.ts","sourceRoot":"","sources":["../../../sources/modules/tree/treeSearchRun.spec.ts"],"names":[],"mappings":""}

package/dist/modules/tree/treeSearchRun.spec.js

@@ -0,0 +1,203 @@
+import { mkdtemp, readFile, rm } from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
+import { text } from "@text";
+import { beforeEach, describe, expect, it, vi } from "vitest";
+const generateTextMock = vi.hoisted(() => vi.fn());
+const treeNodeExpandMock = vi.hoisted(() => vi.fn());
+const treeLeafPickMock = vi.hoisted(() => vi.fn());
+vi.mock("@/modules/ai/generateText.js", () => ({
+    generateText: generateTextMock
+}));
+vi.mock("@/modules/tree/treeNodeExpand.js", () => ({
+    treeNodeExpand: treeNodeExpandMock
+}));
+vi.mock("@/modules/tree/treeLeafPick.js", () => ({
+    treeLeafPick: treeLeafPickMock
+}));
+import { treeNodeRead } from "@/modules/tree/treeNodeRead.js";
+import { treeSearchRun } from "@/modules/tree/treeSearchRun.js";
+describe("treeSearchRun", () => {
+    beforeEach(() => {
+        generateTextMock.mockReset();
+        treeNodeExpandMock.mockReset();
+        treeLeafPickMock.mockReset();
+    });
+    it("phase 1 expands all root children with concurrency limit", async () => {
+        const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-tree-run-"));
+        try {
+            generateTextMock.mockResolvedValue({
+                provider: "pi",
+                text: JSON.stringify([
+                    { title: "Auth", slug: "auth" },
+                    { title: "Database", slug: "database" }
+                ])
+            });
+            let active = 0;
+            let peak = 0;
+            treeNodeExpandMock.mockImplementation(async () => {
+                active += 1;
+                peak = Math.max(peak, active);
+                await new Promise((resolve) => setTimeout(resolve, 15));
+                active -= 1;
+                return { children: [], status: "leaf", skipped: false };
+            });
+            treeLeafPickMock.mockResolvedValue(null);
+            const context = treeSearchRunContext(tempDir);
+            const result = await treeSearchRun(context, {
+                rootDir: path.join(tempDir, "tree"),
+                rootPrompt: "root children",
+                documentPrompt: () => "doc",
+                childrenPrompt: () => "children",
+                concurrency: 2,
+                maxDepth: 4
+            });
+            expect(treeNodeExpandMock).toHaveBeenCalledTimes(2);
+            expect(peak).toBeLessThanOrEqual(2);
+            expect(result.totalExpanded).toBe(2);
+            const rootChildren = JSON.parse(await readFile(path.join(tempDir, "tree", "children.json"), "utf-8"));
+            expect(rootChildren).toEqual([
+                { title: "Auth", slug: "auth" },
+                { title: "Database", slug: "database" }
+            ]);
+        }
+        finally {
+            await rm(tempDir, { recursive: true, force: true });
+        }
+    });
+    it("phase 2 dispatches picker-selected leaves and respects concurrency", async () => {
+        const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-tree-run-"));
+        try {
+            generateTextMock.mockResolvedValue({
+                provider: "pi",
+                text: JSON.stringify([{ title: "Auth", slug: "auth" }])
+            });
+            const rootDir = path.join(tempDir, "tree");
+            const leafA = treeSearchRunNode(path.join(rootDir, "auth", "oauth2"), path.join(rootDir, "auth"), "oauth2", 2);
+            const leafB = treeSearchRunNode(path.join(rootDir, "auth", "sessions"), path.join(rootDir, "auth"), "sessions", 2);
+            treeLeafPickMock
+                .mockResolvedValueOnce(leafA)
+                .mockResolvedValueOnce(leafB)
+                .mockResolvedValueOnce(null)
+                .mockResolvedValueOnce(null);
+            let active = 0;
+            let peak = 0;
+            treeNodeExpandMock.mockImplementation(async (_ctx, node) => {
+                active += 1;
+                peak = Math.max(peak, active);
+                await new Promise((resolve) => setTimeout(resolve, 15));
+                active -= 1;
+                return { children: [], status: "leaf", skipped: node.depth === 1 };
+            });
+            const context = treeSearchRunContext(tempDir);
+            const result = await treeSearchRun(context, {
+                rootDir,
+                rootPrompt: "root children",
+                documentPrompt: () => "doc",
+                childrenPrompt: () => "children",
+                concurrency: 2,
+                maxDepth: 4
+            });
+            expect(peak).toBeLessThanOrEqual(2);
+            expect(treeLeafPickMock).toHaveBeenCalled();
+            expect(result.totalExpanded).toBe(2);
+        }
+        finally {
+            await rm(tempDir, { recursive: true, force: true });
+        }
+    });
+    it("marks nodes as leaf when max depth is reached", async () => {
+        const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-tree-run-"));
+        try {
+            generateTextMock.mockResolvedValue({
+                provider: "pi",
+                text: JSON.stringify([{ title: "Auth", slug: "auth" }])
+            });
+            treeLeafPickMock.mockResolvedValue(null);
+            treeNodeExpandMock.mockResolvedValue({ children: [], status: "leaf", skipped: false });
+            const context = treeSearchRunContext(tempDir);
+            const rootDir = path.join(tempDir, "tree");
+            await treeSearchRun(context, {
+                rootDir,
+                rootPrompt: "root children",
+                documentPrompt: () => "doc",
+                childrenPrompt: () => "children",
+                maxDepth: 1
+            });
+            const node = await treeNodeRead(path.join(rootDir, "auth"));
+            expect(treeNodeExpandMock).not.toHaveBeenCalled();
+            expect(node?.status).toBe("leaf");
+        }
+        finally {
+            await rm(tempDir, { recursive: true, force: true });
+        }
+    });
+    it("counts skipped expansions for resumable nodes", async () => {
+        const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-tree-run-"));
+        try {
+            generateTextMock.mockResolvedValue({
+                provider: "pi",
+                text: JSON.stringify([{ title: "Auth", slug: "auth" }])
+            });
+            treeNodeExpandMock.mockResolvedValue({ children: [], status: "leaf", skipped: true });
+            treeLeafPickMock.mockResolvedValue(null);
+            const context = treeSearchRunContext(tempDir);
+            const result = await treeSearchRun(context, {
+                rootDir: path.join(tempDir, "tree"),
+                rootPrompt: "root children",
+                documentPrompt: () => "doc",
+                childrenPrompt: () => "children"
+            });
+            expect(result.totalSkipped).toBe(1);
+            expect(result.totalExpanded).toBe(0);
+        }
+        finally {
+            await rm(tempDir, { recursive: true, force: true });
+        }
+    });
+    it("reports picker unresolved status when leaves remain but picker returns null", async () => {
+        const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-tree-run-"));
+        try {
+            generateTextMock.mockResolvedValue({
+                provider: "pi",
+                text: JSON.stringify([{ title: "Auth", slug: "auth" }])
+            });
+            treeNodeExpandMock.mockResolvedValue({ children: [], status: "leaf", skipped: true });
+            treeLeafPickMock.mockResolvedValue(null);
+            const progressMessages = [];
+            const context = treeSearchRunContext(tempDir, progressMessages);
+            await treeSearchRun(context, {
+                rootDir: path.join(tempDir, "tree"),
+                rootPrompt: "root children",
+                documentPrompt: () => "doc",
+                childrenPrompt: () => "children"
+            });
+            expect(progressMessages).toContain(text.tree_search_picker_unresolved);
+        }
+        finally {
+            await rm(tempDir, { recursive: true, force: true });
+        }
+    });
+});
+function treeSearchRunNode(dirPath, parentDirPath, slug, depth) {
+    return {
+        slug,
+        title: slug,
+        depth,
+        dirPath,
+        parentDirPath,
+        status: "unexpanded",
+        children: []
+    };
+}
+function treeSearchRunContext(projectPath, progressMessages) {
+    return {
+        projectPath,
+        providers: [],
+        progress: async (initialMessage, operation) => {
+            progressMessages?.push(initialMessage);
+            return operation(() => { });
+        }
+    };
+}
+//# sourceMappingURL=treeSearchRun.spec.js.map

package/dist/modules/tree/treeSearchRun.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeSearchRun.spec.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeSearchRun.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AACzD,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,OAAO,CAAC;AAC7B,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9D,MAAM,gBAAgB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACnD,MAAM,kBAAkB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACrD,MAAM,gBAAgB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAEnD,EAAE,CAAC,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3C,YAAY,EAAE,gBAAgB;CACjC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,kCAAkC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/C,cAAc,EAAE,kBAAkB;CACrC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7C,YAAY,EAAE,gBAAgB;CACjC,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAEhE,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC3B,UAAU,CAAC,GAAG,EAAE;QACZ,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC7B,kBAAkB,CAAC,SAAS,EAAE,CAAC;QAC/B,gBAAgB,CAAC,SAAS,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC;YACD,gBAAgB,CAAC,iBAAiB,CAAC;gBAC/B,QAAQ,EAAE,IAAI;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE;oBAC/B,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE;iBAC1C,CAAC;aACL,CAAC,CAAC;YAEH,IAAI,MAAM,GAAG,CAAC,CAAC;YACf,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,kBAAkB,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBAC7C,MAAM,IAAI,CAAC,CAAC;gBACZ,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBAC9B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;gBACxD,MAAM,IAAI,CAAC,CAAC;gBACZ,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC5D,CAAC,CAAC,CAAC;YACH,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEzC,MAAM,OAAO,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAC9C,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE;gBACxC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC;gBACnC,UAAU,EAAE,eAAe;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;gBAChC,WAAW,EAAE,CAAC;gBACd,QAAQ,EAAE,CAAC;aACd,CAAC,CAAC;YAEH,MAAM,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAErC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAC3B,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,OAAO,CAAC,CAC7B,CAAC;YAC5C,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;gBACzB,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC/B,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE;aAC1C,CAAC,CAAC;QACP,CAAC;gBAAS,CAAC;YACP,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC;YACD,gBAAgB,CAAC,iBAAiB,CAAC;gBAC/B,QAAQ,EAAE,IAAI;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;aAC1D,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC3C,MAAM,KAAK,GAAG,iBAAiB,CAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EACpC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,EAC1B,QAAQ,EACR,CAAC,CACJ,CAAC;YACF,MAAM,KAAK,GAAG,iBAAiB,CAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,EACtC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,EAC1B,UAAU,EACV,CAAC,CACJ,CAAC;YAEF,gBAAgB;iBACX,qBAAqB,CAAC,KAAK,CAAC;iBAC5B,qBAAqB,CAAC,KAAK,CAAC;iBAC5B,qBAAqB,CAAC,IAAI,CAAC;iBAC3B,qBAAqB,CAAC,IAAI,CAAC,CAAC;YAEjC,IAAI,MAAM,GAAG,CAAC,CAAC;YACf,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,kBAAkB,CAAC,kBAAkB,CAAC,KAAK,EAAE,IAAa,EAAE,IAAc,EAAE,EAAE;gBAC1E,MAAM,IAAI,CAAC,CAAC;gBACZ,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBAC9B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;gBACxD,MAAM,IAAI,CAAC,CAAC;gBACZ,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAC9C,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE;gBACxC,OAAO;gBACP,UAAU,EAAE,eAAe;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;gBAChC,WAAW,EAAE,CAAC;gBACd,QAAQ,EAAE,CAAC;aACd,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;YACpC,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC;gBAAS,CAAC;YACP,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC;YACD,gBAAgB,CAAC,iBAAiB,CAAC;gBAC/B,QAAQ,EAAE,IAAI;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;aAC1D,CAAC,CAAC;YACH,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACzC,kBAAkB,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YAEvF,MAAM,OAAO,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC3C,MAAM,aAAa,CAAC,OAAO,EAAE;gBACzB,OAAO;gBACP,UAAU,EAAE,eAAe;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;gBAChC,QAAQ,EAAE,CAAC;aACd,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;YAC5D,MAAM,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAClD,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;gBAAS,CAAC;YACP,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC;YACD,gBAAgB,CAAC,iBAAiB,CAAC;gBAC/B,QAAQ,EAAE,IAAI;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;aAC1D,CAAC,CAAC;YACH,kBAAkB,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACtF,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEzC,MAAM,OAAO,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAC9C,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE;gBACxC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC;gBACnC,UAAU,EAAE,eAAe;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;aACnC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC;gBAAS,CAAC;YACP,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC;YACD,gBAAgB,CAAC,iBAAiB,CAAC;gBAC/B,QAAQ,EAAE,IAAI;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;aAC1D,CAAC,CAAC;YACH,kBAAkB,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACtF,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,gBAAgB,GAAa,EAAE,CAAC;YAEtC,MAAM,OAAO,GAAG,oBAAoB,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;YAChE,MAAM,aAAa,CAAC,OAAO,EAAE;gBACzB,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC;gBACnC,UAAU,EAAE,eAAe;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;aACnC,CAAC,CAAC;YAEH,MAAM,CAAC,gBAAgB,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;QAC3E,CAAC;gBAAS,CAAC;YACP,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,SAAS,iBAAiB,CAAC,OAAe,EAAE,aAAqB,EAAE,IAAY,EAAE,KAAa;IAC1F,OAAO;QACH,IAAI;QACJ,KAAK,EAAE,IAAI;QACX,KAAK;QACL,OAAO;QACP,aAAa;QACb,MAAM,EAAE,YAAY;QACpB,QAAQ,EAAE,EAAE;KACf,CAAC;AACN,CAAC;AAED,SAAS,oBAAoB,CAAC,WAAmB,EAAE,gBAA2B;IAC1E,OAAO;QACH,WAAW;QACX,SAAS,EAAE,EAAE;QACb,QAAQ,EAAE,KAAK,EACX,cAAsB,EACtB,SAA4D,EAClD,EAAE;YACZ,gBAAgB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;YACvC,OAAO,SAAS,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC/B,CAAC;KACkB,CAAC;AAC5B,CAAC"}

package/dist/modules/tree/treeSearchTypes.d.ts

@@ -0,0 +1,40 @@
+import type { ZodTypeAny } from "zod";
+import type { ProviderModelSelectionMode } from "@/types";
+export type TreeNodeStatus = "unexpanded" | "in-progress" | "expanded" | "leaf";
+export interface TreeSearchConfig {
+    rootDir: string;
+    rootPrompt: string;
+    concurrency?: number;
+    maxDepth?: number;
+    modelSelectionMode?: ProviderModelSelectionMode;
+    frontmatterSchema?: ZodTypeAny;
+    documentPrompt: (node: TreeNode) => string;
+    childrenPrompt: (node: TreeNode) => string;
+    pickerPrompt?: (treeOutline: string, leaves: TreeNode[]) => string;
+}
+export interface TreeNode {
+    slug: string;
+    title: string;
+    depth: number;
+    dirPath: string;
+    parentDirPath: string;
+    sessionId?: string;
+    status: TreeNodeStatus;
+    children: TreeNode[];
+}
+export interface TreeChildEntry {
+    title: string;
+    slug: string;
+}
+export interface TreeNodeDisk {
+    title: string;
+    slug: string;
+    sessionId?: string;
+    status: string;
+}
+export interface TreeSearchResult {
+    totalExpanded: number;
+    totalLeaves: number;
+    totalSkipped: number;
+}
+//# sourceMappingURL=treeSearchTypes.d.ts.map

package/dist/modules/tree/treeSearchTypes.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeSearchTypes.d.ts","sourceRoot":"","sources":["../../../sources/modules/tree/treeSearchTypes.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,KAAK,CAAC;AACtC,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,SAAS,CAAC;AAE1D,MAAM,MAAM,cAAc,GAAG,YAAY,GAAG,aAAa,GAAG,UAAU,GAAG,MAAM,CAAC;AAEhF,MAAM,WAAW,gBAAgB;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,kBAAkB,CAAC,EAAE,0BAA0B,CAAC;IAChD,iBAAiB,CAAC,EAAE,UAAU,CAAC;IAC/B,cAAc,EAAE,CAAC,IAAI,EAAE,QAAQ,KAAK,MAAM,CAAC;IAC3C,cAAc,EAAE,CAAC,IAAI,EAAE,QAAQ,KAAK,MAAM,CAAC;IAC3C,YAAY,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,MAAM,CAAC;CACtE;AAED,MAAM,WAAW,QAAQ;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,cAAc,CAAC;IACvB,QAAQ,EAAE,QAAQ,EAAE,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,YAAY;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,gBAAgB;IAC7B,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;CACxB"}

package/dist/modules/tree/treeSearchTypes.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=treeSearchTypes.js.map

package/dist/modules/tree/treeSearchTypes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeSearchTypes.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeSearchTypes.ts"],"names":[],"mappings":""}

package/dist/modules/tree/treeSearchTypes.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=treeSearchTypes.spec.d.ts.map

package/dist/modules/tree/treeSearchTypes.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeSearchTypes.spec.d.ts","sourceRoot":"","sources":["../../../sources/modules/tree/treeSearchTypes.spec.ts"],"names":[],"mappings":""}

package/dist/modules/tree/treeSearchTypes.spec.js

@@ -0,0 +1,40 @@
+import { describe, expect, it } from "vitest";
+describe("treeSearchTypes", () => {
+    it("exposes cross-cutting tree types from @/types", () => {
+        const child = {
+            title: "Auth",
+            slug: "auth"
+        };
+        const node = {
+            slug: "auth",
+            title: "Auth",
+            depth: 1,
+            dirPath: "/tmp/tree/auth",
+            parentDirPath: "/tmp/tree",
+            status: "unexpanded",
+            children: []
+        };
+        const nodeDisk = {
+            title: "Auth",
+            slug: "auth",
+            status: "unexpanded"
+        };
+        const config = {
+            rootDir: "/tmp/tree",
+            rootPrompt: "Create top-level research topics.",
+            documentPrompt: (entry) => `Document: ${entry.title}`,
+            childrenPrompt: (entry) => `Children: ${entry.title}`
+        };
+        const result = {
+            totalExpanded: 1,
+            totalLeaves: 0,
+            totalSkipped: 0
+        };
+        expect(child.slug).toBe("auth");
+        expect(node.status).toBe("unexpanded");
+        expect(nodeDisk.slug).toBe("auth");
+        expect(config.rootDir).toBe("/tmp/tree");
+        expect(result.totalExpanded).toBe(1);
+    });
+});
+//# sourceMappingURL=treeSearchTypes.spec.js.map

package/dist/modules/tree/treeSearchTypes.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeSearchTypes.spec.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeSearchTypes.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9C,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACrD,MAAM,KAAK,GAAmB;YAC1B,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,MAAM;SACf,CAAC;QACF,MAAM,IAAI,GAAa;YACnB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,MAAM;YACb,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,gBAAgB;YACzB,aAAa,EAAE,WAAW;YAC1B,MAAM,EAAE,YAAY;YACpB,QAAQ,EAAE,EAAE;SACf,CAAC;QACF,MAAM,QAAQ,GAAiB;YAC3B,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,YAAY;SACvB,CAAC;QACF,MAAM,MAAM,GAAqB;YAC7B,OAAO,EAAE,WAAW;YACpB,UAAU,EAAE,mCAAmC;YAC/C,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,KAAK,CAAC,KAAK,EAAE;YACrD,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,KAAK,CAAC,KAAK,EAAE;SACxD,CAAC;QACF,MAAM,MAAM,GAAqB;YAC7B,aAAa,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;SAClB,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACvC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/dist/modules/tree/treeStateLeaves.d.ts

@@ -0,0 +1,7 @@
+import type { TreeNode } from "@/types";
+/**
+ * Collects unexpanded leaves whose parent is expanded.
+ * Expects: nodes are root-level tree entries; root is treated as expanded.
+ */
+export declare function treeStateLeaves(nodes: readonly TreeNode[]): TreeNode[];
+//# sourceMappingURL=treeStateLeaves.d.ts.map

package/dist/modules/tree/treeStateLeaves.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeStateLeaves.d.ts","sourceRoot":"","sources":["../../../sources/modules/tree/treeStateLeaves.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAExC;;;GAGG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,SAAS,QAAQ,EAAE,GAAG,QAAQ,EAAE,CAMtE"}

package/dist/modules/tree/treeStateLeaves.js

@@ -0,0 +1,20 @@
+/**
+ * Collects unexpanded leaves whose parent is expanded.
+ * Expects: nodes are root-level tree entries; root is treated as expanded.
+ */
+export function treeStateLeaves(nodes) {
+    const leaves = [];
+    for (const node of nodes) {
+        treeStateLeavesCollect(leaves, node, "expanded");
+    }
+    return leaves;
+}
+function treeStateLeavesCollect(leaves, node, parentStatus) {
+    if (node.status === "unexpanded" && parentStatus === "expanded") {
+        leaves.push(node);
+    }
+    for (const child of node.children) {
+        treeStateLeavesCollect(leaves, child, node.status);
+    }
+}
+//# sourceMappingURL=treeStateLeaves.js.map

package/dist/modules/tree/treeStateLeaves.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeStateLeaves.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeStateLeaves.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,KAA0B;IACtD,MAAM,MAAM,GAAe,EAAE,CAAC;IAC9B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,sBAAsB,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;IACrD,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,sBAAsB,CAAC,MAAkB,EAAE,IAAc,EAAE,YAAgC;IAChG,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,IAAI,YAAY,KAAK,UAAU,EAAE,CAAC;QAC9D,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChC,sBAAsB,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACvD,CAAC;AACL,CAAC"}

package/dist/modules/tree/treeStateLeaves.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=treeStateLeaves.spec.d.ts.map

package/dist/modules/tree/treeStateLeaves.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeStateLeaves.spec.d.ts","sourceRoot":"","sources":["../../../sources/modules/tree/treeStateLeaves.spec.ts"],"names":[],"mappings":""}

package/dist/modules/tree/treeStateLeaves.spec.js

@@ -0,0 +1,57 @@
+import { describe, expect, it } from "vitest";
+import { treeStateLeaves } from "@/modules/tree/treeStateLeaves.js";
+describe("treeStateLeaves", () => {
+    it("returns unexpanded nodes with expanded parents only", () => {
+        const tree = [
+            {
+                slug: "auth",
+                title: "Auth",
+                depth: 1,
+                dirPath: "/tmp/auth",
+                parentDirPath: "/tmp",
+                status: "expanded",
+                children: [
+                    {
+                        slug: "oauth2",
+                        title: "OAuth2",
+                        depth: 2,
+                        dirPath: "/tmp/auth/oauth2",
+                        parentDirPath: "/tmp/auth",
+                        status: "unexpanded",
+                        children: []
+                    },
+                    {
+                        slug: "sessions",
+                        title: "Sessions",
+                        depth: 2,
+                        dirPath: "/tmp/auth/sessions",
+                        parentDirPath: "/tmp/auth",
+                        status: "in-progress",
+                        children: []
+                    }
+                ]
+            },
+            {
+                slug: "database",
+                title: "Database",
+                depth: 1,
+                dirPath: "/tmp/database",
+                parentDirPath: "/tmp",
+                status: "unexpanded",
+                children: [
+                    {
+                        slug: "queries",
+                        title: "Queries",
+                        depth: 2,
+                        dirPath: "/tmp/database/queries",
+                        parentDirPath: "/tmp/database",
+                        status: "unexpanded",
+                        children: []
+                    }
+                ]
+            }
+        ];
+        expect(treeStateLeaves(tree).map((entry) => entry.slug)).toEqual(["oauth2", "database"]);
+    });
+});
+//# sourceMappingURL=treeStateLeaves.spec.js.map

package/dist/modules/tree/treeStateLeaves.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeStateLeaves.spec.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeStateLeaves.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AAGpE,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC3D,MAAM,IAAI,GAAe;YACrB;gBACI,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,MAAM;gBACb,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,WAAW;gBACpB,aAAa,EAAE,MAAM;gBACrB,MAAM,EAAE,UAAU;gBAClB,QAAQ,EAAE;oBACN;wBACI,IAAI,EAAE,QAAQ;wBACd,KAAK,EAAE,QAAQ;wBACf,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE,kBAAkB;wBAC3B,aAAa,EAAE,WAAW;wBAC1B,MAAM,EAAE,YAAY;wBACpB,QAAQ,EAAE,EAAE;qBACf;oBACD;wBACI,IAAI,EAAE,UAAU;wBAChB,KAAK,EAAE,UAAU;wBACjB,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE,oBAAoB;wBAC7B,aAAa,EAAE,WAAW;wBAC1B,MAAM,EAAE,aAAa;wBACrB,QAAQ,EAAE,EAAE;qBACf;iBACJ;aACJ;YACD;gBACI,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,UAAU;gBACjB,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,eAAe;gBACxB,aAAa,EAAE,MAAM;gBACrB,MAAM,EAAE,YAAY;gBACpB,QAAQ,EAAE;oBACN;wBACI,IAAI,EAAE,SAAS;wBACf,KAAK,EAAE,SAAS;wBAChB,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE,uBAAuB;wBAChC,aAAa,EAAE,eAAe;wBAC9B,MAAM,EAAE,YAAY;wBACpB,QAAQ,EAAE,EAAE;qBACf;iBACJ;aACJ;SACJ,CAAC;QAEF,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/dist/modules/tree/treeStateRead.d.ts

@@ -0,0 +1,7 @@
+import type { TreeNode } from "@/types";
+/**
+ * Reads persisted tree state from disk starting at root children.json.
+ * Expects: rootDir is the tree root directory containing root-level children.json.
+ */
+export declare function treeStateRead(rootDir: string): Promise<TreeNode[]>;
+//# sourceMappingURL=treeStateRead.d.ts.map

package/dist/modules/tree/treeStateRead.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"treeStateRead.d.ts","sourceRoot":"","sources":["../../../sources/modules/tree/treeStateRead.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAkB,QAAQ,EAAkB,MAAM,SAAS,CAAC;AAExE;;;GAGG;AACH,wBAAsB,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAOxE"}