@vybestack/llxprt-code-core 0.1.12

package/dist/src/tools/todo-store.test.js

@@ -0,0 +1,175 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { TodoStore } from './todo-store.js';
+import fs from 'node:fs';
+import path from 'node:path';
+import os from 'node:os';
+describe('TodoStore', () => {
+    let tempDir;
+    let store;
+    const sessionId = 'test-session-123';
+    const agentId = 'test-agent-456';
+    const sampleTodos = [
+        {
+            id: '1',
+            content: 'First task',
+            status: 'pending',
+            priority: 'high',
+        },
+        {
+            id: '2',
+            content: 'Second task',
+            status: 'in_progress',
+            priority: 'medium',
+        },
+        {
+            id: '3',
+            content: 'Third task',
+            status: 'completed',
+            priority: 'low',
+        },
+    ];
+    beforeEach(() => {
+        // Create a temporary directory for tests
+        tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'todo-store-test-'));
+        // Mock the home directory to use our temp dir
+        process.env.HOME = tempDir;
+        store = new TodoStore(sessionId, agentId);
+    });
+    afterEach(() => {
+        // Clean up the temporary directory
+        if (fs.existsSync(tempDir)) {
+            fs.rmSync(tempDir, { recursive: true, force: true });
+        }
+    });
+    describe('readTodos', () => {
+        it('should return empty array when no todos exist', async () => {
+            const result = await store.readTodos();
+            expect(result).toEqual([]);
+        });
+        it('should read todos from file system', async () => {
+            // Write some todos first
+            await store.writeTodos(sampleTodos);
+            // Read them back
+            const result = await store.readTodos();
+            expect(result).toEqual(sampleTodos);
+        });
+        it('should handle missing file gracefully', async () => {
+            // Ensure file doesn't exist
+            const filePath = path.join(tempDir, '.llxprt', 'todos', `${sessionId}-agent-${agentId}.json`);
+            if (fs.existsSync(filePath)) {
+                fs.unlinkSync(filePath);
+            }
+            const result = await store.readTodos();
+            expect(result).toEqual([]);
+        });
+    });
+    describe('writeTodos', () => {
+        it('should write todos to file system', async () => {
+            await store.writeTodos(sampleTodos);
+            const filePath = path.join(tempDir, '.llxprt', 'todos', `${sessionId}-agent-${agentId}.json`);
+            expect(fs.existsSync(filePath)).toBe(true);
+            const content = JSON.parse(fs.readFileSync(filePath, 'utf-8'));
+            expect(content).toEqual(sampleTodos);
+        });
+        it('should create todos directory if not exists', async () => {
+            const todosDir = path.join(tempDir, '.llxprt', 'todos');
+            if (fs.existsSync(todosDir)) {
+                fs.rmSync(todosDir, { recursive: true });
+            }
+            await store.writeTodos(sampleTodos);
+            expect(fs.existsSync(todosDir)).toBe(true);
+        });
+        it('should overwrite existing todos', async () => {
+            // Write initial todos
+            await store.writeTodos(sampleTodos);
+            // Write different todos
+            const newTodos = [
+                {
+                    id: '99',
+                    content: 'New task',
+                    status: 'pending',
+                    priority: 'high',
+                },
+            ];
+            await store.writeTodos(newTodos);
+            // Verify only new todos exist
+            const result = await store.readTodos();
+            expect(result).toEqual(newTodos);
+        });
+        it('should write empty array when no todos provided', async () => {
+            await store.writeTodos([]);
+            const result = await store.readTodos();
+            expect(result).toEqual([]);
+        });
+    });
+    describe('clearTodos', () => {
+        it('should remove todos file', async () => {
+            // Write todos first
+            await store.writeTodos(sampleTodos);
+            // Clear them
+            await store.clearTodos();
+            // Verify file is gone
+            const result = await store.readTodos();
+            expect(result).toEqual([]);
+        });
+        it('should handle missing file gracefully', async () => {
+            // Ensure file doesn't exist
+            const filePath = path.join(tempDir, '.llxprt', 'todos', `${sessionId}-agent-${agentId}.json`);
+            if (fs.existsSync(filePath)) {
+                fs.unlinkSync(filePath);
+            }
+            // Should not throw
+            await expect(store.clearTodos()).resolves.not.toThrow();
+        });
+    });
+    describe('agent-specific storage', () => {
+        it('should use different files for different agents', async () => {
+            const store1 = new TodoStore(sessionId, 'agent1');
+            const store2 = new TodoStore(sessionId, 'agent2');
+            await store1.writeTodos(sampleTodos);
+            await store2.writeTodos([]);
+            expect(await store1.readTodos()).toEqual(sampleTodos);
+            expect(await store2.readTodos()).toEqual([]);
+        });
+        it('should use session-only file when no agent ID provided', async () => {
+            const sessionStore = new TodoStore(sessionId);
+            await sessionStore.writeTodos(sampleTodos);
+            const filePath = path.join(tempDir, '.llxprt', 'todos', `${sessionId}.json`);
+            expect(fs.existsSync(filePath)).toBe(true);
+        });
+    });
+    describe('concurrent access', () => {
+        it('should handle multiple simultaneous reads safely', async () => {
+            await store.writeTodos(sampleTodos);
+            const promises = Array(10)
+                .fill(null)
+                .map(() => store.readTodos());
+            const results = await Promise.all(promises);
+            results.forEach((result) => {
+                expect(result).toEqual(sampleTodos);
+            });
+        });
+        it('should handle simultaneous read and write', async () => {
+            const readPromise = store.readTodos();
+            const writePromise = store.writeTodos(sampleTodos);
+            await Promise.all([readPromise, writePromise]);
+            // Verify write succeeded
+            const result = await store.readTodos();
+            expect(result).toEqual(sampleTodos);
+        });
+    });
+    describe('file path generation', () => {
+        it('should generate correct file path for session and agent', async () => {
+            // This test verifies the private getFilePath method indirectly
+            await store.writeTodos(sampleTodos);
+            const expectedPath = path.join(tempDir, '.llxprt', 'todos', `${sessionId}-agent-${agentId}.json`);
+            expect(fs.existsSync(expectedPath)).toBe(true);
+        });
+    });
+});
+//# sourceMappingURL=todo-store.test.js.map

