@vpxa/aikit 0.1.1

package/packages/server/dist/prompts.js

@@ -0,0 +1,13 @@
+import{completeCheckpointNames as e,completeCuratedPaths as t,completeFilePaths as n,completeStashKeys as r,completeSymbolNames as i,completeWorksetNames as a}from"./completions.js";import{completable as o}from"@modelcontextprotocol/sdk/server/completable.js";import{z as s}from"zod";function c(c,l){if(c.registerPrompt(`ready`,{title:`KB Ready`,description:`Knowledge base is ready — quick-start guide for search, onboard, and workflows`},async()=>({messages:[{role:`user`,content:{type:`text`,text:[`Knowledge base is ready. Quick start:`,``,'• **New project?** → `onboard({ path: "." })` for full codebase analysis','• **Returning?** → `status({})` then `search({ query: "SESSION CHECKPOINT", origin: "curated" })`','• **Exploring?** → `guide({ goal: "your task" })` for workflow recommendations','• **Quick lookup?** → `search({ query: "your question" })`'].join(`
+`)}}]})),c.registerPrompt(`onboard`,{title:`Onboard Codebase`,description:`Analyze the codebase for first-time onboarding — runs all analyzers and produces a knowledge summary`,argsSchema:{path:s.string().optional().describe(`Path to analyze (default: workspace root)`)}},async({path:e})=>({messages:[{role:`user`,content:{type:`text`,text:[`Run the full onboarding workflow for "${e??`.`}"`,``,`1. \`onboard({ path: "${e??`.`}" })\` — full codebase analysis`,`2. \`produce_knowledge({ path: "${e??`.`}" })\` — generate synthesis`,"3. `remember` key findings as curated entries","4. `status` to verify index health"].join(`
+`)}}]})),c.registerPrompt(`sessionStart`,{title:`Start KB Session`,description:`Initialize a KB session — check status, list knowledge, and resume from last checkpoint`},async()=>({messages:[{role:`user`,content:{type:`text`,text:[`Run the session start protocol:`,``,"1. `status({})` — check KB health and onboard state","2. `list()` — see stored knowledge entries",'3. `search({ query: "SESSION CHECKPOINT", origin: "curated" })` — resume prior work'].join(`
+`)}}]})),c.registerPrompt(`sessionEnd`,{title:`End KB Session`,description:`Persist decisions and create a session checkpoint before ending`,argsSchema:{summary:s.string().describe(`Brief summary of decisions made, blockers encountered, and next steps`)}},async({summary:e})=>({messages:[{role:`user`,content:{type:`text`,text:[`Run the session end protocol:`,``,'1. `remember({ title: "Session checkpoint: '+e.slice(0,60)+`...", content: "`+e.replace(/"/g,`\\"`)+'", category: "conventions" })` — persist findings',"2. `reindex({})` — refresh search index if files changed",`3. Confirm session data saved`].join(`
+`)}}]})),c.registerPrompt(`search`,{title:`Search Knowledge Base`,description:`Search the knowledge base with hybrid semantic + keyword search`,argsSchema:{query:s.string().describe(`Search query`)}},async({query:e})=>({messages:[{role:`user`,content:{type:`text`,text:`Search the knowledge base for: "${e}"\n\nUse \`search({ query: "${e.replace(/"/g,`\\"`)}" })\` to find relevant code, documentation, and curated knowledge.`}}]})),c.registerPrompt(`remember`,{title:`Remember Knowledge`,description:`Store a decision, convention, or finding as curated knowledge`,argsSchema:{title:s.string().describe(`Title of the knowledge entry`),content:s.string().describe(`Content to remember`),category:s.enum([`conventions`,`decisions`,`patterns`,`blockers`,`tasks`]).optional().describe(`Category (default: conventions)`)}},async({title:e,content:t,category:n})=>({messages:[{role:`user`,content:{type:`text`,text:`Store this knowledge:\n\n\`remember({ title: "${e.replace(/"/g,`\\"`)}", content: "${t.replace(/"/g,`\\"`).slice(0,200)}...", category: "${n??`conventions`}" })\``}}]})),c.registerPrompt(`planTask`,{title:`Plan a Task`,description:`Generate a reading plan and scope map for a development task`,argsSchema:{task:s.string().describe(`Description of the task to plan`)}},async({task:e})=>({messages:[{role:`user`,content:{type:`text`,text:[`Plan implementation for: "${e}"`,``,'1. `search({ query: "'+e.replace(/"/g,`\\"`)+'" })` — find related code and prior decisions','2. `scope_map({ task: "'+e.replace(/"/g,`\\"`)+'" })` — generate a reading plan',"3. For each recommended file, use `file_summary` then `compact` for detail","4. `blast_radius` on planned changes to assess impact"].join(`
+`)}}]})),c.registerPrompt(`investigate`,{title:`Investigate Bug`,description:`Bug investigation workflow — parse error, find symbols, trace data flow, assess impact`,argsSchema:{error:s.string().describe(`Error message, stack trace, or bug description`)}},async({error:e})=>({messages:[{role:`user`,content:{type:`text`,text:[`Investigate this error:`,"```",e,"```",``,`Follow the bug investigation workflow:`,"1. `parse_output({ text: <error> })` — extract structured data from the error","2. `symbol({ name: <relevant symbol> })` — find definition and references",'3. `trace({ symbol: <symbol>, direction: "backward" })` — trace data flow to find root cause',"4. `blast_radius({ changed_files: [<affected files>] })` — assess fix impact"].join(`
+`)}}]})),!l)return;let{curated:u,store:d,graphStore:f}=l;c.registerPrompt(`read_knowledge`,{title:`Read Knowledge`,description:`Read a curated knowledge entry by path — with autocomplete`,argsSchema:{path:o(s.string().describe(`Path of the curated entry (e.g. conventions/naming.md)`),async e=>t(u,e))}},async({path:e})=>({messages:[{role:`user`,content:{type:`text`,text:`Read the curated knowledge entry: \`read({ path: "${e}" })\``}}]})),c.registerPrompt(`update_knowledge`,{title:`Update Knowledge`,description:`Update an existing curated knowledge entry — with autocomplete`,argsSchema:{path:o(s.string().describe(`Path of the curated entry to update`),async e=>t(u,e)),content:s.string().describe(`New content for the entry`)}},async({path:e,content:t})=>({messages:[{role:`user`,content:{type:`text`,text:`Update the curated entry at "${e}":\n\n\`update({ path: "${e}", content: "${t.replace(/"/g,`\\"`).slice(0,200)}..." })\``}}]})),c.registerPrompt(`forget_knowledge`,{title:`Forget Knowledge`,description:`Delete a curated knowledge entry — with autocomplete`,argsSchema:{path:o(s.string().describe(`Path of the curated entry to delete`),async e=>t(u,e))}},async({path:e})=>({messages:[{role:`user`,content:{type:`text`,text:`Delete the curated entry: \`forget({ path: "${e}" })\``}}]})),c.registerPrompt(`lookup_file`,{title:`Lookup File`,description:`Look up an indexed file — with autocomplete`,argsSchema:{path:o(s.string().describe(`Source file path`),async e=>n(d,e))}},async({path:e})=>({messages:[{role:`user`,content:{type:`text`,text:`Look up the indexed file: \`file_summary({ path: "${e}" })\``}}]})),c.registerPrompt(`find_symbol`,{title:`Find Symbol`,description:`Find a symbol definition and references — with autocomplete`,argsSchema:{name:o(s.string().describe(`Symbol name`),async e=>i(f,e))}},async({name:e})=>({messages:[{role:`user`,content:{type:`text`,text:`Find symbol: \`symbol({ name: "${e}" })\``}}]})),c.registerPrompt(`get_stash`,{title:`Get Stash`,description:`Retrieve a stashed value by key — with autocomplete`,argsSchema:{key:o(s.string().describe(`Stash key`),e=>r(e))}},async({key:e})=>({messages:[{role:`user`,content:{type:`text`,text:`Get stashed value: \`stash({ action: "get", key: "${e}" })\``}}]})),c.registerPrompt(`get_workset`,{title:`Get Workset`,description:`Load a saved workset by name — with autocomplete`,argsSchema:{name:o(s.string().describe(`Workset name`),e=>a(e))}},async({name:e})=>({messages:[{role:`user`,content:{type:`text`,text:`Load workset: \`workset({ action: "load", name: "${e}" })\``}}]})),c.registerPrompt(`load_checkpoint`,{title:`Load Checkpoint`,description:`Restore a saved checkpoint by name — with autocomplete`,argsSchema:{name:o(s.string().describe(`Checkpoint name`),t=>e(t))}},async({name:e})=>({messages:[{role:`user`,content:{type:`text`,text:`Load checkpoint: \`checkpoint({ action: "load", name: "${e}" })\``}}]})),c.registerPrompt(`audit_workflow`,{title:`Audit Workflow`,description:`Run a full project quality audit: check → test_run → audit → produce recommendations`},async()=>({messages:[{role:`user`,content:{type:`text`,text:[`## Audit Workflow`,``,`Execute the following tools in sequence:`,``,"1. `check({})` — Run typecheck + lint","2. `test_run({})` — Run all tests","3. `audit({})` — Full project audit","4. Review findings and `remember` important patterns",``,`Report: pass/fail status, error counts, audit recommendations, and suggested next steps.`].join(`
+`)}}]})),c.registerPrompt(`refactor_workflow`,{title:`Refactor Workflow`,description:`Safe refactoring workflow: analyze → blast_radius → implement → verify`},async()=>({messages:[{role:`user`,content:{type:`text`,text:[`## Refactor Workflow`,``,`For the target files/symbols:`,``,"1. `analyze_structure({ path })` — Understand current structure","2. `analyze_dependencies({ path })` — Map dependency graph","3. `blast_radius({ changed_files })` — Assess impact scope",`4. Implement changes carefully`,"5. `check({})` + `test_run({})` — Verify nothing broke","6. `reindex({})` — Update KB with changes",``,`What would you like to refactor?`].join(`
+`)}}]})),c.registerPrompt(`forge_workflow`,{title:`FORGE Workflow`,description:`Full FORGE protocol: classify → ground → implement with quality gates`},async()=>({messages:[{role:`user`,content:{type:`text`,text:[`## FORGE Quality Gate Workflow`,``,"1. `forge_classify({ task })` — Determine complexity tier (floor/standard/critical)","2. `forge_ground({ task, files, root_path })` — Full grounding analysis",`3. Review typed unknowns and constraints`,`4. Implement with evidence tracking`,'5. `evidence_map({ action: "validate" })` — Verify all claims',"6. `check({})` + `test_run({})` — Final verification",``,`Describe the task to classify:`].join(`
+`)}}]})),c.registerPrompt(`investigate_workflow`,{title:`Investigate Issue`,description:`Bug investigation workflow: trace → analyze → find root cause → fix`},async()=>({messages:[{role:`user`,content:{type:`text`,text:[`## Bug Investigation Workflow`,``,"1. `parse_output({ output })` — Parse error messages if available","2. `symbol({ name })` — Find the failing symbol",'3. `trace({ symbol, direction: "backward" })` — Trace data flow to root cause',"4. `blast_radius({ changed_files })` — Assess impact of fix",`5. Implement fix`,"6. `check({})` + `test_run({})` — Verify fix works",``,`Describe the issue:`].join(`
+`)}}]})),c.registerPrompt(`context_workflow`,{title:`Build Context`,description:`Progressive context building: file_summary → compact → digest for deep understanding`},async()=>({messages:[{role:`user`,content:{type:`text`,text:[`## Context Building Workflow`,``,`Build understanding progressively:`,``,"1. `file_summary({ path })` — Structure overview (~50 tokens)","2. `compact({ path, query })` — Extract relevant sections (5-20x reduction)","3. `digest({ sources })` — Multi-file compressed summary","4. `stratum_card({ files })` — Reusable reference card",``,`For project-wide context:`,"- `scope_map({ task })` — Reading plan for a specific task",'- `workset({ action: "save" })` — Save working set for later',``,`What do you need to understand?`].join(`
+`)}}]})),c.registerPrompt(`migrate_workflow`,{title:`Migration Workflow`,description:`Code migration workflow: analyze → plan → migrate → verify → document`},async()=>({messages:[{role:`user`,content:{type:`text`,text:[`## Migration Workflow`,``,"1. `analyze_structure({ path })` — Current codebase state","2. `analyze_patterns({ path })` — Identify patterns to migrate","3. `dead_symbols({ path })` — Find unused code to remove",'4. `scope_map({ task: "migration" })` — Plan migration scope',"5. Use `codemod` for bulk transformations","6. `check({})` + `test_run({})` — Verify after each step","7. `changelog({ from })` — Document changes",``,`What are you migrating from/to?`].join(`
+`)}}]}))}export{c as registerPrompts};

