@slopus/beer 0.1.1 → 0.1.4

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

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

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

@@ -0,0 +1,35 @@
+import { describe, expect, it, vi } from "vitest";
+import { treeInferenceProgressRun } from "../../modules/tree/treeInferenceProgressRun.js";
+describe("treeInferenceProgressRun", () => {
+    it("maps inference events to progress updates", async () => {
+        const updates = [];
+        const ctx = {
+            progress: async (_initialMessage, operation) => operation((message) => updates.push(message))
+        };
+        const runMock = vi.fn(async (onEvent) => {
+            onEvent({
+                providerId: "pi",
+                type: "thinking",
+                status: "started",
+                text: "thinking"
+            });
+            onEvent({
+                providerId: "pi",
+                type: "usage",
+                tokens: {
+                    input: 10,
+                    output: 5,
+                    cacheRead: 0,
+                    cacheWrite: 0,
+                    total: 15
+                }
+            });
+            return "ok";
+        });
+        const result = await treeInferenceProgressRun(ctx, "Generating root children...", runMock);
+        expect(result).toBe("ok");
+        expect(updates.some((entry) => entry.includes("thinking"))).toBe(true);
+        expect(updates.some((entry) => entry.includes("tokens 15"))).toBe(true);
+    });
+});
+//# sourceMappingURL=treeInferenceProgressRun.spec.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"treeInferenceProgressRun.spec.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeInferenceProgressRun.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,wBAAwB,EAAE,MAAM,4CAA4C,CAAC;AAGtF,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG;YACR,QAAQ,EAAE,KAAK,EACX,eAAuB,EACvB,SAA4D,EAClD,EAAE,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;SAC3C,CAAC;QAExB,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,OAAuC,EAAE,EAAE;YACpE,OAAO,CAAC;gBACJ,UAAU,EAAE,IAAI;gBAChB,IAAI,EAAE,UAAU;gBAChB,MAAM,EAAE,SAAS;gBACjB,IAAI,EAAE,UAAU;aACnB,CAAC,CAAC;YACH,OAAO,CAAC;gBACJ,UAAU,EAAE,IAAI;gBAChB,IAAI,EAAE,OAAO;gBACb,MAAM,EAAE;oBACJ,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,SAAS,EAAE,CAAC;oBACZ,UAAU,EAAE,CAAC;oBACb,KAAK,EAAE,EAAE;iBACZ;aACJ,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,GAAG,EAAE,6BAA6B,EAAE,OAAO,CAAC,CAAC;QAE3F,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,7 @@
+import type { Context, TreeNode, TreeSearchConfig } from "../../types.js";
+/**
+ * Picks the next unexpanded leaf using an inference call over the rendered tree state.
+ * Expects: config.rootDir contains persisted tree node metadata.
+ */
+export declare function treeLeafPick(ctx: Context, config: TreeSearchConfig): Promise<TreeNode | null>;
+//# sourceMappingURL=treeLeafPick.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"treeLeafPick.d.ts","sourceRoot":"","sources":["../../../sources/modules/tree/treeLeafPick.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAEnE;;;GAGG;AACH,wBAAsB,YAAY,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC,CAsCnG"}

package/dist/modules/tree/treeLeafPick.js