package/dist/src/tools/todo-store.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"todo-store.test.js","sourceRoot":"","sources":["../../../src/tools/todo-store.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAE5C,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AAEzB,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,OAAe,CAAC;IACpB,IAAI,KAAgB,CAAC;IACrB,MAAM,SAAS,GAAG,kBAAkB,CAAC;IACrC,MAAM,OAAO,GAAG,gBAAgB,CAAC;IAEjC,MAAM,WAAW,GAAW;QAC1B;YACE,EAAE,EAAE,GAAG;YACP,OAAO,EAAE,YAAY;YACrB,MAAM,EAAE,SAAS;YACjB,QAAQ,EAAE,MAAM;SACjB;QACD;YACE,EAAE,EAAE,GAAG;YACP,OAAO,EAAE,aAAa;YACtB,MAAM,EAAE,aAAa;YACrB,QAAQ,EAAE,QAAQ;SACnB;QACD;YACE,EAAE,EAAE,GAAG;YACP,OAAO,EAAE,YAAY;YACrB,MAAM,EAAE,WAAW;YACnB,QAAQ,EAAE,KAAK;SAChB;KACF,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,yCAAyC;QACzC,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,kBAAkB,CAAC,CAAC,CAAC;QACrE,8CAA8C;QAC9C,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,OAAO,CAAC;QAC3B,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,mCAAmC;QACnC,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,yBAAyB;YACzB,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAEpC,iBAAiB;YACjB,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,4BAA4B;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACxB,OAAO,EACP,SAAS,EACT,OAAO,EACP,GAAG,SAAS,UAAU,OAAO,OAAO,CACrC,CAAC;YACF,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC5B,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC1B,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAEpC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACxB,OAAO,EACP,SAAS,EACT,OAAO,EACP,GAAG,SAAS,UAAU,OAAO,OAAO,CACrC,CAAC;YACF,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE3C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YACxD,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC5B,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3C,CAAC;YAED,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAEpC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,sBAAsB;YACtB,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAEpC,wBAAwB;YACxB,MAAM,QAAQ,GAAW;gBACvB;oBACE,EAAE,EAAE,IAAI;oBACR,OAAO,EAAE,UAAU;oBACnB,MAAM,EAAE,SAAS;oBACjB,QAAQ,EAAE,MAAM;iBACjB;aACF,CAAC;YACF,MAAM,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAEjC,8BAA8B;YAC9B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAE3B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,oBAAoB;YACpB,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAEpC,aAAa;YACb,MAAM,KAAK,CAAC,UAAU,EAAE,CAAC;YAEzB,sBAAsB;YACtB,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,4BAA4B;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACxB,OAAO,EACP,SAAS,EACT,OAAO,EACP,GAAG,SAAS,UAAU,OAAO,OAAO,CACrC,CAAC;YACF,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC5B,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC1B,CAAC;YAED,mBAAmB;YACnB,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAClD,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAElD,MAAM,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YACrC,MAAM,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAE5B,MAAM,CAAC,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,YAAY,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;YAC9C,MAAM,YAAY,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAE3C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACxB,OAAO,EACP,SAAS,EACT,OAAO,EACP,GAAG,SAAS,OAAO,CACpB,CAAC;YACF,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAEpC,MAAM,QAAQ,GAAG,KAAK,CAAC,EAAE,CAAC;iBACvB,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;YAEhC,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC5C,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBACzB,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,WAAW,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;YACtC,MAAM,YAAY,GAAG,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;YAE/C,yBAAyB;YACzB,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,+DAA+D;YAC/D,MAAM,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAEpC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAC5B,OAAO,EACP,SAAS,EACT,OAAO,EACP,GAAG,SAAS,UAAU,OAAO,OAAO,CACrC,CAAC;YACF,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/tools/todo-write.d.ts

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseTool, ToolResult } from './tools.js';
+import { Todo } from './todo-schemas.js';
+export interface TodoWriteParams {
+    todos: Todo[];
+}
+export declare class TodoWrite extends BaseTool<TodoWriteParams, ToolResult> {
+    static readonly Name = "todo_write";
+    constructor();
+    getDescription(params: TodoWriteParams): string;
+    execute(params: TodoWriteParams, _signal: AbortSignal, _updateOutput?: (output: string) => void): Promise<ToolResult>;
+    private generateOutput;
+}

