@hanhnd/agent-kit 1.0.32 → 1.0.33

package/dist/{tools/handoffs.js → core/handoffs/index.js}

@@ -12,7 +12,7 @@ const TICKET_ID_PATTERN = /\b[A-Z][A-Z0-9]+-\d+\b/;
 function findTicketId(value) {
     return value.match(TICKET_ID_PATTERN)?.[0].toLowerCase() ?? null;
 }
-export function sanitizeFeatureSlug(value) {
+function sanitizeFeatureSlug(value) {
     return value
         .replace(/[^a-zA-Z0-9-_]/g, '-')
         .replace(/-+/g, '-')
@@ -26,13 +26,13 @@ function contentSlugCandidate(content) {
         .find((line) => line.length > 0);
     return heading ?? content.replace(/\s+/g, ' ').trim().slice(0, 80);
 }
-export function normalizeHandoffType(type) {
+function normalizeHandoffType(type) {
     if (!HANDOFF_TYPES.includes(type)) {
         throw new Error(`Unsupported handoff type: ${type}`);
     }
     return type;
 }
-export function deriveFeatureSlug(input) {
+function deriveFeatureSlug(input) {
     const requestedTicketSlug = findTicketId(input.requestedSlug);
     if (requestedTicketSlug)
         return requestedTicketSlug;

package/dist/{tools/security.d.ts → core/security/index.d.ts}

@@ -18,26 +18,7 @@ export declare function sanitize(input: string): string;
  * Uses stricter path.sep check to prevent prefix matching flaws.
  */
 export declare function validatePath(filePath: string, baseDir?: string): string;
-/**
- * Safe git command execution using execFileSync
- * Includes stderr in error message for better debugging
- *
- * @param timeout Default from GEMINI_KIT_GIT_TIMEOUT env var or 30s
- */
-export declare function safeGit(args: string[], options?: {
-    timeout?: number;
-    cwd?: string;
-}): string;
-/**
- * Check if a command exists (cross-platform)
- * Uses 'where' on Windows, 'which' on macOS/Linux
- */
 export declare function commandExists(cmd: string): boolean;
-/**
- * Async file finder - non-blocking for large repos
- * Uses queue-based approach to prevent stack overflow on deep directories
- */
-export declare function findFilesAsync(dir: string, extensions: string[], maxFiles: number, excludeDirs?: string[]): Promise<string[]>;
 /**
  * Sanitize output text by redacting sensitive tokens (API keys, etc.)
  */

package/dist/{tools/security.js → core/security/index.js}

@@ -3,10 +3,9 @@
  * Exported utilities for safe command execution
  */
 import { execFileSync } from 'child_process';
-import * as fs from 'fs';
 import * as path from 'path';
-import { getWorkspaceRoot } from '../utils/utils.js';
-import { FORBIDDEN_DIRS, FORBIDDEN_FILES, FORBIDDEN_PATTERNS } from './config.js';
+import { getWorkspaceRoot } from '../../utils/utils.js';
+import { FORBIDDEN_DIRS, FORBIDDEN_FILES, FORBIDDEN_PATTERNS } from '../config/index.js';
 /**
  * Sanitize string for safe use with execFileSync
  * Only removes dangerous shell operators - safe chars like !?#* are allowed
@@ -52,42 +51,6 @@ export function validatePath(filePath, baseDir = process.cwd()) {
     }
     return resolved;
 }
-/**
- * Extract stderr from error for better logging
- */
-function extractStderr(error) {
-    if (error && typeof error === 'object' && 'stderr' in error) {
-        return String(error.stderr);
-    }
-    return '';
-}
-// Configurable timeouts via environment variables
-const GIT_TIMEOUT = parseInt(process.env.GEMINI_KIT_GIT_TIMEOUT || '30000', 10);
-/**
- * Safe git command execution using execFileSync
- * Includes stderr in error message for better debugging
- *
- * @param timeout Default from GEMINI_KIT_GIT_TIMEOUT env var or 30s
- */
-export function safeGit(args, options) {
-    try {
-        return execFileSync('git', args, {
-            encoding: 'utf8',
-            timeout: options?.timeout || GIT_TIMEOUT,
-            cwd: options?.cwd || getWorkspaceRoot(),
-            maxBuffer: 10 * 1024 * 1024, // 10MB
-        });
-    }
-    catch (error) {
-        const stderr = extractStderr(error);
-        const baseMsg = error instanceof Error ? error.message : String(error);
-        throw new Error(`Git command failed: ${baseMsg}${stderr ? `\nDetails: ${stderr}` : ''}`);
-    }
-}
-/**
- * Check if a command exists (cross-platform)
- * Uses 'where' on Windows, 'which' on macOS/Linux
- */
 export function commandExists(cmd) {
     try {
         const checkCmd = process.platform === 'win32' ? 'where' : 'which';
@@ -103,42 +66,6 @@ export function commandExists(cmd) {
         return false;
     }
 }
-/**
- * Async file finder - non-blocking for large repos
- * Uses queue-based approach to prevent stack overflow on deep directories
- */
-export async function findFilesAsync(dir, extensions, maxFiles, excludeDirs = ['node_modules', '.git', 'dist', 'build', 'coverage']) {
-    const results = [];
-    // Use queue instead of recursion to prevent stack overflow
-    const queue = [{ fullPath: dir, relativePath: '' }];
-    while (queue.length > 0 && results.length < maxFiles) {
-        const current = queue.shift();
-        let entries;
-        try {
-            entries = await fs.promises.readdir(current.fullPath, { withFileTypes: true });
-        }
-        catch {
-            continue; // Skip directories we can't read
-        }
-        for (const entry of entries) {
-            if (results.length >= maxFiles)
-                break;
-            const entryFullPath = path.join(current.fullPath, entry.name);
-            const entryRelPath = current.relativePath ? path.join(current.relativePath, entry.name) : entry.name;
-            if (entry.isDirectory()) {
-                if (!excludeDirs.includes(entry.name)) {
-                    queue.push({ fullPath: entryFullPath, relativePath: entryRelPath });
-                }
-            }
-            else if (entry.isFile()) {
-                if (extensions.some((ext) => entry.name.endsWith(ext))) {
-                    results.push(entryRelPath);
-                }
-            }
-        }
-    }
-    return results;
-}
 /**
  * Sanitize output text by redacting sensitive tokens (API keys, etc.)
  */
@@ -148,7 +75,7 @@ export function sanitizeOutput(text) {
         /SK-[A-Za-z0-9]{40,}/g, // OpenAI keys
         /AKIA[0-9A-Z]{16}/g, // AWS access keys
         /xox[baprs]-[A-Za-z0-9-]{10,}/g, // Slack tokens
-        /[Aa]tlassian[_\-]?[Aa][Pp][Ii][_\-]?[Tt]oken[^\s]*\s*[:=]\s*\S+/g, // Atlassian token patterns
+        /[Aa]tlassian[_-]?[Aa][Pp][Ii][_-]?[Tt]oken[^\s]*\s*[:=]\s*\S+/g, // Atlassian token patterns
     ];
     let sanitized = text;
     for (const pattern of SECRET_PATTERNS) {

package/dist/entrypoints/cli.js

@@ -0,0 +1,11 @@
+#!/usr/bin/env node
+import { runAgentKitCli } from '../cli/index.js';
+try {
+    const exitCode = await runAgentKitCli(process.argv.slice(2), process.env);
+    process.exit(exitCode);
+}
+catch (err) {
+    const message = err instanceof Error ? err.message : String(err);
+    process.stderr.write('[agent-kit] CLI error: ' + message + '\n');
+    process.exit(1);
+}

package/dist/entrypoints/server.js

@@ -0,0 +1,37 @@
+#!/usr/bin/env node
+/**
+ * Agent-Kit MCP Server
+ * Provides custom tools for agent orchestration
+ *
+ * Modular architecture - tools split into separate modules
+ */
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { registerCoreTools } from '../mcp/core.js';
+import { registerAgentTools } from '../mcp/agent.js';
+import { registerIntegrationTools } from '../mcp/integration.js';
+import { loadProjectSettings } from '../core/config/index.js';
+import { registerMemoryTools } from '../mcp/memory.js';
+import { getWorkspaceRoot } from '../utils/utils.js';
+const server = new McpServer({
+    name: 'kit-agents',
+    version: '1.0.0',
+});
+// ═══════════════════════════════════════════════════════════════
+// REGISTER MODULAR TOOLS
+// ═══════════════════════════════════════════════════════════════
+registerIntegrationTools(server);
+registerCoreTools(server);
+registerAgentTools(server);
+const workspaceRoot = getWorkspaceRoot();
+const settings = loadProjectSettings(workspaceRoot);
+const memoryIndexer = registerMemoryTools(server, settings, workspaceRoot);
+// ═══════════════════════════════════════════════════════════════
+// START SERVER
+// ═══════════════════════════════════════════════════════════════
+const transport = new StdioServerTransport();
+await server.connect(transport);
+// Fire-and-forget startup indexing — must run after server.connect
+if (memoryIndexer) {
+    void memoryIndexer.startupIndex();
+}

package/dist/{tools → mcp}/agent.js

@@ -8,7 +8,7 @@ import * as fs from 'fs';
 import * as path from 'path';
 import { z } from 'zod';
 import { getWorkspaceRoot, mcpJson, mcpText } from '../utils/utils.js';
-import { commandExists, sanitizeOutput, validatePath } from './security.js';
+import { commandExists, sanitizeOutput, validatePath } from '../core/security/index.js';
 // Configurable timeout via environment variable (default: 5 minutes)
 const AGENT_TIMEOUT = parseInt(process.env.KIT_AGENT_TIMEOUT || '300000', 10);
 const jobRegistry = new Map();

package/dist/{tools → mcp}/core.d.ts

@@ -2,7 +2,7 @@
  * Core Tools - Extension info and handoff persistence
  */
 import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
-import { DEFAULT_EXTENSIONS } from './config.js';
+import { DEFAULT_EXTENSIONS } from '../core/config/index.js';
 /**
  * Register core tools with MCP server
  */

package/dist/{tools → mcp}/core.js

@@ -5,8 +5,8 @@ import * as fs from 'fs';
 import * as path from 'path';
 import { z } from 'zod';
 import { getWorkspaceRoot, mcpText } from '../utils/utils.js';
-import { DEFAULT_EXTENSIONS } from './config.js';
-import { HANDOFF_TYPES, resolveHandoffPath } from './handoffs.js';
+import { DEFAULT_EXTENSIONS } from '../core/config/index.js';
+import { HANDOFF_TYPES, resolveHandoffPath } from '../core/handoffs/index.js';
 /**
  * Register core tools with MCP server
  */

package/dist/{tools → mcp}/integration.js

@@ -4,10 +4,10 @@
  */
 import { writeFileSync } from 'fs';
 import { z } from 'zod';
-import { getCredential } from '../utils/credentials.js';
+import { getCredential } from '../services/integration/credentials.js';
 import { mcpText } from '../utils/utils.js';
-import { adfToMarkdown } from '../utils/parser.js';
-import { sanitize } from './security.js';
+import { adfToMarkdown } from '../services/integration/parser/adf.js';
+import { sanitize } from '../core/security/index.js';
 // Zod schema for Bitbucket PR REST API response
 const BitbucketPrSchema = z.object({
     id: z.number(),

package/dist/mcp/memory.d.ts

@@ -0,0 +1,15 @@
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { MemoryIndexer } from '../services/memory/indexer.js';
+import { MemoryStore } from '../services/memory/store.js';
+import type { MemoryConfig } from '../services/memory/types.js';
+import { type ProjectSettings } from '../core/config/index.js';
+/**
+ * Initializes the memory subsystem and registers all memory tools.
+ * Returns the MemoryIndexer so the caller can fire startupIndex() after server.connect().
+ * Returns null when memory is disabled (settings.memory.enabled !== true).
+ */
+export declare function registerMemoryTools(server: McpServer, settings: ProjectSettings, workspaceRoot: string, overrides?: {
+    indexer?: MemoryIndexer;
+    store?: MemoryStore;
+    config?: MemoryConfig;
+}): MemoryIndexer | null;

package/dist/{tools → mcp}/memory.js

@@ -1,17 +1,16 @@
 import * as path from 'path';
 import { z } from 'zod';
-import { Embedder } from '../memory/embedder.js';
-import { MemoryIndexer } from '../memory/indexer.js';
-import { MemoryStore } from '../memory/store.js';
-import { initializeConversationDigestModel } from '../memory/digest/processor.js';
-import { DEFAULT_DIGEST_MODEL_ID } from '../memory/digest/constants.js';
+import { Embedder } from '../services/memory/embedder.js';
+import { MemoryIndexer } from '../services/memory/indexer.js';
+import { MemoryStore } from '../services/memory/store.js';
+import { initializeConversationDigestModel } from '../services/digest/processor.js';
+import { DEFAULT_DIGEST_MODEL_ID } from '../services/digest/constants.js';
 import { mcpJson, mcpText } from '../utils/utils.js';
-import { loadProjectSettings, resolveConversationDigestConfig, resolveMemoryConfig, } from './config.js';
+import { loadProjectSettings, resolveConversationDigestConfig, resolveMemoryConfig, } from '../core/config/index.js';
 /**
  * Registers tool handlers onto an already-constructed indexer/store pair.
- * Exported for unit testing with mocks.
  */
-export function registerMemoryToolHandlers(server, indexer, store, config, workspaceRoot = process.cwd()) {
+function registerMemoryToolHandlers(server, indexer, store, config, workspaceRoot = process.cwd()) {
     server.tool('kit_memory_search', 'Search persistent memory for context relevant to the current question. Returns semantically relevant results using hybrid dense+BM25+RRF search.', {
         query: z.string().min(1).describe('Search query'),
         top_k: z.number().int().positive().optional().describe('Number of results to return'),
@@ -89,13 +88,16 @@ export function registerMemoryToolHandlers(server, indexer, store, config, works
  * Returns the MemoryIndexer so the caller can fire startupIndex() after server.connect().
  * Returns null when memory is disabled (settings.memory.enabled !== true).
  */
-export function registerMemoryTools(server, settings, workspaceRoot) {
+export function registerMemoryTools(server, settings, workspaceRoot, overrides) {
     if (settings.memory?.enabled !== true)
         return null;
-    const config = resolveMemoryConfig(settings, workspaceRoot);
-    const store = new MemoryStore(path.join(config.wikiDir, 'index.db'), config);
-    const embedder = new Embedder(config.embeddingModel);
-    const indexer = new MemoryIndexer(store, embedder, config);
+    const config = overrides?.config ?? resolveMemoryConfig(settings, workspaceRoot);
+    const store = overrides?.store ?? new MemoryStore(path.join(config.wikiDir, 'index.db'), config);
+    const indexer = overrides?.indexer ??
+        (() => {
+            const embedder = new Embedder(config.embeddingModel);
+            return new MemoryIndexer(store, embedder, config);
+        })();
     registerMemoryToolHandlers(server, indexer, store, config, workspaceRoot);
     return indexer;
 }