@@ -0,0 +1,67 @@
+import { text } from "../../text/text.js";
+import { generateText } from "../../modules/ai/generateText.js";
+import { treeInferenceProgressRun } from "../../modules/tree/treeInferenceProgressRun.js";
+import { treeNodePathResolve } from "../../modules/tree/treeNodePathResolve.js";
+import { treeNodeRead } from "../../modules/tree/treeNodeRead.js";
+import { treeStateLeaves } from "../../modules/tree/treeStateLeaves.js";
+import { treeStateRead } from "../../modules/tree/treeStateRead.js";
+import { treeStateRender } from "../../modules/tree/treeStateRender.js";
+/**
+ * Picks the next unexpanded leaf using an inference call over the rendered tree state.
+ * Expects: config.rootDir contains persisted tree node metadata.
+ */
+export async function treeLeafPick(ctx, config) {
+    const maxAttempts = 5;
+    for (let attempt = 0; attempt < maxAttempts; attempt += 1) {
+        const state = await treeStateRead(config.rootDir);
+        const leaves = treeStateLeaves(state);
+        if (leaves.length === 0) {
+            return null;
+        }
+        const treeOutline = treeStateRender(state);
+        const prompt = config.pickerPrompt?.(treeOutline, leaves) ??
+            treeLeafPickPromptDefault(config.rootDir, treeOutline, leaves);
+        const result = await treeInferenceProgressRun(ctx, text.tree_search_picking_leaf, async (onEvent) => {
+            return generateText(ctx, prompt, {
+                modelSelectionMode: config.modelSelectionMode,
+                onEvent
+            });
+        });
+        const selectedPath = treeLeafPathNormalize(result.text);
+        const selected = leaves.find((leaf) => treeNodePathResolve(config.rootDir, leaf) === selectedPath);
+        if (!selected) {
+            continue;
+        }
+        const latestNode = await treeNodeRead(selected.dirPath);
+        if (latestNode?.status === "in-progress") {
+            continue;
+        }
+        if (latestNode && latestNode.status !== "unexpanded") {
+            continue;
+        }
+        return selected;
+    }
+    return null;
+}
+function treeLeafPickPromptDefault(rootDir, treeOutline, leaves) {
+    const candidates = leaves.map((leaf) => `- ${treeNodePathResolve(rootDir, leaf)}`).join("\n");
+    return [
+        "You are a research planner analyzing a tree of topics being explored.",
+        "",
+        "Here is the current state of the research tree:",
+        treeOutline,
+        "",
+        "Candidate unexpanded leaves:",
+        candidates,
+        "",
+        "Select the ONE most important unexpanded leaf to research next.",
+        "Consider: coverage gaps, dependencies between topics, foundational vs advanced.",
+        "",
+        "Return ONLY the slug path inside <output> tags.",
+        "Example: <output>authentication-system/oauth2-flow</output>"
+    ].join("\n");
+}
+function treeLeafPathNormalize(raw) {
+    return raw.trim().replace(/^\/+|\/+$/g, "");
+}
+//# sourceMappingURL=treeLeafPick.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"treeLeafPick.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeLeafPick.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,OAAO,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAC5D,OAAO,EAAE,wBAAwB,EAAE,MAAM,4CAA4C,CAAC;AACtF,OAAO,EAAE,mBAAmB,EAAE,MAAM,uCAAuC,CAAC;AAC5E,OAAO,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AAGpE;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,GAAY,EAAE,MAAwB;IACrE,MAAM,WAAW,GAAG,CAAC,CAAC;IAEtB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,WAAW,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;QACxD,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,MAAM,WAAW,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,MAAM,GACR,MAAM,CAAC,YAAY,EAAE,CAAC,WAAW,EAAE,MAAM,CAAC;YAC1C,yBAAyB,CAAC,MAAM,CAAC,OAAO,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;QACnE,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,GAAG,EAAE,IAAI,CAAC,wBAAyB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACjG,OAAO,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE;gBAC7B,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;gBAC7C,OAAO;aACV,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,YAAY,CAAC,CAAC;QACnG,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,SAAS;QACb,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACxD,IAAI,UAAU,EAAE,MAAM,KAAK,aAAa,EAAE,CAAC;YACvC,SAAS;QACb,CAAC;QACD,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YACnD,SAAS;QACb,CAAC;QAED,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,yBAAyB,CAAC,OAAe,EAAE,WAAmB,EAAE,MAA2B;IAChG,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAE9F,OAAO;QACH,uEAAuE;QACvE,EAAE;QACF,iDAAiD;QACjD,WAAW;QACX,EAAE;QACF,8BAA8B;QAC9B,UAAU;QACV,EAAE;QACF,iEAAiE;QACjE,iFAAiF;QACjF,EAAE;QACF,iDAAiD;QACjD,6DAA6D;KAChE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACjB,CAAC;AAED,SAAS,qBAAqB,CAAC,GAAW;IACtC,OAAO,GAAG,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;AAChD,CAAC"}

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

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

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

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

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