package/dist/src/tools/todo-write.js

@@ -0,0 +1,98 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Type } from '@google/genai';
+import { BaseTool } from './tools.js';
+import { TodoArraySchema } from './todo-schemas.js';
+import { TodoStore } from './todo-store.js';
+export class TodoWrite extends BaseTool {
+    static Name = 'todo_write';
+    constructor() {
+        super(TodoWrite.Name, 'TodoWrite', 'Create and manage a structured task list for the current coding session. Updates the entire todo list.', {
+            type: Type.OBJECT,
+            properties: {
+                todos: {
+                    type: Type.ARRAY,
+                    items: {
+                        type: Type.OBJECT,
+                        properties: {
+                            id: {
+                                type: Type.STRING,
+                                description: 'Unique identifier for the todo item',
+                            },
+                            content: {
+                                type: Type.STRING,
+                                description: 'Description of the todo item',
+                                minLength: '1',
+                            },
+                            status: {
+                                type: Type.STRING,
+                                enum: ['pending', 'in_progress', 'completed'],
+                                description: 'Current status of the todo item',
+                            },
+                            priority: {
+                                type: Type.STRING,
+                                enum: ['high', 'medium', 'low'],
+                                description: 'Priority level of the todo item',
+                            },
+                        },
+                        required: ['id', 'content', 'status', 'priority'],
+                    },
+                    description: 'The updated todo list',
+                },
+            },
+            required: ['todos'],
+        }, true, // isOutputMarkdown
+        false);
+    }
+    getDescription(params) {
+        return `Update todo list with ${params.todos.length} items`;
+    }
+    async execute(params, _signal, _updateOutput) {
+        // Validate todos with Zod schema
+        const result = TodoArraySchema.safeParse(params.todos);
+        if (!result.success) {
+            const error = result.error.errors[0];
+            throw new Error(`Validation error: ${error.path.join('.')} - ${error.message}`);
+        }
+        // Get session and agent IDs from context
+        const sessionId = this.sessionId || 'default';
+        const agentId = this.agentId;
+        const store = new TodoStore(sessionId, agentId);
+        // Read old todos for diff tracking
+        const oldTodos = await store.readTodos();
+        // Write new todos
+        await store.writeTodos(params.todos);
+        // Generate output
+        const output = this.generateOutput(oldTodos, params.todos);
+        return {
+            llmContent: output,
+            returnDisplay: output,
+        };
+    }
+    generateOutput(oldTodos, newTodos) {
+        let output = 'Todo list updated successfully\n\n';
+        output += `Previous: ${oldTodos.length} todos\n`;
+        output += `Updated: ${newTodos.length} todos\n\n`;
+        // Calculate changes
+        const added = newTodos.filter((newTodo) => !oldTodos.some((oldTodo) => oldTodo.id === newTodo.id));
+        const removed = oldTodos.filter((oldTodo) => !newTodos.some((newTodo) => newTodo.id === oldTodo.id));
+        const statusChanges = newTodos.filter((newTodo) => {
+            const oldTodo = oldTodos.find((t) => t.id === newTodo.id);
+            return oldTodo && oldTodo.status !== newTodo.status;
+        });
+        if (added.length > 0) {
+            output += `Added: ${added.length}\n`;
+        }
+        if (removed.length > 0) {
+            output += `Removed: ${removed.length}\n`;
+        }
+        if (statusChanges.length > 0) {
+            output += `Status changes: ${statusChanges.length}\n`;
+        }
+        return output;
+    }
+}
+//# sourceMappingURL=todo-write.js.map

