kibi-mcp 0.2.3 → 0.2.4

package/dist/tools/query.js

@@ -68,13 +68,15 @@ export async function handleKbQuery(prolog, args) {
             goal = `findall(['${safeId}',Type,Props], kb_entity('${safeId}', Type, Props), Results)`;
         }
         else if (tags && tags.length > 0) {
-            const tagList = `[${tags.map((t) => `'${escapeAtomContent(t)}'`).join(",")}]`;
+            // TODO: Reintroduce server-side (Prolog) tag filtering once normalization
+            // issues with tag list formats are resolved, to avoid fetching all entities
+            // before filtering in JS for large knowledge bases.
             if (type) {
                 const safeType = escapeAtomContent(type);
-                goal = `findall([Id,'${safeType}',Props], (kb_entity(Id, '${safeType}', Props), memberchk(tags=Tags, Props), member(Tag, Tags), member(Tag, ${tagList})), Results)`;
+                goal = `findall([Id,'${safeType}',Props], kb_entity(Id, '${safeType}', Props), Results)`;
             }
             else {
-                goal = `findall([Id,Type,Props], (kb_entity(Id, Type, Props), memberchk(tags=Tags, Props), member(Tag, Tags), member(Tag, ${tagList})), Results)`;
+                goal = "findall([Id,Type,Props], kb_entity(Id, Type, Props), Results)";
             }
         }
         else if (type) {
@@ -101,6 +103,9 @@ export async function handleKbQuery(prolog, args) {
         else {
             throw new Error(queryResult.error || "Query failed with unknown error");
         }
+        if (tags && tags.length > 0) {
+            results = dedupeEntities(results.filter((entity) => hasAnyTag(entity, tags)));
+        }
         // Apply pagination
         const paginated = results.slice(offset, offset + limit);
         // Build human-readable text with entity IDs and titles
@@ -138,6 +143,37 @@ export async function handleKbQuery(prolog, args) {
         throw new Error(`Query execution failed: ${message}`);
     }
 }
+function hasAnyTag(entity, requestedTags) {
+    const expected = new Set(requestedTags.map(normalizeTagValue));
+    const rawTags = entity.tags;
+    if (!Array.isArray(rawTags) || rawTags.length === 0) {
+        return false;
+    }
+    for (const tag of rawTags) {
+        if (expected.has(normalizeTagValue(tag))) {
+            return true;
+        }
+    }
+    return false;
+}
+function normalizeTagValue(tag) {
+    return String(tag).trim();
+}
+function dedupeEntities(entities) {
+    const seen = new Set();
+    const deduped = [];
+    for (const entity of entities) {
+        const id = String(entity.id ?? "");
+        const type = String(entity.type ?? "");
+        const key = `${type}::${id}`;
+        if (seen.has(key)) {
+            continue;
+        }
+        seen.add(key);
+        deduped.push(entity);
+    }
+    return deduped;
+}
 /**
  * Parse a Prolog list of lists into a JavaScript array.
  * Input: "[[a,b,c],[d,e,f]]"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kibi-mcp",
-  "version": "0.2.3",
+  "version": "0.2.4",
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.26.0",
     "ajv": "^8.18.0",
@@ -10,7 +10,7 @@
     "gray-matter": "^4.0.3",
     "js-yaml": "^4.1.0",
     "kibi-cli": "^0.2.3",
-    "kibi-core": "^0.1.7",
+    "kibi-core": "^0.1.8",
     "mcpcat": "^0.1.12",
     "ts-morph": "^23.0.0",
     "zod": "^4.3.6"