@@ -0,0 +1,119 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+const generateTextMock = vi.hoisted(() => vi.fn());
+const treeStateReadMock = vi.hoisted(() => vi.fn());
+const treeStateRenderMock = vi.hoisted(() => vi.fn());
+const treeStateLeavesMock = vi.hoisted(() => vi.fn());
+const treeNodeReadMock = vi.hoisted(() => vi.fn());
+vi.mock("@/modules/ai/generateText.js", () => ({
+    generateText: generateTextMock
+}));
+vi.mock("@/modules/tree/treeStateRead.js", () => ({
+    treeStateRead: treeStateReadMock
+}));
+vi.mock("@/modules/tree/treeStateRender.js", () => ({
+    treeStateRender: treeStateRenderMock
+}));
+vi.mock("@/modules/tree/treeStateLeaves.js", () => ({
+    treeStateLeaves: treeStateLeavesMock
+}));
+vi.mock("@/modules/tree/treeNodeRead.js", () => ({
+    treeNodeRead: treeNodeReadMock
+}));
+import { treeLeafPick } from "../../modules/tree/treeLeafPick.js";
+describe("treeLeafPick", () => {
+    beforeEach(() => {
+        generateTextMock.mockReset();
+        treeStateReadMock.mockReset();
+        treeStateRenderMock.mockReset();
+        treeStateLeavesMock.mockReset();
+        treeNodeReadMock.mockReset();
+    });
+    it("returns null when no unexpanded leaves remain", async () => {
+        treeStateReadMock.mockResolvedValue([]);
+        treeStateLeavesMock.mockReturnValue([]);
+        const context = treeLeafPickContext();
+        const result = await treeLeafPick(context, {
+            rootDir: "/tmp/tree",
+            rootPrompt: "root",
+            documentPrompt: () => "doc",
+            childrenPrompt: () => "children"
+        });
+        expect(result).toBeNull();
+        expect(generateTextMock).not.toHaveBeenCalled();
+    });
+    it("uses tree outline prompt and returns selected leaf", async () => {
+        const leafA = treeLeafPickNode("/tmp/tree/auth/oauth2", "/tmp/tree/auth", "oauth2", "OAuth2");
+        const leafB = treeLeafPickNode("/tmp/tree/db/migrations", "/tmp/tree/db", "migrations", "Migrations");
+        treeStateReadMock.mockResolvedValue([leafA, leafB]);
+        treeStateRenderMock.mockReturnValue("- Auth [expanded]");
+        treeStateLeavesMock.mockReturnValue([leafA, leafB]);
+        generateTextMock.mockResolvedValue({ provider: "pi", text: "auth/oauth2" });
+        treeNodeReadMock.mockResolvedValue({ title: "OAuth2", slug: "oauth2", status: "unexpanded" });
+        const context = treeLeafPickContext();
+        const result = await treeLeafPick(context, {
+            rootDir: "/tmp/tree",
+            rootPrompt: "root",
+            documentPrompt: () => "doc",
+            childrenPrompt: () => "children",
+            modelSelectionMode: "codex-xhigh"
+        });
+        expect(result).toBe(leafA);
+        expect(generateTextMock).toHaveBeenCalledWith(context, expect.stringContaining("- Auth [expanded]"), expect.objectContaining({ modelSelectionMode: "codex-xhigh" }));
+    });
+    it("retries when picker returns a node already marked in-progress", async () => {
+        const leaf = treeLeafPickNode("/tmp/tree/auth/oauth2", "/tmp/tree/auth", "oauth2", "OAuth2");
+        treeStateReadMock.mockResolvedValue([leaf]);
+        treeStateRenderMock.mockReturnValue("- Auth [expanded]");
+        treeStateLeavesMock.mockReturnValue([leaf]);
+        generateTextMock
+            .mockResolvedValueOnce({ provider: "pi", text: "auth/oauth2" })
+            .mockResolvedValueOnce({ provider: "pi", text: "auth/oauth2" });
+        treeNodeReadMock
+            .mockResolvedValueOnce({ title: "OAuth2", slug: "oauth2", status: "in-progress" })
+            .mockResolvedValueOnce({ title: "OAuth2", slug: "oauth2", status: "unexpanded" });
+        const context = treeLeafPickContext();
+        const result = await treeLeafPick(context, {
+            rootDir: "/tmp/tree",
+            rootPrompt: "root",
+            documentPrompt: () => "doc",
+            childrenPrompt: () => "children"
+        });
+        expect(result).toBe(leaf);
+        expect(generateTextMock).toHaveBeenCalledTimes(2);
+    });
+    it("returns null when picker keeps returning invalid paths", async () => {
+        const leaf = treeLeafPickNode("/tmp/tree/auth/oauth2", "/tmp/tree/auth", "oauth2", "OAuth2");
+        treeStateReadMock.mockResolvedValue([leaf]);
+        treeStateRenderMock.mockReturnValue("- Auth [expanded]");
+        treeStateLeavesMock.mockReturnValue([leaf]);
+        generateTextMock.mockResolvedValue({ provider: "pi", text: "not-a-real-path" });
+        const context = treeLeafPickContext();
+        const result = await treeLeafPick(context, {
+            rootDir: "/tmp/tree",
+            rootPrompt: "root",
+            documentPrompt: () => "doc",
+            childrenPrompt: () => "children"
+        });
+        expect(result).toBeNull();
+        expect(generateTextMock).toHaveBeenCalledTimes(5);
+    });
+});
+function treeLeafPickNode(dirPath, parentDirPath, slug, title) {
+    return {
+        slug,
+        title,
+        depth: 2,
+        dirPath,
+        parentDirPath,
+        status: "unexpanded",
+        children: []
+    };
+}
+function treeLeafPickContext() {
+    return {
+        projectPath: "/tmp/project",
+        providers: [],
+        progress: async (_initialMessage, operation) => operation(() => { })
+    };
+}
+//# sourceMappingURL=treeLeafPick.spec.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"treeLeafPick.spec.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeLeafPick.spec.ts"],"names":[],"mappings":"AAAA,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,iBAAiB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACpD,MAAM,mBAAmB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACtD,MAAM,mBAAmB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACtD,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,iCAAiC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9C,aAAa,EAAE,iBAAiB;CACnC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC;IAChD,eAAe,EAAE,mBAAmB;CACvC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC;IAChD,eAAe,EAAE,mBAAmB;CACvC,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;AAE9D,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC1B,UAAU,CAAC,GAAG,EAAE;QACZ,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC7B,iBAAiB,CAAC,SAAS,EAAE,CAAC;QAC9B,mBAAmB,CAAC,SAAS,EAAE,CAAC;QAChC,mBAAmB,CAAC,SAAS,EAAE,CAAC;QAChC,gBAAgB,CAAC,SAAS,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC3D,iBAAiB,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QACxC,mBAAmB,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QAExC,MAAM,OAAO,GAAG,mBAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;YACvC,OAAO,EAAE,WAAW;YACpB,UAAU,EAAE,MAAM;YAClB,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;YAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;SACnC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,KAAK,GAAG,gBAAgB,CAAC,uBAAuB,EAAE,gBAAgB,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC9F,MAAM,KAAK,GAAG,gBAAgB,CAAC,yBAAyB,EAAE,cAAc,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;QACtG,iBAAiB,CAAC,iBAAiB,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;QACpD,mBAAmB,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACzD,mBAAmB,CAAC,eAAe,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;QACpD,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;QAC5E,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;QAE9F,MAAM,OAAO,GAAG,mBAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;YACvC,OAAO,EAAE,WAAW;YACpB,UAAU,EAAE,MAAM;YAClB,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;YAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;YAChC,kBAAkB,EAAE,aAAa;SACpC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3B,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CACzC,OAAO,EACP,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAC5C,MAAM,CAAC,gBAAgB,CAAC,EAAE,kBAAkB,EAAE,aAAa,EAAE,CAAC,CACjE,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,IAAI,GAAG,gBAAgB,CAAC,uBAAuB,EAAE,gBAAgB,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC7F,iBAAiB,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5C,mBAAmB,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACzD,mBAAmB,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5C,gBAAgB;aACX,qBAAqB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;aAC9D,qBAAqB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;QACpE,gBAAgB;aACX,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;aACjF,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;QAEtF,MAAM,OAAO,GAAG,mBAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;YACvC,OAAO,EAAE,WAAW;YACpB,UAAU,EAAE,MAAM;YAClB,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;YAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;SACnC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,IAAI,GAAG,gBAAgB,CAAC,uBAAuB,EAAE,gBAAgB,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC7F,iBAAiB,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5C,mBAAmB,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACzD,mBAAmB,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5C,gBAAgB,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC;QAEhF,MAAM,OAAO,GAAG,mBAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;YACvC,OAAO,EAAE,WAAW;YACpB,UAAU,EAAE,MAAM;YAClB,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;YAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;SACnC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,SAAS,gBAAgB,CAAC,OAAe,EAAE,aAAqB,EAAE,IAAY,EAAE,KAAa;IACzF,OAAO;QACH,IAAI;QACJ,KAAK;QACL,KAAK,EAAE,CAAC;QACR,OAAO;QACP,aAAa;QACb,MAAM,EAAE,YAAY;QACpB,QAAQ,EAAE,EAAE;KACf,CAAC;AACN,CAAC;AAED,SAAS,mBAAmB;IACxB,OAAO;QACH,WAAW,EAAE,cAAc;QAC3B,SAAS,EAAE,EAAE;QACb,QAAQ,EAAE,KAAK,EACX,eAAuB,EACvB,SAA4D,EAClD,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;KACjB,CAAC;AAC5B,CAAC"}

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

