kibi-mcp 0.21.1 → 0.22.0

package/dist/server/kb-freshness.js

@@ -1,4 +1,4 @@
-import { stat } from "node:fs/promises";
+import { readFile, stat } from "node:fs/promises";
 import path from "node:path";
 export class KbRefreshError extends Error {
     constructor(message) {
@@ -21,7 +21,17 @@ export async function readBranchKbStamp(branchPath) {
         stamp.dirMissing = true;
         errors.push(formatStatError(branchPath, error));
     }
-    const rdfPath = path.join(branchPath, "kb.rdf");
+    const markerPath = path.join(branchPath, "storage.json");
+    let journaled = false;
+    try {
+        journaled = (await readFile(markerPath, "utf8")).includes("kibi.rdf-journal.v1");
+    }
+    catch {
+        journaled = false;
+    }
+    // Journaled stores expose CURRENT as their cheap, atomic freshness marker;
+    // legacy stores retain kb.rdf for compatibility.
+    const rdfPath = path.join(branchPath, journaled ? "CURRENT" : "kb.rdf");
     try {
         const rdfStat = await stat(rdfPath);
         stamp.rdfDev = rdfStat.dev;

package/dist/server/session.js

@@ -18,6 +18,7 @@
 import fs from "node:fs";
 import { createRequire } from "node:module";
 import process from "node:process";
+import { EngineClient } from "kibi-cli/engine";
 import { PrologProcess } from "kibi-cli/prolog";
 import { copyCleanSnapshot, getBranchDiagnostic, isValidBranchName, resolveActiveBranch, } from "kibi-cli/public/branch-resolver";
 import { getBranchOverride, isMcpDebugEnabled } from "../env.js";
@@ -202,6 +203,7 @@ function usesBranchKbPath(kbPath) {
 async function ensurePrologUnsafe() {
     const generationAtStart = prologResetGeneration;
     const workspaceRoot = sessionDeps.resolveWorkspaceRoot();
+    const useEngine = sessionDeps.PrologProcess === defaultSessionDeps.PrologProcess;
     const assertGeneration = async () => {
         if (generationAtStart !== prologResetGeneration) {
             const current = prologProcess;
@@ -236,6 +238,19 @@ async function ensurePrologUnsafe() {
         targetBranch = branchResult.branch;
     }
     // Check if we need to switch branches
+    if (isInitialized && useEngine && prologProcess?.isRunning()) {
+        const kbPath = sessionDeps.resolveKbPath(workspaceRoot, targetBranch);
+        if (targetBranch === activeBranchName) {
+            attachedBranchKbPath = kbPath;
+            attachedBranchStamp = await readBranchKbStamp(kbPath);
+            return prologProcess;
+        }
+        await prologProcess.terminate();
+        prologProcess = null;
+        isInitialized = false;
+        attachedBranchKbPath = null;
+        attachedBranchStamp = null;
+    }
     if (isInitialized && prologProcess?.isRunning()) {
         const kbPath = sessionDeps.resolveKbPath(workspaceRoot, targetBranch);
         if (targetBranch === activeBranchName) {
@@ -298,7 +313,13 @@ async function ensurePrologUnsafe() {
     }
     // First initialization
     debugLog("[KIBI-MCP] Initializing Prolog process...");
-    prologProcess = new sessionDeps.PrologProcess({ timeout: 120000 });
+    prologProcess = useEngine
+        ? new EngineClient({
+            workspaceRoot,
+            branch: targetBranch,
+            timeout: 120000,
+        })
+        : new sessionDeps.PrologProcess({ timeout: 120000 });
     await prologProcess.start();
     await assertGeneration();
     // Startup debug: resolve which kibi-cli is being used and its version (best-effort).
@@ -340,16 +361,18 @@ async function ensurePrologUnsafe() {
     activeBranchName = targetBranch;
     const createdEmptyBranch = ensureBranchKbExists(workspaceRoot, targetBranch);
     const kbPath = sessionDeps.resolveKbPath(workspaceRoot, targetBranch);
-    const attachResult = await prologProcess.query(`kb_attach('${kbPath}')`);
-    await assertGeneration();
-    if (!attachResult.success) {
-        throw new Error(`Failed to attach KB: ${attachResult.error || "Unknown error"}`);
-    }
-    if (createdEmptyBranch) {
-        const initialSaveResult = await prologProcess.query("kb_save");
+    if (!useEngine) {
+        const attachResult = await prologProcess.query(`kb_attach('${kbPath}')`);
         await assertGeneration();
-        if (!initialSaveResult.success) {
-            throw new Error(`Failed to initialize empty branch KB: ${initialSaveResult.error || "Unknown error"}`);
+        if (!attachResult.success) {
+            throw new Error(`Failed to attach KB: ${attachResult.error || "Unknown error"}`);
+        }
+        if (createdEmptyBranch) {
+            const initialSaveResult = await prologProcess.query("kb_save");
+            await assertGeneration();
+            if (!initialSaveResult.success) {
+                throw new Error(`Failed to initialize empty branch KB: ${initialSaveResult.error || "Unknown error"}`);
+            }
         }
     }
     attachedBranchKbPath = kbPath;

package/dist/server/tools-runtime.js

@@ -62,6 +62,30 @@ function adaptProlog(prolog) {
             return result;
         },
         save: () => prolog.query("kb_save"),
+        ...(typeof prolog.queryEntities ===
+            "function"
+            ? {
+                queryEntities: (input) => prolog.queryEntities(input),
+            }
+            : {}),
+        ...(typeof prolog.searchEntities ===
+            "function"
+            ? {
+                searchEntities: (input) => prolog.searchEntities(input),
+            }
+            : {}),
+        ...(typeof prolog.storageStatus ===
+            "function"
+            ? {
+                storageStatus: () => prolog.storageStatus(),
+            }
+            : {}),
+        ...(typeof prolog.queryStatusJson ===
+            "function"
+            ? {
+                queryStatusJson: () => prolog.queryStatusJson(),
+            }
+            : {}),
     };
     prologPorts.set(prolog, port);
     return port;

package/dist/tools/discovery-adapter.js

@@ -13,6 +13,12 @@ export function createDiscoveryContext(prolog, baseContext) {
             return result;
         },
         save: () => prolog.query("kb_save"),
+        ...(typeof prolog.queryStatusJson ===
+            "function"
+            ? {
+                queryStatusJson: () => prolog.queryStatusJson(),
+            }
+            : {}),
     };
     return {
         ...(baseContext ?? {}),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kibi-mcp",
-  "version": "0.21.1",
+  "version": "0.22.0",
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.26.0",
     "ajv": "^8.18.0",
@@ -9,8 +9,8 @@
     "fast-glob": "^3.2.12",
     "gray-matter": "^4.0.3",
     "js-yaml": "^4.1.0",
-    "kibi-cli": "^0.18.1",
-    "kibi-core": "^0.9.1",
+    "kibi-cli": "^0.19.0",
+    "kibi-core": "^0.10.0",
     "mcpcat": "^0.1.12",
     "ts-morph": "^23.0.0",
     "zod": "^4.3.6"
@@ -30,8 +30,7 @@
   },
   "files": ["dist", "bin"],
   "engines": {
-    "node": ">=18",
-    "bun": ">=1.0"
+    "node": ">=18"
   },
   "exports": {
     ".": "./dist/server.js"