package/packages/server/dist/replay-interceptor.d.ts

@@ -0,0 +1,23 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+
+//#region packages/server/src/replay-interceptor.d.ts
+interface ToolStats {
+  callCount: number;
+  totalDurationMs: number;
+  totalInputChars: number;
+  totalOutputChars: number;
+  errorCount: number;
+}
+/** Get per-tool invocation statistics for the current session. */
+declare function getToolTelemetry(): Array<{
+  tool: string;
+} & ToolStats>;
+/** Get the rolling latency buffer (last N tool call durations in ms). */
+declare function getLatencyBuffer(): readonly number[];
+/**
+ * Wrap the McpServer's registerTool method so that every tool invocation
+ * is logged to the replay JSONL audit trail.
+ */
+declare function installReplayInterceptor(server: McpServer): void;
+//#endregion
+export { getLatencyBuffer, getToolTelemetry, installReplayInterceptor };

package/packages/server/dist/replay-interceptor.js

@@ -0,0 +1 @@
+import{checkAndMaybeGc as e}from"./auto-gc.js";import{replayAppend as t}from"../../tools/dist/index.js";const n=/key|token|secret|auth|password|bearer/i,r=new Set([`eval`,`env`]);function i(e,t){if(r.has(e))return JSON.stringify({_redacted:!0,tool:e});if(e===`http`&&t.headers&&typeof t.headers==`object`){let e={...t},r={};for(let[e,i]of Object.entries(t.headers))r[e]=n.test(e)?`[REDACTED]`:i;return e.headers=r,JSON.stringify(e).slice(0,200)}return JSON.stringify(t).slice(0,200)}const a=new Map,o=[];function s(t,n,r,i,s){let c=a.get(t);c||(c={callCount:0,totalDurationMs:0,totalInputChars:0,totalOutputChars:0,errorCount:0},a.set(t,c)),c.callCount++,c.totalDurationMs+=n,c.totalInputChars+=r,c.totalOutputChars+=i,s&&c.errorCount++,o.push(n),o.length>100&&o.shift(),o.length>=10&&o.length%20==0&&e()}function c(){return Array.from(a.entries()).map(([e,t])=>({tool:e,...t}))}function l(){return o}function u(e){let n=e.registerTool.bind(e);e.registerTool=(e,r,a)=>n(e,r,async(n,r)=>{let o=Date.now(),c=JSON.stringify(n??{}).length;try{let l=await a(n,r),u=Date.now()-o,d=JSON.stringify(l);return t({ts:new Date().toISOString(),source:`mcp`,tool:e,input:i(e,n),durationMs:u,status:`ok`,output:d.slice(0,200)}),s(e,u,c,d.length,!1),l}catch(r){let a=Date.now()-o;throw t({ts:new Date().toISOString(),source:`mcp`,tool:e,input:i(e,n),durationMs:a,status:`error`,output:r instanceof Error?r.message:String(r)}),s(e,a,c,0,!0),r}})}export{l as getLatencyBuffer,c as getToolTelemetry,u as installReplayInterceptor};

package/packages/server/dist/resource-links.d.ts

@@ -0,0 +1,34 @@
+//#region packages/server/src/resource-links.d.ts
+/**
+ * Helpers for emitting resource_link content blocks in tool results.
+ * Centralises the `kb://curated/{path}` URI scheme so all tools stay consistent.
+ */
+/** Shape of a resource_link content block (matches MCP SDK ResourceLink). */
+interface ResourceLinkBlock {
+  type: 'resource_link';
+  uri: string;
+  name: string;
+  description?: string;
+  mimeType?: string;
+}
+/**
+ * Build a resource_link content block for a single curated knowledge entry.
+ * Returns `undefined` for invalid paths so callers can safely filter.
+ */
+declare function curatedResourceLink(path: string | undefined | null, name?: string, description?: string): ResourceLinkBlock | undefined;
+/**
+ * Build resource_link blocks for multiple curated entries.
+ * Deduplicates by URI and filters out invalid paths.
+ */
+declare function curatedResourceLinks(entries: ReadonlyArray<{
+  path: string;
+  title?: string;
+  category?: string;
+}>): ResourceLinkBlock[];
+/**
+ * Extract the curated path from a store `sourcePath`.
+ * Returns `undefined` for non-curated records.
+ */
+declare function extractCuratedPath(sourcePath: string): string | undefined;
+//#endregion
+export { ResourceLinkBlock, curatedResourceLink, curatedResourceLinks, extractCuratedPath };

package/packages/server/dist/resource-links.js

@@ -0,0 +1 @@
+function e(e,t,n){if(!(!e||typeof e!=`string`)&&!(e.includes(`..`)||e.startsWith(`/`)||e.startsWith(`[`)))return{type:`resource_link`,uri:`kb://curated/${e}`,name:t??e,mimeType:`text/markdown`,...n?{description:n}:{}}}function t(t){let n=new Set,r=[];for(let i of t){let t=e(i.path,i.title,i.category?`[${i.category}]`:void 0);t&&!n.has(t.uri)&&(n.add(t.uri),r.push(t))}return r}function n(e){if(e.startsWith(`.ai/curated/`))return e.slice(12)}export{e as curatedResourceLink,t as curatedResourceLinks,n as extractCuratedPath};

package/packages/server/dist/resources/curated-resources.d.ts

@@ -0,0 +1,13 @@
+import { CuratedKnowledgeManager } from "../curated-manager.js";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+
+//#region packages/server/src/resources/curated-resources.d.ts
+/**
+ * Register curated knowledge as browsable MCP resources.
+ *
+ * - `kb://curated` — index listing all curated entries
+ * - `kb://curated/{path}` — individual entry by path (e.g. `decisions/use-lancedb.md`)
+ */
+declare function registerCuratedResources(server: McpServer, curated: CuratedKnowledgeManager): void;
+//#endregion
+export { registerCuratedResources };

package/packages/server/dist/resources/curated-resources.js

@@ -0,0 +1,2 @@
+import{ResourceTemplate as e}from"@modelcontextprotocol/sdk/server/mcp.js";function t(t,n){t.resource(`aikit-curated-index`,`kb://curated`,{description:`Index of all curated knowledge entries`,mimeType:`text/markdown`},async()=>{let e=(await n.list()).map(e=>`- [${e.title}](kb://curated/${e.path}) — ${e.category}`);return{contents:[{uri:`kb://curated`,text:`# Curated Knowledge Index\n\n${e.length>0?e.join(`
+`):`_No curated entries yet._`}`,mimeType:`text/markdown`}]}});let r=new e(`kb://curated/{+path}`,{list:async()=>({resources:(await n.list()).map(e=>({uri:`kb://curated/${e.path}`,name:e.title,description:`[${e.category}] ${e.contentPreview?.slice(0,80)??``}`,mimeType:`text/markdown`}))})});t.resource(`aikit-curated-entry`,r,{description:`A curated knowledge entry`,mimeType:`text/markdown`},async(e,t)=>{let r=t.path;if(!r)throw Error(`Missing path variable in curated resource URI`);let i=await n.read(r);return{contents:[{uri:e.toString(),text:`---\ntitle: ${i.title}\ncategory: ${i.category}\ntags: ${i.tags?.join(`, `)??``}\nversion: ${i.version??1}\n---\n\n${i.content??i.contentPreview??``}`,mimeType:`text/markdown`}]}})}export{t as registerCuratedResources};

package/packages/server/dist/resources/resource-notifier.d.ts

@@ -0,0 +1,45 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+
+//#region packages/server/src/resources/resource-notifier.d.ts
+declare class ResourceNotifier {
+  private readonly mcpServer;
+  constructor(mcpServer: McpServer);
+  /**
+   * Notify that the KB status resource has changed.
+   * Call after: reindex, onboard, remember, forget operations.
+   */
+  notifyStatusChanged(): Promise<void>;
+  /**
+   * Notify that the file tree resource has changed.
+   * Call after: reindex operations.
+   */
+  notifyFileTreeChanged(): Promise<void>;
+  /**
+   * Notify that the curated knowledge index has changed.
+   * Call after: remember, update, forget operations.
+   */
+  notifyCuratedIndexChanged(): Promise<void>;
+  /**
+   * Notify that a specific curated entry has changed.
+   * Call after: update or remember operations for a specific path.
+   */
+  notifyCuratedEntryChanged(path: string): Promise<void>;
+  /**
+   * Notify that the resource list itself changed (new/removed resources).
+   * Call after: curated entries are added or removed.
+   */
+  notifyResourceListChanged(): Promise<void>;
+  /**
+   * Convenience: notify all relevant resources after a reindex.
+   */
+  notifyAfterReindex(): Promise<void>;
+  /**
+   * Convenience: notify all relevant resources after a curated write.
+   */
+  notifyAfterCuratedWrite(path?: string): Promise<void>;
+  private sendUpdate;
+}
+/** No-op resource notifier — all methods silently succeed. */
+declare const noopResourceNotifier: ResourceNotifier;
+//#endregion
+export { ResourceNotifier, noopResourceNotifier };

package/packages/server/dist/resources/resource-notifier.js