@@ -0,0 +1,12 @@
+import type { Context, TreeChildEntry, TreeNode, TreeSearchConfig } from "../../types.js";
+export interface TreeNodeExpandResult {
+    children: TreeChildEntry[];
+    status: "expanded" | "leaf";
+    skipped: boolean;
+}
+/**
+ * Expands a single node by generating its document and children list in one session.
+ * Expects: node points to an on-disk tree directory with writable permissions.
+ */
+export declare function treeNodeExpand(ctx: Context, node: TreeNode, config: TreeSearchConfig): Promise<TreeNodeExpandResult>;
+//# sourceMappingURL=treeNodeExpand.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"treeNodeExpand.d.ts","sourceRoot":"","sources":["../../../sources/modules/tree/treeNodeExpand.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAEnF,MAAM,WAAW,oBAAoB;IACjC,QAAQ,EAAE,cAAc,EAAE,CAAC;IAC3B,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC;IAC5B,OAAO,EAAE,OAAO,CAAC;CACpB;AAED;;;GAGG;AACH,wBAAsB,cAAc,CAChC,GAAG,EAAE,OAAO,EACZ,IAAI,EAAE,QAAQ,EACd,MAAM,EAAE,gBAAgB,GACzB,OAAO,CAAC,oBAAoB,CAAC,CAsG/B"}

package/dist/modules/tree/treeNodeExpand.js

