@vpxa/aikit 0.1.58 → 0.1.60

package/packages/server/dist/output-schemas.d.ts

@@ -1,244 +0,0 @@
-import { z } from "zod";
-
-//#region packages/server/src/output-schemas.d.ts
-/**
- * Centralized output schemas for MCP tools that return naturally structured data.
- *
- * Phase 1: status, list, health, measure, env, time
- * These schemas are used with `outputSchema` in tool registration and validated
- * by the SDK at runtime. Keep them intentionally flat and stable — do not leak
- * internal storage record shapes.
- */
-declare const StatusOutputSchema: z.ZodObject<{
-  totalRecords: z.ZodNumber;
-  totalFiles: z.ZodNumber;
-  lastIndexedAt: z.ZodNullable<z.ZodString>;
-  onboarded: z.ZodBoolean;
-  onboardDir: z.ZodString;
-  contentTypes: z.ZodRecord<z.ZodString, z.ZodNumber>;
-  wasmAvailable: z.ZodBoolean;
-  graphStats: z.ZodNullable<z.ZodObject<{
-    nodes: z.ZodNumber;
-    edges: z.ZodNumber;
-  }, z.core.$strip>>;
-  curatedCount: z.ZodNumber;
-  serverVersion: z.ZodString;
-  scaffoldVersion: z.ZodNullable<z.ZodString>;
-  workspaceScaffoldVersion: z.ZodNullable<z.ZodString>;
-  upgradeAvailable: z.ZodBoolean;
-  contextPressure: z.ZodNumber;
-}, z.core.$strip>;
-declare const ListOutputSchema: z.ZodObject<{
-  entries: z.ZodArray<z.ZodObject<{
-    path: z.ZodString;
-    title: z.ZodString;
-    category: z.ZodString;
-    tags: z.ZodArray<z.ZodString>;
-    version: z.ZodNumber;
-    preview: z.ZodString;
-  }, z.core.$strip>>;
-  totalCount: z.ZodNumber;
-}, z.core.$strip>;
-declare const HealthOutputSchema: z.ZodObject<{
-  ok: z.ZodBoolean;
-  checks: z.ZodArray<z.ZodObject<{
-    name: z.ZodString;
-    ok: z.ZodBoolean;
-    message: z.ZodOptional<z.ZodString>;
-  }, z.core.$strip>>;
-}, z.core.$strip>;
-declare const MeasureOutputSchema: z.ZodObject<{
-  summary: z.ZodObject<{
-    totalFiles: z.ZodNumber;
-    totalLines: z.ZodNumber;
-    totalCodeLines: z.ZodNumber;
-    totalFunctions: z.ZodNumber;
-    avgComplexity: z.ZodNumber;
-    maxComplexity: z.ZodObject<{
-      value: z.ZodNumber;
-      file: z.ZodString;
-    }, z.core.$strip>;
-  }, z.core.$strip>;
-  files: z.ZodArray<z.ZodObject<{
-    path: z.ZodString;
-    lines: z.ZodNumber;
-    code: z.ZodNumber;
-    complexity: z.ZodNumber;
-    functions: z.ZodNumber;
-  }, z.core.$strip>>;
-}, z.core.$strip>;
-declare const EnvOutputSchema: z.ZodObject<{
-  platform: z.ZodString;
-  arch: z.ZodString;
-  nodeVersion: z.ZodString;
-  cwd: z.ZodString;
-  cpus: z.ZodNumber;
-  memoryFreeGb: z.ZodNumber;
-  memoryTotalGb: z.ZodNumber;
-}, z.core.$strip>;
-declare const TimeOutputSchema: z.ZodObject<{
-  iso: z.ZodString;
-  unix: z.ZodNumber;
-  timezone: z.ZodString;
-  formatted: z.ZodString;
-}, z.core.$strip>;
-declare const CheckOutputSchema: z.ZodObject<{
-  passed: z.ZodBoolean;
-  tsc: z.ZodObject<{
-    passed: z.ZodBoolean;
-    errorCount: z.ZodNumber;
-    warningCount: z.ZodNumber;
-    topErrors: z.ZodArray<z.ZodString>;
-  }, z.core.$strip>;
-  biome: z.ZodObject<{
-    passed: z.ZodBoolean;
-    errorCount: z.ZodNumber;
-    warningCount: z.ZodNumber;
-    topErrors: z.ZodArray<z.ZodString>;
-  }, z.core.$strip>;
-}, z.core.$strip>;
-declare const SymbolOutputSchema: z.ZodObject<{
-  name: z.ZodString;
-  definedIn: z.ZodNullable<z.ZodObject<{
-    path: z.ZodString;
-    line: z.ZodNumber;
-    kind: z.ZodString;
-    signature: z.ZodOptional<z.ZodString>;
-  }, z.core.$strip>>;
-  importedBy: z.ZodArray<z.ZodObject<{
-    path: z.ZodString;
-    line: z.ZodNumber;
-    importStatement: z.ZodString;
-  }, z.core.$strip>>;
-  referencedIn: z.ZodArray<z.ZodObject<{
-    path: z.ZodString;
-    line: z.ZodNumber;
-    context: z.ZodString;
-    scope: z.ZodOptional<z.ZodString>;
-  }, z.core.$strip>>;
-  graphContext: z.ZodNullable<z.ZodObject<{
-    definingModule: z.ZodOptional<z.ZodString>;
-    importedByModules: z.ZodArray<z.ZodString>;
-    siblingSymbols: z.ZodArray<z.ZodString>;
-  }, z.core.$strip>>;
-}, z.core.$strip>;
-declare const SearchOutputSchema: z.ZodObject<{
-  results: z.ZodArray<z.ZodObject<{
-    sourcePath: z.ZodString;
-    contentType: z.ZodString;
-    score: z.ZodNumber;
-    headingPath: z.ZodOptional<z.ZodString>;
-    startLine: z.ZodOptional<z.ZodNumber>;
-    endLine: z.ZodOptional<z.ZodNumber>;
-    origin: z.ZodOptional<z.ZodString>;
-    category: z.ZodOptional<z.ZodString>;
-    tags: z.ZodOptional<z.ZodArray<z.ZodString>>;
-  }, z.core.$strip>>;
-  totalResults: z.ZodNumber;
-  searchMode: z.ZodString;
-  query: z.ZodString;
-}, z.core.$strip>;
-declare const FindOutputSchema: z.ZodObject<{
-  matches: z.ZodArray<z.ZodObject<{
-    path: z.ZodString;
-    line: z.ZodOptional<z.ZodNumber>;
-    matchType: z.ZodString;
-    preview: z.ZodString;
-  }, z.core.$strip>>;
-  totalMatches: z.ZodNumber;
-  pattern: z.ZodString;
-  truncated: z.ZodBoolean;
-}, z.core.$strip>;
-declare const ScopeMapOutputSchema: z.ZodObject<{
-  files: z.ZodArray<z.ZodObject<{
-    path: z.ZodString;
-    relevance: z.ZodNumber;
-    estimatedTokens: z.ZodNumber;
-    focusLines: z.ZodOptional<z.ZodArray<z.ZodString>>;
-  }, z.core.$strip>>;
-  totalFiles: z.ZodNumber;
-  totalEstimatedTokens: z.ZodNumber;
-  task: z.ZodString;
-}, z.core.$strip>;
-declare const BlastRadiusOutputSchema: z.ZodObject<{
-  changedFiles: z.ZodArray<z.ZodString>;
-  affectedFiles: z.ZodArray<z.ZodObject<{
-    path: z.ZodString;
-    impact: z.ZodString;
-    reason: z.ZodString;
-  }, z.core.$strip>>;
-  totalAffected: z.ZodNumber;
-  riskLevel: z.ZodString;
-}, z.core.$strip>;
-declare const AuditOutputSchema: z.ZodObject<{
-  passed: z.ZodBoolean;
-  score: z.ZodNumber;
-  checks: z.ZodArray<z.ZodObject<{
-    name: z.ZodString;
-    passed: z.ZodBoolean;
-    message: z.ZodOptional<z.ZodString>;
-    severity: z.ZodOptional<z.ZodString>;
-  }, z.core.$strip>>;
-  summary: z.ZodString;
-}, z.core.$strip>;
-declare const GraphOutputSchema: z.ZodObject<{
-  nodes: z.ZodArray<z.ZodObject<{
-    id: z.ZodString;
-    name: z.ZodString;
-    type: z.ZodString;
-    sourcePath: z.ZodOptional<z.ZodString>;
-  }, z.core.$strip>>;
-  edges: z.ZodArray<z.ZodObject<{
-    fromId: z.ZodString;
-    toId: z.ZodString;
-    type: z.ZodString;
-  }, z.core.$strip>>;
-  totalNodes: z.ZodNumber;
-  totalEdges: z.ZodNumber;
-  query: z.ZodString;
-}, z.core.$strip>;
-declare const DeadSymbolsOutputSchema: z.ZodObject<{
-  symbols: z.ZodArray<z.ZodObject<{
-    name: z.ZodString;
-    path: z.ZodString;
-    line: z.ZodOptional<z.ZodNumber>;
-    kind: z.ZodString;
-  }, z.core.$strip>>;
-  totalDead: z.ZodNumber;
-}, z.core.$strip>;
-declare const AnalyzeStructureOutputSchema: z.ZodObject<{
-  files: z.ZodNumber;
-  packages: z.ZodNumber;
-  languages: z.ZodRecord<z.ZodString, z.ZodNumber>;
-  tree: z.ZodString;
-}, z.core.$strip>;
-declare const FileSummaryOutputSchema: z.ZodObject<{
-  path: z.ZodString;
-  language: z.ZodString;
-  lines: z.ZodNumber;
-  imports: z.ZodNumber;
-  exports: z.ZodNumber;
-  functions: z.ZodNumber;
-  classes: z.ZodNumber;
-}, z.core.$strip>;
-declare const GitContextOutputSchema: z.ZodObject<{
-  gitRoot: z.ZodString;
-  branch: z.ZodString;
-  commitCount: z.ZodNumber;
-  hasUncommitted: z.ZodBoolean;
-  recentCommits: z.ZodArray<z.ZodObject<{
-    hash: z.ZodString;
-    message: z.ZodString;
-    author: z.ZodString;
-    date: z.ZodString;
-  }, z.core.$strip>>;
-}, z.core.$strip>;
-declare const CompactOutputSchema: z.ZodObject<{
-  originalChars: z.ZodNumber;
-  compressedChars: z.ZodNumber;
-  ratio: z.ZodNumber;
-  segmentsKept: z.ZodNumber;
-  segmentsTotal: z.ZodNumber;
-}, z.core.$strip>;
-//#endregion
-export { AnalyzeStructureOutputSchema, AuditOutputSchema, BlastRadiusOutputSchema, CheckOutputSchema, CompactOutputSchema, DeadSymbolsOutputSchema, EnvOutputSchema, FileSummaryOutputSchema, FindOutputSchema, GitContextOutputSchema, GraphOutputSchema, HealthOutputSchema, ListOutputSchema, MeasureOutputSchema, ScopeMapOutputSchema, SearchOutputSchema, StatusOutputSchema, SymbolOutputSchema, TimeOutputSchema };