@@ -0,0 +1 @@
+import{createLogger as e}from"../../../core/dist/index.js";const t=e(`resource-notifier`);var n=class{constructor(e){this.mcpServer=e}async notifyStatusChanged(){await this.sendUpdate(`kb://status`)}async notifyFileTreeChanged(){await this.sendUpdate(`kb://file-tree`)}async notifyCuratedIndexChanged(){await this.sendUpdate(`kb://curated`)}async notifyCuratedEntryChanged(e){await this.sendUpdate(`kb://curated/${e}`)}async notifyResourceListChanged(){try{await this.mcpServer.server.sendResourceListChanged()}catch(e){t.debug(`sendResourceListChanged failed`,{error:String(e)})}}async notifyAfterReindex(){await Promise.allSettled([this.notifyStatusChanged(),this.notifyFileTreeChanged()])}async notifyAfterCuratedWrite(e){let t=[this.notifyStatusChanged(),this.notifyCuratedIndexChanged()];e&&t.push(this.notifyCuratedEntryChanged(e)),await Promise.allSettled(t)}async sendUpdate(e){try{await this.mcpServer.server.sendResourceUpdated({uri:e})}catch(n){t.debug(`sendResourceUpdated failed`,{uri:e,error:String(n)})}}};const r={notifyStatusChanged:async()=>{},notifyFileTreeChanged:async()=>{},notifyCuratedIndexChanged:async()=>{},notifyCuratedEntryChanged:async()=>{},notifyResourceListChanged:async()=>{},notifyAfterReindex:async()=>{},notifyAfterCuratedWrite:async()=>{}};export{n as ResourceNotifier,r as noopResourceNotifier};

package/packages/server/dist/resources/resources.d.ts

@@ -0,0 +1,8 @@
+import { CuratedKnowledgeManager } from "../curated-manager.js";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { IKnowledgeStore } from "../../../store/dist/index.js";
+
+//#region packages/server/src/resources/resources.d.ts
+declare function registerResources(server: McpServer, store: IKnowledgeStore, curated: CuratedKnowledgeManager): void;
+//#endregion
+export { registerResources };

package/packages/server/dist/resources/resources.js

@@ -0,0 +1,2 @@
+import{registerCuratedResources as e}from"./curated-resources.js";function t(t,n,r){t.resource(`aikit-status`,`kb://status`,{description:`Current knowledge base status and statistics`,mimeType:`text/plain`},async()=>{let e=await n.getStats();return{contents:[{uri:`kb://status`,text:`Knowledge Base: ${e.totalRecords} records from ${e.totalFiles} files. Last indexed: ${e.lastIndexedAt??`Never`}`,mimeType:`text/plain`}]}}),t.resource(`aikit-file-tree`,`kb://file-tree`,{description:`List of all indexed source files`,mimeType:`text/plain`},async()=>({contents:[{uri:`kb://file-tree`,text:(await n.listSourcePaths()).sort().join(`
+`),mimeType:`text/plain`}]})),e(t,r)}export{t as registerResources};

package/packages/server/dist/sampling.d.ts

@@ -0,0 +1,41 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+
+//#region packages/server/src/sampling.d.ts
+interface SamplingRequest {
+  /** The user prompt to send. */
+  prompt: string;
+  /** Optional system prompt. */
+  systemPrompt?: string;
+  /** Optional context to prepend to the prompt. */
+  context?: string;
+  /** Max tokens in the response (default: 4000). */
+  maxTokens?: number;
+  /** Model preferences. */
+  modelPreferences?: {
+    costPriority?: number;
+    speedPriority?: number;
+    intelligencePriority?: number;
+  };
+}
+interface SamplingResponse {
+  /** The model's text response. */
+  text: string;
+  /** Which model was used. */
+  model?: string;
+  /** Stop reason. */
+  stopReason?: string;
+}
+interface ISamplingClient {
+  /** Whether the sampling capability is available. */
+  readonly available: boolean;
+  /** Send a prompt to the host LLM. */
+  createMessage(request: SamplingRequest): Promise<SamplingResponse>;
+}
+/**
+ * Create a sampling client backed by the MCP server.
+ */
+declare function createSamplingClient(mcpServer: McpServer): ISamplingClient;
+/** No-op sampling client — always reports unavailable. */
+declare const noopSamplingClient: ISamplingClient;
+//#endregion
+export { ISamplingClient, SamplingRequest, SamplingResponse, createSamplingClient, noopSamplingClient };

package/packages/server/dist/sampling.js

@@ -0,0 +1,2 @@
+import{createLogger as e}from"../../core/dist/index.js";const t=e(`sampling`);function n(e){if(!e.modelPreferences)return;let t={...e.modelPreferences.costPriority===void 0?{}:{costPriority:e.modelPreferences.costPriority},...e.modelPreferences.speedPriority===void 0?{}:{speedPriority:e.modelPreferences.speedPriority},...e.modelPreferences.intelligencePriority===void 0?{}:{intelligencePriority:e.modelPreferences.intelligencePriority}};return Object.keys(t).length>0?t:void 0}function r(e){if(Array.isArray(e))return e.map(e=>r(e)).filter(e=>e.length>0).join(`
+`);if(e&&typeof e==`object`&&`type`in e){let t=e;if(t.type===`text`&&typeof t.text==`string`)return t.text}return JSON.stringify(e)}function i(e){let i=e.server,a=typeof i?.createMessage==`function`;return{get available(){return a},async createMessage(e){if(!a)throw Error(`Sampling not available: client does not support createMessage`);let o=e.context?`${e.context}\n\n---\n\n${e.prompt}`:e.prompt;try{let t=await i.createMessage({messages:[{role:`user`,content:{type:`text`,text:o}}],systemPrompt:e.systemPrompt,modelPreferences:n(e),maxTokens:e.maxTokens??4e3});return{text:r(t.content),model:t.model,stopReason:t.stopReason}}catch(e){throw t.warn(`Sampling createMessage failed`,{error:String(e)}),e}}}}const a={available:!1,async createMessage(){throw Error(`Sampling not available`)}};export{i as createSamplingClient,a as noopSamplingClient};

package/packages/server/dist/server.d.ts

@@ -0,0 +1,47 @@
+import { CuratedKnowledgeManager } from "./curated-manager.js";
+import { ResourceNotifier } from "./resources/resource-notifier.js";
+import { ISamplingClient } from "./sampling.js";
+import { BridgeComponents } from "./tools/bridge.tools.js";
+import { KBConfig } from "../../core/dist/index.js";
+import { OnnxEmbedder } from "../../embeddings/dist/index.js";
+import { EvolutionCollector, PolicyStore } from "../../enterprise-bridge/dist/index.js";
+import { IncrementalIndexer } from "../../indexer/dist/index.js";
+import { IGraphStore, createStore } from "../../store/dist/index.js";
+import { FileCache } from "../../tools/dist/index.js";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { Elicitor } from "../../elicitation/dist/index.js";
+
+//#region packages/server/src/server.d.ts
+interface KnowledgeBaseComponents {
+  embedder: OnnxEmbedder;
+  store: Awaited<ReturnType<typeof createStore>>;
+  indexer: IncrementalIndexer;
+  curated: CuratedKnowledgeManager;
+  graphStore: IGraphStore;
+  fileCache: FileCache;
+  bridge?: BridgeComponents;
+  policyStore?: PolicyStore;
+  evolutionCollector?: EvolutionCollector;
+  onboardComplete: boolean;
+  onboardTimestamp?: string;
+}
+declare function initializeKnowledgeBase(config: KBConfig): Promise<KnowledgeBaseComponents>;
+declare function createMcpServer(kb: KnowledgeBaseComponents, config: KBConfig): McpServer;
+declare function registerMcpTools(server: McpServer, kb: KnowledgeBaseComponents, config: KBConfig, elicitor?: Elicitor, resourceNotifier?: ResourceNotifier, samplingClient?: ISamplingClient): void;
+/**
+ * Create a fully-initialized server (HTTP mode or when immediate readiness is needed).
+ */
+declare function createServer(config: KBConfig): Promise<{
+  server: McpServer;
+  runInitialIndex: () => Promise<void>;
+  shutdown: () => Promise<void>;
+}>;
+declare const ALL_TOOL_NAMES: readonly ["analyze_dependencies", "analyze_diagram", "analyze_entry_points", "analyze_patterns", "analyze_structure", "analyze_symbols", "audit", "batch", "blast_radius", "brainstorm", "changelog", "check", "checkpoint", "codemod", "compact", "data_transform", "dead_symbols", "delegate", "diff_parse", "digest", "encode", "env", "eval", "evidence_map", "file_summary", "find", "forge_classify", "forge_ground", "forget", "git_context", "graph", "guide", "health", "http", "lane", "list", "lookup", "measure", "onboard", "parse_output", "present", "process", "produce_knowledge", "queue", "read", "regex_test", "reindex", "remember", "rename", "replay", "restore", "schema_validate", "scope_map", "search", "snippet", "stash", "status", "stratum_card", "symbol", "test_run", "time", "trace", "update", "watch", "web_fetch", "web_search", "workset"];
+declare function createLazyServer(config: KBConfig): {
+  server: McpServer; /** Call after MCP roots are resolved (or fallback decided) to start heavy init. */
+  startInit: () => void;
+  ready: Promise<void>;
+  runInitialIndex: () => Promise<void>;
+};
+//#endregion
+export { ALL_TOOL_NAMES, KnowledgeBaseComponents, createLazyServer, createMcpServer, createServer, initializeKnowledgeBase, registerMcpTools };

package/packages/server/dist/server.js