@@ -0,0 +1,128 @@
+import { mkdir, stat } from "node:fs/promises";
+import path from "node:path";
+import { textFormatKey } from "../../text/text.js";
+import matter from "gray-matter";
+import { generateSessionCreate } from "../../modules/ai/generateSessionCreate.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 { treeNodeRead } from "../../modules/tree/treeNodeRead.js";
+import { treeNodeWrite } from "../../modules/tree/treeNodeWrite.js";
+/**
+ * Expands a single node by generating its document and children list in one session.
+ * Expects: node points to an on-disk tree directory with writable permissions.
+ */
+export async function treeNodeExpand(ctx, node, config) {
+    await mkdir(node.dirPath, { recursive: true });
+    const existingNode = await treeNodeRead(node.dirPath);
+    const existingChildren = await treeChildrenRead(node.dirPath);
+    const nodeStatus = treeNodeExpandStatusResolve(existingNode?.status ?? node.status);
+    const documentPath = path.join(node.dirPath, "document.md");
+    if ((nodeStatus === "expanded" || nodeStatus === "leaf") &&
+        existingChildren !== null &&
+        (await treeNodeExpandFileExists(documentPath))) {
+        return {
+            children: existingChildren,
+            status: existingChildren.length > 0 ? "expanded" : "leaf",
+            skipped: true
+        };
+    }
+    const parentSessionId = await treeNodeExpandParentSessionResolve(node.parentDirPath, config.rootDir);
+    const session = generateSessionCreate(ctx, {
+        sessionId: parentSessionId
+    });
+    const documentPrompt = config.documentPrompt(node);
+    await treeInferenceProgressRun(ctx, textFormatKey("tree_search_document_generating", { title: node.title }), async (onEvent) => session.generate(documentPrompt, {
+        modelSelectionMode: config.modelSelectionMode,
+        onEvent,
+        expectedOutput: config.frontmatterSchema
+            ? {
+                type: "file",
+                filePath: documentPath,
+                verify: ({ filePath, fileContent }) => {
+                    treeNodeExpandFrontmatterVerify(documentPath, filePath, fileContent, config.frontmatterSchema);
+                }
+            }
+            : {
+                type: "file",
+                filePath: documentPath
+            },
+        writePolicy: {
+            mode: "write-whitelist",
+            writablePaths: [documentPath]
+        }
+    }));
+    const childrenResult = await treeInferenceProgressRun(ctx, textFormatKey("tree_search_children_generating", { title: node.title }), async (onEvent) => session.generate(config.childrenPrompt(node), {
+        modelSelectionMode: config.modelSelectionMode,
+        onEvent,
+        expectedOutput: {
+            type: "text",
+            verify: ({ text: outputText }) => {
+                treeChildrenParse(outputText);
+            }
+        }
+    }));
+    const children = treeChildrenParse(childrenResult.text);
+    await treeChildrenWrite(node.dirPath, children);
+    const status = children.length > 0 ? "expanded" : "leaf";
+    await treeNodeWrite(node.dirPath, {
+        title: node.title,
+        slug: node.slug,
+        sessionId: session.sessionId,
+        status
+    });
+    for (const child of children) {
+        const childDirPath = path.join(node.dirPath, child.slug);
+        const childNode = await treeNodeRead(childDirPath);
+        if (childNode) {
+            continue;
+        }
+        await treeNodeWrite(childDirPath, {
+            title: child.title,
+            slug: child.slug,
+            status: "unexpanded"
+        });
+    }
+    return {
+        children,
+        status,
+        skipped: false
+    };
+}
+function treeNodeExpandStatusResolve(status) {
+    if (status === "unexpanded" || status === "in-progress" || status === "expanded" || status === "leaf") {
+        return status;
+    }
+    return "unexpanded";
+}
+async function treeNodeExpandParentSessionResolve(parentDirPath, rootDir) {
+    if (path.resolve(parentDirPath) === path.resolve(rootDir)) {
+        return undefined;
+    }
+    const parentNode = await treeNodeRead(parentDirPath);
+    return parentNode?.sessionId;
+}
+async function treeNodeExpandFileExists(filePath) {
+    try {
+        const fileStat = await stat(filePath);
+        return fileStat.isFile();
+    }
+    catch {
+        return false;
+    }
+}
+function treeNodeExpandFrontmatterVerify(expectedFilePath, filePath, fileContent, schema) {
+    if (path.resolve(filePath) !== path.resolve(expectedFilePath)) {
+        throw new Error(`File must be generated at path: ${expectedFilePath}`);
+    }
+    if (!matter.test(fileContent)) {
+        throw new Error(`Generated file must include frontmatter: ${expectedFilePath}`);
+    }
+    const parsed = matter(fileContent);
+    const parsedFrontmatter = schema.safeParse(parsed.data);
+    if (!parsedFrontmatter.success) {
+        throw new Error(`Generated frontmatter does not match schema: ${parsedFrontmatter.error.message}`);
+    }
+}
+//# sourceMappingURL=treeNodeExpand.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"treeNodeExpand.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeNodeExpand.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,OAAO,CAAC;AACtC,OAAO,MAAM,MAAM,aAAa,CAAC;AAEjC,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAC9E,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,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAShE;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAChC,GAAY,EACZ,IAAc,EACd,MAAwB;IAExB,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAE/C,MAAM,YAAY,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtD,MAAM,gBAAgB,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9D,MAAM,UAAU,GAAG,2BAA2B,CAAC,YAAY,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC;IACpF,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;IAE5D,IACI,CAAC,UAAU,KAAK,UAAU,IAAI,UAAU,KAAK,MAAM,CAAC;QACpD,gBAAgB,KAAK,IAAI;QACzB,CAAC,MAAM,wBAAwB,CAAC,YAAY,CAAC,CAAC,EAChD,CAAC;QACC,OAAO;YACH,QAAQ,EAAE,gBAAgB;YAC1B,MAAM,EAAE,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM;YACzD,OAAO,EAAE,IAAI;SAChB,CAAC;IACN,CAAC;IAED,MAAM,eAAe,GAAG,MAAM,kCAAkC,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IACrG,MAAM,OAAO,GAAG,qBAAqB,CAAC,GAAG,EAAE;QACvC,SAAS,EAAE,eAAe;KAC7B,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACnD,MAAM,wBAAwB,CAC1B,GAAG,EACH,aAAa,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,EACvE,KAAK,EAAE,OAAO,EAAE,EAAE,CACd,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE;QAC7B,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;QAC7C,OAAO;QACP,cAAc,EAAE,MAAM,CAAC,iBAAiB;YACpC,CAAC,CAAC;gBACI,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,YAAY;gBACtB,MAAM,EAAE,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,EAAE,EAAE;oBAClC,+BAA+B,CAC3B,YAAY,EACZ,QAAQ,EACR,WAAW,EACX,MAAM,CAAC,iBAAkB,CAC5B,CAAC;gBACN,CAAC;aACJ;YACH,CAAC,CAAC;gBACI,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,YAAY;aACzB;QACP,WAAW,EAAE;YACT,IAAI,EAAE,iBAAiB;YACvB,aAAa,EAAE,CAAC,YAAY,CAAC;SAChC;KACJ,CAAC,CACT,CAAC;IAEF,MAAM,cAAc,GAAG,MAAM,wBAAwB,CACjD,GAAG,EACH,aAAa,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,EACvE,KAAK,EAAE,OAAO,EAAE,EAAE,CACd,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE;QAC1C,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;QAC7C,OAAO;QACP,cAAc,EAAE;YACZ,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE;gBAC7B,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAClC,CAAC;SACJ;KACJ,CAAC,CACT,CAAC;IAEF,MAAM,QAAQ,GAAG,iBAAiB,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACxD,MAAM,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAEhD,MAAM,MAAM,GAAwB,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC;IAC9E,MAAM,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE;QAC9B,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,MAAM;KACT,CAAC,CAAC;IAEH,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;QAC3B,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,SAAS,EAAE,CAAC;YACZ,SAAS;QACb,CAAC;QACD,MAAM,aAAa,CAAC,YAAY,EAAE;YAC9B,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,MAAM,EAAE,YAAY;SACvB,CAAC,CAAC;IACP,CAAC;IAED,OAAO;QACH,QAAQ;QACR,MAAM;QACN,OAAO,EAAE,KAAK;KACjB,CAAC;AACN,CAAC;AAED,SAAS,2BAA2B,CAAC,MAAc;IAC/C,IAAI,MAAM,KAAK,YAAY,IAAI,MAAM,KAAK,aAAa,IAAI,MAAM,KAAK,UAAU,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;QACpG,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,OAAO,YAAY,CAAC;AACxB,CAAC;AAED,KAAK,UAAU,kCAAkC,CAAC,aAAqB,EAAE,OAAe;IACpF,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACxD,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,aAAa,CAAC,CAAC;IACrD,OAAO,UAAU,EAAE,SAAS,CAAC;AACjC,CAAC;AAED,KAAK,UAAU,wBAAwB,CAAC,QAAgB;IACpD,IAAI,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,OAAO,QAAQ,CAAC,MAAM,EAAE,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,KAAK,CAAC;IACjB,CAAC;AACL,CAAC;AAED,SAAS,+BAA+B,CACpC,gBAAwB,EACxB,QAAgB,EAChB,WAAmB,EACnB,MAAkB;IAElB,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC5D,MAAM,IAAI,KAAK,CAAC,mCAAmC,gBAAgB,EAAE,CAAC,CAAC;IAC3E,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,4CAA4C,gBAAgB,EAAE,CAAC,CAAC;IACpF,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;IACnC,MAAM,iBAAiB,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACxD,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CAAC,gDAAgD,iBAAiB,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IACvG,CAAC;AACL,CAAC"}

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

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

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

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

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

