@wanshi-kg/wanshi 0.1.0

package/dist/core/export/strategies/LoraExportStrategy.js

@@ -0,0 +1,46 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LoraExportStrategy = void 0;
+const kbTriples_1 = require("./kbTriples");
+/**
+ * Harvests a graph into a LoRA / SFT fine-tuning set: chat-format instruction
+ * examples (Q → A) derived from the same `(name, property, value)` triples as the
+ * KBLaM export, but **quality-filtered** — observations whose `groundingScore`
+ * falls below the threshold are dropped *before* aggregation, so only grounded
+ * facts join an entity's `description`. Relations (structural, no per-fact score)
+ * are always included. One example per unique (name, property) key (KG-09), so no
+ * two examples share a question with conflicting answers.
+ *
+ * Output is JSONL of `{ messages: [{role:"user"}, {role:"assistant"}] }`.
+ */
+class LoraExportStrategy {
+    export(graph, options) {
+        var _a, _b;
+        const min = (_b = (_a = options === null || options === void 0 ? void 0 : options.grounding) === null || _a === void 0 ? void 0 : _a.minScore) !== null && _b !== void 0 ? _b : 0.5;
+        return (0, kbTriples_1.toKbTriples)(graph, (obs) => this.isGroundedEnough(obs, min))
+            .map((t) => this.example(t.name, t.property, t.value))
+            .join("\n");
+    }
+    getFormat() {
+        return "lora";
+    }
+    supportsFormat(format) {
+        return format === "lora";
+    }
+    /** Keep when there's no score (gate off) or the score meets the threshold. */
+    isGroundedEnough(obs, min) {
+        if (typeof obs === "string")
+            return true;
+        return obs.groundingScore === undefined || obs.groundingScore >= min;
+    }
+    example(name, property, value) {
+        return JSON.stringify({
+            messages: [
+                { role: "user", content: `What is the ${property} of ${name}?` },
+                { role: "assistant", content: `The ${property} of ${name} is ${value}.` },
+            ],
+        });
+    }
+}
+exports.LoraExportStrategy = LoraExportStrategy;
+//# sourceMappingURL=LoraExportStrategy.js.map

package/dist/core/export/strategies/LoraExportStrategy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LoraExportStrategy.js","sourceRoot":"","sources":["../../../../src/core/export/strategies/LoraExportStrategy.ts"],"names":[],"mappings":";;;AAIA,2CAA0C;AAE1C;;;;;;;;;;GAUG;AACH,MAAa,kBAAkB;IAC7B,MAAM,CAAC,KAAqB,EAAE,OAA2B;;QACvD,MAAM,GAAG,GAAG,MAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,0CAAE,QAAQ,mCAAI,GAAG,CAAC;QAChD,OAAO,IAAA,uBAAW,EAAC,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;aAChE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;aACrD,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,SAAS;QACP,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,OAAO,MAAM,KAAK,MAAM,CAAC;IAC3B,CAAC;IAED,8EAA8E;IACtE,gBAAgB,CAAC,GAAyB,EAAE,GAAW;QAC7D,IAAI,OAAO,GAAG,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC;QACzC,OAAO,GAAG,CAAC,cAAc,KAAK,SAAS,IAAI,GAAG,CAAC,cAAc,IAAI,GAAG,CAAC;IACvE,CAAC;IAEO,OAAO,CAAC,IAAY,EAAE,QAAgB,EAAE,KAAa;QAC3D,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,QAAQ,OAAO,IAAI,GAAG,EAAE;gBAChE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,QAAQ,OAAO,IAAI,OAAO,KAAK,GAAG,EAAE;aAC1E;SACF,CAAC,CAAC;IACL,CAAC;CACF;AA9BD,gDA8BC"}

package/dist/core/export/strategies/McpExportStrategy.js

