(mcp-neo4j-memory) load/save entire kg

Author: akollegger

package-lock.json

@@ -6,7 +6,8 @@
   "private": true,
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.0.1",
-    "axios": "^1.6.0"
+    "axios": "^1.6.0",
+    "neo4j-driver": "^5.27.0"
   },
   "devDependencies": {
     "@eslint/js": "^9.8.0",

package.json

@@ -2,7 +2,6 @@
 import path from 'path';
 import { fileURLToPath } from 'url';
 import { promises as fs } from 'fs';
-// import { KnowledgeGraphMemory, Entity, KnowledgeGraph, Relation } from "./types";
 import { KnowledgeGraphMemory, Entity, KnowledgeGraph, Relation } from "@neo4j/graphrag-memory";
 
 // Default path to the JSONL file, you can change this to your desired local path 
@@ -18,7 +17,6 @@ export class JsonMemory implements KnowledgeGraphMemory {
   }
 
   private async loadGraph(): Promise<KnowledgeGraph> {
-    console.log(`loadGraph(${this.memoryFilePath})`);
     try {
       const data = await fs.readFile(this.memoryFilePath, "utf-8");
       const lines = data.split("\n").filter(line => line.trim() !== "");

servers/mcp-json-memory/src/json-memory.ts

@@ -4,7 +4,6 @@ import {
   CallToolRequestSchema,
   ListToolsRequestSchema,
 } from "@modelcontextprotocol/sdk/types.js";
-// import { KnowledgeGraphMemory, Entity, KnowledgeGraph, Relation } from "./types.js";
 import { KnowledgeGraphMemory, Entity, KnowledgeGraph, Relation } from "@neo4j/graphrag-memory";
 import { JsonMemory } from './json-memory.js'
 
@@ -238,7 +237,7 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
 async function main() {
   const transport = new StdioServerTransport();
   await server.connect(transport);
-  console.error("MCP Knowledge Graph Memory using JSON running on stdio");
+  console.error("MCP Knowledge Graph Memory using Neo4j running on stdio");
 }
 
 main().catch((error) => {

servers/mcp-json-memory/src/main.ts

@@ -3,7 +3,6 @@
   "files": [],
   "include": [],
   "references": [
-    // @ts-ignore -> https://github.com/nrwl/nx/issues/27292
     {
       "path": "./tsconfig.app.json"
     },

servers/mcp-json-memory/src/types.ts

@@ -12,7 +12,7 @@
       "options": {
         "platform": "node",
         "outputPath": "dist/servers/mcp-neo4j-memory",
-        "format": ["cjs"],
+        "format": ["esm"],
         "bundle": false,
         "main": "servers/mcp-neo4j-memory/src/main.ts",
         "tsConfig": "servers/mcp-neo4j-memory/tsconfig.app.json",