package/packages/server/dist/output-schemas.js

@@ -1 +0,0 @@
-import{z as e}from"zod";const t=e.object({totalRecords:e.number(),totalFiles:e.number(),lastIndexedAt:e.string().nullable(),onboarded:e.boolean(),onboardDir:e.string(),contentTypes:e.record(e.string(),e.number()),wasmAvailable:e.boolean(),graphStats:e.object({nodes:e.number(),edges:e.number()}).nullable(),curatedCount:e.number(),serverVersion:e.string(),scaffoldVersion:e.string().nullable(),workspaceScaffoldVersion:e.string().nullable(),upgradeAvailable:e.boolean(),contextPressure:e.number().min(0).max(100).describe(`0–100 score indicating knowledge base saturation`)}),n=e.object({entries:e.array(e.object({path:e.string(),title:e.string(),category:e.string(),tags:e.array(e.string()),version:e.number(),preview:e.string()})),totalCount:e.number()}),r=e.object({ok:e.boolean(),checks:e.array(e.object({name:e.string(),ok:e.boolean(),message:e.string().optional()}))}),i=e.object({summary:e.object({totalFiles:e.number(),totalLines:e.number(),totalCodeLines:e.number(),totalFunctions:e.number(),avgComplexity:e.number(),maxComplexity:e.object({value:e.number(),file:e.string()})}),files:e.array(e.object({path:e.string(),lines:e.number(),code:e.number(),complexity:e.number(),functions:e.number()}))}),a=e.object({platform:e.string(),arch:e.string(),nodeVersion:e.string(),cwd:e.string(),cpus:e.number(),memoryFreeGb:e.number(),memoryTotalGb:e.number()}),o=e.object({iso:e.string(),unix:e.number(),timezone:e.string(),formatted:e.string()}),s=e.object({passed:e.boolean(),errorCount:e.number(),warningCount:e.number(),topErrors:e.array(e.string())}),c=e.object({passed:e.boolean(),tsc:s,biome:s}),l=e.object({name:e.string(),definedIn:e.object({path:e.string(),line:e.number(),kind:e.string(),signature:e.string().optional()}).nullable(),importedBy:e.array(e.object({path:e.string(),line:e.number(),importStatement:e.string()})),referencedIn:e.array(e.object({path:e.string(),line:e.number(),context:e.string(),scope:e.string().optional()})),graphContext:e.object({definingModule:e.string().optional(),importedByModules:e.array(e.string()),siblingSymbols:e.array(e.string())}).nullable()}),u=e.object({sourcePath:e.string(),contentType:e.string(),score:e.number(),headingPath:e.string().optional(),startLine:e.number().optional(),endLine:e.number().optional(),origin:e.string().optional(),category:e.string().optional(),tags:e.array(e.string()).optional()}),d=e.object({results:e.array(u),totalResults:e.number(),searchMode:e.string(),query:e.string()}),f=e.object({path:e.string(),line:e.number().optional(),matchType:e.string(),preview:e.string()}),p=e.object({matches:e.array(f),totalMatches:e.number(),pattern:e.string(),truncated:e.boolean()}),m=e.object({path:e.string(),relevance:e.number(),estimatedTokens:e.number(),focusLines:e.array(e.string()).optional()}),h=e.object({files:e.array(m),totalFiles:e.number(),totalEstimatedTokens:e.number(),task:e.string()}),g=e.object({path:e.string(),impact:e.string(),reason:e.string()}),_=e.object({changedFiles:e.array(e.string()),affectedFiles:e.array(g),totalAffected:e.number(),riskLevel:e.string()}),v=e.object({name:e.string(),passed:e.boolean(),message:e.string().optional(),severity:e.string().optional()}),y=e.object({passed:e.boolean(),score:e.number(),checks:e.array(v),summary:e.string()}),b=e.object({id:e.string(),name:e.string(),type:e.string(),sourcePath:e.string().optional()}),x=e.object({fromId:e.string(),toId:e.string(),type:e.string()}),S=e.object({nodes:e.array(b),edges:e.array(x),totalNodes:e.number(),totalEdges:e.number(),query:e.string()}),C=e.object({symbols:e.array(e.object({name:e.string(),path:e.string(),line:e.number().optional(),kind:e.string()})),totalDead:e.number()}),w=e.object({files:e.number(),packages:e.number(),languages:e.record(e.string(),e.number()),tree:e.string()}),T=e.object({path:e.string(),language:e.string(),lines:e.number(),imports:e.number(),exports:e.number(),functions:e.number(),classes:e.number()}),E=e.object({gitRoot:e.string(),branch:e.string(),commitCount:e.number(),hasUncommitted:e.boolean(),recentCommits:e.array(e.object({hash:e.string(),message:e.string(),author:e.string(),date:e.string()}))}),D=e.object({originalChars:e.number(),compressedChars:e.number(),ratio:e.number(),segmentsKept:e.number(),segmentsTotal:e.number()});export{w as AnalyzeStructureOutputSchema,y as AuditOutputSchema,_ as BlastRadiusOutputSchema,c as CheckOutputSchema,D as CompactOutputSchema,C as DeadSymbolsOutputSchema,a as EnvOutputSchema,T as FileSummaryOutputSchema,p as FindOutputSchema,E as GitContextOutputSchema,S as GraphOutputSchema,r as HealthOutputSchema,n as ListOutputSchema,i as MeasureOutputSchema,h as ScopeMapOutputSchema,d as SearchOutputSchema,t as StatusOutputSchema,l as SymbolOutputSchema,o as TimeOutputSchema};

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

