@hiveai/mcp 0.2.9 → 0.2.11

package/dist/index.js

@@ -526,10 +526,14 @@ async function memForFiles(input, ctx) {
       seen.add(loaded.memory.frontmatter.id);
     }
   }
+  const pathSegments = extractPathSegments(input.files);
   for (const loaded of all) {
     if (seen.has(loaded.memory.frontmatter.id)) continue;
     const fm = loaded.memory.frontmatter;
-    const moduleHit = fm.module && inferred.includes(fm.module) || fm.tags.some((t) => inferred.includes(t));
+    const moduleHit = fm.module && inferred.includes(fm.module) || fm.tags.some((t) => inferred.includes(t)) || fm.tags.some((t) => {
+      const tl = t.toLowerCase();
+      return pathSegments.has(tl) || pathSegments.has(tl.replace(/[-_]/g, ""));
+    });
     if (moduleHit) {
       byModule.push(toMatch(loaded, "module", usage));
       seen.add(fm.id);
@@ -571,6 +575,52 @@ function toMatch(loaded, reason, usage) {
     body: loaded.memory.body
   };
 }
+function extractPathSegments(files) {
+  const GENERIC = /* @__PURE__ */ new Set([
+    "src",
+    "main",
+    "java",
+    "kotlin",
+    "python",
+    "go",
+    "lib",
+    "libs",
+    "com",
+    "org",
+    "net",
+    "io",
+    "app",
+    "apps",
+    "pkg",
+    "internal",
+    "test",
+    "tests",
+    "spec",
+    "specs",
+    "impl",
+    "domain",
+    "shared",
+    "resources",
+    "static",
+    "assets",
+    "config",
+    "configs"
+  ]);
+  const out = /* @__PURE__ */ new Set();
+  for (const file of files) {
+    const parts = file.replace(/\\/g, "/").split("/");
+    for (const part of parts) {
+      const seg = part.toLowerCase().replace(/\.[^.]+$/, "");
+      if (seg.length >= 3 && !GENERIC.has(seg) && /^[a-z]/.test(seg)) {
+        out.add(seg);
+        for (const sub of seg.split(/[-_]/).filter((s) => s.length >= 3)) {
+          out.add(sub);
+        }
+      }
+    }
+  }
+  return out;
+}
 async function loadModuleContexts(ctx, modules, enabled) {
   if (!enabled || modules.length === 0) return [];
   if (!existsSync8(ctx.paths.modulesContextDir)) return [];
@@ -850,10 +900,62 @@ async function memTried(input, ctx) {
   return { id: frontmatter.id, scope: frontmatter.scope, file_path: file };
 }
 
-// src/tools/get-briefing.ts
-import { readFile as readFile3, readdir as readdir3 } from "fs/promises";
+// src/tools/mem-observe.ts
+import { mkdir as mkdir4, writeFile as writeFile8 } from "fs/promises";
 import { existsSync as existsSync15 } from "fs";
 import path6 from "path";
+import {
+  buildFrontmatter as buildFrontmatter3,
+  memoryFilePath as memoryFilePath3,
+  serializeMemory as serializeMemory7
+} from "@hiveai/core";
+import { z as z15 } from "zod";
+var MemObserveInputSchema = {
+  what: z15.string().min(1).describe("Short title: what did you observe? (e.g. 'MobilePaymentController has two @RequestBody on handleWebhook')"),
+  where: z15.string().min(1).describe("File path(s) where the issue lives \u2014 be specific"),
+  impact: z15.string().min(1).describe("What breaks or could break because of this (e.g. 'Spring MVC rejects the handler at startup')"),
+  fix: z15.string().optional().describe("Suggested fix or workaround (optional \u2014 leave empty if unknown)"),
+  scope: z15.enum(["personal", "team", "module"]).default("team").describe("Visibility scope \u2014 defaults to team since discoveries benefit everyone"),
+  module: z15.string().optional().describe("Module name (required when scope=module)"),
+  tags: z15.array(z15.string()).default([]).describe("Tags for filtering"),
+  author: z15.string().optional().describe("Author handle or email")
+};
+async function memObserve(input, ctx) {
+  if (!existsSync15(ctx.paths.haiveDir)) {
+    throw new Error(`No .ai/ directory at ${ctx.paths.root}. Run 'haive init' first.`);
+  }
+  const slug = input.what.toLowerCase().replace(/[^a-z0-9\s]/g, "").trim().split(/\s+/).slice(0, 6).join("-");
+  const anchorPaths = input.where.split(/[,\n]/).map((s) => s.trim()).filter(Boolean);
+  const baseFm = buildFrontmatter3({
+    type: "gotcha",
+    slug,
+    scope: input.scope,
+    module: input.module,
+    tags: input.tags,
+    paths: anchorPaths,
+    author: input.author
+  });
+  const frontmatter = { ...baseFm, status: "validated" };
+  const lines = [`# ${input.what}`, ""];
+  lines.push(`**Where:** \`${input.where}\``);
+  lines.push("", `**Impact:** ${input.impact}`);
+  if (input.fix) {
+    lines.push("", `**Fix/workaround:** ${input.fix}`);
+  }
+  const body = lines.join("\n") + "\n";
+  const file = memoryFilePath3(ctx.paths, frontmatter.scope, frontmatter.id, frontmatter.module);
+  await mkdir4(path6.dirname(file), { recursive: true });
+  if (existsSync15(file)) {
+    throw new Error(`Memory already exists at ${file}`);
+  }
+  await writeFile8(file, serializeMemory7({ frontmatter, body }), "utf8");
+  return { id: frontmatter.id, scope: frontmatter.scope, file_path: file };
+}
+
+// src/tools/get-briefing.ts
+import { readFile as readFile3, readdir as readdir3 } from "fs/promises";
+import { existsSync as existsSync16 } from "fs";
+import path7 from "path";
 import {
   allocateBudget,
   deriveConfidence as deriveConfidence4,
@@ -869,24 +971,24 @@ import {
   trackReads as trackReads3,
   truncateToTokens
 } from "@hiveai/core";
-import { z as z15 } from "zod";
+import { z as z16 } from "zod";
 var GetBriefingInputSchema = {
-  task: z15.string().optional().describe(
+  task: z16.string().optional().describe(
     "What you are about to do, in 1\u20132 sentences. Used to rank relevant memories semantically."
   ),
-  files: z15.array(z15.string()).default([]).describe("Project-relative file paths the agent is currently looking at or about to edit"),
-  max_tokens: z15.number().int().positive().default(8e3).describe(
+  files: z16.array(z16.string()).default([]).describe("Project-relative file paths the agent is currently looking at or about to edit"),
+  max_tokens: z16.number().int().positive().default(8e3).describe(
     "Approximate token budget for the entire briefing. Each section is allocated a share and truncated to fit."
   ),
-  max_memories: z15.number().int().positive().default(8).describe("Cap on memories surfaced regardless of token budget"),
-  include_project_context: z15.boolean().default(true),
-  include_module_contexts: z15.boolean().default(true),
-  semantic: z15.boolean().default(true).describe(
+  max_memories: z16.number().int().positive().default(8).describe("Cap on memories surfaced regardless of token budget"),
+  include_project_context: z16.boolean().default(true),
+  include_module_contexts: z16.boolean().default(true),
+  semantic: z16.boolean().default(true).describe(
     "Use semantic ranking when a task is provided (requires `haive embeddings index`)."
   ),
-  include_stale: z15.boolean().default(false).describe("Include stale memories (excluded by default \u2014 they may be outdated)"),
-  track: z15.boolean().default(true).describe("Increment read_count on returned memories"),
-  format: z15.enum(["full", "compact"]).default("full").describe(
+  include_stale: z16.boolean().default(false).describe("Include stale memories (excluded by default \u2014 they may be outdated)"),
+  track: z16.boolean().default(true).describe("Increment read_count on returned memories"),
+  format: z16.enum(["full", "compact"]).default("full").describe(
     "Output format: 'full' returns complete memory bodies; 'compact' returns id + 1-line summary only (call mem_get for details)."
   )
 };
@@ -896,7 +998,7 @@ async function getBriefing(input, ctx) {
   let searchMode = "literal";
   let usage = { version: 1, updated_at: "", by_id: {} };
   let byId = /* @__PURE__ */ new Map();
-  if (existsSync15(ctx.paths.memoriesDir)) {
+  if (existsSync16(ctx.paths.memoriesDir)) {
     const allLoaded = await loadMemoriesFromDir12(ctx.paths.memoriesDir);
     const allMemories = allLoaded.filter(({ memory }) => {
       const s = memory.frontmatter.status;
@@ -991,7 +1093,7 @@ async function getBriefing(input, ctx) {
       await trackReads3(ctx.paths, memories.map((m) => m.id));
     }
   }
-  const projectContext = input.include_project_context && existsSync15(ctx.paths.projectContext) ? await readFile3(ctx.paths.projectContext, "utf8") : "";
+  const projectContext = input.include_project_context && existsSync16(ctx.paths.projectContext) ? await readFile3(ctx.paths.projectContext, "utf8") : "";
   const moduleContents = input.include_module_contexts ? await loadModuleContexts2(ctx, inferred) : [];
   const memoriesText = memories.map((m) => {
     const unverified = m.status === "proposed" ? " [UNVERIFIED \u2014 not yet validated]" : "";
@@ -1092,15 +1194,15 @@ async function trySemanticHits(ctx, task, limit) {
 }
 async function loadModuleContexts2(ctx, modules) {
   if (modules.length === 0) return [];
-  if (!existsSync15(ctx.paths.modulesContextDir)) return [];
+  if (!existsSync16(ctx.paths.modulesContextDir)) return [];
   const available = new Set(
     (await readdir3(ctx.paths.modulesContextDir, { withFileTypes: true })).filter((d) => d.isDirectory()).map((d) => d.name)
   );
   const out = [];
   for (const m of modules) {
     if (!available.has(m)) continue;
-    const file = path6.join(ctx.paths.modulesContextDir, m, "context.md");
-    if (existsSync15(file)) {
+    const file = path7.join(ctx.paths.modulesContextDir, m, "context.md");
+    if (existsSync16(file)) {
       out.push({ name: m, content: await readFile3(file, "utf8") });
     }
   }
@@ -1109,11 +1211,11 @@ async function loadModuleContexts2(ctx, modules) {
 
 // src/tools/code-map.ts
 import { loadCodeMap, queryCodeMap } from "@hiveai/core";
-import { z as z16 } from "zod";
+import { z as z17 } from "zod";
 var CodeMapInputSchema = {
-  file: z16.string().optional().describe("Filter to files whose path contains this substring"),
-  symbol: z16.string().optional().describe("Filter to files exporting a symbol whose name contains this substring"),
-  max_files: z16.number().int().positive().default(40).describe("Cap on returned files")
+  file: z17.string().optional().describe("Filter to files whose path contains this substring"),
+  symbol: z17.string().optional().describe("Filter to files exporting a symbol whose name contains this substring"),
+  max_files: z17.number().int().positive().default(40).describe("Cap on returned files")
 };
 async function codeMapTool(input, ctx) {
   const map = await loadCodeMap(ctx.paths);
@@ -1139,15 +1241,15 @@ async function codeMapTool(input, ctx) {
 }
 
 // src/tools/mem-diff.ts
-import { existsSync as existsSync16 } from "fs";
+import { existsSync as existsSync17 } from "fs";
 import { loadMemoriesFromDir as loadMemoriesFromDir13 } from "@hiveai/core";
-import { z as z17 } from "zod";
+import { z as z18 } from "zod";
 var MemDiffInputSchema = {
-  id_a: z17.string().min(1).describe("First memory id"),
-  id_b: z17.string().min(1).describe("Second memory id")
+  id_a: z18.string().min(1).describe("First memory id"),
+  id_b: z18.string().min(1).describe("Second memory id")
 };
 async function memDiff(input, ctx) {
-  if (!existsSync16(ctx.paths.memoriesDir)) {
+  if (!existsSync17(ctx.paths.memoriesDir)) {
     throw new Error(`No .ai/memories at ${ctx.paths.root}.`);
   }
   const all = await loadMemoriesFromDir13(ctx.paths.memoriesDir);
@@ -1184,12 +1286,12 @@ async function memDiff(input, ctx) {
 }
 
 // src/prompts/bootstrap-project.ts
-import { z as z18 } from "zod";
+import { z as z19 } from "zod";
 var BootstrapProjectArgsSchema = {
-  module: z18.string().optional().describe(
+  module: z19.string().optional().describe(
     "Optional module name to scope the analysis to (writes to .ai/modules/<module>/context.md)"
   ),
-  focus: z18.string().optional().describe("Optional area to emphasize (e.g. 'data layer', 'API surface')")
+  focus: z19.string().optional().describe("Optional area to emphasize (e.g. 'data layer', 'API surface')")
 };
 var ROOT_TEMPLATE = `# Project context
 
@@ -1271,10 +1373,10 @@ ${template}\`\`\`
 }
 
 // src/prompts/post-task.ts
-import { z as z19 } from "zod";
+import { z as z20 } from "zod";
 var PostTaskArgsSchema = {
-  task_summary: z19.string().optional().describe("One sentence describing what you just did"),
-  files_touched: z19.array(z19.string()).optional().describe("Files you created or modified during the task")
+  task_summary: z20.string().optional().describe("One sentence describing what you just did"),
+  files_touched: z20.array(z20.string()).optional().describe("Files you created or modified during the task")
 };
 function postTaskPrompt(args, ctx) {
   const taskLine = args.task_summary ? `
@@ -1290,6 +1392,19 @@ Project root: \`${ctx.paths.root}\`
 
 Go through each item. If the answer is yes, call the corresponding tool immediately.
 
+### 0. Did you read existing code and discover bugs, inconsistencies, or security gaps that weren't in the briefing?
+This is the most important question. Deep code reading surfaces issues that no memory captures yet.
+Examples of things to look for:
+- A method with an invalid signature (e.g. two \`@RequestBody\` on the same handler)
+- A configuration that looks wrong or missing (e.g. webhook path not whitelisted in SecurityConfig)
+- A component scan / DI issue (e.g. a Spring bean not picked up because the package isn't scanned)
+- A DB constraint that will break when you add a new enum value
+- A hardcoded value that should be dynamic (e.g. hardcoded tenant id "default-tenant")
+- Anything that will silently break in production
+
+\u2192 If yes, call **\`mem_save\`** with \`type="gotcha"\`, \`scope="team"\`, and **anchor it to the file** with \`paths\`.
+  This transforms your discovery into institutional knowledge that protects every future agent.
+
 ### 1. Did you try an approach that failed?
 \u2192 If yes, call **\`mem_tried\`** with:
   - \`what\`: the approach you tried (e.g. "importing gray-matter with ESM dynamic import")
@@ -1304,7 +1419,7 @@ Go through each item. If the answer is yes, call the corresponding tool immediat
 ### 3. Did you make an architectural decision?
 \u2192 If yes, call **\`mem_save\`** with \`type="decision"\` and document the WHY (constraints, tradeoffs), not just the what
 
-### 4. Did you hit a non-obvious bug or surprising behavior?
+### 4. Did you hit a non-obvious bug or surprising behavior in a library or framework?
 \u2192 If yes, call **\`mem_save\`** with \`type="gotcha"\` and anchor it to the relevant file paths
 
 ### 5. Did you find that an existing memory is outdated or wrong?
@@ -1316,6 +1431,7 @@ Go through each item. If the answer is yes, call the corresponding tool immediat
 - Anchor memories to file paths when possible (the \`paths\` field) \u2014 this enables staleness detection.
 - Prefer \`scope="team"\` for anything a teammate or future agent would benefit from.
 - Skip sections where you genuinely have nothing to add. Don't fabricate memories.
+- **Question 0 is not optional** \u2014 always scan your exploration history for code-level discoveries.
 
 When done, respond with a brief summary: "Saved N memories: [list of IDs]" or "Nothing new to save."
 `;
@@ -1331,12 +1447,12 @@ When done, respond with a brief summary: "Saved N memories: [list of IDs]" or "N
 }
 
 // src/prompts/import-docs.ts
-import { z as z20 } from "zod";
+import { z as z21 } from "zod";
 var ImportDocsArgsSchema = {
-  content: z20.string().describe("The documentation content to analyze and import as memories (Markdown, README, ADR, etc.)"),
-  source: z20.string().optional().describe("Origin of the content (file path, URL, or document title) \u2014 used to anchor memories"),
-  scope: z20.enum(["personal", "team"]).default("team").describe("Scope to assign to created memories"),
-  dry_run: z20.boolean().default(false).describe("If true, describe what would be saved without actually calling mem_save")
+  content: z21.string().describe("The documentation content to analyze and import as memories (Markdown, README, ADR, etc.)"),
+  source: z21.string().optional().describe("Origin of the content (file path, URL, or document title) \u2014 used to anchor memories"),
+  scope: z21.enum(["personal", "team"]).default("team").describe("Scope to assign to created memories"),
+  dry_run: z21.boolean().default(false).describe("If true, describe what would be saved without actually calling mem_save")
 };
 function importDocsPrompt(args, ctx) {
   const sourceLine = args.source ? `
@@ -1401,7 +1517,7 @@ When done, respond with: "Imported N memories: [list of IDs]" or "Nothing action
 
 // src/server.ts
 var SERVER_NAME = "haive";
-var SERVER_VERSION = "0.2.9";
+var SERVER_VERSION = "0.2.11";
 function jsonResult(data) {
   return {
     content: [
@@ -1520,6 +1636,12 @@ function createHaiveServer(options = {}) {
     MemDiffInputSchema,
     async (input) => jsonResult(await memDiff(input, context))
   );
+  server.tool(
+    "mem_observe",
+    "Capture a code-level discovery made during exploration: a bug, inconsistency, missing config, or security gap found by reading existing code that was NOT in the briefing. Use this whenever you read code and spot something that could silently break in production. Auto-validated, anchored to file paths for staleness detection. Prefer this over mem_save for reactive discoveries during code reading.",
+    MemObserveInputSchema,
+    async (input) => jsonResult(await memObserve(input, context))
+  );
   server.prompt(
     "bootstrap_project",
     "Instructions for the AI client to analyze the project and save the context.",