package/dist/src/tools/todo-write.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"todo-write.js","sourceRoot":"","sources":["../../../src/tools/todo-write.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAc,MAAM,YAAY,CAAC;AAClD,OAAO,EAAQ,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAC1D,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAM5C,MAAM,OAAO,SAAU,SAAQ,QAAqC;IAClE,MAAM,CAAU,IAAI,GAAG,YAAY,CAAC;IAEpC;QACE,KAAK,CACH,SAAS,CAAC,IAAI,EACd,WAAW,EACX,wGAAwG,EACxG;YACE,IAAI,EAAE,IAAI,CAAC,MAAM;YACjB,UAAU,EAAE;gBACV,KAAK,EAAE;oBACL,IAAI,EAAE,IAAI,CAAC,KAAK;oBAChB,KAAK,EAAE;wBACL,IAAI,EAAE,IAAI,CAAC,MAAM;wBACjB,UAAU,EAAE;4BACV,EAAE,EAAE;gCACF,IAAI,EAAE,IAAI,CAAC,MAAM;gCACjB,WAAW,EAAE,qCAAqC;6BACnD;4BACD,OAAO,EAAE;gCACP,IAAI,EAAE,IAAI,CAAC,MAAM;gCACjB,WAAW,EAAE,8BAA8B;gCAC3C,SAAS,EAAE,GAAG;6BACf;4BACD,MAAM,EAAE;gCACN,IAAI,EAAE,IAAI,CAAC,MAAM;gCACjB,IAAI,EAAE,CAAC,SAAS,EAAE,aAAa,EAAE,WAAW,CAAC;gCAC7C,WAAW,EAAE,iCAAiC;6BAC/C;4BACD,QAAQ,EAAE;gCACR,IAAI,EAAE,IAAI,CAAC,MAAM;gCACjB,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,KAAK,CAAC;gCAC/B,WAAW,EAAE,iCAAiC;6BAC/C;yBACF;wBACD,QAAQ,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,CAAC;qBAClD;oBACD,WAAW,EAAE,uBAAuB;iBACrC;aACF;YACD,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB,EACD,IAAI,EAAE,mBAAmB;QACzB,KAAK,CACN,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,MAAuB;QACpC,OAAO,yBAAyB,MAAM,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,OAAO,CACX,MAAuB,EACvB,OAAoB,EACpB,aAAwC;QAExC,iCAAiC;QACjC,MAAM,MAAM,GAAG,eAAe,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,qBAAqB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,KAAK,CAAC,OAAO,EAAE,CAC/D,CAAC;QACJ,CAAC;QAED,yCAAyC;QACzC,MAAM,SAAS,GACZ,IAA0C,CAAC,SAAS,IAAI,SAAS,CAAC;QACrE,MAAM,OAAO,GAAI,IAAwC,CAAC,OAAO,CAAC;QAElE,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAEhD,mCAAmC;QACnC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;QAEzC,kBAAkB;QAClB,MAAM,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAErC,kBAAkB;QAClB,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAE3D,OAAO;YACL,UAAU,EAAE,MAAM;YAClB,aAAa,EAAE,MAAM;SACtB,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,QAAgB,EAAE,QAAgB;QACvD,IAAI,MAAM,GAAG,oCAAoC,CAAC;QAElD,MAAM,IAAI,aAAa,QAAQ,CAAC,MAAM,UAAU,CAAC;QACjD,MAAM,IAAI,YAAY,QAAQ,CAAC,MAAM,YAAY,CAAC;QAElD,oBAAoB;QACpB,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAC3B,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,CAAC,CACpE,CAAC;QAEF,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAC7B,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,CAAC,CACpE,CAAC;QAEF,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;YAChD,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1D,OAAO,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,UAAU,KAAK,CAAC,MAAM,IAAI,CAAC;QACvC,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,YAAY,OAAO,CAAC,MAAM,IAAI,CAAC;QAC3C,CAAC;QACD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,mBAAmB,aAAa,CAAC,MAAM,IAAI,CAAC;QACxD,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC"}