@@ -0,0 +1,3 @@
+import{CuratedKnowledgeManager as e}from"./curated-manager.js";import{createElicitor as t,noopElicitor as n}from"./elicitor.js";import{bridgeMcpLogging as r}from"./mcp-logging.js";import{registerPrompts as i}from"./prompts.js";import{installReplayInterceptor as a}from"./replay-interceptor.js";import{ResourceNotifier as o}from"./resources/resource-notifier.js";import{registerResources as s}from"./resources/resources.js";import{createSamplingClient as c}from"./sampling.js";import{installStructuredContentGuard as l}from"./structured-content-guard.js";import{getToolMeta as u}from"./tool-metadata.js";import{installToolPrefix as d}from"./tool-prefix.js";import{registerAnalyzeDependenciesTool as f,registerAnalyzeDiagramTool as p,registerAnalyzeEntryPointsTool as m,registerAnalyzePatternsTool as h,registerAnalyzeStructureTool as g,registerAnalyzeSymbolsTool as _,registerBlastRadiusTool as v}from"./tools/analyze.tools.js";import{registerAuditTool as y}from"./tools/audit.tool.js";import{registerBrainstormTool as b}from"./tools/brainstorm.tool.js";import{initBridgeComponents as x,registerErPullTool as S,registerErPushTool as C,registerErSyncStatusTool as w}from"./tools/bridge.tools.js";import{registerCompactTool as T,registerDeadSymbolsTool as ee,registerFileSummaryTool as te,registerFindTool as ne,registerScopeMapTool as re,registerSymbolTool as ie,registerTraceTool as ae}from"./tools/context.tools.js";import{registerErEvolveReviewTool as oe}from"./tools/evolution.tools.js";import{registerBatchTool as se,registerCheckTool as E,registerDelegateTool as ce,registerEvalTool as D,registerParseOutputTool as O,registerTestRunTool as k}from"./tools/execution.tools.js";import{registerFlowTools as le}from"./tools/flow.tools.js";import{registerDigestTool as ue,registerEvidenceMapTool as A,registerForgeClassifyTool as j,registerForgeGroundTool as de,registerStratumCardTool as fe}from"./tools/forge.tools.js";import{registerForgetTool as pe}from"./tools/forget.tool.js";import{registerGraphTool as me}from"./tools/graph.tool.js";import{registerGuideTool as M,registerHealthTool as N,registerProcessTool as P,registerWatchTool as F,registerWebFetchTool as I}from"./tools/infra.tools.js";import{registerListTool as he}from"./tools/list.tool.js";import{registerLookupTool as ge}from"./tools/lookup.tool.js";import{registerCodemodTool as L,registerDataTransformTool as R,registerDiffParseTool as z,registerGitContextTool as B,registerRenameTool as V}from"./tools/manipulation.tools.js";import{registerOnboardTool as _e}from"./tools/onboard.tool.js";import{registerCheckpointTool as H,registerLaneTool as U,registerQueueTool as W,registerStashTool as G,registerWorksetTool as K}from"./tools/persistence.tools.js";import{registerErUpdatePolicyTool as ve}from"./tools/policy.tools.js";import{registerPresentTool as q}from"./tools/present/tool.js";import"./tools/present/index.js";import{registerProduceKnowledgeTool as ye}from"./tools/produce.tool.js";import{registerReadTool as be}from"./tools/read.tool.js";import{registerReindexTool as xe}from"./tools/reindex.tool.js";import{registerRememberTool as Se}from"./tools/remember.tool.js";import{registerReplayTool as Ce}from"./tools/replay.tool.js";import{registerRestoreTool as we}from"./tools/restore.tool.js";import{registerSearchTool as Te}from"./tools/search.tool.js";import{getCurrentVersion as Ee}from"./version-check.js";import{registerEarlyStatusTool as De,registerStatusTool as Oe}from"./tools/status.tool.js";import{registerUpdateTool as ke}from"./tools/update.tool.js";import{registerChangelogTool as Ae,registerEncodeTool as je,registerEnvTool as Me,registerHttpTool as Ne,registerMeasureTool as Pe,registerRegexTestTool as Fe,registerSchemaValidateTool as Ie,registerSnippetTool as Le,registerTimeTool as Re,registerWebSearchTool as ze}from"./tools/utility.tools.js";import{existsSync as Be,statSync as Ve}from"node:fs";import{resolve as He}from"node:path";import{AIKIT_PATHS as Ue,createLogger as We,serializeError as J}from"../../core/dist/index.js";import{initializeWasm as Ge}from"../../chunker/dist/index.js";import{OnnxEmbedder as Ke}from"../../embeddings/dist/index.js";import{EvolutionCollector as qe,PolicyStore as Je}from"../../enterprise-bridge/dist/index.js";import{FileHashCache as Ye,IncrementalIndexer as Xe}from"../../indexer/dist/index.js";import{SqliteGraphStore as Ze,createStore as Qe}from"../../store/dist/index.js";import{FileCache as $e}from"../../tools/dist/index.js";import{McpServer as et}from"@modelcontextprotocol/sdk/server/mcp.js";const Y=We(`server`);async function X(t){Y.info(`Initializing knowledge base components`);let[n,r,i,a]=await Promise.all([(async()=>{let e=new Ke({model:t.embedding.model,dimensions:t.embedding.dimensions});return await e.initialize(),Y.info(`Embedder loaded`,{modelId:e.modelId,dimensions:e.dimensions}),e})(),(async()=>{let e=await Qe({backend:t.store.backend,path:t.store.path});return await e.initialize(),Y.info(`Store initialized`),e})(),(async()=>{let e=new Ze({path:t.store.path});return await e.initialize(),Y.info(`Graph store initialized`),e})(),(async()=>{let e=await Ge();return e?Y.info(`WASM tree-sitter enabled for AST analysis`):Y.warn(`WASM tree-sitter not available; analyzers will use regex fallback`),e})()]),o=new Xe(n,r),s=new Ye(t.store.path);s.load(),o.setHashCache(s);let c=t.curated.path,l=new e(c,r,n);o.setGraphStore(i);let u=x(t.er),d=u?new Je(t.curated.path):void 0;d&&Y.info(`Policy store initialized`,{ruleCount:d.getRules().length});let f=u?new qe:void 0,p=He(t.sources[0]?.path??process.cwd(),Ue.aiKb),m=Be(p),h=t.onboardDir?Be(t.onboardDir):!1,g=m||h,_,v=m?p:t.onboardDir;if(g&&v)try{_=Ve(v).mtime.toISOString()}catch{}return Y.info(`Onboard state detected`,{onboardComplete:g,onboardTimestamp:_,aiKbExists:m,onboardDirExists:h}),{embedder:n,store:r,indexer:o,curated:l,graphStore:i,fileCache:new $e,bridge:u,policyStore:d,evolutionCollector:f,onboardComplete:g,onboardTimestamp:_}}function tt(e,n){let a=new et({name:n.serverName??`knowledge-base`,version:Ee()},{capabilities:{logging:{}}});return r(a),d(a,n.toolPrefix??``),Z(a,e,n,t(a),new o(a),c(a)),i(a,{curated:e.curated,store:e.store,graphStore:e.graphStore}),a}function Z(e,t,n,r,i,o){a(e),l(e),Te(e,t.embedder,t.store,t.graphStore,t.bridge,t.evolutionCollector,o),ge(e,t.store),Oe(e,t.store,t.graphStore,t.curated,{onboardComplete:t.onboardComplete,onboardTimestamp:t.onboardTimestamp},n),xe(e,t.indexer,n,t.curated,t.store,i),Se(e,t.curated,t.policyStore,t.evolutionCollector,i),ke(e,t.curated,i),pe(e,t.curated,i),be(e,t.curated),he(e,t.curated),g(e,t.store,t.embedder),f(e,t.store,t.embedder),_(e,t.store,t.embedder),h(e,t.store,t.embedder),m(e,t.store,t.embedder),p(e,t.store,t.embedder),v(e,t.store,t.embedder,t.graphStore),ye(e,n),_e(e,t.store,t.embedder,n),me(e,t.graphStore),y(e,t.store,t.embedder);let c=n.sources[0]?.path??process.cwd();T(e,t.embedder,t.fileCache,c),re(e,t.embedder,t.store),ne(e,t.embedder,t.store),O(e),K(e),E(e),se(e,t.embedder,t.store),ie(e,t.embedder,t.store,t.graphStore),D(e),k(e),G(e),B(e),z(e),V(e),L(e),we(e),te(e,t.fileCache,c),H(e),R(e),ae(e,t.embedder,t.store),P(e),F(e),ee(e,t.embedder,t.store),ce(e,o),N(e),U(e),W(e),I(e),M(e),A(e),ue(e,t.embedder),j(e),fe(e,t.embedder,t.fileCache),de(e,t.embedder,t.store),q(e,r),r&&b(e,r),ze(e),Ne(e),Fe(e),je(e),Pe(e),Ae(e),Ie(e),Le(e),Me(e),Re(e),le(e,n),t.bridge&&(C(e,t.bridge,t.evolutionCollector),S(e,t.bridge),w(e,t.bridge)),t.policyStore&&ve(e,t.policyStore),t.evolutionCollector&&oe(e,t.evolutionCollector),s(e,t.store,t.curated),Ce(e)}async function nt(e){let t=await X(e),n=tt(t,e);Y.info(`MCP server configured`,{toolCount:$.length,resourceCount:2});let r=async()=>{try{let n=e.sources.map(e=>e.path).join(`, `);Y.info(`Running initial index`,{sourcePaths:n});let r=await t.indexer.index(e,e=>{e.phase===`crawling`||e.phase===`done`||(e.phase===`chunking`&&e.currentFile&&Y.debug(`Indexing file`,{current:e.filesProcessed+1,total:e.filesTotal,file:e.currentFile}),e.phase===`cleanup`&&Y.debug(`Index cleanup`,{staleEntries:e.filesTotal-e.filesProcessed}))});Y.info(`Initial index complete`,{filesProcessed:r.filesProcessed,filesSkipped:r.filesSkipped,chunksCreated:r.chunksCreated,durationMs:r.durationMs});try{await t.store.createFtsIndex()}catch(e){Y.warn(`FTS index creation failed`,J(e))}try{let e=await t.curated.reindexAll();Y.info(`Curated re-index complete`,{indexed:e.indexed})}catch(e){Y.error(`Curated re-index failed`,J(e))}}catch(e){Y.error(`Initial index failed; will retry on aikit_reindex`,J(e))}},i=async()=>{Y.info(`Shutting down`),await t.embedder.shutdown().catch(()=>{}),await t.graphStore.close().catch(()=>{}),await t.store.close(),process.exit(0)};process.on(`SIGINT`,i),process.on(`SIGTERM`,i);let a=process.ppid,o=setInterval(()=>{try{process.kill(a,0)}catch{Y.info(`Parent process died; shutting down`,{parentPid:a}),clearInterval(o),i()}},5e3);return o.unref(),{server:n,runInitialIndex:r,shutdown:i}}const rt=new Set(`batch.brainstorm.changelog.check.checkpoint.codemod.compact.data_transform.delegate.diff_parse.digest.encode.env.eval.evidence_map.file_summary.forge_classify.git_context.graph.guide.health.http.lane.measure.onboard.parse_output.present.process.produce_knowledge.queue.read.regex_test.reindex.remember.rename.replay.restore.schema_validate.scope_map.snippet.stash.status.stratum_card.test_run.time.update.forget.list.watch.web_fetch.web_search.workset`.split(`.`)),it=5e3,Q=new Set(`brainstorm.changelog.check.checkpoint.codemod.data_transform.delegate.diff_parse.encode.env.eval.evidence_map.forge_classify.git_context.guide.present.health.http.lane.measure.parse_output.process.produce_knowledge.queue.regex_test.rename.replay.restore.schema_validate.snippet.stash.status.test_run.time.watch.web_fetch.web_search.workset`.split(`.`));function at(e){E(e),D(e),k(e),O(e),ce(e),B(e),z(e),V(e),L(e),R(e),K(e),G(e),H(e),we(e),U(e),W(e),N(e),P(e),F(e),I(e),M(e),A(e),j(e),q(e),b(e,n),ye(e),Ce(e),De(e),ze(e),Ne(e),Fe(e),je(e),Pe(e),Ae(e),Ie(e),Le(e),Me(e),Re(e)}const $=`analyze_dependencies.analyze_diagram.analyze_entry_points.analyze_patterns.analyze_structure.analyze_symbols.audit.batch.blast_radius.brainstorm.changelog.check.checkpoint.codemod.compact.data_transform.dead_symbols.delegate.diff_parse.digest.encode.env.eval.evidence_map.file_summary.find.forge_classify.forge_ground.forget.git_context.graph.guide.health.http.lane.list.lookup.measure.onboard.parse_output.present.process.produce_knowledge.queue.read.regex_test.reindex.remember.rename.replay.restore.schema_validate.scope_map.search.snippet.stash.status.stratum_card.symbol.test_run.time.trace.update.watch.web_fetch.web_search.workset`.split(`.`);function ot(e){let n=new et({name:e.serverName??`knowledge-base`,version:Ee()},{capabilities:{logging:{}}}),a=`initializing`,s=``,l=!1,f=()=>a===`failed`?[`❌ KB initialization failed — this tool is unavailable.`,``,s?`Error: ${s}`:``,``,`**35 tools are still available** and fully functional:`,`check, eval, test_run, git_context, health, measure, web_fetch, web_search,`,`regex_test, encode, stash, checkpoint, lane, process, time, env, and more.`,``,`Try restarting the MCP server to retry initialization.`].filter(Boolean).join(`
+`):[`KB is still initializing (loading embeddings model & store).`,``,`**35 tools are already available** while initialization completes — including:`,`check, eval, test_run, git_context, health, measure, web_fetch, web_search,`,`regex_test, encode, stash, checkpoint, lane, process, time, env, and more.`,``,`This tool requires the knowledge base index. Please retry in a few seconds,`,`or use one of the available tools above in the meantime.`].join(`
+`);r(n),d(n,e.toolPrefix??``);let p=n.sendToolListChanged.bind(n);n.sendToolListChanged=()=>{};let m=[];for(let e of $){let t=u(e),r=n.registerTool(e,{title:t.title,description:`${t.title} — initializing, available shortly`,inputSchema:{},annotations:t.annotations},async()=>({content:[{type:`text`,text:f()}]}));Q.has(e)?r.remove():m.push(r)}at(n),n.sendToolListChanged=p;let h=n.resource(`aikit-status`,`kb://status`,{description:`Knowledge base status (initializing...)`,mimeType:`text/plain`},async()=>({contents:[{uri:`kb://status`,text:`KB is initializing...`,mimeType:`text/plain`}]})),g=n.prompt(`_init`,`KB is initializing prompts...`,async()=>({messages:[{role:`user`,content:{type:`text`,text:f()}}]})),_,v=new Promise(e=>{_=e}),y,b=new Promise(e=>{y=e}),x=()=>y?.(),S=(async()=>{await b;let r;try{r=await X(e)}catch(e){a=`failed`,s=e instanceof Error?e.message:String(e),Y.error(`KB initialization failed — server continuing with zero-dep tools only`,{error:s});return}let u=n.sendToolListChanged.bind(n);n.sendToolListChanged=()=>{};let d=n.sendPromptListChanged.bind(n);n.sendPromptListChanged=()=>{};let f=n.sendResourceListChanged.bind(n);n.sendResourceListChanged=()=>{};for(let e of m)e.remove();h.remove(),g.remove();let p=n._registeredTools??{};for(let e of Q)p[e]?.remove();let v=new o(n),y=c(n);Z(n,r,e,t(n),v,y),i(n),n.sendToolListChanged=u,n.sendPromptListChanged=d,n.sendResourceListChanged=f,Promise.resolve(n.sendToolListChanged()).catch(()=>{}),Promise.resolve(n.sendPromptListChanged()).catch(()=>{}),Promise.resolve(n.sendResourceListChanged()).catch(()=>{});let x=n._registeredTools??{};for(let[e,t]of Object.entries(x)){if(rt.has(e))continue;let n=t.handler;t.handler=async(...t)=>{if(!r.indexer.isIndexing)return n(...t);let i=l?`re-indexing`:`running initial index`,a=new Promise(t=>setTimeout(()=>t({content:[{type:`text`,text:`⏳ KB is ${i}. The tool "${e}" timed out waiting for index data (${it/1e3}s).\n\nThe existing index may be temporarily locked. Please retry shortly — indexing will complete automatically.`}]}),it));return Promise.race([n(...t),a])}}let S=Object.keys(x).length;S!==$.length&&Y.warn(`ALL_TOOL_NAMES count mismatch`,{expectedToolCount:$.length,registeredToolCount:S}),Y.info(`MCP server configured`,{toolCount:$.length,resourceCount:4}),_?.(r)})(),C=async()=>{let t=await v;try{let n=e.sources.map(e=>e.path).join(`, `);Y.info(`Running initial index`,{sourcePaths:n});let r=await t.indexer.index(e,e=>{e.phase===`crawling`||e.phase===`done`||(e.phase===`chunking`&&e.currentFile&&Y.debug(`Indexing file`,{current:e.filesProcessed+1,total:e.filesTotal,file:e.currentFile}),e.phase===`cleanup`&&Y.debug(`Index cleanup`,{staleEntries:e.filesTotal-e.filesProcessed}))});l=!0,Y.info(`Initial index complete`,{filesProcessed:r.filesProcessed,filesSkipped:r.filesSkipped,chunksCreated:r.chunksCreated,durationMs:r.durationMs});try{await t.store.createFtsIndex()}catch(e){Y.warn(`FTS index creation failed`,J(e))}try{let e=await t.curated.reindexAll();Y.info(`Curated re-index complete`,{indexed:e.indexed})}catch(e){Y.error(`Curated re-index failed`,J(e))}}catch(e){Y.error(`Initial index failed; will retry on aikit_reindex`,J(e))}},w=process.ppid,T=setInterval(()=>{try{process.kill(w,0)}catch{Y.info(`Parent process died; shutting down`,{parentPid:w}),clearInterval(T),v.then(async e=>{await e.embedder.shutdown().catch(()=>{}),await e.graphStore.close().catch(()=>{}),await e.store.close().catch(()=>{})}).catch(()=>{}).finally(()=>process.exit(0))}},5e3);return T.unref(),{server:n,startInit:x,ready:S,runInitialIndex:C}}export{$ as ALL_TOOL_NAMES,ot as createLazyServer,tt as createMcpServer,nt as createServer,X as initializeKnowledgeBase,Z as registerMcpTools};