@@ -0,0 +1,146 @@
+import { mkdtemp, readFile, rm, writeFile } from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
+import { beforeEach, describe, expect, it, vi } from "vitest";
+const generateSessionCreateMock = vi.hoisted(() => vi.fn());
+vi.mock("@/modules/ai/generateSessionCreate.js", () => ({
+    generateSessionCreate: generateSessionCreateMock
+}));
+import { treeNodeExpand } from "../../modules/tree/treeNodeExpand.js";
+import { treeNodeRead } from "../../modules/tree/treeNodeRead.js";
+import { treeNodeWrite } from "../../modules/tree/treeNodeWrite.js";
+describe("treeNodeExpand", () => {
+    beforeEach(() => {
+        generateSessionCreateMock.mockReset();
+    });
+    it("runs document + children generation in one forked session and persists files", async () => {
+        const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-tree-expand-"));
+        try {
+            const rootDir = path.join(tempDir, "tree");
+            const parentDir = path.join(rootDir, "auth");
+            const nodeDir = path.join(parentDir, "oauth2");
+            await treeNodeWrite(parentDir, {
+                title: "Authentication",
+                slug: "auth",
+                sessionId: "sess-parent",
+                status: "expanded"
+            });
+            await treeNodeWrite(nodeDir, {
+                title: "OAuth2 Flow",
+                slug: "oauth2",
+                status: "unexpanded"
+            });
+            const generateMock = vi
+                .fn()
+                .mockImplementationOnce(async (_prompt, permissions) => {
+                const expectedOutput = permissions.expectedOutput;
+                await writeFile(expectedOutput.filePath, "---\ntitle: OAuth2\n---\n\nBody\n", "utf-8");
+                expectedOutput.verify?.({
+                    text: "ok",
+                    filePath: expectedOutput.filePath,
+                    fileContent: "---\ntitle: OAuth2\n---\n\nBody\n"
+                });
+                return { provider: "pi", sessionId: "sess-child", text: "wrote doc" };
+            })
+                .mockImplementationOnce(async () => {
+                return {
+                    provider: "pi",
+                    sessionId: "sess-child",
+                    text: JSON.stringify([{ title: "Token Refresh", slug: "token-refresh" }])
+                };
+            });
+            generateSessionCreateMock.mockReturnValue({
+                sessionId: "sess-child",
+                generate: generateMock
+            });
+            const context = treeNodeExpandContext(tempDir);
+            const node = {
+                title: "OAuth2 Flow",
+                slug: "oauth2",
+                depth: 2,
+                dirPath: nodeDir,
+                parentDirPath: parentDir,
+                status: "unexpanded",
+                children: []
+            };
+            const result = await treeNodeExpand(context, node, {
+                rootDir,
+                rootPrompt: "unused",
+                documentPrompt: (entry) => `Document ${entry.title}`,
+                childrenPrompt: (entry) => `Children ${entry.title}`,
+                modelSelectionMode: "codex-high"
+            });
+            expect(generateSessionCreateMock).toHaveBeenCalledWith(context, { sessionId: "sess-parent" });
+            expect(generateMock).toHaveBeenCalledTimes(2);
+            expect(result).toEqual({
+                children: [{ title: "Token Refresh", slug: "token-refresh" }],
+                status: "expanded",
+                skipped: false
+            });
+            const childrenRaw = await readFile(path.join(nodeDir, "children.json"), "utf-8");
+            expect(JSON.parse(childrenRaw)).toEqual([{ title: "Token Refresh", slug: "token-refresh" }]);
+            await expect(treeNodeRead(nodeDir)).resolves.toEqual({
+                title: "OAuth2 Flow",
+                slug: "oauth2",
+                sessionId: "sess-child",
+                status: "expanded"
+            });
+            await expect(treeNodeRead(path.join(nodeDir, "token-refresh"))).resolves.toEqual({
+                title: "Token Refresh",
+                slug: "token-refresh",
+                status: "unexpanded"
+            });
+        }
+        finally {
+            await rm(tempDir, { recursive: true, force: true });
+        }
+    });
+    it("skips inference when node is already expanded with persisted document and children", async () => {
+        const tempDir = await mkdtemp(path.join(os.tmpdir(), "holdmybeer-tree-expand-"));
+        try {
+            const rootDir = path.join(tempDir, "tree");
+            const nodeDir = path.join(rootDir, "auth");
+            await treeNodeWrite(nodeDir, {
+                title: "Authentication",
+                slug: "auth",
+                sessionId: "sess-auth",
+                status: "expanded"
+            });
+            await writeFile(path.join(nodeDir, "document.md"), "---\ntitle: Auth\n---\n", "utf-8");
+            await writeFile(path.join(nodeDir, "children.json"), "[]\n", "utf-8");
+            const context = treeNodeExpandContext(tempDir);
+            const node = {
+                title: "Authentication",
+                slug: "auth",
+                depth: 1,
+                dirPath: nodeDir,
+                parentDirPath: rootDir,
+                status: "expanded",
+                children: []
+            };
+            const result = await treeNodeExpand(context, node, {
+                rootDir,
+                rootPrompt: "unused",
+                documentPrompt: () => "doc",
+                childrenPrompt: () => "children"
+            });
+            expect(result).toEqual({
+                children: [],
+                status: "leaf",
+                skipped: true
+            });
+            expect(generateSessionCreateMock).not.toHaveBeenCalled();
+        }
+        finally {
+            await rm(tempDir, { recursive: true, force: true });
+        }
+    });
+});
+function treeNodeExpandContext(projectPath) {
+    return {
+        projectPath,
+        providers: [],
+        progress: async (_initialMessage, operation) => operation(() => { })
+    };
+}
+//# sourceMappingURL=treeNodeExpand.spec.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"treeNodeExpand.spec.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeNodeExpand.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACpE,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9D,MAAM,yBAAyB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAE5D,EAAE,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACpD,qBAAqB,EAAE,yBAAyB;CACnD,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAEhE,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC5B,UAAU,CAAC,GAAG,EAAE;QACZ,yBAAyB,CAAC,SAAS,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,yBAAyB,CAAC,CAAC,CAAC;QACjF,IAAI,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAE/C,MAAM,aAAa,CAAC,SAAS,EAAE;gBAC3B,KAAK,EAAE,gBAAgB;gBACvB,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,aAAa;gBACxB,MAAM,EAAE,UAAU;aACrB,CAAC,CAAC;YACH,MAAM,aAAa,CAAC,OAAO,EAAE;gBACzB,KAAK,EAAE,aAAa;gBACpB,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAE,YAAY;aACvB,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,EAAE;iBAClB,EAAE,EAAE;iBACJ,sBAAsB,CAAC,KAAK,EAAE,OAAe,EAAE,WAAoC,EAAE,EAAE;gBACpF,MAAM,cAAc,GAAG,WAAW,CAAC,cAIlC,CAAC;gBACF,MAAM,SAAS,CAAC,cAAc,CAAC,QAAQ,EAAE,mCAAmC,EAAE,OAAO,CAAC,CAAC;gBACvF,cAAc,CAAC,MAAM,EAAE,CAAC;oBACpB,IAAI,EAAE,IAAI;oBACV,QAAQ,EAAE,cAAc,CAAC,QAAQ;oBACjC,WAAW,EAAE,mCAAmC;iBACnD,CAAC,CAAC;gBACH,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;YAC1E,CAAC,CAAC;iBACD,sBAAsB,CAAC,KAAK,IAAI,EAAE;gBAC/B,OAAO;oBACH,QAAQ,EAAE,IAAI;oBACd,SAAS,EAAE,YAAY;oBACvB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;iBAC5E,CAAC;YACN,CAAC,CAAC,CAAC;YAEP,yBAAyB,CAAC,eAAe,CAAC;gBACtC,SAAS,EAAE,YAAY;gBACvB,QAAQ,EAAE,YAAY;aACzB,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;YAC/C,MAAM,IAAI,GAAa;gBACnB,KAAK,EAAE,aAAa;gBACpB,IAAI,EAAE,QAAQ;gBACd,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,OAAO;gBAChB,aAAa,EAAE,SAAS;gBACxB,MAAM,EAAE,YAAY;gBACpB,QAAQ,EAAE,EAAE;aACf,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,IAAI,EAAE;gBAC/C,OAAO;gBACP,UAAU,EAAE,QAAQ;gBACpB,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,YAAY,KAAK,CAAC,KAAK,EAAE;gBACpD,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,YAAY,KAAK,CAAC,KAAK,EAAE;gBACpD,kBAAkB,EAAE,YAAY;aACnC,CAAC,CAAC;YAEH,MAAM,CAAC,yBAAyB,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,CAAC;YAC9F,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACnB,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;gBAC7D,MAAM,EAAE,UAAU;gBAClB,OAAO,EAAE,KAAK;aACjB,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,eAAe,CAAC,EAAE,OAAO,CAAC,CAAC;YACjF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC;YAC7F,MAAM,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;gBACjD,KAAK,EAAE,aAAa;gBACpB,IAAI,EAAE,QAAQ;gBACd,SAAS,EAAE,YAAY;gBACvB,MAAM,EAAE,UAAU;aACrB,CAAC,CAAC;YACH,MAAM,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAC7E,KAAK,EAAE,eAAe;gBACtB,IAAI,EAAE,eAAe;gBACrB,MAAM,EAAE,YAAY;aACvB,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,oFAAoF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,yBAAyB,CAAC,CAAC,CAAC;QACjF,IAAI,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC3C,MAAM,aAAa,CAAC,OAAO,EAAE;gBACzB,KAAK,EAAE,gBAAgB;gBACvB,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,WAAW;gBACtB,MAAM,EAAE,UAAU;aACrB,CAAC,CAAC;YACH,MAAM,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,EAAE,yBAAyB,EAAE,OAAO,CAAC,CAAC;YACvF,MAAM,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,eAAe,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAEtE,MAAM,OAAO,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;YAC/C,MAAM,IAAI,GAAa;gBACnB,KAAK,EAAE,gBAAgB;gBACvB,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,OAAO;gBAChB,aAAa,EAAE,OAAO;gBACtB,MAAM,EAAE,UAAU;gBAClB,QAAQ,EAAE,EAAE;aACf,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,IAAI,EAAE;gBAC/C,OAAO;gBACP,UAAU,EAAE,QAAQ;gBACpB,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;gBAC3B,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU;aACnC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACnB,QAAQ,EAAE,EAAE;gBACZ,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,IAAI;aAChB,CAAC,CAAC;YACH,MAAM,CAAC,yBAAyB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC7D,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,qBAAqB,CAAC,WAAmB;IAC9C,OAAO;QACH,WAAW;QACX,SAAS,EAAE,EAAE;QACb,QAAQ,EAAE,KAAK,EACX,eAAuB,EACvB,SAA4D,EAClD,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;KACjB,CAAC;AAC5B,CAAC"}

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