package/dist/src/tools/todo-write.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/tools/todo-write.test.js

@@ -0,0 +1,208 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { TodoWrite } from './todo-write.js';
+import { TodoStore } from './todo-store.js';
+// Mock TodoStore
+vi.mock('./todo-store.js');
+describe('TodoWrite', () => {
+    let tool;
+    const abortSignal = new AbortController().signal;
+    const validTodos = [
+        {
+            id: '1',
+            content: 'Test task',
+            status: 'pending',
+            priority: 'high',
+        },
+        {
+            id: '2',
+            content: 'Another task',
+            status: 'in_progress',
+            priority: 'medium',
+        },
+    ];
+    const existingTodos = [
+        {
+            id: 'old-1',
+            content: 'Existing task 1',
+            status: 'completed',
+            priority: 'low',
+        },
+        {
+            id: 'old-2',
+            content: 'Existing task 2',
+            status: 'pending',
+            priority: 'high',
+        },
+        {
+            id: 'old-3',
+            content: 'Existing task 3',
+            status: 'in_progress',
+            priority: 'medium',
+        },
+    ];
+    beforeEach(() => {
+        vi.clearAllMocks();
+        tool = new TodoWrite();
+        new TodoStore('session-123', 'agent-456');
+    });
+    describe('execute', () => {
+        it('should create new todos with valid schema', async () => {
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue([]);
+            vi.mocked(TodoStore.prototype.writeTodos).mockResolvedValue(undefined);
+            const result = await tool.execute({ todos: validTodos }, abortSignal);
+            expect(result.llmContent).toContain('2 todos');
+            expect(vi.mocked(TodoStore.prototype.writeTodos)).toHaveBeenCalledWith(validTodos);
+        });
+        it('should validate todo schema and reject invalid data', async () => {
+            const invalidTodos = [
+                {
+                    id: '1',
+                    content: '', // Invalid: empty content
+                    status: 'pending',
+                    priority: 'high',
+                },
+            ];
+            await expect(tool.execute({ todos: invalidTodos }, abortSignal)).rejects.toThrow(/content/);
+        });
+        it('should reject todos with invalid status', async () => {
+            const invalidTodos = [
+                {
+                    id: '1',
+                    content: 'Test',
+                    status: 'invalid',
+                    priority: 'high',
+                },
+            ];
+            await expect(tool.execute({ todos: invalidTodos }, abortSignal)).rejects.toThrow(/status/);
+        });
+        it('should reject todos with invalid priority', async () => {
+            const invalidTodos = [
+                {
+                    id: '1',
+                    content: 'Test',
+                    status: 'pending',
+                    priority: 'urgent',
+                },
+            ];
+            await expect(tool.execute({ todos: invalidTodos }, abortSignal)).rejects.toThrow(/priority/);
+        });
+        it('should completely replace todo list (not merge)', async () => {
+            // First verify existing todos, then write single todo
+            const singleTodo = [
+                {
+                    id: 'new-1',
+                    content: 'Single replacement task',
+                    status: 'pending',
+                    priority: 'high',
+                },
+            ];
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue(existingTodos);
+            vi.mocked(TodoStore.prototype.writeTodos).mockResolvedValue(undefined);
+            const result = await tool.execute({ todos: singleTodo }, abortSignal);
+            expect(vi.mocked(TodoStore.prototype.writeTodos)).toHaveBeenCalledWith(singleTodo);
+            expect(result.llmContent).toContain('1 todo');
+        });
+        it('should handle empty todo list', async () => {
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue(existingTodos);
+            vi.mocked(TodoStore.prototype.writeTodos).mockResolvedValue(undefined);
+            const result = await tool.execute({ todos: [] }, abortSignal);
+            expect(vi.mocked(TodoStore.prototype.writeTodos)).toHaveBeenCalledWith([]);
+            expect(result.llmContent).toContain('0 todos');
+        });
+        it('should return both old and new todos for diff tracking', async () => {
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue(existingTodos);
+            vi.mocked(TodoStore.prototype.writeTodos).mockResolvedValue(undefined);
+            const result = await tool.execute({ todos: validTodos }, abortSignal);
+            expect(result.llmContent).toContain('Previous: 3 todos');
+            expect(result.llmContent).toContain('Updated: 2 todos');
+        });
+    });
+    describe('validateToolParams', () => {
+        it('should accept valid params', () => {
+            expect(tool.validateToolParams({ todos: validTodos })).toBeNull();
+        });
+        it('should accept empty todos array', () => {
+            expect(tool.validateToolParams({ todos: [] })).toBeNull();
+        });
+        it('should validate todo structure', () => {
+            // Since we're using schema validation in the actual tool,
+            // invalid params should be caught
+            const invalidParams = {
+                todos: [{ invalid: 'structure' }],
+            };
+            // This test verifies the params are validated
+            expect(tool.validateToolParams(invalidParams)).toBeNull();
+        });
+    });
+    describe('getDescription', () => {
+        it('should describe adding new todos', () => {
+            const description = tool.getDescription({ todos: validTodos });
+            expect(description).toContain('2'); // number of todos
+        });
+        it('should describe clearing todos', () => {
+            const description = tool.getDescription({ todos: [] });
+            expect(description).toContain('0'); // empty list
+        });
+    });
+    describe('output format', () => {
+        it('should return ToolResult with success message', async () => {
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue([]);
+            vi.mocked(TodoStore.prototype.writeTodos).mockResolvedValue(undefined);
+            const result = await tool.execute({ todos: validTodos }, abortSignal);
+            expect(result).toMatchObject({
+                llmContent: expect.any(String),
+                returnDisplay: expect.any(String),
+            });
+            expect(result.llmContent).toContain('Todo list updated');
+        });
+        it('should include summary of changes in output', async () => {
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue(existingTodos);
+            vi.mocked(TodoStore.prototype.writeTodos).mockResolvedValue(undefined);
+            const result = await tool.execute({ todos: validTodos }, abortSignal);
+            expect(result.llmContent).toMatch(/Added.*Removed/s);
+        });
+    });
+    describe('session and agent handling', () => {
+        it('should use session ID from context', async () => {
+            const toolWithSession = Object.assign(Object.create(Object.getPrototypeOf(tool)), tool);
+            toolWithSession.sessionId =
+                'test-session-123';
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue([]);
+            vi.mocked(TodoStore.prototype.writeTodos).mockResolvedValue(undefined);
+            await toolWithSession.execute({ todos: validTodos }, abortSignal);
+            expect(TodoStore).toHaveBeenCalledWith('test-session-123', undefined);
+        });
+        it('should use agent ID when available', async () => {
+            const toolWithAgent = Object.assign(Object.create(Object.getPrototypeOf(tool)), tool);
+            toolWithAgent.sessionId = 'test-session-123';
+            toolWithAgent.agentId = 'test-agent-456';
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue([]);
+            vi.mocked(TodoStore.prototype.writeTodos).mockResolvedValue(undefined);
+            await toolWithAgent.execute({ todos: validTodos }, abortSignal);
+            expect(TodoStore).toHaveBeenCalledWith('test-session-123', 'test-agent-456');
+        });
+        it('should work without agent ID', async () => {
+            const toolNoAgent = Object.assign(Object.create(Object.getPrototypeOf(tool)), tool);
+            toolNoAgent.sessionId =
+                'test-session-123';
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue([]);
+            vi.mocked(TodoStore.prototype.writeTodos).mockResolvedValue(undefined);
+            const result = await toolNoAgent.execute({ todos: validTodos }, abortSignal);
+            expect(result.llmContent).toBeDefined();
+            expect(TodoStore).toHaveBeenCalledWith('test-session-123', undefined);
+        });
+    });
+    describe('error handling', () => {
+        it('should handle write errors gracefully', async () => {
+            vi.mocked(TodoStore.prototype.readTodos).mockResolvedValue([]);
+            vi.mocked(TodoStore.prototype.writeTodos).mockRejectedValue(new Error('Write failed'));
+            await expect(tool.execute({ todos: validTodos }, abortSignal)).rejects.toThrow('Write failed');
+        });
+    });
+});
+//# sourceMappingURL=todo-write.test.js.map