package/packages/server/dist/structured-content-guard.d.ts

@@ -0,0 +1,26 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+
+//#region packages/server/src/structured-content-guard.d.ts
+/**
+ * Build a minimal `structuredContent` placeholder when the handler didn't
+ * provide one.  The SDK only checks that the value is a non-null object and
+ * validates it against the tool's Zod-driven JSON-Schema.
+ *
+ * We extract the error message from the text content blocks and store it
+ * under an `_error` key.  Because `outputSchema` Zod schemas are compiled
+ * with `strict: false` by default via `zodToJsonSchema`, additional
+ * properties are tolerated and the SDK validator will not reject the
+ * placeholder.
+ *
+ * UPDATE: The SDK uses passthrough-style validation which DOES reject
+ * unknown keys on strict schemas.  Instead, we build a "zero-value"
+ * object from the registered outputSchema (if available) so it always
+ * passes validation.
+ */
+/**
+ * Wrap `server.registerTool` so that every handler result is guaranteed
+ * to include `structuredContent` when the tool declares `outputSchema`.
+ */
+declare function installStructuredContentGuard(server: McpServer): void;
+//#endregion
+export { installStructuredContentGuard };

package/packages/server/dist/structured-content-guard.js

@@ -0,0 +1 @@
+function e(e){let n=e.registerTool.bind(e);e.registerTool=(e,r,i)=>r?.outputSchema==null?n(e,r,i):n(e,r,async(e,n)=>{let a=await i(e,n);return a.structuredContent??=t(r.outputSchema),a})}function t(e){if(!e)return{};if(e.anyOf){let n=e.anyOf.find(e=>e.type!==`null`);return n?t(n):null}switch(e.type){case`object`:{let n={};if(e.properties)for(let[r,i]of Object.entries(e.properties))n[r]=t(i);return n}case`array`:return[];case`string`:return``;case`number`:case`integer`:return 0;case`boolean`:return!1;default:return{}}}export{e as installStructuredContentGuard};

package/packages/server/dist/task-manager.d.ts

@@ -0,0 +1,40 @@
+//#region packages/server/src/task-manager.d.ts
+/**
+ * Lightweight progress reporting for long-running tool operations.
+ *
+ * Wraps MCP progress notifications (`notifications/progress`) with a simple
+ * TaskHandle API. Degrades gracefully when the client doesn't support progress.
+ *
+ * @see E-027 — Tasks + Progress enhancement
+ */
+/**
+ * Minimal type for the tool handler `extra` parameter we need.
+ * Avoids importing full SDK types for testing flexibility.
+ */
+interface ToolHandlerExtra {
+  sendNotification?: (notification: {
+    method: string;
+    params?: Record<string, unknown>;
+  }) => Promise<void>;
+}
+interface TaskHandle {
+  /** Report progress. `current` is 0-based step number. */
+  progress(current: number, message?: string): void;
+  /** Mark the task as complete. */
+  complete(message?: string): void;
+  /** Mark the task as failed. */
+  fail(error?: string): void;
+}
+interface ITaskRunner {
+  /** Create a task with a descriptive title and expected total steps. */
+  createTask(title: string, totalSteps: number): TaskHandle;
+}
+/**
+ * Create a task runner that sends MCP progress notifications.
+ * If `extra` is undefined or lacks sendNotification, returns a no-op runner.
+ */
+declare function createTaskRunner(extra?: ToolHandlerExtra): ITaskRunner;
+/** No-op task runner — all methods silently succeed. */
+declare const noopTaskRunner: ITaskRunner;
+//#endregion
+export { ITaskRunner, TaskHandle, ToolHandlerExtra, createTaskRunner, noopTaskRunner };

package/packages/server/dist/task-manager.js

@@ -0,0 +1 @@
+import{createLogger as e}from"../../core/dist/index.js";const t=e(`task-manager`);let n=0;function r(e){let r=typeof e?.sendNotification==`function`;return{createTask(i,a){let o=`aikit-task-${++n}`,s=(n,i,a)=>{r&&e.sendNotification?.({method:`notifications/progress`,params:{progressToken:o,progress:n,total:i,...a?{message:a}:{}}}).catch(e=>{t.debug(`Progress notification failed`,{token:o,error:String(e)})})};return r&&t.debug(`Task created`,{token:o,title:i,totalSteps:a}),{progress(e,t){s(e,a,t??`${i}: step ${e+1}/${a}`)},complete(e){s(a,a,e??`${i}: complete`)},fail(e){s(a,a,e??`${i}: failed`)}}}}}const i={createTask(){return{progress(){},complete(){},fail(){}}}};export{r as createTaskRunner,i as noopTaskRunner};

package/packages/server/dist/tool-metadata.d.ts

