@dexto/tools-plan 1.5.8 → 1.6.1

package/dist/tools/plan-update-tool.js

@@ -1,72 +1,68 @@
-/**
- * Plan Update Tool
- *
- * Updates the implementation plan for the current session.
- * Shows a diff preview for approval before saving.
- */
-import { z } from 'zod';
-import { createPatch } from 'diff';
-import { PlanError } from '../errors.js';
-const PlanUpdateInputSchema = z
-    .object({
-    content: z.string().describe('Updated plan content in markdown format'),
-})
-    .strict();
-/**
- * Generate diff preview for plan update
- */
+import { z } from "zod";
+import { createPatch } from "diff";
+import { createLocalToolCallHeader, defineTool } from "@dexto/core";
+import { PlanError } from "../errors.js";
+const PlanUpdateInputSchema = z.object({
+  content: z.string().describe("Updated plan content in markdown format")
+}).strict();
 function generateDiffPreview(filePath, originalContent, newContent) {
-    const unified = createPatch(filePath, originalContent, newContent, 'before', 'after', {
-        context: 3,
-    });
-    const additions = (unified.match(/^\+[^+]/gm) || []).length;
-    const deletions = (unified.match(/^-[^-]/gm) || []).length;
-    return {
-        type: 'diff',
-        unified,
-        filename: filePath,
-        additions,
-        deletions,
-    };
+  const unified = createPatch(filePath, originalContent, newContent, "before", "after", {
+    context: 3
+  });
+  const additions = (unified.match(/^\+[^+]/gm) || []).length;
+  const deletions = (unified.match(/^-[^-]/gm) || []).length;
+  return {
+    type: "diff",
+    title: "Update Plan",
+    unified,
+    filename: filePath,
+    additions,
+    deletions
+  };
 }
