@hiveai/mcp 0.9.6 → 0.9.8

package/dist/server.d.ts

@@ -167,16 +167,26 @@ interface BriefingOutput {
 }
 declare function getBriefing(input: GetBriefingInput, ctx: HaiveContext): Promise<BriefingOutput>;
 
-declare const CodeMapInputSchema: {
+declare const CodeMapInputZod: z.ZodObject<{
     file: z.ZodOptional<z.ZodString>;
     symbol: z.ZodOptional<z.ZodString>;
     paths: z.ZodDefault<z.ZodArray<z.ZodString, "many">>;
     max_files: z.ZodDefault<z.ZodNumber>;
     max_tokens: z.ZodOptional<z.ZodNumber>;
-};
-type CodeMapInput = {
-    [K in keyof typeof CodeMapInputSchema]: z.infer<(typeof CodeMapInputSchema)[K]>;
-};
+}, "strip", z.ZodTypeAny, {
+    paths: string[];
+    max_files: number;
+    symbol?: string | undefined;
+    max_tokens?: number | undefined;
+    file?: string | undefined;
+}, {
+    symbol?: string | undefined;
+    paths?: string[] | undefined;
+    max_tokens?: number | undefined;
+    file?: string | undefined;
+    max_files?: number | undefined;
+}>;
+type CodeMapInput = z.infer<typeof CodeMapInputZod>;
 interface CodeMapToolOutput {
     available: boolean;
     generated_at?: string;

package/dist/server.js

@@ -1910,6 +1910,7 @@ var CodeMapInputSchema = {
     "Approximate token budget for the response. When the matching set exceeds it, files are ranked by export density (exports per LOC) and the highest-signal ones are kept first. Omit to disable budgeting (legacy behavior)."
   )
 };