@@ -0,0 +1,38 @@
+//#region packages/server/src/tool-metadata.d.ts
+/**
+ * Centralized tool metadata registry for all KB MCP tools.
+ *
+ * Provides `title` and `annotations` (readOnlyHint, destructiveHint) for each tool.
+ * Used by both placeholder registration (lazy startup) and real tool registration
+ * to keep metadata in sync.
+ *
+ * @see E-020 — Tool Annotations enhancement
+ */
+interface ToolMeta {
+  /** Human-readable display name shown in the VS Code tools picker. */
+  title: string;
+  /** MCP tool annotations. */
+  annotations: {
+    readOnlyHint?: boolean;
+    destructiveHint?: boolean;
+    idempotentHint?: boolean;
+    openWorldHint?: boolean;
+  };
+}
+/**
+ * Master metadata map keyed by tool name.
+ *
+ * Classification rules:
+ * - `readOnlyHint: true`  — tool NEVER modifies state (files, store, index, processes)
+ * - `readOnlyHint: false` — tool CAN modify state (even if only via certain actions)
+ * - `destructiveHint: true` — tool deletes or overwrites data
+ * - Mixed-mode tools (read+write via `action` param) → `readOnlyHint: false`
+ */
+declare const TOOL_METADATA: Record<string, ToolMeta>;
+/**
+ * Get tool metadata for a given tool name.
+ * Returns a default (no title, readOnlyHint: false) for unknown tools.
+ */
+declare function getToolMeta(name: string): ToolMeta;
+//#endregion
+export { TOOL_METADATA, ToolMeta, getToolMeta };

package/packages/server/dist/tool-metadata.js

@@ -0,0 +1 @@
+const e={search:{title:`Hybrid Search`,annotations:{readOnlyHint:!0,idempotentHint:!0}},find:{title:`Federated Find`,annotations:{readOnlyHint:!0,idempotentHint:!0}},symbol:{title:`Symbol Resolver`,annotations:{readOnlyHint:!0,idempotentHint:!0}},trace:{title:`Data Flow Tracer`,annotations:{readOnlyHint:!0,idempotentHint:!0}},scope_map:{title:`Task Scope Map`,annotations:{readOnlyHint:!0,idempotentHint:!0}},lookup:{title:`Chunk Lookup`,annotations:{readOnlyHint:!0,idempotentHint:!0}},dead_symbols:{title:`Dead Symbol Finder`,annotations:{readOnlyHint:!0,idempotentHint:!0}},file_summary:{title:`File Summary`,annotations:{readOnlyHint:!0,idempotentHint:!0}},analyze_structure:{title:`Analyze Structure`,annotations:{readOnlyHint:!0,idempotentHint:!0}},analyze_dependencies:{title:`Analyze Dependencies`,annotations:{readOnlyHint:!0,idempotentHint:!0}},analyze_symbols:{title:`Analyze Symbols`,annotations:{readOnlyHint:!0,idempotentHint:!0}},analyze_patterns:{title:`Analyze Patterns`,annotations:{readOnlyHint:!0,idempotentHint:!0}},analyze_entry_points:{title:`Analyze Entry Points`,annotations:{readOnlyHint:!0,idempotentHint:!0}},analyze_diagram:{title:`Analyze Diagram`,annotations:{readOnlyHint:!0,idempotentHint:!0}},blast_radius:{title:`Blast Radius`,annotations:{readOnlyHint:!0,idempotentHint:!0}},brainstorm:{title:`Brainstorm Session`,annotations:{readOnlyHint:!0,idempotentHint:!0}},remember:{title:`Remember Knowledge`,annotations:{readOnlyHint:!1}},read:{title:`Read Knowledge`,annotations:{readOnlyHint:!0,idempotentHint:!0}},update:{title:`Update Knowledge`,annotations:{readOnlyHint:!1}},forget:{title:`Forget Knowledge`,annotations:{readOnlyHint:!1,destructiveHint:!0}},list:{title:`List Knowledge`,annotations:{readOnlyHint:!0,idempotentHint:!0}},produce_knowledge:{title:`Produce Knowledge`,annotations:{readOnlyHint:!0,idempotentHint:!0}},compact:{title:`Semantic Compactor`,annotations:{readOnlyHint:!0,idempotentHint:!0}},digest:{title:`Multi-Source Digest`,annotations:{readOnlyHint:!0,idempotentHint:!0}},stratum_card:{title:`Stratum Card`,annotations:{readOnlyHint:!0,idempotentHint:!0}},forge_ground:{title:`FORGE Ground`,annotations:{readOnlyHint:!0,idempotentHint:!0}},forge_classify:{title:`FORGE Classify`,annotations:{readOnlyHint:!0,idempotentHint:!0}},evidence_map:{title:`Evidence Map`,annotations:{readOnlyHint:!1}},present:{title:`Rich Content Presenter`,annotations:{readOnlyHint:!0,idempotentHint:!0}},check:{title:`Typecheck & Lint`,annotations:{readOnlyHint:!0,openWorldHint:!0}},test_run:{title:`Run Tests`,annotations:{readOnlyHint:!0,openWorldHint:!0}},eval:{title:`Evaluate Code`,annotations:{readOnlyHint:!1,openWorldHint:!0}},batch:{title:`Batch Operations`,annotations:{readOnlyHint:!0,idempotentHint:!0}},audit:{title:`Project Audit`,annotations:{readOnlyHint:!0,idempotentHint:!0}},rename:{title:`Rename Symbol`,annotations:{readOnlyHint:!1,destructiveHint:!0}},restore:{title:`Restore`,annotations:{readOnlyHint:!1}},codemod:{title:`Codemod`,annotations:{readOnlyHint:!1,destructiveHint:!0}},data_transform:{title:`Data Transform`,annotations:{readOnlyHint:!0,idempotentHint:!0}},stash:{title:`Stash Values`,annotations:{readOnlyHint:!1}},checkpoint:{title:`Session Checkpoint`,annotations:{readOnlyHint:!1}},workset:{title:`Workset Manager`,annotations:{readOnlyHint:!1}},lane:{title:`Exploration Lane`,annotations:{readOnlyHint:!1}},git_context:{title:`Git Context`,annotations:{readOnlyHint:!0,idempotentHint:!0}},diff_parse:{title:`Diff Parser`,annotations:{readOnlyHint:!0,idempotentHint:!0}},parse_output:{title:`Parse Build Output`,annotations:{readOnlyHint:!0,idempotentHint:!0}},process:{title:`Process Manager`,annotations:{readOnlyHint:!1,openWorldHint:!0}},watch:{title:`File Watcher`,annotations:{readOnlyHint:!1,openWorldHint:!0}},delegate:{title:`Delegate Task`,annotations:{readOnlyHint:!1,openWorldHint:!0}},status:{title:`KB Status`,annotations:{readOnlyHint:!0,idempotentHint:!0}},health:{title:`Health Check`,annotations:{readOnlyHint:!0,idempotentHint:!0}},reindex:{title:`Reindex`,annotations:{readOnlyHint:!1}},onboard:{title:`Onboard Codebase`,annotations:{readOnlyHint:!1}},graph:{title:`Knowledge Graph`,annotations:{readOnlyHint:!1}},guide:{title:`Tool Guide`,annotations:{readOnlyHint:!0,idempotentHint:!0}},replay:{title:`Replay History`,annotations:{readOnlyHint:!0,idempotentHint:!0}},changelog:{title:`Generate Changelog`,annotations:{readOnlyHint:!0,idempotentHint:!0}},regex_test:{title:`Regex Tester`,annotations:{readOnlyHint:!0,idempotentHint:!0}},encode:{title:`Encode / Decode`,annotations:{readOnlyHint:!0,idempotentHint:!0}},measure:{title:`Code Metrics`,annotations:{readOnlyHint:!0,idempotentHint:!0}},schema_validate:{title:`Schema Validator`,annotations:{readOnlyHint:!0,idempotentHint:!0}},snippet:{title:`Code Snippets`,annotations:{readOnlyHint:!1}},env:{title:`Environment Info`,annotations:{readOnlyHint:!0,idempotentHint:!0}},time:{title:`Date & Time`,annotations:{readOnlyHint:!0,idempotentHint:!0}},web_fetch:{title:`Web Fetch`,annotations:{readOnlyHint:!0,openWorldHint:!0}},web_search:{title:`Web Search`,annotations:{readOnlyHint:!0,openWorldHint:!0}},http:{title:`HTTP Request`,annotations:{readOnlyHint:!1,openWorldHint:!0}},queue:{title:`Operation Queue`,annotations:{readOnlyHint:!1}},bridge_push:{title:`Bridge Push`,annotations:{readOnlyHint:!1}},bridge_pull:{title:`Bridge Pull`,annotations:{readOnlyHint:!0,idempotentHint:!0}},bridge_sync:{title:`Bridge Sync Status`,annotations:{readOnlyHint:!0,idempotentHint:!0}},evolution_state:{title:`Evolution State`,annotations:{readOnlyHint:!1}},policy_check:{title:`Policy Check`,annotations:{readOnlyHint:!0,idempotentHint:!0}},flow_list:{title:`Flow List`,annotations:{readOnlyHint:!0,idempotentHint:!0}},flow_info:{title:`Flow Info`,annotations:{readOnlyHint:!0,idempotentHint:!0}},flow_start:{title:`Flow Start`,annotations:{readOnlyHint:!1}},flow_step:{title:`Flow Step`,annotations:{readOnlyHint:!1}},flow_status:{title:`Flow Status`,annotations:{readOnlyHint:!0,idempotentHint:!0}},flow_reset:{title:`Flow Reset`,annotations:{readOnlyHint:!1}}};function t(t){return e[t]??{title:t,annotations:{readOnlyHint:!1}}}export{e as TOOL_METADATA,t as getToolMeta};

package/packages/server/dist/tool-prefix.d.ts

@@ -0,0 +1,12 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+
+//#region packages/server/src/tool-prefix.d.ts
+/**
+ * Install a tool-name prefix interceptor on the given McpServer.
+ *
+ * @param server  The MCP server whose `registerTool` method will be wrapped.
+ * @param prefix  A string prepended to every tool name. Empty string = no-op.
+ */
+declare function installToolPrefix(server: McpServer, prefix: string): void;
+//#endregion
+export { installToolPrefix };

package/packages/server/dist/tool-prefix.js

@@ -0,0 +1 @@
+function e(e,t){if(!t)return;let n=e.registerTool.bind(e);e.registerTool=(e,r,i)=>{let a=`${t}${e}`;return i===void 0?n(a,r):n(a,r,i)}}export{e as installToolPrefix};

package/packages/server/dist/tools/analyze.tools.d.ts

@@ -0,0 +1,14 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { IGraphStore, IKnowledgeStore } from "../../../store/dist/index.js";
+import { IEmbedder } from "../../../embeddings/dist/index.js";
+
+//#region packages/server/src/tools/analyze.tools.d.ts
+declare function registerAnalyzeStructureTool(server: McpServer, store: IKnowledgeStore, embedder: IEmbedder): void;
+declare function registerAnalyzeDependenciesTool(server: McpServer, store: IKnowledgeStore, embedder: IEmbedder): void;
+declare function registerAnalyzeSymbolsTool(server: McpServer, store: IKnowledgeStore, embedder: IEmbedder): void;
+declare function registerAnalyzePatternsTool(server: McpServer, store: IKnowledgeStore, embedder: IEmbedder): void;
+declare function registerAnalyzeEntryPointsTool(server: McpServer, store: IKnowledgeStore, embedder: IEmbedder): void;
+declare function registerAnalyzeDiagramTool(server: McpServer, store: IKnowledgeStore, embedder: IEmbedder): void;
+declare function registerBlastRadiusTool(server: McpServer, store: IKnowledgeStore, embedder: IEmbedder, graphStore?: IGraphStore): void;
+//#endregion
+export { registerAnalyzeDependenciesTool, registerAnalyzeDiagramTool, registerAnalyzeEntryPointsTool, registerAnalyzePatternsTool, registerAnalyzeStructureTool, registerAnalyzeSymbolsTool, registerBlastRadiusTool };