@@ -0,0 +1,67 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.McpExportStrategy = void 0;
+const Observation_1 = require("../../../types/Observation");
+/**
+ * Strategy for handling MCP (Model Context Protocol) format conversions and exports
+ */
+class McpExportStrategy {
+    export(graph, processingOptions) {
+        const mcpGraph = this.toMCP(graph);
+        const lines = [];
+        mcpGraph.entities.forEach((entity) => {
+            lines.push(JSON.stringify(Object.assign({ type: "entity" }, entity)));
+        });
+        mcpGraph.relations.forEach((relation) => {
+            lines.push(JSON.stringify(Object.assign({ type: "relation" }, relation)));
+        });
+        return lines.join("\n");
+    }
+    getFormat() {
+        return "mcp-jsonl";
+    }
+    supportsFormat(format) {
+        return format === "mcp-jsonl";
+    }
+    /**
+     * Convert our KnowledgeGraph format to MCP format
+     */
+    toMCP(graph) {
+        return {
+            entities: graph.entities.map((entity) => ({
+                name: entity.name,
+                entityType: entity.entityType,
+                // MCP memory server stores observations as bare strings.
+                observations: (entity.observations || []).map(Observation_1.obsText),
+            })),
+            relations: graph.relations.map((relation) => ({
+                from: relation.from,
+                to: relation.to,
+                relationType: Array.isArray(relation.relationType)
+                    ? relation.relationType.join(",")
+                    : relation.relationType,
+            })),
+        };
+    }
+    /**
+     * Convert MCP format to our KnowledgeGraph format
+     */
+    static fromMCP(mcpGraph) {
+        return {
+            entities: mcpGraph.entities.map((entity) => ({
+                name: entity.name,
+                entityType: entity.entityType,
+                observations: (0, Observation_1.normalizeObservations)(entity.observations || []),
+            })),
+            relations: mcpGraph.relations.map((relation) => ({
+                from: relation.from,
+                to: relation.to,
+                relationType: relation.relationType.includes(",")
+                    ? relation.relationType.split(",").map((s) => s.trim())
+                    : relation.relationType,
+            })),
+        };
+    }
+}
+exports.McpExportStrategy = McpExportStrategy;
+//# sourceMappingURL=McpExportStrategy.js.map

package/dist/core/export/strategies/McpExportStrategy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpExportStrategy.js","sourceRoot":"","sources":["../../../../src/core/export/strategies/McpExportStrategy.ts"],"names":[],"mappings":";;;AACA,4DAA4E;AAK5E;;GAEG;AACH,MAAa,iBAAiB;IAC5B,MAAM,CAAC,KAAqB,EAAE,iBAAqC;QACjE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACnC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,iBAAG,IAAI,EAAE,QAAQ,IAAK,MAAM,EAAG,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YACtC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,iBAAG,IAAI,EAAE,UAAU,IAAK,QAAQ,EAAG,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED,SAAS;QACP,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,OAAO,MAAM,KAAK,WAAW,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAqB;QACzB,OAAO;YACL,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACxC,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,yDAAyD;gBACzD,YAAY,EAAE,CAAC,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,qBAAO,CAAC;aACvD,CAAC,CAAC;YACH,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAC5C,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;oBAChD,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC;oBACjC,CAAC,CAAC,QAAQ,CAAC,YAAY;aAC1B,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,OAAO,CAAC,QAA2B;QACxC,OAAO;YACL,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAC7B,CAAC,MAAM,EAAE,EAAE,CACT,CAAC;gBACC,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,YAAY,EAAE,IAAA,mCAAqB,EAAC,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC;aACpD,CAAA,CACf;YACD,SAAS,EAAE,QAAQ,CAAC,SAAS,CAAC,GAAG,CAC/B,CAAC,QAAQ,EAAE,EAAE,CACX,CAAC;gBACC,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,YAAY,EAAE,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC;oBAC/C,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACvD,CAAC,CAAC,QAAQ,CAAC,YAAY;aACb,CAAA,CACjB;SACF,CAAC;IACJ,CAAC;CACF;AAtED,8CAsEC"}

package/dist/core/export/strategies/index.js

@@ -0,0 +1,25 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./IExportStrategy"), exports);
+__exportStar(require("./JsonExportStrategy"), exports);
+__exportStar(require("./JsonlExportStrategy"), exports);
+__exportStar(require("./McpExportStrategy"), exports);
+__exportStar(require("./GraphvizDotExportStrategy"), exports);
+__exportStar(require("./KblamExportStrategy"), exports);
+__exportStar(require("./LoraExportStrategy"), exports);
+__exportStar(require("./GraphitiExportStrategy"), exports);
+//# sourceMappingURL=index.js.map

package/dist/core/export/strategies/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/core/export/strategies/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,oDAAkC;AAClC,uDAAqC;AACrC,wDAAsC;AACtC,sDAAoC;AACpC,8DAA4C;AAC5C,wDAAsC;AACtC,uDAAqC;AACrC,2DAAyC"}

package/dist/core/export/strategies/kbTriples.js