@@ -0,0 +1,7 @@
+import type { TreeNode } from "../../types.js";
+/**
+ * Resolves a node path relative to the tree root using slash-separated slugs.
+ * Expects: node.dirPath is located under rootDir.
+ */
+export declare function treeNodePathResolve(rootDir: string, node: TreeNode): string;
+//# sourceMappingURL=treeNodePathResolve.d.ts.map

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

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

package/dist/modules/tree/treeNodePathResolve.js

@@ -0,0 +1,9 @@
+import path from "node:path";
+/**
+ * Resolves a node path relative to the tree root using slash-separated slugs.
+ * Expects: node.dirPath is located under rootDir.
+ */
+export function treeNodePathResolve(rootDir, node) {
+    return path.relative(rootDir, node.dirPath).split(path.sep).join("/");
+}
+//# sourceMappingURL=treeNodePathResolve.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"treeNodePathResolve.js","sourceRoot":"","sources":["../../../sources/modules/tree/treeNodePathResolve.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAG7B;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,OAAe,EAAE,IAAc;IAC/D,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC1E,CAAC"}

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

@@ -0,0 +1,7 @@
+import type { TreeNodeDisk } from "../../types.js";
+/**
+ * Reads node metadata from `<dir>/node.json`.
+ * Expects: dirPath points to a node directory; returns null when file does not exist.
+ */
+export declare function treeNodeRead(dirPath: string): Promise<TreeNodeDisk | null>;
+//# sourceMappingURL=treeNodeRead.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"treeNodeRead.d.ts","sourceRoot":"","sources":["../../../sources/modules/tree/treeNodeRead.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAS5C;;;GAGG;AACH,wBAAsB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,CAShF"}