package/packages/server/dist/tools/analyze.tools.js

@@ -0,0 +1,8 @@
+import{getToolMeta as e}from"../tool-metadata.js";import{AnalyzeStructureOutputSchema as t,BlastRadiusOutputSchema as n}from"../output-schemas.js";import{createHash as r}from"node:crypto";import{z as i}from"zod";import{createLogger as a,serializeError as o}from"../../../core/dist/index.js";import{BlastRadiusAnalyzer as s,DependencyAnalyzer as c,DiagramGenerator as l,EntryPointAnalyzer as u,PatternAnalyzer as d,StructureAnalyzer as f,SymbolAnalyzer as p}from"../../../analyzers/dist/index.js";import{WasmRuntime as m}from"../../../chunker/dist/index.js";import{gitContext as h,truncateToTokenBudget as g}from"../../../tools/dist/index.js";const _=a(`tools`),v=i.number().min(100).max(5e4).optional().describe(`Maximum token budget for the response. When set, output is truncated to fit.`);function y(e,t){return t?g(e,t):e}function b(){let e=[];return m.get()||e.push(`Tree-sitter unavailable — using regex fallback, symbol/pattern confidence reduced`),e.length===0?``:`\n\n> **⚠ Caveats:** ${e.join(`; `)}`}function x(e){return(e??[]).map(e=>{if(typeof e==`string`)return e;if(e&&typeof e==`object`&&`path`in e)return typeof e.path==`string`?e.path:void 0}).filter(e=>!!e)}function S(e){let t=[],n=e.filter(e=>/\.(ts|tsx|js|jsx)$/.test(e)&&/(service|store|model|schema|migration)/i.test(e)),r=e.filter(e=>/\.(ts|tsx|js|jsx)$/.test(e)&&!n.includes(e)),i=e.filter(e=>!/\.(ts|tsx|js|jsx)$/.test(e));return(n.length>0||r.length>0||i.length>0)&&(t.push(`
+
+### Risk Assessment`),n.length>0&&t.push(`- 🔴 **High risk** (${n.length}): ${n.slice(0,5).map(e=>`\`${e}\``).join(`, `)}`),r.length>0&&t.push(`- 🟡 **Medium risk** (${r.length}): source files`),i.length>0&&t.push(`- 🟢 **Low risk** (${i.length}): non-source files`)),t.join(`
+`)}function C(e){let t=e.replace(/\\/g,`/`);return/(^|\/)__tests__\/|\.(test|spec)\.[jt]sx?$/i.test(t)?`Tests`:/(controller|route|handler|api)/i.test(t)?`API`:/(service|store|model|schema|migration|repo|repository|db|database)/i.test(t)?`Core/Data`:/\.(ts|tsx|js|jsx)$/.test(t)?`Source`:`Config/Docs`}function w(e){if(e.length===0)return``;let t=[`Core/Data`,`API`,`Source`,`Tests`,`Config/Docs`],n=new Map;for(let t of e){let e=C(t),r=n.get(e)??[];r.push(t),n.set(e,r)}let r=[`
+
+### Layer Classification`];for(let e of t){let t=n.get(e);if(!t?.length)continue;let i=t.slice(0,5).map(e=>`\`${e}\``).join(`, `),a=t.length>5?`, ... and ${t.length-5} more`:``;r.push(`- **${e}** (${t.length}): ${i}${a}`)}return r.join(`
+`)}async function T(e,t,n,i,a){try{let o=`produced/analysis/${n}/${r(`sha256`).update(i).digest(`hex`).slice(0,12)}.md`,s=r(`sha256`).update(a).digest(`hex`).slice(0,16),c=new Date().toISOString(),l=a.length>2e3?a.split(/(?=^## )/m).filter(e=>e.trim().length>0):[a],u=l.map((e,t)=>({id:r(`sha256`).update(`${o}::${t}`).digest(`hex`).slice(0,16),content:e.trim(),sourcePath:o,contentType:`produced-knowledge`,chunkIndex:t,totalChunks:l.length,startLine:0,endLine:0,fileHash:s,indexedAt:c,origin:`produced`,tags:[`analysis`,n],category:`analysis`,version:1})),d=await t.embedBatch(u.map(e=>e.content));await e.upsert(u,d),_.info(`Auto-persisted analysis`,{analyzerName:n,chunkCount:u.length})}catch(e){_.warn(`Auto-persist analysis failed`,{analyzerName:n,...o(e)})}}function E(n,r,a){let s=new f,c=e(`analyze_structure`);n.registerTool(`analyze_structure`,{title:c.title,description:`Analyze the file/directory structure of a codebase. Returns an annotated tree with language stats.`,outputSchema:t,inputSchema:{path:i.string().describe(`Root path to analyze`),max_depth:i.number().min(1).max(10).default(6).describe(`Maximum directory depth`),format:i.enum([`json`,`markdown`]).default(`markdown`).describe(`Output format`),max_tokens:v},annotations:c.annotations},async({path:e,max_depth:t,format:n,max_tokens:i})=>{try{let o=await s.analyze(e,{format:n,maxDepth:t,maxTokens:i});T(r,a,`structure`,e,o.output);let c=o.data;return{content:[{type:`text`,text:y(o.output+"\n\n---\n_Analysis auto-saved to KB. Next: Use `analyze_dependencies` for import graphs, or `analyze_patterns` to detect architecture patterns._",i)}],structuredContent:{files:c.fileCount??o.meta.fileCount,packages:c.packageCount??0,languages:c.languages??{},tree:o.output.slice(0,2e3)}}}catch(e){return _.error(`Analysis failed`,o(e)),{content:[{type:`text`,text:`Analysis failed: ${e instanceof Error?e.message:String(e)}`}],isError:!0}}})}function D(t,n,r){let a=new c,s=e(`analyze_dependencies`);t.registerTool(`analyze_dependencies`,{title:s.title,description:`Analyze import/require dependencies across a codebase. Shows external packages and internal module graph.`,inputSchema:{path:i.string().describe(`Root path to analyze`),format:i.enum([`json`,`markdown`,`mermaid`]).default(`markdown`).describe(`Output format`),max_tokens:v},annotations:s.annotations},async({path:e,format:t,max_tokens:i})=>{try{let o=await a.analyze(e,{format:t});return T(n,r,`dependencies`,e,o.output),{content:[{type:`text`,text:y(o.output+"\n\n---\n_Analysis auto-saved to KB. Next: Use `analyze_symbols` to explore exported symbols, or `analyze_diagram` for visual representation._",i)}]}}catch(e){return _.error(`Analysis failed`,o(e)),{content:[{type:`text`,text:`Analysis failed: ${e instanceof Error?e.message:String(e)}`}],isError:!0}}})}function O(t,n,r){let a=new p,s=e(`analyze_symbols`);t.registerTool(`analyze_symbols`,{title:s.title,description:`Extract exported and local symbols (functions, classes, interfaces, types, constants) from a codebase.`,inputSchema:{path:i.string().describe(`Root path to analyze`),filter:i.string().optional().describe(`Filter symbols by name substring`),format:i.enum([`json`,`markdown`]).default(`markdown`).describe(`Output format`)},annotations:s.annotations},async({path:e,filter:t,format:i})=>{try{let o=await a.analyze(e,{format:i,filter:t});return T(n,r,`symbols`,e,o.output),{content:[{type:`text`,text:o.output+b()+"\n\n---\n_Analysis auto-saved to KB. Next: Use `analyze_dependencies` to see import relationships, or `search` to find usage patterns._"}]}}catch(e){return _.error(`Analysis failed`,o(e)),{content:[{type:`text`,text:`Analysis failed: ${e instanceof Error?e.message:String(e)}`}],isError:!0}}})}function k(t,n,r){let a=new d,s=e(`analyze_patterns`);t.registerTool(`analyze_patterns`,{title:s.title,description:`Detect architectural patterns, frameworks, and conventions in a codebase using directory structure and code heuristics.`,inputSchema:{path:i.string().describe(`Root path to analyze`)},annotations:s.annotations},async({path:e})=>{try{let t=await a.analyze(e);return T(n,r,`patterns`,e,t.output),{content:[{type:`text`,text:t.output+b()+"\n\n---\n_Analysis auto-saved to KB. Next: Use `analyze_entry_points` to find Lambda handlers and main exports, or `produce_knowledge` for full analysis._"}]}}catch(e){return _.error(`Analysis failed`,o(e)),{content:[{type:`text`,text:`Analysis failed: ${e instanceof Error?e.message:String(e)}`}],isError:!0}}})}function A(t,n,r){let a=new u,s=e(`analyze_entry_points`);t.registerTool(`analyze_entry_points`,{title:s.title,description:`Find entry points: Lambda handlers, main exports, CLI bins, and server start scripts.`,inputSchema:{path:i.string().describe(`Root path to analyze`)},annotations:s.annotations},async({path:e})=>{try{let t=await a.analyze(e);return T(n,r,`entry-points`,e,t.output),{content:[{type:`text`,text:t.output+b()+"\n\n---\n_Analysis auto-saved to KB. Next: Use `analyze_dependencies` to see what each entry point imports, or `produce_knowledge` for comprehensive analysis._"}]}}catch(e){return _.error(`Analysis failed`,o(e)),{content:[{type:`text`,text:`Analysis failed: ${e instanceof Error?e.message:String(e)}`}],isError:!0}}})}function j(t,n,r){let a=new l,s=e(`analyze_diagram`);t.registerTool(`analyze_diagram`,{title:s.title,description:`Generate a Mermaid diagram of the codebase architecture or dependency graph. Returns Mermaid flowchart syntax — paste into any Markdown renderer or Mermaid editor to visualize.`,inputSchema:{path:i.string().describe(`Root path to analyze`),diagram_type:i.enum([`architecture`,`dependencies`]).default(`architecture`).describe(`Type of diagram`)},annotations:s.annotations},async({path:e,diagram_type:t})=>{try{let i=await a.analyze(e,{diagramType:t});return T(n,r,`diagram`,e,i.output),{content:[{type:`text`,text:i.output+"\n\n---\n_Analysis auto-saved to KB. Next: Use `analyze_structure` for detailed file tree, or `produce_knowledge` for comprehensive analysis._"}]}}catch(e){return _.error(`Diagram generation failed`,o(e)),{content:[{type:`text`,text:`Diagram generation failed: ${e instanceof Error?e.message:String(e)}`}],isError:!0}}})}function M(t,r,a,c){let l=new s,u=e(`blast_radius`);t.registerTool(`blast_radius`,{title:u.title,description:`Given a list of changed files, trace the dependency graph to find all affected files (direct + transitive importers) and their tests. Useful for scoping code reviews and impact analysis.`,outputSchema:n,inputSchema:{path:i.string().describe(`Root path of the codebase`),files:i.array(i.string()).optional().describe(`Changed file paths (relative to root). If omitted, auto-detects from git status.`),max_depth:i.number().min(1).max(20).default(5).describe(`Maximum transitive dependency depth`),format:i.enum([`json`,`markdown`]).default(`markdown`).describe(`Output format`),max_tokens:v},annotations:u.annotations},async({path:e,files:t,max_depth:n,format:i,max_tokens:s})=>{try{let o=t??[];if(o.length===0)try{let t=await h({cwd:e,includeDiff:!1});if(o=Array.from(new Set([...x(t.status?.staged),...x(t.status?.modified),...x(t.status?.untracked)])),o.length===0)return{content:[{type:`text`,text:"No changed files detected from git status. Provide `files` explicitly or make changes first."}]}}catch{return{content:[{type:`text`,text:"Could not detect changed files from git. Provide `files` explicitly."}],isError:!0}}let u=await l.analyze(e,{files:o,maxDepth:n,format:i}),d=``;if(c)try{let e=new Set;for(let t of o){let n=await c.findNodes({sourcePath:t,limit:10});for(let t of n){let n=await c.getNeighbors(t.id,{direction:`incoming`,edgeType:`imports`});for(let t of n.nodes){let n=t.sourcePath??t.name;o.includes(n)||e.add(n)}}}e.size>0&&(d=`\n\n### Graph-discovered importers (${e.size})\n`+[...e].slice(0,20).map(e=>`- \`${e}\``).join(`
+`),e.size>20&&(d+=`\n- ... and ${e.size-20} more`))}catch{}let f=w(o),p=S(o),m=u.output+d+f+p+b()+"\n\n---\n_Analysis auto-saved to KB. Next: Use `analyze_dependencies` to see the full import graph, or `analyze_symbols` to inspect affected exports._";T(r,a,`blast-radius`,e,m);let g=new Set;if(c)for(let e of o)try{let t=await c.findNodes({sourcePath:e,limit:10});for(let e of t){let t=await c.getNeighbors(e.id,{direction:`incoming`,edgeType:`imports`});for(let e of t.nodes){let t=e.sourcePath??e.name;o.includes(t)||g.add(t)}}}catch{}let _=[...g].map(e=>({path:e,impact:`transitive`,reason:`imports changed file`})),v=o.length>10?`high`:o.length>3?`medium`:`low`;return{content:[{type:`text`,text:y(m,s)}],structuredContent:{changedFiles:o,affectedFiles:_,totalAffected:_.length,riskLevel:v}}}catch(e){return _.error(`Blast radius analysis failed`,o(e)),{content:[{type:`text`,text:`Blast radius analysis failed: ${e instanceof Error?e.message:String(e)}`}],isError:!0}}})}export{D as registerAnalyzeDependenciesTool,j as registerAnalyzeDiagramTool,A as registerAnalyzeEntryPointsTool,k as registerAnalyzePatternsTool,E as registerAnalyzeStructureTool,O as registerAnalyzeSymbolsTool,M as registerBlastRadiusTool};

package/packages/server/dist/tools/audit.tool.d.ts

@@ -0,0 +1,8 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { IKnowledgeStore } from "../../../store/dist/index.js";
+import { IEmbedder } from "../../../embeddings/dist/index.js";
+
+//#region packages/server/src/tools/audit.tool.d.ts
+declare function registerAuditTool(server: McpServer, store: IKnowledgeStore, embedder: IEmbedder): void;
+//#endregion
+export { registerAuditTool };

package/packages/server/dist/tools/audit.tool.js

@@ -0,0 +1 @@
+import{getToolMeta as e}from"../tool-metadata.js";import{AuditOutputSchema as t}from"../output-schemas.js";import{createTaskRunner as n}from"../task-manager.js";import{z as r}from"zod";import{createLogger as i,serializeError as a}from"../../../core/dist/index.js";import{audit as o}from"../../../tools/dist/index.js";const s=i(`tools`);function c(i,c,l){let u=e(`audit`);i.registerTool(`audit`,{title:u.title,description:`Run a unified audit: structure, dependencies, patterns, health, dead_symbols, check, entry_points. Returns synthesized report with score, recommendations, and next steps. 6 round-trips → 1.`,outputSchema:t,inputSchema:{path:r.string().default(`.`).describe(`Root path to audit`),checks:r.array(r.enum([`structure`,`dependencies`,`patterns`,`health`,`dead_symbols`,`check`,`entry_points`])).optional().describe(`Which checks to run (default: all)`),detail:r.enum([`summary`,`full`]).default(`summary`).describe(`'summary' for overview (~500 tokens), 'full' includes pattern table`)},annotations:u.annotations},async({path:e,checks:t,detail:r},i)=>{try{let a=await o(c,l,{path:e,checks:t,detail:r});return n(i).createTask(`Audit`,1).complete(`Audit complete: ${a.data?.recommendations?.length??0} recommendations`),{content:[{type:`text`,text:a.ok?`${a.summary}\n\n---\n_Audit score: ${a.data?.score}/100 | ${a.meta.durationMs}ms${a.next&&a.next.length>0?` | Next: ${a.next.map(e=>`\`${e.tool}\` (${e.reason})`).join(`, `)}`:``}_`:a.error?.message??`Audit failed`}],isError:!a.ok,...a.ok&&a.data?{structuredContent:{passed:a.ok,score:a.data.score??0,checks:(a.data.recommendations??[]).map(e=>({name:e.area,passed:e.priority!==`high`,message:e.message,severity:e.priority})),summary:a.summary??``}}:{}}}catch(e){return s.error(`Audit failed`,a(e)),{content:[{type:`text`,text:`Audit failed: ${e instanceof Error?e.message:String(e)}`}],isError:!0}}})}export{c as registerAuditTool};