+var CodeMapInputZod = z18.object(CodeMapInputSchema);
 async function codeMapTool(input, ctx) {
   const map = await loadCodeMap2(ctx.paths);
   if (!map) {
@@ -3448,8 +3449,9 @@ When done, respond with: "Imported N memories: [list of IDs]" or "Nothing action
 }
 
 // src/server.ts
+import { loadConfigSync } from "@hiveai/core";
 var SERVER_NAME = "haive";
-var SERVER_VERSION = "0.9.6";
+var SERVER_VERSION = "0.9.8";
 function jsonResult(data) {
   return {
     content: [
@@ -3460,15 +3462,70 @@ function jsonResult(data) {
     ]
   };
 }
+var ENFORCEMENT_PROFILE_TOOLS = /* @__PURE__ */ new Set([
+  "get_briefing",
+  "mem_save",
+  "mem_tried",
+  "mem_search",
+  "mem_get",
+  "mem_update",
+  "mem_verify",
+  "mem_relevant_to",
+  "code_map",
+  "pre_commit_check"
+]);
+var BRIEFING_TOOLS = /* @__PURE__ */ new Set(["get_briefing", "mem_relevant_to"]);
+var MUTATING_TOOLS = /* @__PURE__ */ new Set([
+  "mem_save",
+  "mem_tried",
+  "mem_observe",
+  "mem_session_end",
+  "bootstrap_project_save",
+  "mem_update",
+  "mem_approve",
+  "mem_reject",
+  "mem_delete",
+  "runtime_journal_append",
+  "pattern_detect"
+]);
 function createHaiveServer(options = {}) {
   const context = createContext(options);
+  const config = loadConfigSync(context.paths);
+  const toolProfile = options.env?.HAIVE_TOOL_PROFILE ?? config.enforcement?.toolProfile ?? "enforcement";
+  const requireBriefingFirst = options.env?.HAIVE_REQUIRE_BRIEFING_FIRST === "0" ? false : config.enforcement?.requireBriefingFirst ?? true;
+  let briefingLoaded = false;
   const tracker = new SessionTracker(context);
   void tracker.init();
   const server = new McpServer(
     { name: SERVER_NAME, version: SERVER_VERSION },
     { capabilities: { tools: {}, prompts: {} } }
   );
-  server.tool(
+  const shouldRegisterTool = (name) => toolProfile === "full" || ENFORCEMENT_PROFILE_TOOLS.has(name);
+  const registerTool = (name, description, schema, handler) => {
+    if (!shouldRegisterTool(name)) return;
+    const tool = server.tool.bind(server);
+    tool(
+      name,
+      description,
+      schema,
+      async (input) => {
+        if (BRIEFING_TOOLS.has(name)) {
+          briefingLoaded = true;
+          return await handler(input);
+        }
+        if (requireBriefingFirst && MUTATING_TOOLS.has(name) && !briefingLoaded) {
+          return jsonResult({
+            error: "haive_briefing_required",
+            message: "This hAIve project requires get_briefing or mem_relevant_to before state-changing hAIve tools. Call get_briefing({ task: '...' }) first.",
+            tool: name
+          });
+        }
+        return await handler(input);
+      }
+    );
+  };
+  const shouldRegisterPrompt = (name) => toolProfile === "full" || name === "bootstrap_project" || name === "post_task";
+  registerTool(
     "mem_save",
     [
       "Save a piece of knowledge as a persistent memory that survives across AI sessions.",
@@ -3500,7 +3557,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memSave(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_suggest_topic",
     [
       "Propose a stable `topic` key (topic-upsert) from type + short title.",
@@ -3517,7 +3574,7 @@ function createHaiveServer(options = {}) {
     MemSuggestTopicInputSchema,
     async (input) => jsonResult(await memSuggestTopic(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_tried",
     [
       "Record a FAILED approach so future agents don't repeat the same mistake.",
@@ -3544,7 +3601,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memTried(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_observe",
     [
       "Capture a code-level discovery made WHILE READING existing code.",
@@ -3576,7 +3633,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memObserve(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_session_end",
     [
       "Save an end-of-session recap so the NEXT session starts with fresh context.",
@@ -3605,7 +3662,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memSessionEnd(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "get_briefing",
     [
       "\u2B50 DEFAULT-FIRST for coding agents on any repo where `haive init` ran: call this BEFORE",
@@ -3656,7 +3713,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await getBriefing(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_search",
     [
       "Search memories by keyword or semantic similarity.",
@@ -3688,7 +3745,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memSearch(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_timeline",
     [
       "Chronological view of related memories: by shared frontmatter.topic OR expanded from a seed id",
@@ -3704,7 +3761,7 @@ function createHaiveServer(options = {}) {
     MemTimelineInputSchema,
     async (input) => jsonResult(await memTimeline(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_for_files",
     [
       "Surface memories relevant to the files you are currently editing.",
@@ -3728,7 +3785,7 @@ function createHaiveServer(options = {}) {
     MemForFilesInputSchema,
     async (input) => jsonResult(await memForFiles(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_get",
     [
       "Fetch a single memory by its full id with all details.",
@@ -3744,7 +3801,7 @@ function createHaiveServer(options = {}) {
     MemGetInputSchema,
     async (input) => jsonResult(await memGet(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_list",
     [
       "List memories with optional filters. Use for browsing, not for task onboarding.",
@@ -3764,7 +3821,7 @@ function createHaiveServer(options = {}) {
     MemListInputSchema,
     async (input) => jsonResult(await memList(input, context))
   );
-  server.tool(
+  registerTool(
     "get_project_context",
     [
       "Read .ai/project-context.md (and optionally a module context) directly.",
@@ -3782,7 +3839,7 @@ function createHaiveServer(options = {}) {
     GetProjectContextInputSchema,
     async (input) => jsonResult(await getProjectContext(input, context))
   );
-  server.tool(
+  registerTool(
     "bootstrap_project_save",
     [
       "Persist the project context document (.ai/project-context.md) or a module",
@@ -3800,7 +3857,7 @@ function createHaiveServer(options = {}) {
     BootstrapProjectSaveInputSchema,
     async (input) => jsonResult(await bootstrapProjectSave(input, context))
   );
-  server.tool(
+  registerTool(
     "code_map",
     [
       "Look up where symbols (classes, functions, interfaces) are defined in the codebase.",
@@ -3821,7 +3878,7 @@ function createHaiveServer(options = {}) {
     CodeMapInputSchema,
     async (input) => jsonResult(await codeMapTool(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_resolve_project",
     [
       "Diagnostics: resolve which project root hAIve is using (never throws).",
@@ -3837,7 +3894,7 @@ function createHaiveServer(options = {}) {
     MemResolveProjectInputSchema,
     async (input) => jsonResult(await memResolveProject(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_update",
     [
       "Update the body, tags, or anchor of an existing memory in-place.",
@@ -3860,7 +3917,7 @@ function createHaiveServer(options = {}) {
     MemUpdateInputSchema,
     async (input) => jsonResult(await memUpdate(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_verify",
     [
       "Check whether memory anchor paths and symbols still exist in the current code.",
@@ -3879,7 +3936,7 @@ function createHaiveServer(options = {}) {
     MemVerifyInputSchema,
     async (input) => jsonResult(await memVerify(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_approve",
     [
       "Mark a memory as validated (trusted, approved by a human or the team).",
@@ -3895,7 +3952,7 @@ function createHaiveServer(options = {}) {
     MemApproveInputSchema,
     async (input) => jsonResult(await memApprove(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_reject",
     [
       "Mark a memory as rejected and record a reason.",
@@ -3913,7 +3970,7 @@ function createHaiveServer(options = {}) {
     MemRejectInputSchema,
     async (input) => jsonResult(await memReject(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_pending",
     [
       "List memories in 'proposed' status awaiting review, sorted by read count.",
@@ -3929,7 +3986,7 @@ function createHaiveServer(options = {}) {
     MemPendingInputSchema,
     async (input) => jsonResult(await memPending(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_delete",
     [
       "Permanently delete a memory by id.",
@@ -3946,7 +4003,7 @@ function createHaiveServer(options = {}) {
     MemDeleteInputSchema,
     async (input) => jsonResult(await memDelete(input, context))
   );
-  server.tool(
+  registerTool(
     "get_recap",
     [
       "Return ONLY the most recent session_recap. Cheaper than get_briefing when",
@@ -3964,7 +4021,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await getRecap(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_relevant_to",
     [
       "One-shot ranked memories for a task \u2014 use instead of get_briefing when",
@@ -3990,7 +4047,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memRelevantTo(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "code_search",
     [
       "Semantic search over the codebase \u2014 finds exported symbols (functions, classes,",
@@ -4013,7 +4070,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await codeSearch(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "why_this_file",
     [
       "One-shot file-context lookup: combines recent git history, memories anchored",
@@ -4033,7 +4090,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await whyThisFile(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "anti_patterns_check",
     [
       "Scan a diff (or set of paths) against documented attempt/gotcha memories.",
@@ -4056,7 +4113,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await antiPatternsCheck(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_distill",
     [
       "Cluster recurring observations / failed attempts so a human can collapse",
@@ -4081,7 +4138,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memDistill(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "why_this_decision",
     [
       "Trace the genealogy of a memory (especially decision/architecture):",
@@ -4103,7 +4160,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await whyThisDecision(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_conflicts_with",
     [
       "Detect memories that potentially CONTRADICT a given memory.",
@@ -4129,7 +4186,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memConflicts(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_conflict_candidates",
     [
       "Bulk scan for conflict CANDIDATES (not proof):",
@@ -4150,7 +4207,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memConflictCandidates(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "runtime_journal_append",
     [
       "Append one line to `.ai/.runtime/session-journal.ndjson` \u2014 machine-local session continuity.",
@@ -4164,7 +4221,7 @@ function createHaiveServer(options = {}) {
     RuntimeJournalAppendInputSchema,
     async (input) => jsonResult(await runtimeJournalAppend(input, context))
   );
-  server.tool(
+  registerTool(
     "runtime_journal_tail",
     [
       "Read the last N entries from the runtime session journal (parsed JSON lines).",
@@ -4176,7 +4233,7 @@ function createHaiveServer(options = {}) {
     RuntimeJournalTailInputSchema,
     async (input) => jsonResult(await runtimeJournalTail(input, context))
   );
-  server.tool(
+  registerTool(
     "pre_commit_check",
     [
       "One-shot 'should I block this commit?' check. Combines three signals:",
@@ -4201,7 +4258,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await preCommitCheck(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "pattern_detect",
     [
       "Heuristic memory detector \u2014 finds knowledge worth saving WITHOUT calling an LLM.",
@@ -4232,7 +4289,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await patternDetect(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_diff",
     [
       "Compare two memories side-by-side to decide if they should be merged.",
@@ -4249,40 +4306,46 @@ function createHaiveServer(options = {}) {
     MemDiffInputSchema,
     async (input) => jsonResult(await memDiff(input, context))
   );
-  server.prompt(
-    "bootstrap_project",
-    [
-      "Analyze the project codebase and write .ai/project-context.md \u2014 run once after haive init.",
-      "The AI explores the directory structure, reads key files (package.json, README, config),",
-      "identifies the tech stack, architectural patterns, key modules, and conventions,",
-      "then persists everything via bootstrap_project_save.",
-      "For multi-component projects, run with module param to create .ai/modules/<name>/context.md."
-    ].join(" "),
-    BootstrapProjectArgsSchema,
-    (args) => bootstrapProjectPrompt(args, context)
-  );
-  server.prompt(
-    "post_task",
-    [
-      "\u2B50 Post-task reflection \u2014 run at the end of every session to capture what you learned:",
-      "failed approaches (mem_tried), new conventions/decisions/gotchas (mem_save),",
-      "code discoveries (mem_observe), and an end-of-session recap (mem_session_end).",
-      "In autopilot mode a minimal recap saves automatically; calling this produces a richer one."
-    ].join(" "),
-    PostTaskArgsSchema,
-    (args) => postTaskPrompt(args, context)
-  );
-  server.prompt(
-    "import_docs",
-    [
-      "Import knowledge from a document (README, ADR, wiki, API spec) as hAIve memories.",
-      "Pass the full document content; the AI extracts up to 10 actionable memories",
-      "(conventions, decisions, gotchas, architecture) and saves them via mem_save.",
-      "Good candidates: ADRs, onboarding docs, runbooks, team wikis."
-    ].join(" "),
-    ImportDocsArgsSchema,
-    (args) => importDocsPrompt(args, context)
-  );
+  if (shouldRegisterPrompt("bootstrap_project")) {
+    server.prompt(
+      "bootstrap_project",
+      [
+        "Analyze the project codebase and write .ai/project-context.md \u2014 run once after haive init.",
+        "The AI explores the directory structure, reads key files (package.json, README, config),",
+        "identifies the tech stack, architectural patterns, key modules, and conventions,",
+        "then persists everything via bootstrap_project_save.",
+        "For multi-component projects, run with module param to create .ai/modules/<name>/context.md."
+      ].join(" "),
+      BootstrapProjectArgsSchema,
+      (args) => bootstrapProjectPrompt(args, context)
+    );
+  }
+  if (shouldRegisterPrompt("post_task")) {
+    server.prompt(
+      "post_task",
+      [
+        "\u2B50 Post-task reflection \u2014 run at the end of every session to capture what you learned:",
+        "failed approaches (mem_tried), new conventions/decisions/gotchas (mem_save),",
+        "code discoveries (mem_observe), and an end-of-session recap (mem_session_end).",
+        "In autopilot mode a minimal recap saves automatically; calling this produces a richer one."
+      ].join(" "),
+      PostTaskArgsSchema,
+      (args) => postTaskPrompt(args, context)
+    );
+  }
+  if (shouldRegisterPrompt("import_docs")) {
+    server.prompt(
+      "import_docs",
+      [
+        "Import knowledge from a document (README, ADR, wiki, API spec) as hAIve memories.",
+        "Pass the full document content; the AI extracts up to 10 actionable memories",
+        "(conventions, decisions, gotchas, architecture) and saves them via mem_save.",
+        "Good candidates: ADRs, onboarding docs, runbooks, team wikis."
+      ].join(" "),
+      ImportDocsArgsSchema,
+      (args) => importDocsPrompt(args, context)
+    );
+  }
   return { server, context, tracker };
 }
 function parseMcpCliArgs(argv) {
@@ -4307,7 +4370,7 @@ function printHaiveMcpVersion() {
   console.log(SERVER_VERSION);
 }
 async function runHaiveMcpStdio(options) {
-  const { server, context } = createHaiveServer({ root: options.root });
+  const { server, context } = createHaiveServer({ root: options.root, env: process.env });
   console.error(
     `[haive-mcp] starting server v${SERVER_VERSION} (project root: ${context.paths.root})`
   );