@@ -1,13 +0,0 @@
-import { CuratedKnowledgeManager } from "./curated-manager.js";
-import { IGraphStore, IKnowledgeStore } from "../../store/dist/index.js";
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-
-//#region packages/server/src/prompts.d.ts
-interface PromptDependencies {
-  curated: CuratedKnowledgeManager;
-  store: IKnowledgeStore;
-  graphStore: IGraphStore;
-}
-declare function registerPrompts(server: McpServer, deps?: PromptDependencies, indexMode?: string): void;
-//#endregion
-export { PromptDependencies, registerPrompts };

package/packages/server/dist/prompts.js

@@ -1,13 +0,0 @@
-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,u){if(c.registerPrompt(`ready`,{title:`AI Kit Ready`,description:`AI Kit is ready — quick-start guide for search, onboard, and workflows`},async()=>({messages:[{role:`user`,content:{type:`text`,text:[`AI Kit 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 AI Kit Session`,description:`Initialize an AI Kit 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 AI Kit 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 AI Kit 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',u===`smart`?`2. Smart indexing is active — index refreshes automatically`:"2. `reindex({})` — refresh search index if files changed",`3. Confirm session data saved`].join(`
-`)}}]})),c.registerPrompt(`search`,{title:`Search AI Kit`,description:`Search AI Kit with hybrid semantic + keyword search`,argsSchema:{query:s.string().describe(`Search query`)}},async({query:e})=>({messages:[{role:`user`,content:{type:`text`,text:`Search AI Kit 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:d,store:f,graphStore:p}=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(d,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(d,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(d,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(f,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(p,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",u===`smart`?`6. Smart indexing is active — changes will be indexed automatically`:"6. `reindex({})` — Update AI Kit 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

@@ -1,23 +0,0 @@
-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

@@ -1 +0,0 @@
-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

@@ -1,34 +0,0 @@
-//#region packages/server/src/resource-links.d.ts
-/**
- * Helpers for emitting resource_link content blocks in tool results.
- * Centralises the `aikit://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

@@ -1 +0,0 @@
-function e(e,t,n){if(!(!e||typeof e!=`string`)&&!(e.includes(`..`)||e.startsWith(`/`)||e.startsWith(`[`)))return{type:`resource_link`,uri:`aikit://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

@@ -1,13 +0,0 @@
-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.
- *
- * - `aikit://curated` — index listing all curated entries
- * - `aikit://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

@@ -1,2 +0,0 @@
-import{ResourceTemplate as e}from"@modelcontextprotocol/sdk/server/mcp.js";function t(t,n){t.resource(`aikit-curated-index`,`aikit://curated`,{description:`Index of all curated knowledge entries`,mimeType:`text/markdown`},async()=>{let e=(await n.list()).map(e=>`- [${e.title}](aikit://curated/${e.path}) — ${e.category}`);return{contents:[{uri:`aikit://curated`,text:`# Curated Knowledge Index\n\n${e.length>0?e.join(`
-`):`_No curated entries yet._`}`,mimeType:`text/markdown`}]}});let r=new e(`aikit://curated/{+path}`,{list:async()=>({resources:(await n.list()).map(e=>({uri:`aikit://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

@@ -1,45 +0,0 @@
-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 AI Kit 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

@@ -1 +0,0 @@
-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(`aikit://status`)}async notifyFileTreeChanged(){await this.sendUpdate(`aikit://file-tree`)}async notifyCuratedIndexChanged(){await this.sendUpdate(`aikit://curated`)}async notifyCuratedEntryChanged(e){await this.sendUpdate(`aikit://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

@@ -1,8 +0,0 @@
-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

@@ -1,2 +0,0 @@
-import{registerCuratedResources as e}from"./curated-resources.js";function t(t,n,r){t.resource(`aikit-status`,`aikit://status`,{description:`Current AI Kit status and statistics`,mimeType:`text/plain`},async()=>{let e=await n.getStats();return{contents:[{uri:`aikit://status`,text:`AI Kit: ${e.totalRecords} records from ${e.totalFiles} files. Last indexed: ${e.lastIndexedAt??`Never`}`,mimeType:`text/plain`}]}}),t.resource(`aikit-file-tree`,`aikit://file-tree`,{description:`List of all indexed source files`,mimeType:`text/plain`},async()=>({contents:[{uri:`aikit://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

@@ -1,41 +0,0 @@
-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

@@ -1,2 +0,0 @@
-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

@@ -1,63 +0,0 @@
-import { BackgroundTaskScheduler } from "./background-task.js";
-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
-/** Shape returned by SmartIndexScheduler.getState() */
-interface SmartSchedulerState {
-  mode: string;
-  queueSize: number;
-  changedFilesSize: number;
-  intervalMs: number;
-  batchSize: number;
-  running: boolean;
-}
-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, indexMode?: string, getSmartState?: (() => SmartSchedulerState) | null): 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", "config", "data_transform", "dead_symbols", "delegate", "diff_parse", "digest", "encode", "env", "eval", "evidence_map", "file_summary", "find", "flow_info", "flow_list", "flow_reset", "flow_start", "flow_status", "flow_step", "flow_add", "flow_update", "flow_remove", "flow_read_instruction", "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, indexMode?: string): {
-  server: McpServer; /** Call after MCP roots are resolved (or fallback decided) to start heavy init. */
-  startInit: () => void;
-  ready: Promise<void>;
-  runInitialIndex: () => Promise<void>; /** Initialized AI Kit components after lazy startup completes. */
-  readonly kb: KnowledgeBaseComponents | null; /** Background task scheduler — use to queue heavy operations and check status. */
-  scheduler: BackgroundTaskScheduler; /** Set the smart index scheduler for tool invocation prioritization. */
-  setSmartScheduler: (s: {
-    prioritize: (...paths: string[]) => void;
-    getState?: () => SmartSchedulerState;
-  } | null) => void;
-};
-//#endregion
-export { ALL_TOOL_NAMES, KnowledgeBaseComponents, createLazyServer, createMcpServer, createServer, initializeKnowledgeBase, registerMcpTools };

package/packages/server/dist/server.js

@@ -1,3 +0,0 @@
-import{BackgroundTaskScheduler as e}from"./background-task.js";import{clearCompletionCache as t}from"./completions.js";import{installCompressionInterceptor as n}from"./compression-interceptor.js";import{CuratedKnowledgeManager as r}from"./curated-manager.js";import{createElicitor as i,noopElicitor as a}from"./elicitor.js";import{IdleTimer as o}from"./idle-timer.js";import{bridgeMcpLogging as s}from"./mcp-logging.js";import{MemoryMonitor as c}from"./memory-monitor.js";import{registerPrompts as l}from"./prompts.js";import{installReplayInterceptor as u}from"./replay-interceptor.js";import{ResourceNotifier as d}from"./resources/resource-notifier.js";import{registerResources as f}from"./resources/resources.js";import{createSamplingClient as p}from"./sampling.js";import{installStructuredContentGuard as m}from"./structured-content-guard.js";import{getToolMeta as h}from"./tool-metadata.js";import{installToolPrefix as g}from"./tool-prefix.js";import{ToolTimeoutError as _,getToolTimeout as v,withTimeout as ee}from"./tool-timeout.js";import{registerAnalyzeDependenciesTool as y,registerAnalyzeDiagramTool as b,registerAnalyzeEntryPointsTool as x,registerAnalyzePatternsTool as te,registerAnalyzeStructureTool as S,registerAnalyzeSymbolsTool as C,registerBlastRadiusTool as w}from"./tools/analyze.tools.js";import{registerAuditTool as ne}from"./tools/audit.tool.js";import{registerBrainstormTool as T}from"./tools/brainstorm.tool.js";import{initBridgeComponents as E,registerErPullTool as D,registerErPushTool as O,registerErSyncStatusTool as re}from"./tools/bridge.tools.js";import{registerConfigTool as k}from"./tools/config.tool.js";import{registerCompactTool as A,registerDeadSymbolsTool as j,registerFileSummaryTool as M,registerFindTool as ie,registerScopeMapTool as N,registerSymbolTool as P,registerTraceTool as ae}from"./tools/context.tools.js";import{registerErEvolveReviewTool as oe}from"./tools/evolution.tools.js";import{registerBatchTool as se,registerCheckTool as F,registerDelegateTool as I,registerEvalTool as L,registerParseOutputTool as R,registerTestRunTool as ce}from"./tools/execution.tools.js";import{registerFlowTools as z}from"./tools/flow.tools.js";import{registerDigestTool as le,registerEvidenceMapTool as B,registerForgeClassifyTool as V,registerForgeGroundTool as ue,registerStratumCardTool as de}from"./tools/forge.tools.js";import{registerForgetTool as fe}from"./tools/forget.tool.js";import{registerGraphTool as pe}from"./tools/graph.tool.js";import{registerGuideTool as H,registerHealthTool as U,registerProcessTool as W,registerWatchTool as G,registerWebFetchTool as me}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 _e,registerDataTransformTool as K,registerDiffParseTool as ve,registerGitContextTool as ye,registerRenameTool as be}from"./tools/manipulation.tools.js";import{registerOnboardTool as xe}from"./tools/onboard.tool.js";import{registerCheckpointTool as Se,registerLaneTool as Ce,registerQueueTool as we,registerStashTool as Te,registerWorksetTool as Ee}from"./tools/persistence.tools.js";import{registerErUpdatePolicyTool as De}from"./tools/policy.tools.js";import{registerPresentTool as Oe}from"./tools/present/tool.js";import"./tools/present/index.js";import{registerProduceKnowledgeTool as ke}from"./tools/produce.tool.js";import{registerReadTool as Ae}from"./tools/read.tool.js";import{registerReindexTool as je}from"./tools/reindex.tool.js";import{registerRememberTool as Me}from"./tools/remember.tool.js";import{registerReplayTool as Ne}from"./tools/replay.tool.js";import{registerRestoreTool as Pe}from"./tools/restore.tool.js";import{registerSearchTool as Fe}from"./tools/search.tool.js";import{getCurrentVersion as Ie}from"./version-check.js";import{registerEarlyStatusTool as Le,registerStatusTool as Re}from"./tools/status.tool.js";import{registerUpdateTool as ze}from"./tools/update.tool.js";import{registerChangelogTool as Be,registerEncodeTool as Ve,registerEnvTool as He,registerHttpTool as Ue,registerMeasureTool as We,registerRegexTestTool as Ge,registerSchemaValidateTool as Ke,registerSnippetTool as qe,registerTimeTool as Je,registerWebSearchTool as Ye}from"./tools/utility.tools.js";import{existsSync as Xe,statSync as Ze}from"node:fs";import{join as q,resolve as Qe}from"node:path";import{AIKIT_PATHS as $e,EMBEDDING_DEFAULTS as et,createLogger as tt,serializeError as J}from"../../core/dist/index.js";import{homedir as nt}from"node:os";import{initializeWasm as rt}from"../../chunker/dist/index.js";import{OnnxEmbedder as it}from"../../embeddings/dist/index.js";import{EvolutionCollector as at,PolicyStore as ot}from"../../enterprise-bridge/dist/index.js";import{FileHashCache as st,IncrementalIndexer as ct}from"../../indexer/dist/index.js";import{SqliteGraphStore as lt,createStore as ut}from"../../store/dist/index.js";import{FileCache as dt}from"../../tools/dist/index.js";import{completable as ft}from"@modelcontextprotocol/sdk/server/completable.js";import{McpServer as pt}from"@modelcontextprotocol/sdk/server/mcp.js";import{z as mt}from"zod";const Y=tt(`server`);function ht(e){let t=e.toLowerCase();return[`protobuf`,`invalid model`,`invalid onnx`,`unexpected end`,`unexpected token`,`failed to load`,`failed to initialize embedding`,`checksum`,`corrupt`,`malformed`,`could not load`,`onnx`,`database disk image is malformed`,`file is not a database`,`lance`].some(e=>t.includes(e))}async function gt(e,t){let n=t.toLowerCase(),r;try{({rm:r}=await import(`node:fs/promises`))}catch{return}if(n.includes(`embedding`)||n.includes(`onnx`)||n.includes(`protobuf`)||n.includes(`model`)){let t=e.embedding?.model??et.model,n=q(nt(),`.cache`,`huggingface`,`transformers-js`,t);try{await r(n,{recursive:!0,force:!0}),Y.info(`Auto-heal: cleared embedding model cache`,{path:n})}catch{}}if(n.includes(`lance`)||n.includes(`database`)||n.includes(`store`)){let t=q(e.store.path,`lance`);try{await r(t,{recursive:!0,force:!0}),Y.info(`Auto-heal: cleared LanceDB store`,{path:t})}catch{}}if(n.includes(`sqlite`)||n.includes(`database disk image`)||n.includes(`graph`)){let t=q(e.store.path,`graph.db`);try{await r(t,{force:!0}),Y.info(`Auto-heal: cleared graph database`,{path:t})}catch{}}}async function X(e){Y.info(`Initializing AI Kit components`);let[t,n,i,a]=await Promise.all([(async()=>{let t=new it({model:e.embedding.model,dimensions:e.embedding.dimensions});return await t.initialize(),Y.info(`Embedder loaded`,{modelId:t.modelId,dimensions:t.dimensions}),t})(),(async()=>{let t=await ut({backend:e.store.backend,path:e.store.path});return await t.initialize(),Y.info(`Store initialized`),t})(),(async()=>{let t=new lt({path:e.store.path});return await t.initialize(),Y.info(`Graph store initialized`),t})(),(async()=>{let e=await rt();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 ct(t,n),s=new st(e.store.path);s.load(),o.setHashCache(s);let c=e.curated.path,l=new r(c,n,t);o.setGraphStore(i);let u=E(e.er),d=u?new ot(e.curated.path):void 0;d&&Y.info(`Policy store initialized`,{ruleCount:d.getRules().length});let f=u?new at:void 0,p=Qe(e.sources[0]?.path??process.cwd(),$e.aiContext),m=Xe(p),h=e.onboardDir?Xe(e.onboardDir):!1,g=m||h,_,v=m?p:e.onboardDir;if(g&&v)try{_=Ze(v).mtime.toISOString()}catch{}return Y.info(`Onboard state detected`,{onboardComplete:g,onboardTimestamp:_,aiKbExists:m,onboardDirExists:h}),{embedder:t,store:n,indexer:o,curated:l,graphStore:i,fileCache:new dt,bridge:u,policyStore:d,evolutionCollector:f,onboardComplete:g,onboardTimestamp:_}}function _t(e,t){let n=new pt({name:t.serverName??`aikit`,version:Ie()},{capabilities:{logging:{},completions:{},prompts:{}}});return s(n),g(n,t.toolPrefix??``),Z(n,e,t,i(n),new d(n),p(n)),l(n,{curated:e.curated,store:e.store,graphStore:e.graphStore},t.indexMode),n}function Z(e,t,r,i,a,o,s,c){u(e),m(e),n(e),Fe(e,t.embedder,t.store,t.graphStore,t.bridge,t.evolutionCollector,o),ge(e,t.store);let l={onboardComplete:t.onboardComplete,onboardTimestamp:t.onboardTimestamp};Re(e,t.store,t.graphStore,t.curated,l,r,s,c),k(e,r),je(e,t.indexer,r,t.curated,t.store,a,s),Me(e,t.curated,t.policyStore,t.evolutionCollector,a),ze(e,t.curated,a),fe(e,t.curated,a),Ae(e,t.curated),he(e,t.curated),S(e,t.store,t.embedder),y(e,t.store,t.embedder),C(e,t.store,t.embedder),te(e,t.store,t.embedder),x(e,t.store,t.embedder),b(e,t.store,t.embedder),w(e,t.store,t.embedder,t.graphStore),ke(e,r),xe(e,t.store,t.embedder,r,l),pe(e,t.graphStore),ne(e,t.store,t.embedder);let d=r.sources[0]?.path??process.cwd();A(e,t.embedder,t.fileCache,d),N(e,t.embedder,t.store),ie(e,t.embedder,t.store),R(e),Ee(e),F(e),se(e,t.embedder,t.store),P(e,t.embedder,t.store,t.graphStore),L(e),ce(e),Te(e),ye(e),ve(e),be(e),_e(e),Pe(e),M(e,t.fileCache,d),Se(e),K(e),ae(e,t.embedder,t.store,t.graphStore),W(e),G(e),j(e,t.embedder,t.store),I(e,o),U(e),Ce(e),we(e),me(e),H(e,s),B(e),le(e,t.embedder),V(e),de(e,t.embedder,t.fileCache),ue(e,t.embedder,t.store),Oe(e,i),i&&T(e,i),Ye(e),Ue(e),Ge(e),Ve(e),We(e),Be(e),Ke(e),qe(e),He(e),Je(e),z(e,r),t.bridge&&(O(e,t.bridge,t.evolutionCollector),D(e,t.bridge),re(e,t.bridge)),t.policyStore&&De(e,t.policyStore),t.evolutionCollector&&oe(e,t.evolutionCollector),f(e,t.store,t.curated),Ne(e)}async function vt(e){let t=await X(e),n=_t(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 Promise.all([t.embedder.shutdown().catch(()=>{}),t.graphStore.close().catch(()=>{}),t.store.close().catch(()=>{})]),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 yt=new Set(`batch.brainstorm.changelog.check.checkpoint.codemod.compact.config.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(`.`)),bt=5e3,Q=new Set(`brainstorm.changelog.check.checkpoint.codemod.data_transform.delegate.diff_parse.encode.env.eval.evidence_map.flow_info.flow_list.flow_reset.flow_start.flow_status.flow_step.flow_add.flow_update.flow_remove.flow_read_instruction.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 xt(e,t){F(e),L(e),ce(e),R(e),I(e),ye(e),ve(e),be(e),_e(e),K(e),Ee(e),Te(e),Se(e),Pe(e),Ce(e),we(e),U(e),W(e),G(e),me(e),H(e),B(e),V(e),Oe(e),T(e,a),ke(e),Ne(e),Le(e),z(e,t),Ye(e),Ue(e),Ge(e),Ve(e),We(e),Be(e),Ke(e),qe(e),He(e),Je(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.config.data_transform.dead_symbols.delegate.diff_parse.digest.encode.env.eval.evidence_map.file_summary.find.flow_info.flow_list.flow_reset.flow_start.flow_status.flow_step.flow_add.flow_update.flow_remove.flow_read_instruction.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 St(n,r){let a=new pt({name:n.serverName??`aikit`,version:Ie()},{capabilities:{logging:{},completions:{},prompts:{}}}),u=`initializing`,f=``,m=!1,y=null,b=null,x=null;function te(e){if(!e||typeof e!=`object`)return[];let t=e,n=[];for(let e of[`path`,`file`,`source_path`,`sourcePath`,`filePath`]){let r=t[e];typeof r==`string`&&r&&n.push(r)}for(let e of[`changed_files`,`paths`,`files`]){let r=t[e];if(Array.isArray(r))for(let e of r){if(typeof e==`string`){n.push(e);continue}e&&typeof e==`object`&&typeof e.path==`string`&&n.push(e.path)}}if(Array.isArray(t.sources))for(let e of t.sources)e&&typeof e==`object`&&typeof e.path==`string`&&n.push(e.path);return n}let S=()=>u===`failed`?[`❌ AI Kit initialization failed — this tool is unavailable.`,``,f?`Error: ${f}`:``,``,`**${Q.size} tools are still available** and fully functional:`,`check, eval, test_run, git_context, health, measure, web_fetch, web_search,`,`flow_list, flow_status, flow_start, flow_read_instruction, regex_test, encode,`,`stash, checkpoint, lane, process, time, env, and more.`,``,`To fix embedding errors, try deleting the cached model:`,`  rm -rf ~/.cache/huggingface/transformers-js/mixedbread-ai/`,`Then restart the server to re-download a fresh copy.`,``,`Try restarting the MCP server to retry initialization.`].filter(Boolean).join(`
-`):[`AI Kit is still initializing (loading embeddings model & store).`,``,`**${Q.size} tools are already available** while initialization completes — including:`,`check, eval, test_run, git_context, health, measure, web_fetch, web_search,`,`flow_list, flow_status, flow_start, flow_read_instruction, regex_test, encode,`,`stash, checkpoint, lane, process, time, env, and more.`,``,`This tool requires the AI Kit index. Please retry in a few seconds,`,`or use one of the available tools above in the meantime.`].join(`
-`);s(a),g(a,n.toolPrefix??``);let C=a.sendToolListChanged.bind(a);a.sendToolListChanged=()=>{};let w=[];for(let e of $){let t=h(e),n=a.registerTool(e,{title:t.title,description:`${t.title} — initializing, available shortly`,inputSchema:{},annotations:t.annotations},async()=>({content:[{type:`text`,text:S()}]}));Q.has(e)?n.remove():w.push(n)}xt(a,n),a.sendToolListChanged=C;let ne=a.registerResource(`aikit-status`,`aikit://status`,{description:`AI Kit status (initializing...)`,mimeType:`text/plain`},async()=>({contents:[{uri:`aikit://status`,text:`AI Kit is initializing...`,mimeType:`text/plain`}]})),T=a.registerPrompt(`_init`,{description:`Initializing AI Kit…`,argsSchema:{_dummy:ft(mt.string(),()=>[])}},async()=>({messages:[]})),E,D=new Promise(e=>{E=e}),O,re=new Promise(e=>{O=e}),k=()=>O?.(),A=(async()=>{await re;let e;try{e=await X(n)}catch(t){let r=t instanceof Error?t.message:String(t);if(ht(r)){Y.warn(`AI Kit initialization failed with recoverable error — attempting auto-heal retry`,{error:r}),await gt(n,r);try{e=await X(n),Y.info(`AI Kit auto-heal successful — initialization recovered after retry`)}catch(e){u=`failed`,f=e instanceof Error?e.message:String(e),Y.error(`AI Kit initialization failed after auto-heal attempt — server continuing with zero-dep tools only`,{error:f,originalError:r});return}}else{u=`failed`,f=r,Y.error(`AI Kit initialization failed — server continuing with zero-dep tools only`,{error:f});return}}let s=a.sendToolListChanged.bind(a);a.sendToolListChanged=()=>{};let h=a.sendPromptListChanged.bind(a);a.sendPromptListChanged=()=>{};let g=a.sendResourceListChanged.bind(a);a.sendResourceListChanged=()=>{};for(let e of w)e.remove();ne.remove(),T.remove();let S=a._registeredTools??{};for(let e of Q)S[e]?.remove();let C=new d(a),D=p(a);Z(a,e,n,i(a),C,D,r,r===`smart`?(()=>{let e=x;return e?.getState?e.getState():null}):null),l(a,{curated:e.curated,store:e.store,graphStore:e.graphStore},r),a.sendToolListChanged=s,a.sendPromptListChanged=h,a.sendResourceListChanged=g,Promise.resolve(a.sendToolListChanged()).catch(()=>{}),Promise.resolve(a.sendPromptListChanged()).catch(()=>{}),Promise.resolve(a.sendResourceListChanged()).catch(()=>{});let O=a._registeredTools??{};for(let[t,n]of Object.entries(O)){if(yt.has(t))continue;let r=n.handler;n.handler=async(...n)=>{if(!e.indexer.isIndexing)return r(...n);let i=m?`re-indexing`:`running initial index`,a=new Promise(e=>setTimeout(()=>e({content:[{type:`text`,text:`⏳ AI Kit is ${i}. The tool "${t}" timed out waiting for index data (${bt/1e3}s).\n\nThe existing index may be temporarily locked. Please retry shortly — indexing will complete automatically.`}]}),bt));return Promise.race([r(...n),a])}}for(let[e,t]of Object.entries(O)){let n=t.handler,r=v(e);t.handler=async(...t)=>{try{return await ee(()=>n(...t),r,e)}catch(t){if(t instanceof _)return{content:[{type:`text`,text:`⏳ Tool "${e}" timed out after ${r/1e3}s. This may indicate a long-running operation. Please retry or break the task into smaller steps.`}]};throw t}}}let k=Object.keys(O).length;k<$.length&&Y.warn(`ALL_TOOL_NAMES count mismatch`,{expectedToolCount:$.length,registeredToolCount:k}),Y.info(`MCP server configured`,{toolCount:$.length,resourceCount:4});let A=new c;A.onPressure((e,n)=>{e===`warning`&&t(),e===`critical`&&(Y.warn(`Memory pressure critical — consider restarting`,{rssMB:Math.round(n/1024/1024)}),t())}),A.start();let j=new o;b=j,j.onIdle(async()=>{if(M.isRunning||e.indexer.isIndexing){Y.info(`Idle cleanup deferred — background tasks still running`),j.touch();return}Y.info(`Idle cleanup: closing store and graph connections`);try{await Promise.all([e.store.close().catch(()=>{}),e.graphStore.close().catch(()=>{})])}catch{}}),j.touch();for(let e of Object.values(O)){let t=e.handler;e.handler=async(...e)=>{if(j.touch(),x){let t=te(e[0]);t.length>0&&x.prioritize(...t)}return t(...e)}}y=e,E?.(e)})(),j=async()=>{let e=await D;b?.setBusy(!0);try{let t=n.sources.map(e=>e.path).join(`, `);Y.info(`Running initial index`,{sourcePaths:t});let r=await e.indexer.index(n,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}))});m=!0,Y.info(`Initial index complete`,{filesProcessed:r.filesProcessed,filesSkipped:r.filesSkipped,chunksCreated:r.chunksCreated,durationMs:r.durationMs});try{await e.store.createFtsIndex()}catch(e){Y.warn(`FTS index creation failed`,J(e))}try{let t=await e.curated.reindexAll();Y.info(`Curated re-index complete`,{indexed:t.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))}finally{b?.setBusy(!1)}},M=new e,ie=()=>M.schedule({name:`initial-index`,fn:j}),N=process.ppid,P=setInterval(()=>{try{process.kill(N,0)}catch{Y.info(`Parent process died; shutting down`,{parentPid:N}),clearInterval(P),D.then(async e=>{await Promise.all([e.embedder.shutdown().catch(()=>{}),e.graphStore.close().catch(()=>{}),e.store.close().catch(()=>{})])}).catch(()=>{}).finally(()=>process.exit(0))}},5e3);return P.unref(),{server:a,startInit:k,ready:A,runInitialIndex:ie,get kb(){return y},scheduler:M,setSmartScheduler(e){x=e}}}export{$ as ALL_TOOL_NAMES,St as createLazyServer,_t as createMcpServer,vt as createServer,X as initializeKnowledgeBase,Z as registerMcpTools};