-/**
- * Creates the plan_update tool
- */
-export function createPlanUpdateTool(planService) {
-    return {
-        id: 'plan_update',
-        description: 'Update the existing implementation plan for this session. Shows a diff preview for approval before saving. The plan must already exist (use plan_create first).',
-        inputSchema: PlanUpdateInputSchema,
-        /**
-         * Generate diff preview for approval UI
-         */
-        generatePreview: async (input, context) => {
-            const { content: newContent } = input;
-            if (!context?.sessionId) {
-                throw PlanError.sessionIdRequired();
-            }
-            // Read existing plan
-            const existing = await planService.read(context.sessionId);
-            if (!existing) {
-                throw PlanError.planNotFound(context.sessionId);
-            }
-            // Generate diff preview
-            const planPath = planService.getPlanPath(context.sessionId);
-            return generateDiffPreview(planPath, existing.content, newContent);
-        },
-        execute: async (input, context) => {
-            const { content } = input;
-            if (!context?.sessionId) {
-                throw PlanError.sessionIdRequired();
-            }
-            const result = await planService.update(context.sessionId, content);
-            const planPath = planService.getPlanPath(context.sessionId);
-            return {
-                success: true,
-                path: planPath,
-                status: result.meta.status,
-                _display: generateDiffPreview(planPath, result.oldContent, result.newContent),
-            };
-        },
-    };
+function createPlanUpdateTool(getPlanService) {
+  return defineTool({
+    id: "plan_update",
+    description: "Update the existing implementation plan for this session. Shows a diff preview for approval before saving. The plan must already exist (use plan_create first).",
+    inputSchema: PlanUpdateInputSchema,
+    presentation: {
+      describeHeader: () => createLocalToolCallHeader({
+        title: "Update Plan"
+      }),
+      /**
+       * Generate diff preview for approval UI
+       */
+      preview: async (input, context) => {
+        const resolvedPlanService = await getPlanService(context);
+        const { content: newContent } = input;
+        if (!context.sessionId) {
+          throw PlanError.sessionIdRequired();
+        }
+        const existing = await resolvedPlanService.read(context.sessionId);
+        if (!existing) {
+          throw PlanError.planNotFound(context.sessionId);
+        }
+        const planPath = resolvedPlanService.getPlanPath(context.sessionId);
+        return generateDiffPreview(planPath, existing.content, newContent);
+      }
+    },
+    async execute(input, context) {
+      const resolvedPlanService = await getPlanService(context);
+      const { content } = input;
+      if (!context.sessionId) {
+        throw PlanError.sessionIdRequired();
+      }
+      const result = await resolvedPlanService.update(context.sessionId, content);
+      const planPath = resolvedPlanService.getPlanPath(context.sessionId);
+      return {
+        success: true,
+        path: planPath,
+        status: result.meta.status,
+        _display: generateDiffPreview(planPath, result.oldContent, result.newContent)
+      };
+    }
+  });
 }
+export {
+  createPlanUpdateTool
+};

package/dist/tools/plan-update-tool.test.cjs

@@ -0,0 +1,203 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var import_vitest = require("vitest");
+var path = __toESM(require("node:path"), 1);
+var fs = __toESM(require("node:fs/promises"), 1);
+var os = __toESM(require("node:os"), 1);
+var import_plan_update_tool = require("./plan-update-tool.js");
+var import_plan_service = require("../plan-service.js");
+var import_errors = require("../errors.js");
+var import_core = require("@dexto/core");
+const createMockLogger = () => {
+  const logger = {
+    debug: import_vitest.vi.fn(),
+    silly: import_vitest.vi.fn(),
+    info: import_vitest.vi.fn(),
+    warn: import_vitest.vi.fn(),
+    error: import_vitest.vi.fn(),
+    trackException: import_vitest.vi.fn(),
+    createChild: import_vitest.vi.fn(() => logger),
+    createFileOnlyChild: import_vitest.vi.fn(() => logger),
+    setLevel: import_vitest.vi.fn(),
+    getLevel: import_vitest.vi.fn(() => "debug"),
+    getLogFilePath: import_vitest.vi.fn(() => null),
+    destroy: import_vitest.vi.fn(async () => void 0)
+  };
+  return logger;
+};
+function createToolContext(logger, overrides = {}) {
+  return { logger, ...overrides };
+}
+(0, import_vitest.describe)("plan_update tool", () => {
+  let logger;
+  let tempDir;
+  let planService;
+  (0, import_vitest.beforeEach)(async () => {
+    logger = createMockLogger();
+    const rawTempDir = await fs.mkdtemp(path.join(os.tmpdir(), "dexto-plan-update-test-"));
+    tempDir = await fs.realpath(rawTempDir);
+    planService = new import_plan_service.PlanService({ basePath: tempDir }, logger);
+    import_vitest.vi.clearAllMocks();
+  });
+  (0, import_vitest.afterEach)(async () => {
+    try {
+      await fs.rm(tempDir, { recursive: true, force: true });
+    } catch {
+    }
+  });
+  (0, import_vitest.describe)("generatePreview", () => {
+    (0, import_vitest.it)("should return DiffDisplayData with unified diff", async () => {
+      const tool = (0, import_plan_update_tool.createPlanUpdateTool)(async () => planService);
+      const sessionId = "test-session";
+      const originalContent = "# Plan\n\n## Steps\n1. First step";
+      const newContent = "# Plan\n\n## Steps\n1. First step\n2. Second step";
+      const previewFn = tool.presentation?.preview;
+      (0, import_vitest.expect)(previewFn).toBeDefined();
+      await planService.create(sessionId, originalContent);
+      const preview = await previewFn(
+        { content: newContent },
+        createToolContext(logger, { sessionId })
+      );
+      (0, import_vitest.expect)(preview.type).toBe("diff");
+      (0, import_vitest.expect)(preview.title).toBe("Update Plan");
+      (0, import_vitest.expect)(preview.filename).toContain(sessionId);
+      (0, import_vitest.expect)(preview.filename).toMatch(/plan\.md$/);
+      (0, import_vitest.expect)(preview.unified).toContain("-1. First step");
+      (0, import_vitest.expect)(preview.unified).toContain("+1. First step");
+      (0, import_vitest.expect)(preview.unified).toContain("+2. Second step");
+      (0, import_vitest.expect)(preview.additions).toBeGreaterThan(0);
+    });
+    (0, import_vitest.it)("should throw error when plan does not exist", async () => {
+      const tool = (0, import_plan_update_tool.createPlanUpdateTool)(async () => planService);
+      const sessionId = "test-session";
+      const previewFn = tool.presentation?.preview;
+      (0, import_vitest.expect)(previewFn).toBeDefined();
+      try {
+        await previewFn(
+          { content: "# New Content" },
+          createToolContext(logger, { sessionId })
+        );
+        import_vitest.expect.fail("Should have thrown an error");
+      } catch (error) {
+        (0, import_vitest.expect)(error).toBeInstanceOf(import_core.DextoRuntimeError);
+        (0, import_vitest.expect)(error.code).toBe(import_errors.PlanErrorCode.PLAN_NOT_FOUND);
+      }
+    });
+    (0, import_vitest.it)("should throw error when sessionId is missing", async () => {
+      const tool = (0, import_plan_update_tool.createPlanUpdateTool)(async () => planService);
+      const previewFn = tool.presentation?.preview;
+      (0, import_vitest.expect)(previewFn).toBeDefined();
+      try {
+        await previewFn({ content: "# Content" }, createToolContext(logger));
+        import_vitest.expect.fail("Should have thrown an error");
+      } catch (error) {
+        (0, import_vitest.expect)(error).toBeInstanceOf(import_core.DextoRuntimeError);
+        (0, import_vitest.expect)(error.code).toBe(import_errors.PlanErrorCode.SESSION_ID_REQUIRED);
+      }
+    });
+    (0, import_vitest.it)("should show deletions in diff", async () => {
+      const tool = (0, import_plan_update_tool.createPlanUpdateTool)(async () => planService);
+      const sessionId = "test-session";
+      const originalContent = "# Plan\n\nLine to remove\nKeep this";
+      const newContent = "# Plan\n\nKeep this";
+      const previewFn = tool.presentation?.preview;
+      (0, import_vitest.expect)(previewFn).toBeDefined();
+      await planService.create(sessionId, originalContent);
+      const preview = await previewFn(
+        { content: newContent },
+        createToolContext(logger, { sessionId })
+      );
+      (0, import_vitest.expect)(preview.deletions).toBeGreaterThan(0);
+      (0, import_vitest.expect)(preview.unified).toContain("-Line to remove");
+    });
+  });
+  (0, import_vitest.describe)("execute", () => {
+    (0, import_vitest.it)("should update plan content and return success", async () => {
+      const tool = (0, import_plan_update_tool.createPlanUpdateTool)(async () => planService);
+      const sessionId = "test-session";
+      const originalContent = "# Original Plan";
+      const newContent = "# Updated Plan";
+      await planService.create(sessionId, originalContent);
+      const result = await tool.execute(
+        { content: newContent },
+        createToolContext(logger, { sessionId })
+      );
+      (0, import_vitest.expect)(result.success).toBe(true);
+      (0, import_vitest.expect)(result.path).toContain(sessionId);
+      (0, import_vitest.expect)(result.path).toMatch(/plan\.md$/);
+      const plan = await planService.read(sessionId);
+      (0, import_vitest.expect)(plan.content).toBe(newContent);
+    });
+    (0, import_vitest.it)("should include _display data with diff", async () => {
+      const tool = (0, import_plan_update_tool.createPlanUpdateTool)(async () => planService);
+      const sessionId = "test-session";
+      await planService.create(sessionId, "# Original");
+      const result = await tool.execute(
+        { content: "# Updated" },
+        createToolContext(logger, { sessionId })
+      );
+      (0, import_vitest.expect)(result._display).toBeDefined();
+      (0, import_vitest.expect)(result._display.type).toBe("diff");
+      (0, import_vitest.expect)(result._display.title).toBe("Update Plan");
+      (0, import_vitest.expect)(result._display.unified).toContain("-# Original");
+      (0, import_vitest.expect)(result._display.unified).toContain("+# Updated");
+    });
+    (0, import_vitest.it)("should throw error when plan does not exist", async () => {
+      const tool = (0, import_plan_update_tool.createPlanUpdateTool)(async () => planService);
+      const sessionId = "non-existent";
+      try {
+        await tool.execute(
+          { content: "# Content" },
+          createToolContext(logger, { sessionId })
+        );
+        import_vitest.expect.fail("Should have thrown an error");
+      } catch (error) {
+        (0, import_vitest.expect)(error).toBeInstanceOf(import_core.DextoRuntimeError);
+        (0, import_vitest.expect)(error.code).toBe(import_errors.PlanErrorCode.PLAN_NOT_FOUND);
+      }
+    });
+    (0, import_vitest.it)("should throw error when sessionId is missing", async () => {
+      const tool = (0, import_plan_update_tool.createPlanUpdateTool)(async () => planService);
+      try {
+        await tool.execute({ content: "# Content" }, createToolContext(logger));
+        import_vitest.expect.fail("Should have thrown an error");
+      } catch (error) {
+        (0, import_vitest.expect)(error).toBeInstanceOf(import_core.DextoRuntimeError);
+        (0, import_vitest.expect)(error.code).toBe(import_errors.PlanErrorCode.SESSION_ID_REQUIRED);
+      }
+    });
+    (0, import_vitest.it)("should preserve plan status after update", async () => {
+      const tool = (0, import_plan_update_tool.createPlanUpdateTool)(async () => planService);
+      const sessionId = "test-session";
+      await planService.create(sessionId, "# Plan");
+      await planService.updateMeta(sessionId, { status: "approved" });
+      await tool.execute(
+        { content: "# Updated Plan" },
+        createToolContext(logger, { sessionId })
+      );
+      const plan = await planService.read(sessionId);
+      (0, import_vitest.expect)(plan.meta.status).toBe("approved");
+    });
+  });
+});

package/dist/tools/plan-update-tool.test.js

@@ -1,151 +1,180 @@
-/**
- * Plan Update Tool Tests
- *
- * Tests for the plan_update tool including diff preview generation.
- */
-import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
-import * as path from 'node:path';
-import * as fs from 'node:fs/promises';
-import * as os from 'node:os';
-import { createPlanUpdateTool } from './plan-update-tool.js';
-import { PlanService } from '../plan-service.js';
-import { PlanErrorCode } from '../errors.js';
-import { DextoRuntimeError } from '@dexto/core';
-// Create mock logger
-const createMockLogger = () => ({
+import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
+import * as path from "node:path";
+import * as fs from "node:fs/promises";
+import * as os from "node:os";
+import { createPlanUpdateTool } from "./plan-update-tool.js";
+import { PlanService } from "../plan-service.js";
+import { PlanErrorCode } from "../errors.js";
+import { DextoRuntimeError } from "@dexto/core";
+const createMockLogger = () => {
+  const logger = {
     debug: vi.fn(),
+    silly: vi.fn(),
     info: vi.fn(),
     warn: vi.fn(),
     error: vi.fn(),
-    createChild: vi.fn().mockReturnThis(),
-});
-describe('plan_update tool', () => {
-    let mockLogger;
-    let tempDir;
-    let planService;
-    beforeEach(async () => {
-        mockLogger = createMockLogger();
-        const rawTempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'dexto-plan-update-test-'));
-        tempDir = await fs.realpath(rawTempDir);
-        planService = new PlanService({ basePath: tempDir }, mockLogger);
-        vi.clearAllMocks();
+    trackException: vi.fn(),
+    createChild: vi.fn(() => logger),
+    createFileOnlyChild: vi.fn(() => logger),
+    setLevel: vi.fn(),
+    getLevel: vi.fn(() => "debug"),
+    getLogFilePath: vi.fn(() => null),
+    destroy: vi.fn(async () => void 0)
+  };
+  return logger;
+};
+function createToolContext(logger, overrides = {}) {
+  return { logger, ...overrides };
+}
+describe("plan_update tool", () => {
+  let logger;
+  let tempDir;
+  let planService;
+  beforeEach(async () => {
+    logger = createMockLogger();
+    const rawTempDir = await fs.mkdtemp(path.join(os.tmpdir(), "dexto-plan-update-test-"));
+    tempDir = await fs.realpath(rawTempDir);
+    planService = new PlanService({ basePath: tempDir }, logger);
+    vi.clearAllMocks();
+  });
+  afterEach(async () => {
+    try {
+      await fs.rm(tempDir, { recursive: true, force: true });
+    } catch {
+    }
+  });
+  describe("generatePreview", () => {
+    it("should return DiffDisplayData with unified diff", async () => {
+      const tool = createPlanUpdateTool(async () => planService);
+      const sessionId = "test-session";
+      const originalContent = "# Plan\n\n## Steps\n1. First step";
+      const newContent = "# Plan\n\n## Steps\n1. First step\n2. Second step";
+      const previewFn = tool.presentation?.preview;
+      expect(previewFn).toBeDefined();
+      await planService.create(sessionId, originalContent);
+      const preview = await previewFn(
+        { content: newContent },
+        createToolContext(logger, { sessionId })
+      );
+      expect(preview.type).toBe("diff");
+      expect(preview.title).toBe("Update Plan");
+      expect(preview.filename).toContain(sessionId);
+      expect(preview.filename).toMatch(/plan\.md$/);
+      expect(preview.unified).toContain("-1. First step");
+      expect(preview.unified).toContain("+1. First step");
+      expect(preview.unified).toContain("+2. Second step");
+      expect(preview.additions).toBeGreaterThan(0);
+    });
+    it("should throw error when plan does not exist", async () => {
+      const tool = createPlanUpdateTool(async () => planService);
+      const sessionId = "test-session";
+      const previewFn = tool.presentation?.preview;
+      expect(previewFn).toBeDefined();
+      try {
+        await previewFn(
+          { content: "# New Content" },
+          createToolContext(logger, { sessionId })
+        );
+        expect.fail("Should have thrown an error");
+      } catch (error) {
+        expect(error).toBeInstanceOf(DextoRuntimeError);
+        expect(error.code).toBe(PlanErrorCode.PLAN_NOT_FOUND);
+      }
+    });
+    it("should throw error when sessionId is missing", async () => {
+      const tool = createPlanUpdateTool(async () => planService);
+      const previewFn = tool.presentation?.preview;
+      expect(previewFn).toBeDefined();
+      try {
+        await previewFn({ content: "# Content" }, createToolContext(logger));
+        expect.fail("Should have thrown an error");
+      } catch (error) {
+        expect(error).toBeInstanceOf(DextoRuntimeError);
+        expect(error.code).toBe(PlanErrorCode.SESSION_ID_REQUIRED);
+      }
+    });
+    it("should show deletions in diff", async () => {
+      const tool = createPlanUpdateTool(async () => planService);
+      const sessionId = "test-session";
+      const originalContent = "# Plan\n\nLine to remove\nKeep this";
+      const newContent = "# Plan\n\nKeep this";
+      const previewFn = tool.presentation?.preview;
+      expect(previewFn).toBeDefined();
+      await planService.create(sessionId, originalContent);
+      const preview = await previewFn(
+        { content: newContent },
+        createToolContext(logger, { sessionId })
+      );
+      expect(preview.deletions).toBeGreaterThan(0);
+      expect(preview.unified).toContain("-Line to remove");
+    });
+  });
+  describe("execute", () => {
+    it("should update plan content and return success", async () => {
+      const tool = createPlanUpdateTool(async () => planService);
+      const sessionId = "test-session";
+      const originalContent = "# Original Plan";
+      const newContent = "# Updated Plan";
+      await planService.create(sessionId, originalContent);
+      const result = await tool.execute(
+        { content: newContent },
+        createToolContext(logger, { sessionId })
+      );
+      expect(result.success).toBe(true);
+      expect(result.path).toContain(sessionId);
+      expect(result.path).toMatch(/plan\.md$/);
+      const plan = await planService.read(sessionId);
+      expect(plan.content).toBe(newContent);
+    });
+    it("should include _display data with diff", async () => {
+      const tool = createPlanUpdateTool(async () => planService);
+      const sessionId = "test-session";
+      await planService.create(sessionId, "# Original");
+      const result = await tool.execute(
+        { content: "# Updated" },
+        createToolContext(logger, { sessionId })
+      );
+      expect(result._display).toBeDefined();
+      expect(result._display.type).toBe("diff");
+      expect(result._display.title).toBe("Update Plan");
+      expect(result._display.unified).toContain("-# Original");
+      expect(result._display.unified).toContain("+# Updated");
     });
-    afterEach(async () => {
-        try {
-            await fs.rm(tempDir, { recursive: true, force: true });
-        }
-        catch {
-            // Ignore cleanup errors
-        }
+    it("should throw error when plan does not exist", async () => {
+      const tool = createPlanUpdateTool(async () => planService);
+      const sessionId = "non-existent";
+      try {
+        await tool.execute(
+          { content: "# Content" },
+          createToolContext(logger, { sessionId })
+        );
+        expect.fail("Should have thrown an error");
+      } catch (error) {
+        expect(error).toBeInstanceOf(DextoRuntimeError);
+        expect(error.code).toBe(PlanErrorCode.PLAN_NOT_FOUND);
+      }
     });
-    describe('generatePreview', () => {
-        it('should return DiffDisplayData with unified diff', async () => {
-            const tool = createPlanUpdateTool(planService);
-            const sessionId = 'test-session';
-            const originalContent = '# Plan\n\n## Steps\n1. First step';
-            const newContent = '# Plan\n\n## Steps\n1. First step\n2. Second step';
-            await planService.create(sessionId, originalContent);
-            const preview = (await tool.generatePreview({ content: newContent }, { sessionId }));
-            expect(preview.type).toBe('diff');
-            // Path is now absolute, check it ends with the expected suffix
-            expect(preview.filename).toContain(sessionId);
-            expect(preview.filename).toMatch(/plan\.md$/);
-            expect(preview.unified).toContain('-1. First step');
-            expect(preview.unified).toContain('+1. First step');
-            expect(preview.unified).toContain('+2. Second step');
-            expect(preview.additions).toBeGreaterThan(0);
-        });
-        it('should throw error when plan does not exist', async () => {
-            const tool = createPlanUpdateTool(planService);
-            const sessionId = 'test-session';
-            try {
-                await tool.generatePreview({ content: '# New Content' }, { sessionId });
-                expect.fail('Should have thrown an error');
-            }
-            catch (error) {
-                expect(error).toBeInstanceOf(DextoRuntimeError);
-                expect(error.code).toBe(PlanErrorCode.PLAN_NOT_FOUND);
-            }
-        });
-        it('should throw error when sessionId is missing', async () => {
-            const tool = createPlanUpdateTool(planService);
-            try {
-                await tool.generatePreview({ content: '# Content' }, {});
-                expect.fail('Should have thrown an error');
-            }
-            catch (error) {
-                expect(error).toBeInstanceOf(DextoRuntimeError);
-                expect(error.code).toBe(PlanErrorCode.SESSION_ID_REQUIRED);
-            }
-        });
-        it('should show deletions in diff', async () => {
-            const tool = createPlanUpdateTool(planService);
-            const sessionId = 'test-session';
-            const originalContent = '# Plan\n\nLine to remove\nKeep this';
-            const newContent = '# Plan\n\nKeep this';
-            await planService.create(sessionId, originalContent);
-            const preview = (await tool.generatePreview({ content: newContent }, { sessionId }));
-            expect(preview.deletions).toBeGreaterThan(0);
-            expect(preview.unified).toContain('-Line to remove');
-        });
+    it("should throw error when sessionId is missing", async () => {
+      const tool = createPlanUpdateTool(async () => planService);
+      try {
+        await tool.execute({ content: "# Content" }, createToolContext(logger));
+        expect.fail("Should have thrown an error");
+      } catch (error) {
+        expect(error).toBeInstanceOf(DextoRuntimeError);
+        expect(error.code).toBe(PlanErrorCode.SESSION_ID_REQUIRED);
+      }
     });
-    describe('execute', () => {
-        it('should update plan content and return success', async () => {
-            const tool = createPlanUpdateTool(planService);
-            const sessionId = 'test-session';
-            const originalContent = '# Original Plan';
-            const newContent = '# Updated Plan';
-            await planService.create(sessionId, originalContent);
-            const result = (await tool.execute({ content: newContent }, { sessionId }));
-            expect(result.success).toBe(true);
-            // Path is now absolute, check it ends with the expected suffix
-            expect(result.path).toContain(sessionId);
-            expect(result.path).toMatch(/plan\.md$/);
-            // Verify content was updated
-            const plan = await planService.read(sessionId);
-            expect(plan.content).toBe(newContent);
-        });
-        it('should include _display data with diff', async () => {
-            const tool = createPlanUpdateTool(planService);
-            const sessionId = 'test-session';
-            await planService.create(sessionId, '# Original');
-            const result = (await tool.execute({ content: '# Updated' }, { sessionId }));
-            expect(result._display).toBeDefined();
-            expect(result._display.type).toBe('diff');
-            expect(result._display.unified).toContain('-# Original');
-            expect(result._display.unified).toContain('+# Updated');
-        });
-        it('should throw error when plan does not exist', async () => {
-            const tool = createPlanUpdateTool(planService);
-            const sessionId = 'non-existent';
-            try {
-                await tool.execute({ content: '# Content' }, { sessionId });
-                expect.fail('Should have thrown an error');
-            }
-            catch (error) {
-                expect(error).toBeInstanceOf(DextoRuntimeError);
-                expect(error.code).toBe(PlanErrorCode.PLAN_NOT_FOUND);
-            }
-        });
-        it('should throw error when sessionId is missing', async () => {
-            const tool = createPlanUpdateTool(planService);
-            try {
-                await tool.execute({ content: '# Content' }, {});
-                expect.fail('Should have thrown an error');
-            }
-            catch (error) {
-                expect(error).toBeInstanceOf(DextoRuntimeError);
-                expect(error.code).toBe(PlanErrorCode.SESSION_ID_REQUIRED);
-            }
-        });
-        it('should preserve plan status after update', async () => {
-            const tool = createPlanUpdateTool(planService);
-            const sessionId = 'test-session';
-            await planService.create(sessionId, '# Plan');
-            await planService.updateMeta(sessionId, { status: 'approved' });
-            await tool.execute({ content: '# Updated Plan' }, { sessionId });
-            const plan = await planService.read(sessionId);
-            expect(plan.meta.status).toBe('approved');
-        });
+    it("should preserve plan status after update", async () => {
+      const tool = createPlanUpdateTool(async () => planService);
+      const sessionId = "test-session";
+      await planService.create(sessionId, "# Plan");
+      await planService.updateMeta(sessionId, { status: "approved" });
+      await tool.execute(
+        { content: "# Updated Plan" },
+        createToolContext(logger, { sessionId })
+      );
+      const plan = await planService.read(sessionId);
+      expect(plan.meta.status).toBe("approved");
     });
+  });
 });