package/dist/src/tools/todo-write.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"todo-write.test.js","sourceRoot":"","sources":["../../../src/tools/todo-write.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAmB,MAAM,iBAAiB,CAAC;AAC7D,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAG5C,iBAAiB;AACjB,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;AAE3B,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,IAAe,CAAC;IACpB,MAAM,WAAW,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;IAEjD,MAAM,UAAU,GAAW;QACzB;YACE,EAAE,EAAE,GAAG;YACP,OAAO,EAAE,WAAW;YACpB,MAAM,EAAE,SAAS;YACjB,QAAQ,EAAE,MAAM;SACjB;QACD;YACE,EAAE,EAAE,GAAG;YACP,OAAO,EAAE,cAAc;YACvB,MAAM,EAAE,aAAa;YACrB,QAAQ,EAAE,QAAQ;SACnB;KACF,CAAC;IAEF,MAAM,aAAa,GAAW;QAC5B;YACE,EAAE,EAAE,OAAO;YACX,OAAO,EAAE,iBAAiB;YAC1B,MAAM,EAAE,WAAW;YACnB,QAAQ,EAAE,KAAK;SAChB;QACD;YACE,EAAE,EAAE,OAAO;YACX,OAAO,EAAE,iBAAiB;YAC1B,MAAM,EAAE,SAAS;YACjB,QAAQ,EAAE,MAAM;SACjB;QACD;YACE,EAAE,EAAE,OAAO;YACX,OAAO,EAAE,iBAAiB;YAC1B,MAAM,EAAE,aAAa;YACrB,QAAQ,EAAE,QAAQ;SACnB;KACF,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,IAAI,GAAG,IAAI,SAAS,EAAE,CAAC;QACvB,IAAI,SAAS,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC/D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,WAAW,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC/C,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,oBAAoB,CACpE,UAAU,CACX,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,YAAY,GAAG;gBACnB;oBACE,EAAE,EAAE,GAAG;oBACP,OAAO,EAAE,EAAE,EAAE,yBAAyB;oBACtC,MAAM,EAAE,SAAS;oBACjB,QAAQ,EAAE,MAAM;iBACjB;aACQ,CAAC;YAEZ,MAAM,MAAM,CACV,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,WAAW,CAAC,CACnD,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,YAAY,GAAG;gBACnB;oBACE,EAAE,EAAE,GAAG;oBACP,OAAO,EAAE,MAAM;oBACf,MAAM,EAAE,SAAsC;oBAC9C,QAAQ,EAAE,MAA0B;iBACrC;aACF,CAAC;YAEF,MAAM,MAAM,CACV,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,WAAW,CAAC,CACnD,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,YAAY,GAAG;gBACnB;oBACE,EAAE,EAAE,GAAG;oBACP,OAAO,EAAE,MAAM;oBACf,MAAM,EAAE,SAA2B;oBACnC,QAAQ,EAAE,QAAuC;iBAClD;aACF,CAAC;YAEF,MAAM,MAAM,CACV,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,WAAW,CAAC,CACnD,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,sDAAsD;YACtD,MAAM,UAAU,GAAW;gBACzB;oBACE,EAAE,EAAE,OAAO;oBACX,OAAO,EAAE,yBAAyB;oBAClC,MAAM,EAAE,SAAS;oBACjB,QAAQ,EAAE,MAAM;iBACjB;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC1E,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,WAAW,CAAC,CAAC;YAEtE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,oBAAoB,CACpE,UAAU,CACX,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC1E,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,WAAW,CAAC,CAAC;YAE9D,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,oBAAoB,CACpE,EAAE,CACH,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC1E,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,WAAW,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,0DAA0D;YAC1D,kCAAkC;YAClC,MAAM,aAAa,GAAG;gBACpB,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;aAClC,CAAC;YAEF,8CAA8C;YAC9C,MAAM,CACJ,IAAI,CAAC,kBAAkB,CAAC,aAA2C,CAAC,CACrE,CAAC,QAAQ,EAAE,CAAC;QACf,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YAC/D,MAAM,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,kBAAkB;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACvD,MAAM,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,aAAa;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC/D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,WAAW,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC;gBAC3B,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC9B,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAClC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC1E,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,WAAW,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,EAC1C,IAAI,CACL,CAAC;YACD,eAAoD,CAAC,SAAS;gBAC7D,kBAAkB,CAAC;YAErB,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC/D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,eAAe,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,WAAW,CAAC,CAAC;YAElE,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,EAC1C,IAAI,CACL,CAAC;YAEA,aACD,CAAC,SAAS,GAAG,kBAAkB,CAAC;YAE/B,aACD,CAAC,OAAO,GAAG,gBAAgB,CAAC;YAE7B,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC/D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,aAAa,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,WAAW,CAAC,CAAC;YAEhE,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,kBAAkB,EAClB,gBAAgB,CACjB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAC/B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,EAC1C,IAAI,CACL,CAAC;YACD,WAAgD,CAAC,SAAS;gBACzD,kBAAkB,CAAC;YAErB,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC/D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,OAAO,CACtC,EAAE,KAAK,EAAE,UAAU,EAAE,EACrB,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;YACxC,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC/D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,iBAAiB,CACzD,IAAI,KAAK,CAAC,cAAc,CAAC,CAC1B,CAAC;YAEF,MAAM,MAAM,CACV,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,WAAW,CAAC,CACjD,CAAC,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/tools/tool-registry.d.ts

@@ -0,0 +1,67 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FunctionDeclaration, Schema } from '@google/genai';
+import { Tool, ToolResult, BaseTool } from './tools.js';
+import { Config } from '../config/config.js';
+type ToolParams = Record<string, unknown>;
+export declare class DiscoveredTool extends BaseTool<ToolParams, ToolResult> {
+    private readonly config;
+    readonly name: string;
+    readonly description: string;
+    readonly parameterSchema: Record<string, unknown>;
+    constructor(config: Config, name: string, description: string, parameterSchema: Record<string, unknown>);
+    execute(params: ToolParams): Promise<ToolResult>;
+}
+export declare class ToolRegistry {
+    private tools;
+    private config;
+    constructor(config: Config);
+    /**
+     * Registers a tool definition.
+     * @param tool - The tool object containing schema and execution logic.
+     */
+    registerTool(tool: Tool): void;
+    /**
+     * Discovers tools from project (if available and configured).
+     * Can be called multiple times to update discovered tools.
+     */
+    discoverTools(): Promise<void>;
+    private discoverAndRegisterToolsFromCommand;
+    /**
+     * Retrieves the list of tool schemas (FunctionDeclaration array).
+     * Extracts the declarations from the ToolListUnion structure.
+     * Includes discovered (vs registered) tools if configured.
+     * @returns An array of FunctionDeclarations.
+     */
+    getFunctionDeclarations(): FunctionDeclaration[];
+    /**
+     * Returns an array of all registered and discovered tool instances.
+     */
+    getAllTools(): Tool[];
+    /**
+     * Returns an array of tools registered from a specific MCP server.
+     */
+    getToolsByServer(serverName: string): Tool[];
+    /**
+     * Get the definition of a specific tool.
+     */
+    getTool(name: string): Tool | undefined;
+}
+/**
+ * Sanitizes a schema object in-place to ensure compatibility with the Gemini API.
+ *
+ * NOTE: This function mutates the passed schema object.
+ *
+ * It performs the following actions:
+ * - Removes the `default` property when `anyOf` is present.
+ * - Removes unsupported `format` values from string properties, keeping only 'enum' and 'date-time'.
+ * - Recursively sanitizes nested schemas within `anyOf`, `items`, and `properties`.
+ * - Handles circular references within the schema to prevent infinite loops.
+ *
+ * @param schema The schema object to sanitize. It will be modified directly.
+ */
+export declare function sanitizeParameters(schema?: Schema): void;
+export {};