@@ -0,0 +1,60 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DESCRIPTION_PROPERTY = void 0;
+exports.toKbTriples = toKbTriples;
+const Observation_1 = require("../../../types/Observation");
+/** The property all free-text observations aggregate under (KBLaM-released name). */
+exports.DESCRIPTION_PROPERTY = "description";
+/**
+ * Turn a graph into KBLaM-shaped triples with unique `(name, property)` keys:
+ *  - **observations** → a single `description` property per entity (free-text facts
+ *    have no inherent property), values joined with `"; "`;
+ *  - **relations** → `property = predicate`, with same-`(from, predicate)` targets
+ *    aggregated into one `", "`-joined value.
+ * `keepObservation` lets a caller (LoRA) drop facts before aggregation; KBLaM passes
+ * none. Values are de-duplicated and emptiness-checked, so each key maps to exactly
+ * one non-empty value.
+ */
+function toKbTriples(graph, keepObservation) {
+    // name → property → values (insertion order preserved for deterministic output).
+    const byEntity = new Map();
+    const add = (name, property, value) => {
+        var _a;
+        const v = value.trim();
+        if (!v)
+            return;
+        let props = byEntity.get(name);
+        if (!props)
+            byEntity.set(name, (props = new Map()));
+        const values = (_a = props.get(property)) !== null && _a !== void 0 ? _a : [];
+        if (values.length === 0)
+            props.set(property, values);
+        values.push(v);
+    };
+    for (const entity of graph.entities) {
+        for (const obs of entity.observations || []) {
+            if (keepObservation && !keepObservation(obs))
+                continue;
+            add(entity.name, exports.DESCRIPTION_PROPERTY, (0, Observation_1.obsText)(obs));
+        }
+    }
+    for (const relation of graph.relations) {
+        const types = Array.isArray(relation.relationType)
+            ? relation.relationType
+            : [relation.relationType];
+        for (const type of types) {
+            if (!type)
+                continue;
+            add(relation.from, String(type), relation.to);
+        }
+    }
+    const triples = [];
+    for (const [name, props] of byEntity) {
+        for (const [property, values] of props) {
+            const sep = property === exports.DESCRIPTION_PROPERTY ? "; " : ", ";
+            triples.push({ name, property, value: [...new Set(values)].join(sep) });
+        }
+    }
+    return triples;
+}
+//# sourceMappingURL=kbTriples.js.map

package/dist/core/export/strategies/kbTriples.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kbTriples.js","sourceRoot":"","sources":["../../../../src/core/export/strategies/kbTriples.ts"],"names":[],"mappings":";;;AA6BA,kCAyCC;AArED,4DAAkE;AAelE,qFAAqF;AACxE,QAAA,oBAAoB,GAAG,aAAa,CAAC;AAElD;;;;;;;;;GASG;AACH,SAAgB,WAAW,CACzB,KAAqB,EACrB,eAAwD;IAExD,iFAAiF;IACjF,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAiC,CAAC;IAC1D,MAAM,GAAG,GAAG,CAAC,IAAY,EAAE,QAAgB,EAAE,KAAa,EAAE,EAAE;;QAC5D,MAAM,CAAC,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QACvB,IAAI,CAAC,CAAC;YAAE,OAAO;QACf,IAAI,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC/B,IAAI,CAAC,KAAK;YAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,MAAA,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,mCAAI,EAAE,CAAC;QACzC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACrD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,CAAC,CAAC;IAEF,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACpC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,YAAY,IAAI,EAAE,EAAE,CAAC;YAC5C,IAAI,eAAe,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC;gBAAE,SAAS;YACvD,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,4BAAoB,EAAE,IAAA,qBAAO,EAAC,GAAG,CAAC,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;YAChD,CAAC,CAAC,QAAQ,CAAC,YAAY;YACvB,CAAC,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAC5B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,MAAM,OAAO,GAAe,EAAE,CAAC;IAC/B,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,QAAQ,EAAE,CAAC;QACrC,KAAK,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;YACvC,MAAM,GAAG,GAAG,QAAQ,KAAK,4BAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;YAC5D,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/core/index.js

@@ -0,0 +1,22 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./DirectoryProcessor"), exports);
+__exportStar(require("./processor"), exports);
+__exportStar(require("./llm"), exports);
+__exportStar(require("./knowledge"), exports);
+__exportStar(require("./export"), exports);
+//# sourceMappingURL=index.js.map

package/dist/core/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/core/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,uDAAqC;AACrC,8CAA4B;AAC5B,wCAAsB;AACtB,8CAA4B;AAC5B,2CAAyB"}