@hanhnd/agent-kit 1.0.40 → 1.0.42

package/dist/cli.js

@@ -15136,26 +15136,26 @@ var DigestModelRegistryError = class extends Error {
 var DIGEST_MODEL_REGISTRY = {
   ["tiny" /* TINY */]: {
     id: "tiny" /* TINY */,
-    ggufUri: "hf:bartowski/Qwen2.5-0.5B-Instruct-GGUF/Qwen2.5-0.5B-Instruct-Q4_K_M.gguf",
-    approxSizeBytes: 4e8,
+    ggufUri: "hf:lmstudio-community/Qwen3.5-0.8B-GGUF/Qwen3.5-0.8B-Q4_K_M.gguf",
+    approxSizeBytes: 528e6,
     license: "Apache-2.0",
-    sourceUrl: "https://huggingface.co/bartowski/Qwen2.5-0.5B-Instruct-GGUF",
+    sourceUrl: "https://huggingface.co/lmstudio-community/Qwen3.5-0.8B-GGUF",
     enabled: true
   },
   ["base" /* BASE */]: {
     id: "base" /* BASE */,
-    ggufUri: "hf:bartowski/Qwen2.5-1.5B-Instruct-GGUF/Qwen2.5-1.5B-Instruct-Q4_K_M.gguf",
-    approxSizeBytes: 1e9,
+    ggufUri: "hf:lmstudio-community/Qwen3.5-2B-GGUF/Qwen3.5-2B-Q4_K_M.gguf",
+    approxSizeBytes: 127e7,
     license: "Apache-2.0",
-    sourceUrl: "https://huggingface.co/bartowski/Qwen2.5-1.5B-Instruct-GGUF",
+    sourceUrl: "https://huggingface.co/lmstudio-community/Qwen3.5-2B-GGUF",
     enabled: true
   },
   ["large" /* LARGE */]: {
     id: "large" /* LARGE */,
-    ggufUri: "hf:bartowski/Qwen2.5-3B-Instruct-GGUF/Qwen2.5-3B-Instruct-Q4_K_M.gguf",
-    approxSizeBytes: 19e8,
+    ggufUri: "hf:lmstudio-community/Qwen3.5-4B-GGUF/Qwen3.5-4B-Q4_K_M.gguf",
+    approxSizeBytes: 271e7,
     license: "Apache-2.0",
-    sourceUrl: "https://huggingface.co/bartowski/Qwen2.5-3B-Instruct-GGUF",
+    sourceUrl: "https://huggingface.co/lmstudio-community/Qwen3.5-4B-GGUF",
     enabled: true
   }
 };
@@ -15315,7 +15315,8 @@ async function createLlamaLocalDigestProvider(modelId) {
         const response = await withTimeout(
           session.prompt(userContent, {
             maxTokens: LLAMA_MAX_GENERATED_TOKENS,
-            temperature: LLAMA_TEMPERATURE
+            temperature: LLAMA_TEMPERATURE,
+            budgets: { thoughtTokens: 0 }
           }),
           options.timeoutMs,
           () => new LlamaLocalDigestProviderError(`Llama provider timed out after ${options.timeoutMs}ms`)

package/dist/core/handoffs/index.d.ts

@@ -1,6 +1,6 @@
-export declare const HANDOFF_TYPES: readonly ["brainstorm", "clarification", "plan", "ticket", "research", "scenario", "investigation", "code"];
+export declare const HANDOFF_TYPES: readonly ["brainstorm", "clarification", "plan", "ticket", "research", "investigation", "code"];
 type HandoffType = (typeof HANDOFF_TYPES)[number];
-type CanonicalHandoffType = 'brainstorm' | 'clarification' | 'plan' | 'ticket' | 'research' | 'scenario' | 'investigation' | 'code';
+type CanonicalHandoffType = 'brainstorm' | 'clarification' | 'plan' | 'ticket' | 'research' | 'investigation' | 'code';
 export interface SavedHandoffFolderLocation {
     featureSlug: string;
     canonicalType: CanonicalHandoffType;

package/dist/core/handoffs/index.js

@@ -5,7 +5,6 @@ export const HANDOFF_TYPES = [
     'plan',
     'ticket',
     'research',
-    'scenario',
     'investigation',
     'code',
 ];

package/dist/mcp/agent.js

@@ -49,11 +49,18 @@ function initAgentLog(workspaceRoot, slug, agent) {
     return { logPath: path.relative(workspaceRoot, logFile), logStream };
 }
 export function registerAgentTools(server) {
-    server.tool('kit_trigger_agent', '!Important: Trigger only when user explicitly asks to delegate a task to an external agent CLI (gemini, claude, or codex). The task can be a direct message or a path to a handoff folder (.agent-kit/handoffs/feature-slug/plan).', {
-        agent: z.enum(['gemini', 'claude', 'codex']).describe('The agent CLI to invoke: "gemini", "claude", or "codex"'),
-        task: z
-            .string()
-            .describe('Task message or path to a handoff folder (e.g., ".agent-kit/handoffs/feature-slug/plan")'),
+    server.registerTool('kit_trigger_agent', {
+        title: 'Delegate to External Agent',
+        description: '!Important: Trigger only when user explicitly asks to delegate a task to an external agent CLI (gemini, claude, or codex). The task can be a direct message or a path to a handoff folder (.agent-kit/handoffs/feature-slug/plan).',
+        inputSchema: {
+            agent: z
+                .enum(['gemini', 'claude', 'codex'])
+                .describe('The agent CLI to invoke: "gemini", "claude", or "codex"'),
+            task: z
+                .string()
+                .describe('Task message or path to a handoff folder (e.g., ".agent-kit/handoffs/feature-slug/plan")'),
+        },
+        annotations: { readOnlyHint: false, destructiveHint: true, idempotentHint: false, openWorldHint: true },
     }, async ({ agent, task }, extra) => {
         const workspaceRoot = getWorkspaceRoot();
         let job;

package/dist/mcp/core.js

@@ -46,14 +46,19 @@ export function registerCoreTools(server) {
     // Writes handoff artifacts to the workspace as a folder of files
     // Returns the saved folder path to use in next-step instructions
     // ═══════════════════════════════════════════════════════════════
-    server.tool('kit_save_handoff', `Save a handoff artifact to .agent-kit/handoffs/. Returns the saved folder path to use in next-step instructions. Do NOT append version numbers (v2, v3, etc.) to the slug.`, {
-        type: z.enum(HANDOFF_TYPES).describe('Handoff type'),
-        files: z
-            .record(z.string(), z.string())
-            .describe('Markdown files to write into the handoff folder. May include one or many files per call; repeated calls with the same type and slug add or overwrite the provided files. Filenames must match /^[A-Za-z0-9_-]+\\.md$/.'),
-        slug: z
-            .string()
-            .describe('Short identifier for the folder, e.g. "user-auth" or "PROJ-123". Do NOT append version numbers (v2, v3, etc.).'),
+    server.registerTool('kit_save_handoff', {
+        title: 'Save Handoff Artifact',
+        description: `Save a handoff artifact to .agent-kit/handoffs/. Returns the saved folder path to use in next-step instructions. Do NOT append version numbers (v2, v3, etc.) to the slug.`,
+        inputSchema: {
+            type: z.enum(HANDOFF_TYPES).describe('Handoff type'),
+            files: z
+                .record(z.string(), z.string())
+                .describe('Markdown files to write into the handoff folder. May include one or many files per call; repeated calls with the same type and slug add or overwrite the provided files. Filenames must match /^[A-Za-z0-9_-]+\\.md$/.'),
+            slug: z
+                .string()
+                .describe('Short identifier for the folder, e.g. "user-auth" or "PROJ-123". Do NOT append version numbers (v2, v3, etc.).'),
+        },
+        annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: true, openWorldHint: false },
     }, async (args) => handleSaveHandoff(args, getWorkspaceRoot()));
 }
 // Export DEFAULT_EXTENSIONS for backward compatibility

package/dist/mcp/integration.js

@@ -74,14 +74,19 @@ async function callRestApi(url, auth, accept = 'application/json') {
 }
 export function registerIntegrationTools(server) {
     // TOOL: GET BITBUCKET PR
-    server.tool('kit_get_bitbucket_pr', 'Get Bitbucket PR details and optionally the diff. Accepts a full PR URL or a numeric PR ID with workspace + repoSlug.', {
-        input: z.string().describe('Bitbucket PR URL or numeric PR ID'),
-        workspace: z
-            .string()
-            .optional()
-            .describe('Bitbucket workspace slug (required for numeric ID if BITBUCKET_DEFAULT_WORKSPACE not set)'),
-        repoSlug: z.string().optional().describe('Bitbucket repo slug (required for numeric ID)'),
-        includeDiff: z.boolean().optional().default(true).describe('Include unified diff in response'),
+    server.registerTool('kit_get_bitbucket_pr', {
+        title: 'Get Bitbucket PR',
+        description: 'Get Bitbucket PR details and optionally the diff. Accepts a full PR URL or a numeric PR ID with workspace + repoSlug.',
+        inputSchema: {
+            input: z.string().describe('Bitbucket PR URL or numeric PR ID'),
+            workspace: z
+                .string()
+                .optional()
+                .describe('Bitbucket workspace slug (required for numeric ID if BITBUCKET_DEFAULT_WORKSPACE not set)'),
+            repoSlug: z.string().optional().describe('Bitbucket repo slug (required for numeric ID)'),
+            includeDiff: z.boolean().optional().default(true).describe('Include unified diff in response'),
+        },
+        annotations: { readOnlyHint: true, idempotentHint: true, openWorldHint: true },
     }, async ({ input, workspace, repoSlug, includeDiff }) => {
         try {
             let ws;
@@ -145,8 +150,13 @@ ${pr.description || 'No description'}`;
         }
     });
     // TOOL: JIRA GET TICKET
-    server.tool('kit_jira_get_ticket', 'Get ticket details from Jira using the Atlassian REST API', {
-        ticketId: z.string().describe('Jira ticket ID (e.g., PROJ-123)'),
+    server.registerTool('kit_jira_get_ticket', {
+        title: 'Get Jira Ticket',
+        description: 'Get ticket details from Jira using the Atlassian REST API',
+        inputSchema: {
+            ticketId: z.string().describe('Jira ticket ID (e.g., PROJ-123)'),
+        },
+        annotations: { readOnlyHint: true, idempotentHint: true, openWorldHint: true },
     }, async ({ ticketId }) => {
         try {
             const safeTicketId = ticketId.match(/^[A-Z]+-\d+$/)?.[0];

package/dist/mcp/memory.js

@@ -12,9 +12,14 @@ import { loadGlobalSettings, resolveConversationDigestConfig } from '../core/con
  */
 function registerMemoryToolHandlers(server, subsystem, workspaceRoot = process.cwd()) {
     const { config } = subsystem;
-    server.tool('kit_memory_search', 'Search persistent memory for factual or semantic matches (decisions, concepts, entities). Extract concise keywords only, removing stopwords and filler words from the user request before querying. Use kit_memory_recent for temporal or recency questions.', {
-        query: z.string().min(1).describe('Keyword-only search query with stopwords removed'),
-        top_k: z.number().int().positive().optional().describe('Number of results to return'),
+    server.registerTool('kit_memory_search', {
+        title: 'Search Memory',
+        description: 'Search persistent memory for factual or semantic matches (decisions, concepts, entities). Extract concise keywords only, removing stopwords and filler words from the user request before querying. Use kit_memory_recent for temporal or recency questions.',
+        inputSchema: {
+            query: z.string().min(1).describe('Keyword-only search query with stopwords removed'),
+            top_k: z.number().int().positive().optional().describe('Number of results to return'),
+        },
+        annotations: { readOnlyHint: true, idempotentHint: true, openWorldHint: false },
     }, async ({ query, top_k }) => {
         try {
             if (!subsystem.indexer || !subsystem.store) {
@@ -44,9 +49,14 @@ function registerMemoryToolHandlers(server, subsystem, workspaceRoot = process.c
             return mcpText(`kit_memory_search failed: ${formatError(err)}`);
         }
     });
-    server.tool('kit_memory_recent', 'Return the N most recently updated memory sources. Optionally filter by source type. Use for temporal questions like "what did we do last session" or "recent".', {
-        n: z.number().int().positive().max(50).optional().describe('How many recent sources to return (default 5)'),
-        source_type: z.enum(SOURCE_TYPES).optional().describe('Optional source type filter'),
+    server.registerTool('kit_memory_recent', {
+        title: 'Recent Memory Sources',
+        description: 'Return the N most recently updated memory sources. Optionally filter by source type. Use for temporal questions like "what did we do last session" or "recent".',
+        inputSchema: {
+            n: z.number().int().positive().max(50).optional().describe('How many recent sources to return (default 5)'),
+            source_type: z.enum(SOURCE_TYPES).optional().describe('Optional source type filter'),
+        },
+        annotations: { readOnlyHint: true, idempotentHint: true, openWorldHint: false },
     }, async ({ n, source_type }) => {
         try {
             if (!subsystem.store) {
@@ -75,8 +85,13 @@ function registerMemoryToolHandlers(server, subsystem, workspaceRoot = process.c
             return mcpText(`kit_memory_recent failed: ${formatError(err)}`);
         }
     });
-    server.tool('kit_memory_save', 'Save content to wiki/raw for inclusion after the next /wiki compile.', {
-        content: z.string().min(1).describe('Content to save to memory'),
+    server.registerTool('kit_memory_save', {
+        title: 'Save to Memory',
+        description: 'Save content to wiki/raw for inclusion after the next /wiki compile.',
+        inputSchema: {
+            content: z.string().min(1).describe('Content to save to memory'),
+        },
+        annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: false, openWorldHint: false },
     }, async ({ content }) => {
         try {
             if (!subsystem.indexer) {
@@ -100,9 +115,14 @@ function registerMemoryToolHandlers(server, subsystem, workspaceRoot = process.c
             });
         }
     });
-    server.tool('kit_memory_digest_init', 'Initialize or toggle local conversation digesting. Downloads/cache-loads the pinned local model only when explicitly invoked.', {
-        model_id: z.string().optional().describe('Digest model id'),
-        enabled: z.boolean().optional().describe('Enable or disable conversation digesting'),
+    server.registerTool('kit_memory_digest_init', {
+        title: 'Initialize Conversation Digest',
+        description: 'Initialize or toggle local conversation digesting. Downloads/cache-loads the pinned local model only when explicitly invoked.',
+        inputSchema: {
+            model_id: z.string().optional().describe('Digest model id'),
+            enabled: z.boolean().optional().describe('Enable or disable conversation digesting'),
+        },
+        annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: true, openWorldHint: true },
     }, async ({ model_id, enabled }) => {
         try {
             const existingConfig = resolveConversationDigestConfig(loadGlobalSettings());

package/dist/mcp/registration.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/mcp/registration.test.js

@@ -0,0 +1,77 @@
+import * as assert from 'node:assert/strict';
+import { describe, test } from 'node:test';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory.js';
+import { registerCoreTools } from './core.js';
+import { registerAgentTools } from './agent.js';
+import { registerIntegrationTools } from './integration.js';
+const EXPECTED_NON_MEMORY_TOOLS = [
+    'kit_save_handoff',
+    'kit_trigger_agent',
+    'kit_get_bitbucket_pr',
+    'kit_jira_get_ticket',
+];
+async function buildClientWithTools() {
+    const server = new McpServer({ name: 'test-kit', version: '0.0.0' });
+    registerIntegrationTools(server);
+    registerCoreTools(server);
+    registerAgentTools(server);
+    const [serverTransport, clientTransport] = InMemoryTransport.createLinkedPair();
+    const client = new Client({ name: 'test-client', version: '0.0.0' });
+    await Promise.all([server.connect(serverTransport), client.connect(clientTransport)]);
+    return client;
+}
+describe('MCP tool registration smoke tests', () => {
+    test('R1: all 4 non-memory tool names are registered', async () => {
+        const client = await buildClientWithTools();
+        const { tools } = await client.listTools();
+        const names = tools.map((t) => t.name);
+        for (const expected of EXPECTED_NON_MEMORY_TOOLS) {
+            assert.ok(names.includes(expected), `expected tool "${expected}" to be registered, got: ${names.join(', ')}`);
+        }
+    });
+    test('R2: each non-memory tool has a non-empty description and inputSchema (BC3)', async () => {
+        const client = await buildClientWithTools();
+        const { tools } = await client.listTools();
+        const byName = Object.fromEntries(tools.map((t) => [t.name, t]));
+        for (const name of EXPECTED_NON_MEMORY_TOOLS) {
+            const tool = byName[name];
+            assert.ok(tool, `tool "${name}" not found`);
+            assert.ok(tool.description && tool.description.length > 0, `tool "${name}" must have a non-empty description`);
+            assert.ok(tool.inputSchema, `tool "${name}" must have an inputSchema`);
+        }
+    });
+    test('R3: kit_trigger_agent (readOnlyHint:false) has correct annotations (BC5)', async () => {
+        const client = await buildClientWithTools();
+        const { tools } = await client.listTools();
+        const tool = tools.find((t) => t.name === 'kit_trigger_agent');
+        assert.ok(tool, 'kit_trigger_agent must be registered');
+        assert.ok(tool.annotations, 'kit_trigger_agent must have annotations');
+        assert.equal(tool.annotations.readOnlyHint, false, 'readOnlyHint must be false');
+        assert.equal(tool.annotations.destructiveHint, true, 'destructiveHint must be true');
+        assert.equal(tool.annotations.idempotentHint, false, 'idempotentHint must be false');
+        assert.equal(tool.annotations.openWorldHint, true, 'openWorldHint must be true');
+    });
+    test('R4: kit_save_handoff (readOnlyHint:false) has correct annotations (BC5)', async () => {
+        const client = await buildClientWithTools();
+        const { tools } = await client.listTools();
+        const tool = tools.find((t) => t.name === 'kit_save_handoff');
+        assert.ok(tool, 'kit_save_handoff must be registered');
+        assert.ok(tool.annotations, 'kit_save_handoff must have annotations');
+        assert.equal(tool.annotations.readOnlyHint, false, 'readOnlyHint must be false');
+        assert.equal(tool.annotations.destructiveHint, false, 'destructiveHint must be false');
+        assert.equal(tool.annotations.idempotentHint, true, 'idempotentHint must be true');
+        assert.equal(tool.annotations.openWorldHint, false, 'openWorldHint must be false');
+    });
+    test('R5: kit_get_bitbucket_pr (readOnlyHint:true) has correct annotations (BC5)', async () => {
+        const client = await buildClientWithTools();
+        const { tools } = await client.listTools();
+        const tool = tools.find((t) => t.name === 'kit_get_bitbucket_pr');
+        assert.ok(tool, 'kit_get_bitbucket_pr must be registered');
+        assert.ok(tool.annotations, 'kit_get_bitbucket_pr must have annotations');
+        assert.equal(tool.annotations.readOnlyHint, true, 'readOnlyHint must be true');
+        assert.equal(tool.annotations.idempotentHint, true, 'idempotentHint must be true');
+        assert.equal(tool.annotations.openWorldHint, true, 'openWorldHint must be true');
+    });
+});

package/dist/server.js

@@ -36078,7 +36078,6 @@ var HANDOFF_TYPES = [
   "plan",
   "ticket",
   "research",
-  "scenario",
   "investigation",
   "code"
 ];
@@ -36161,17 +36160,21 @@ function handleSaveHandoff(args, workspaceRoot2) {
   }
 }
 function registerCoreTools(server2) {
-  server2.tool(
+  server2.registerTool(
     "kit_save_handoff",
-    `Save a handoff artifact to .agent-kit/handoffs/. Returns the saved folder path to use in next-step instructions. Do NOT append version numbers (v2, v3, etc.) to the slug.`,
     {
-      type: external_exports.enum(HANDOFF_TYPES).describe("Handoff type"),
-      files: external_exports.record(external_exports.string(), external_exports.string()).describe(
-        "Markdown files to write into the handoff folder. May include one or many files per call; repeated calls with the same type and slug add or overwrite the provided files. Filenames must match /^[A-Za-z0-9_-]+\\.md$/."
-      ),
-      slug: external_exports.string().describe(
-        'Short identifier for the folder, e.g. "user-auth" or "PROJ-123". Do NOT append version numbers (v2, v3, etc.).'
-      )
+      title: "Save Handoff Artifact",
+      description: `Save a handoff artifact to .agent-kit/handoffs/. Returns the saved folder path to use in next-step instructions. Do NOT append version numbers (v2, v3, etc.) to the slug.`,
+      inputSchema: {
+        type: external_exports.enum(HANDOFF_TYPES).describe("Handoff type"),
+        files: external_exports.record(external_exports.string(), external_exports.string()).describe(
+          "Markdown files to write into the handoff folder. May include one or many files per call; repeated calls with the same type and slug add or overwrite the provided files. Filenames must match /^[A-Za-z0-9_-]+\\.md$/."
+        ),
+        slug: external_exports.string().describe(
+          'Short identifier for the folder, e.g. "user-auth" or "PROJ-123". Do NOT append version numbers (v2, v3, etc.).'
+        )
+      },
+      annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: true, openWorldHint: false }
     },
     async (args) => handleSaveHandoff(args, getWorkspaceRoot())
   );
@@ -36281,12 +36284,16 @@ ${"\u2500".repeat(60)}
   return { logPath: path6.relative(workspaceRoot2, logFile), logStream };
 }
 function registerAgentTools(server2) {
-  server2.tool(
+  server2.registerTool(
     "kit_trigger_agent",
-    "!Important: Trigger only when user explicitly asks to delegate a task to an external agent CLI (gemini, claude, or codex). The task can be a direct message or a path to a handoff folder (.agent-kit/handoffs/feature-slug/plan).",
     {
-      agent: external_exports.enum(["gemini", "claude", "codex"]).describe('The agent CLI to invoke: "gemini", "claude", or "codex"'),
-      task: external_exports.string().describe('Task message or path to a handoff folder (e.g., ".agent-kit/handoffs/feature-slug/plan")')
+      title: "Delegate to External Agent",
+      description: "!Important: Trigger only when user explicitly asks to delegate a task to an external agent CLI (gemini, claude, or codex). The task can be a direct message or a path to a handoff folder (.agent-kit/handoffs/feature-slug/plan).",
+      inputSchema: {
+        agent: external_exports.enum(["gemini", "claude", "codex"]).describe('The agent CLI to invoke: "gemini", "claude", or "codex"'),
+        task: external_exports.string().describe('Task message or path to a handoff folder (e.g., ".agent-kit/handoffs/feature-slug/plan")')
+      },
+      annotations: { readOnlyHint: false, destructiveHint: true, idempotentHint: false, openWorldHint: true }
     },
     async ({ agent, task }, extra) => {
       const workspaceRoot2 = getWorkspaceRoot();
@@ -36641,14 +36648,18 @@ async function callRestApi(url, auth, accept = "application/json") {
   return accept === "text/plain" ? resp.text() : resp.json();
 }
 function registerIntegrationTools(server2) {
-  server2.tool(
+  server2.registerTool(
     "kit_get_bitbucket_pr",
-    "Get Bitbucket PR details and optionally the diff. Accepts a full PR URL or a numeric PR ID with workspace + repoSlug.",
     {
-      input: external_exports.string().describe("Bitbucket PR URL or numeric PR ID"),
-      workspace: external_exports.string().optional().describe("Bitbucket workspace slug (required for numeric ID if BITBUCKET_DEFAULT_WORKSPACE not set)"),
-      repoSlug: external_exports.string().optional().describe("Bitbucket repo slug (required for numeric ID)"),
-      includeDiff: external_exports.boolean().optional().default(true).describe("Include unified diff in response")
+      title: "Get Bitbucket PR",
+      description: "Get Bitbucket PR details and optionally the diff. Accepts a full PR URL or a numeric PR ID with workspace + repoSlug.",
+      inputSchema: {
+        input: external_exports.string().describe("Bitbucket PR URL or numeric PR ID"),
+        workspace: external_exports.string().optional().describe("Bitbucket workspace slug (required for numeric ID if BITBUCKET_DEFAULT_WORKSPACE not set)"),
+        repoSlug: external_exports.string().optional().describe("Bitbucket repo slug (required for numeric ID)"),
+        includeDiff: external_exports.boolean().optional().default(true).describe("Include unified diff in response")
+      },
+      annotations: { readOnlyHint: true, idempotentHint: true, openWorldHint: true }
     },
     async ({ input, workspace, repoSlug, includeDiff }) => {
       try {
@@ -36725,11 +36736,15 @@ ${diff}
       }
     }
   );
-  server2.tool(
+  server2.registerTool(
     "kit_jira_get_ticket",
-    "Get ticket details from Jira using the Atlassian REST API",
     {
-      ticketId: external_exports.string().describe("Jira ticket ID (e.g., PROJ-123)")
+      title: "Get Jira Ticket",
+      description: "Get ticket details from Jira using the Atlassian REST API",
+      inputSchema: {
+        ticketId: external_exports.string().describe("Jira ticket ID (e.g., PROJ-123)")
+      },
+      annotations: { readOnlyHint: true, idempotentHint: true, openWorldHint: true }
     },
     async ({ ticketId }) => {
       try {
@@ -37796,26 +37811,26 @@ var DigestModelRegistryError = class extends Error {
 var DIGEST_MODEL_REGISTRY = {
   ["tiny" /* TINY */]: {
     id: "tiny" /* TINY */,
-    ggufUri: "hf:bartowski/Qwen2.5-0.5B-Instruct-GGUF/Qwen2.5-0.5B-Instruct-Q4_K_M.gguf",
-    approxSizeBytes: 4e8,
+    ggufUri: "hf:lmstudio-community/Qwen3.5-0.8B-GGUF/Qwen3.5-0.8B-Q4_K_M.gguf",
+    approxSizeBytes: 528e6,
     license: "Apache-2.0",
-    sourceUrl: "https://huggingface.co/bartowski/Qwen2.5-0.5B-Instruct-GGUF",
+    sourceUrl: "https://huggingface.co/lmstudio-community/Qwen3.5-0.8B-GGUF",
     enabled: true
   },
   ["base" /* BASE */]: {
     id: "base" /* BASE */,
-    ggufUri: "hf:bartowski/Qwen2.5-1.5B-Instruct-GGUF/Qwen2.5-1.5B-Instruct-Q4_K_M.gguf",
-    approxSizeBytes: 1e9,
+    ggufUri: "hf:lmstudio-community/Qwen3.5-2B-GGUF/Qwen3.5-2B-Q4_K_M.gguf",
+    approxSizeBytes: 127e7,
     license: "Apache-2.0",
-    sourceUrl: "https://huggingface.co/bartowski/Qwen2.5-1.5B-Instruct-GGUF",
+    sourceUrl: "https://huggingface.co/lmstudio-community/Qwen3.5-2B-GGUF",
     enabled: true
   },
   ["large" /* LARGE */]: {
     id: "large" /* LARGE */,
-    ggufUri: "hf:bartowski/Qwen2.5-3B-Instruct-GGUF/Qwen2.5-3B-Instruct-Q4_K_M.gguf",
-    approxSizeBytes: 19e8,
+    ggufUri: "hf:lmstudio-community/Qwen3.5-4B-GGUF/Qwen3.5-4B-Q4_K_M.gguf",
+    approxSizeBytes: 271e7,
     license: "Apache-2.0",
-    sourceUrl: "https://huggingface.co/bartowski/Qwen2.5-3B-Instruct-GGUF",
+    sourceUrl: "https://huggingface.co/lmstudio-community/Qwen3.5-4B-GGUF",
     enabled: true
   }
 };
@@ -37933,7 +37948,8 @@ async function createLlamaLocalDigestProvider(modelId) {
         const response = await withTimeout(
           session.prompt(userContent, {
             maxTokens: LLAMA_MAX_GENERATED_TOKENS,
-            temperature: LLAMA_TEMPERATURE
+            temperature: LLAMA_TEMPERATURE,
+            budgets: { thoughtTokens: 0 }
           }),
           options.timeoutMs,
           () => new LlamaLocalDigestProviderError(`Llama provider timed out after ${options.timeoutMs}ms`)
@@ -38143,12 +38159,16 @@ async function initializeConversationDigestModel(input) {
 // src/mcp/memory.ts
 function registerMemoryToolHandlers(server2, subsystem, workspaceRoot2 = process.cwd()) {
   const { config: config2 } = subsystem;
-  server2.tool(
+  server2.registerTool(
     "kit_memory_search",
-    "Search persistent memory for factual or semantic matches (decisions, concepts, entities). Extract concise keywords only, removing stopwords and filler words from the user request before querying. Use kit_memory_recent for temporal or recency questions.",
     {
-      query: external_exports.string().min(1).describe("Keyword-only search query with stopwords removed"),
-      top_k: external_exports.number().int().positive().optional().describe("Number of results to return")
+      title: "Search Memory",
+      description: "Search persistent memory for factual or semantic matches (decisions, concepts, entities). Extract concise keywords only, removing stopwords and filler words from the user request before querying. Use kit_memory_recent for temporal or recency questions.",
+      inputSchema: {
+        query: external_exports.string().min(1).describe("Keyword-only search query with stopwords removed"),
+        top_k: external_exports.number().int().positive().optional().describe("Number of results to return")
+      },
+      annotations: { readOnlyHint: true, idempotentHint: true, openWorldHint: false }
     },
     async ({ query, top_k }) => {
       try {
@@ -38175,12 +38195,16 @@ ${content}`;
       }
     }
   );
-  server2.tool(
+  server2.registerTool(
     "kit_memory_recent",
-    'Return the N most recently updated memory sources. Optionally filter by source type. Use for temporal questions like "what did we do last session" or "recent".',
     {
-      n: external_exports.number().int().positive().max(50).optional().describe("How many recent sources to return (default 5)"),
-      source_type: external_exports.enum(SOURCE_TYPES).optional().describe("Optional source type filter")
+      title: "Recent Memory Sources",
+      description: 'Return the N most recently updated memory sources. Optionally filter by source type. Use for temporal questions like "what did we do last session" or "recent".',
+      inputSchema: {
+        n: external_exports.number().int().positive().max(50).optional().describe("How many recent sources to return (default 5)"),
+        source_type: external_exports.enum(SOURCE_TYPES).optional().describe("Optional source type filter")
+      },
+      annotations: { readOnlyHint: true, idempotentHint: true, openWorldHint: false }
     },
     async ({ n, source_type }) => {
       try {
@@ -38208,11 +38232,15 @@ ${content}`);
       }
     }
   );
-  server2.tool(
+  server2.registerTool(
     "kit_memory_save",
-    "Save content to wiki/raw for inclusion after the next /wiki compile.",
     {
-      content: external_exports.string().min(1).describe("Content to save to memory")
+      title: "Save to Memory",
+      description: "Save content to wiki/raw for inclusion after the next /wiki compile.",
+      inputSchema: {
+        content: external_exports.string().min(1).describe("Content to save to memory")
+      },
+      annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: false, openWorldHint: false }
     },
     async ({ content }) => {
       try {
@@ -38237,12 +38265,16 @@ ${content}`);
       }
     }
   );
-  server2.tool(
+  server2.registerTool(
     "kit_memory_digest_init",
-    "Initialize or toggle local conversation digesting. Downloads/cache-loads the pinned local model only when explicitly invoked.",
     {
-      model_id: external_exports.string().optional().describe("Digest model id"),
-      enabled: external_exports.boolean().optional().describe("Enable or disable conversation digesting")
+      title: "Initialize Conversation Digest",
+      description: "Initialize or toggle local conversation digesting. Downloads/cache-loads the pinned local model only when explicitly invoked.",
+      inputSchema: {
+        model_id: external_exports.string().optional().describe("Digest model id"),
+        enabled: external_exports.boolean().optional().describe("Enable or disable conversation digesting")
+      },
+      annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: true, openWorldHint: true }
     },
     async ({ model_id, enabled }) => {
       try {

package/dist/services/digest/digest-llama-local.test.js

@@ -1,21 +1,64 @@
 import * as assert from 'node:assert/strict';
 import { describe, test } from 'node:test';
-import { createLlamaLocalDigestProvider } from '../digest/providers/llama-local.js';
-const INPUT = {
-    sourcePath: '.agent-kit/wiki/archive/conversations/conv_2026-05-19T00-45-17-697Z.md',
-    contentHash: 'tiny',
-    content: '**User:** use local digest generation for archived conversations',
-};
-describe('createLlamaLocalDigestProvider', () => {
-    test('runs an opt-in smoke test against a real model', { skip: !process.env.AGENT_KIT_DIGEST_MODEL_ID }, async () => {
+import { createLlamaLocalDigestProvider, sanitizeConversationDigestMarkdown, titleFromSource, trimConversationExport, } from '../digest/providers/llama-local.js';
+describe('titleFromSource', () => {
+    test('strips the extension and turns separators into spaces', () => {
+        assert.equal(titleFromSource('/a/b/conv_claude_1bda-2b2b.md'), 'conv claude 1bda 2b2b');
+    });
+    test('handles a bare filename with no directory', () => {
+        assert.equal(titleFromSource('inbox.md'), 'inbox');
+    });
+});
+describe('trimConversationExport', () => {
+    test('returns the trimmed content unchanged when within the limit', () => {
+        assert.equal(trimConversationExport('  hello world  ', 100), 'hello world');
+    });
+    test('keeps the tail and prepends an omission notice when over the limit', () => {
+        const content = 'A'.repeat(50) + 'TAIL_MARKER';
+        const out = trimConversationExport(content, 11);
+        assert.match(out, /^\[Earlier conversation content omitted/);
+        assert.ok(out.endsWith('TAIL_MARKER'), 'keeps the most recent content');
+        assert.equal(out.split('\n\n')[1], 'TAIL_MARKER');
+    });
+});
+describe('sanitizeConversationDigestMarkdown', () => {
+    test('unwraps a fenced markdown block', () => {
+        const out = sanitizeConversationDigestMarkdown('```markdown\n# Title\n\nbody\n```');
+        assert.equal(out, '# Title\n\nbody\n');
+    });
+    test('leaves unfenced text intact and guarantees a single trailing newline', () => {
+        const out = sanitizeConversationDigestMarkdown('  plain summary  ');
+        assert.equal(out, 'plain summary\n');
+    });
+    test('does not strip text that merely mentions a code fence inline', () => {
+        const out = sanitizeConversationDigestMarkdown('Use ```bash``` to run it.');
+        assert.equal(out, 'Use ```bash``` to run it.\n');
+    });
+});
+describe('createLlamaLocalDigestProvider (opt-in real model)', () => {
+    const INPUT = {
+        sourcePath: '.agent-kit/wiki/raw/conv_2026-05-19T00-45-17-697Z.md',
+        contentHash: 'tiny',
+        content: '**User:** use local digest generation for archived conversations',
+    };
+    test('generates a digest with the documented metadata header and a non-empty body', { skip: !process.env.AGENT_KIT_DIGEST_MODEL_ID }, async () => {
         const modelId = process.env.AGENT_KIT_DIGEST_MODEL_ID;
         const provider = await createLlamaLocalDigestProvider(modelId);
-        const output = await provider.generateDigestMarkdown(INPUT, {
-            modelId,
-            maxInputChars: 1000,
-            timeoutMs: 120_000,
-        });
-        assert.match(output, /^# Conversation Digest:/);
-        assert.match(output, /Status: provisional/);
+        try {
+            const output = await provider.generateDigestMarkdown(INPUT, {
+                modelId,
+                maxInputChars: 1000,
+                timeoutMs: 120_000,
+            });
+            assert.match(output, /^## Digest: conv 2026/);
+            assert.match(output, /\| \*\*Source\*\* \| .*conv_2026-05-19/);
+            assert.match(output, new RegExp(`\\| \\*\\*Model\\*\\* \\| ${modelId} \\|`));
+            assert.doesNotMatch(output, /<\/?think>/i);
+            const body = output.replace(/^##[\s\S]*?\n\n/, '').trim();
+            assert.ok(body.length > 0, 'expected a non-empty digest body');
+        }
+        finally {
+            await provider.dispose?.();
+        }
     });
 });

package/dist/services/digest/model-registry.js

@@ -8,26 +8,26 @@ class DigestModelRegistryError extends Error {
 const DIGEST_MODEL_REGISTRY = {
     [DigestModelId.TINY]: {
         id: DigestModelId.TINY,
-        ggufUri: 'hf:bartowski/Qwen2.5-0.5B-Instruct-GGUF/Qwen2.5-0.5B-Instruct-Q4_K_M.gguf',
-        approxSizeBytes: 400_000_000,
+        ggufUri: 'hf:lmstudio-community/Qwen3.5-0.8B-GGUF/Qwen3.5-0.8B-Q4_K_M.gguf',
+        approxSizeBytes: 528_000_000,
         license: 'Apache-2.0',
-        sourceUrl: 'https://huggingface.co/bartowski/Qwen2.5-0.5B-Instruct-GGUF',
+        sourceUrl: 'https://huggingface.co/lmstudio-community/Qwen3.5-0.8B-GGUF',
         enabled: true,
     },
     [DigestModelId.BASE]: {
         id: DigestModelId.BASE,
-        ggufUri: 'hf:bartowski/Qwen2.5-1.5B-Instruct-GGUF/Qwen2.5-1.5B-Instruct-Q4_K_M.gguf',
-        approxSizeBytes: 1_000_000_000,
+        ggufUri: 'hf:lmstudio-community/Qwen3.5-2B-GGUF/Qwen3.5-2B-Q4_K_M.gguf',
+        approxSizeBytes: 1_270_000_000,
         license: 'Apache-2.0',
-        sourceUrl: 'https://huggingface.co/bartowski/Qwen2.5-1.5B-Instruct-GGUF',
+        sourceUrl: 'https://huggingface.co/lmstudio-community/Qwen3.5-2B-GGUF',
         enabled: true,
     },
     [DigestModelId.LARGE]: {
         id: DigestModelId.LARGE,
-        ggufUri: 'hf:bartowski/Qwen2.5-3B-Instruct-GGUF/Qwen2.5-3B-Instruct-Q4_K_M.gguf',
-        approxSizeBytes: 1_900_000_000,
+        ggufUri: 'hf:lmstudio-community/Qwen3.5-4B-GGUF/Qwen3.5-4B-Q4_K_M.gguf',
+        approxSizeBytes: 2_710_000_000,
         license: 'Apache-2.0',
-        sourceUrl: 'https://huggingface.co/bartowski/Qwen2.5-3B-Instruct-GGUF',
+        sourceUrl: 'https://huggingface.co/lmstudio-community/Qwen3.5-4B-GGUF',
         enabled: true,
     },
 };

package/dist/services/digest/providers/llama-local.d.ts

@@ -1,2 +1,5 @@
 import type { ConversationDigestProvider } from '../types.js';
+export declare function titleFromSource(sourcePath: string): string;
+export declare function trimConversationExport(content: string, maxInputChars: number): string;
+export declare function sanitizeConversationDigestMarkdown(generatedText: string): string;
 export declare function createLlamaLocalDigestProvider(modelId: string): Promise<ConversationDigestProvider>;

package/dist/services/digest/providers/llama-local.js

@@ -11,13 +11,13 @@ class LlamaLocalDigestProviderError extends Error {
         this.name = 'LlamaLocalDigestProviderError';
     }
 }
-function titleFromSource(sourcePath) {
+export function titleFromSource(sourcePath) {
     return path
         .basename(sourcePath)
         .replace(/\.[^.]+$/, '')
         .replace(/[_-]+/g, ' ');
 }
-function trimConversationExport(content, maxInputChars) {
+export function trimConversationExport(content, maxInputChars) {
     const trimmed = content.trim();
     if (trimmed.length <= maxInputChars)
         return trimmed;
@@ -54,7 +54,7 @@ function stripCodeFence(text) {
     const match = trimmed.match(/^```(?:markdown|md)?\s*\n([\s\S]*?)\n```$/i);
     return match ? match[1].trim() : trimmed;
 }
-function sanitizeConversationDigestMarkdown(generatedText) {
+export function sanitizeConversationDigestMarkdown(generatedText) {
     return stripCodeFence(generatedText).trim() + '\n';
 }
 export async function createLlamaLocalDigestProvider(modelId) {
@@ -104,6 +104,7 @@ export async function createLlamaLocalDigestProvider(modelId) {
                 const response = await withTimeout(session.prompt(userContent, {
                     maxTokens: LLAMA_MAX_GENERATED_TOKENS,
                     temperature: LLAMA_TEMPERATURE,
+                    budgets: { thoughtTokens: 0 },
                 }), options.timeoutMs, () => new LlamaLocalDigestProviderError(`Llama provider timed out after ${options.timeoutMs}ms`));
                 const metadata = `## Digest: ${titleFromSource(input.sourcePath)}
 

package/dist/services/memory/memory.test.js

@@ -11,8 +11,9 @@ function makeMockServer() {
     const tools = new Map();
     const descriptions = new Map();
     const server = {
-        tool(name, desc, _schema, handler) {
-            descriptions.set(name, desc);
+        registerTool(name, config, handler) {
+            if (config.description)
+                descriptions.set(name, config.description);
             tools.set(name, handler);
         },
     };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hanhnd/agent-kit",
-  "version": "1.0.40",
+  "version": "1.0.42",
   "description": "Agent Kit MCP server for software development agent workflows",
   "type": "module",
   "main": "dist/server.js",