package/packages/server/dist/tools/brainstorm.tool.d.ts

@@ -0,0 +1,7 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { Elicitor } from "../../../elicitation/dist/index.js";
+
+//#region packages/server/src/tools/brainstorm.tool.d.ts
+declare function registerBrainstormTool(server: McpServer, elicitor: Elicitor): void;
+//#endregion
+export { registerBrainstormTool };

package/packages/server/dist/tools/brainstorm.tool.js

@@ -0,0 +1,9 @@
+import{getToolMeta as e}from"../tool-metadata.js";import{z as t}from"zod";import{createLogger as n}from"../../../core/dist/index.js";const r=n(`tools`);function i(n,i){let s=e(`brainstorm`);n.registerTool(`brainstorm`,{title:s.title,description:`Interactive brainstorming tool for exploring ideas, making design decisions, and gathering requirements. Supports multiple actions: ask_choice (single selection), ask_multi (multiple selections), ask_freeform (text input), ask_confirm (yes/no), present_design (show summary and confirm). Falls back to text-only mode when elicitation is unavailable.`,inputSchema:{action:t.enum([`ask_choice`,`ask_multi`,`ask_freeform`,`ask_confirm`,`present_design`]).describe(`The type of brainstorm interaction`),message:t.string().describe(`The question or prompt to present`),options:t.array(t.string()).optional().describe(`Options for ask_choice/ask_multi actions`),context:t.string().optional().describe(`Additional context to include in the prompt`)},annotations:s.annotations},async({action:e,message:t,options:n,context:s})=>{let c=s?`${s}\n\n${t}`:t;if(!i.available)return o(e,c,n);try{switch(e){case`ask_choice`:{if(!n?.length)return a(`Error: ask_choice requires options array`);let e=await i.selectOne(c,n);return a(e===null?`No selection made (declined or cancelled).`:`Selected: **${e}**`)}case`ask_multi`:{if(!n?.length)return a(`Error: ask_multi requires options array`);let e=await i.selectMany(c,n);return e.length===0?a(`No selections made (declined or cancelled).`):a(`Selected:\n${e.map(e=>`- ${e}`).join(`
+`)}`)}case`ask_freeform`:{let e=await i.promptText(c);return a(e===null?`No input provided (declined or cancelled).`:`Response:\n\n${e}`)}case`ask_confirm`:return a(await i.confirm(c)?`Confirmed: **Yes**`:`Confirmed: **No**`);case`present_design`:return a(await i.confirm(`## Design Summary\n\n${c}\n\nDo you approve this design?`)?`## Design Approved ✓
+
+Proceeding with the design as described.`:`## Design Not Approved
+
+Please provide feedback for revisions.`);default:return a(`Unknown action: ${e}`)}}catch(t){return r.warn(`Brainstorm elicitation failed, falling back to text`,{action:e,error:String(t)}),o(e,c,n)}})}function a(e){return{content:[{type:`text`,text:e}]}}function o(e,t,n){let r=`## Brainstorm: ${e}\n\n${t}`;return n?.length&&(r+=`\n\n**Options:**\n${n.map((e,t)=>`${t+1}. ${e}`).join(`
+`)}`),r+=`
+
+> *Elicitation unavailable — please respond with your choice in the next message.*`,{content:[{type:`text`,text:r}]}}export{i as registerBrainstormTool};

package/packages/server/dist/tools/bridge.tools.d.ts

@@ -0,0 +1,34 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { ERBridgeConfig, ERCache, ERClient, EvolutionCollector, PushAdapter } from "../../../enterprise-bridge/dist/index.js";
+
+//#region packages/server/src/tools/bridge.tools.d.ts
+/** Resolve ER bridge config from KBConfig.er + env */
+declare function resolveErBridgeConfig(erConfig?: {
+  enabled: boolean;
+  baseUrl: string;
+  timeoutMs?: number;
+  cacheTtlMs?: number;
+  cacheMaxEntries?: number;
+  fallbackThreshold?: number;
+}): ERBridgeConfig | undefined;
+/** Shared bridge components (initialized once, reused across tools) */
+interface BridgeComponents {
+  client: ERClient;
+  cache: ERCache;
+  pushAdapter: PushAdapter;
+  config: ERBridgeConfig;
+}
+/** Initialize bridge components from config. Returns undefined if ER disabled or init fails. */
+declare function initBridgeComponents(erConfig?: {
+  enabled: boolean;
+  baseUrl: string;
+  timeoutMs?: number;
+  cacheTtlMs?: number;
+  cacheMaxEntries?: number;
+  fallbackThreshold?: number;
+}): BridgeComponents | undefined;
+declare function registerErPushTool(server: McpServer, bridge: BridgeComponents, evolutionCollector?: EvolutionCollector): void;
+declare function registerErPullTool(server: McpServer, bridge: BridgeComponents): void;
+declare function registerErSyncStatusTool(server: McpServer, bridge: BridgeComponents): void;
+//#endregion
+export { BridgeComponents, initBridgeComponents, registerErPullTool, registerErPushTool, registerErSyncStatusTool, resolveErBridgeConfig };