@levalicious/server-memory 0.0.0

package/README.md

@@ -0,0 +1,281 @@
+# Knowledge Graph Memory Server
+
+A basic implementation of persistent memory using a local knowledge graph. This lets Claude remember information about the user across chats.
+
+## Core Concepts
+
+### Entities
+Entities are the primary nodes in the knowledge graph. Each entity has:
+- A unique name (identifier)
+- An entity type (e.g., "person", "organization", "event")
+- A list of observations
+
+Example:
+```json
+{
+  "name": "John_Smith",
+  "entityType": "person",
+  "observations": ["Speaks fluent Spanish"]
+}
+```
+
+### Relations
+Relations define directed connections between entities. They are always stored in active voice and describe how entities interact or relate to each other.
+
+Example:
+```json
+{
+  "from": "John_Smith",
+  "to": "Anthropic",
+  "relationType": "works_at"
+}
+```
+### Observations
+Observations are discrete pieces of information about an entity. They are:
+
+- Stored as strings
+- Attached to specific entities
+- Can be added or removed independently
+- Should be atomic (one fact per observation)
+
+Example:
+```json
+{
+  "entityName": "John_Smith",
+  "observations": [
+    "Speaks fluent Spanish",
+    "Graduated in 2019",
+    "Prefers morning meetings"
+  ]
+}
+```
+
+## API
+
+### Tools
+- **create_entities**
+  - Create multiple new entities in the knowledge graph
+  - Input: `entities` (array of objects)
+    - Each object contains:
+      - `name` (string): Entity identifier
+      - `entityType` (string): Type classification
+      - `observations` (string[]): Associated observations
+  - Ignores entities with existing names
+
+- **create_relations**
+  - Create multiple new relations between entities
+  - Input: `relations` (array of objects)
+    - Each object contains:
+      - `from` (string): Source entity name
+      - `to` (string): Target entity name
+      - `relationType` (string): Relationship type in active voice
+  - Skips duplicate relations
+
+- **add_observations**
+  - Add new observations to existing entities
+  - Input: `observations` (array of objects)
+    - Each object contains:
+      - `entityName` (string): Target entity
+      - `contents` (string[]): New observations to add
+  - Returns added observations per entity
+  - Fails if entity doesn't exist
+
+- **delete_entities**
+  - Remove entities and their relations
+  - Input: `entityNames` (string[])
+  - Cascading deletion of associated relations
+  - Silent operation if entity doesn't exist
+
+- **delete_observations**
+  - Remove specific observations from entities
+  - Input: `deletions` (array of objects)
+    - Each object contains:
+      - `entityName` (string): Target entity
+      - `observations` (string[]): Observations to remove
+  - Silent operation if observation doesn't exist
+
+- **delete_relations**
+  - Remove specific relations from the graph
+  - Input: `relations` (array of objects)
+    - Each object contains:
+      - `from` (string): Source entity name
+      - `to` (string): Target entity name
+      - `relationType` (string): Relationship type
+  - Silent operation if relation doesn't exist
+
+- **read_graph**
+  - Read the entire knowledge graph
+  - No input required
+  - Returns complete graph structure with all entities and relations
+
+- **search_nodes**
+  - Search for nodes based on query
+  - Input: `query` (string)
+  - Searches across:
+    - Entity names
+    - Entity types
+    - Observation content
+  - Returns matching entities and their relations
+
+- **open_nodes**
+  - Retrieve specific nodes by name
+  - Input: `names` (string[])
+  - Returns:
+    - Requested entities
+    - Relations between requested entities
+  - Silently skips non-existent nodes
+
+# Usage with Claude Desktop
+
+### Setup
+
+Add this to your claude_desktop_config.json:
+
+#### Docker
+
+```json
+{
+  "mcpServers": {
+    "memory": {
+      "command": "docker",
+      "args": ["run", "-i", "-v", "claude-memory:/app/dist", "--rm", "mcp/memory"]
+    }
+  }
+}
+```
+
+#### NPX
+```json
+{
+  "mcpServers": {
+    "memory": {
+      "command": "npx",
+      "args": [
+        "-y",
+        "@levalicious/server-memory"
+      ]
+    }
+  }
+}
+```
+
+#### NPX with custom setting
+
+The server can be configured using the following environment variables:
+
+```json
+{
+  "mcpServers": {
+    "memory": {
+      "command": "npx",
+      "args": [
+        "-y",
+        "@levalicious/server-memory"
+      ],
+      "env": {
+        "MEMORY_FILE_PATH": "/path/to/custom/memory.json"
+      }
+    }
+  }
+}
+```
+
+- `MEMORY_FILE_PATH`: Path to the memory storage JSON file (default: `memory.json` in the server directory)
+
+# VS Code Installation Instructions
+
+For quick installation, use one of the one-click installation buttons below:
+
+[![Install with NPX in VS Code](https://img.shields.io/badge/VS_Code-NPM-0098FF?style=flat-square&logo=visualstudiocode&logoColor=white)](https://insiders.vscode.dev/redirect/mcp/install?name=memory&config=%7B%22command%22%3A%22npx%22%2C%22args%22%3A%5B%22-y%22%2C%22%40modelcontextprotocol%2Fserver-memory%22%5D%7D) [![Install with NPX in VS Code Insiders](https://img.shields.io/badge/VS_Code_Insiders-NPM-24bfa5?style=flat-square&logo=visualstudiocode&logoColor=white)](https://insiders.vscode.dev/redirect/mcp/install?name=memory&config=%7B%22command%22%3A%22npx%22%2C%22args%22%3A%5B%22-y%22%2C%22%40modelcontextprotocol%2Fserver-memory%22%5D%7D&quality=insiders)
+
+[![Install with Docker in VS Code](https://img.shields.io/badge/VS_Code-Docker-0098FF?style=flat-square&logo=visualstudiocode&logoColor=white)](https://insiders.vscode.dev/redirect/mcp/install?name=memory&config=%7B%22command%22%3A%22docker%22%2C%22args%22%3A%5B%22run%22%2C%22-i%22%2C%22-v%22%2C%22claude-memory%3A%2Fapp%2Fdist%22%2C%22--rm%22%2C%22mcp%2Fmemory%22%5D%7D) [![Install with Docker in VS Code Insiders](https://img.shields.io/badge/VS_Code_Insiders-Docker-24bfa5?style=flat-square&logo=visualstudiocode&logoColor=white)](https://insiders.vscode.dev/redirect/mcp/install?name=memory&config=%7B%22command%22%3A%22docker%22%2C%22args%22%3A%5B%22run%22%2C%22-i%22%2C%22-v%22%2C%22claude-memory%3A%2Fapp%2Fdist%22%2C%22--rm%22%2C%22mcp%2Fmemory%22%5D%7D&quality=insiders)
+
+For manual installation, add the following JSON block to your User Settings (JSON) file in VS Code. You can do this by pressing `Ctrl + Shift + P` and typing `Preferences: Open Settings (JSON)`.
+
+Optionally, you can add it to a file called `.vscode/mcp.json` in your workspace. This will allow you to share the configuration with others. 
+
+> Note that the `mcp` key is not needed in the `.vscode/mcp.json` file.
+
+#### NPX
+
+```json
+{
+  "mcp": {
+    "servers": {
+      "memory": {
+        "command": "npx",
+        "args": [
+          "-y",
+          "@levalicious/server-memory"
+        ]
+      }
+    }
+  }
+}
+```
+
+#### Docker
+
+```json
+{
+  "mcp": {
+    "servers": {
+      "memory": {
+        "command": "docker",
+        "args": [
+          "run",
+          "-i",
+          "-v",
+          "claude-memory:/app/dist",
+          "--rm",
+          "mcp/memory"
+        ]
+      }
+    }
+  }
+}
+```
+
+### System Prompt
+
+The prompt for utilizing memory depends on the use case. Changing the prompt will help the model determine the frequency and types of memories created.
+
+Here is an example prompt for chat personalization. You could use this prompt in the "Custom Instructions" field of a [Claude.ai Project](https://www.anthropic.com/news/projects). 
+
+```
+Follow these steps for each interaction:
+
+1. User Identification:
+   - You should assume that you are interacting with default_user
+   - If you have not identified default_user, proactively try to do so.
+
+2. Memory Retrieval:
+   - Always begin your chat by saying only "Remembering..." and retrieve all relevant information from your knowledge graph
+   - Always refer to your knowledge graph as your "memory"
+
+3. Memory
+   - While conversing with the user, be attentive to any new information that falls into these categories:
+     a) Basic Identity (age, gender, location, job title, education level, etc.)
+     b) Behaviors (interests, habits, etc.)
+     c) Preferences (communication style, preferred language, etc.)
+     d) Goals (goals, targets, aspirations, etc.)
+     e) Relationships (personal and professional relationships up to 3 degrees of separation)
+
+4. Memory Update:
+   - If any new information was gathered during the interaction, update your memory as follows:
+     a) Create entities for recurring organizations, people, and significant events
+     b) Connect them to the current entities using relations
+     c) Store facts about them as observations
+```
+
+## Building
+
+Docker:
+
+```sh
+docker build -t mcp/memory -f src/memory/Dockerfile . 
+```
+
+## License
+
+This MCP server is licensed under the MIT License. This means you are free to use, modify, and distribute the software, subject to the terms and conditions of the MIT License. For more details, please see the LICENSE file in the project repository.

package/dist/index.js

@@ -0,0 +1,579 @@
+#!/usr/bin/env node
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
+import { promises as fs } from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+// Define memory file path using environment variable with fallback
+const defaultMemoryPath = path.join(path.dirname(fileURLToPath(import.meta.url)), 'memory.json');
+// If MEMORY_FILE_PATH is just a filename, put it in the same directory as the script
+const MEMORY_FILE_PATH = process.env.MEMORY_FILE_PATH
+    ? path.isAbsolute(process.env.MEMORY_FILE_PATH)
+        ? process.env.MEMORY_FILE_PATH
+        : path.join(path.dirname(fileURLToPath(import.meta.url)), process.env.MEMORY_FILE_PATH)
+    : defaultMemoryPath;
+// The KnowledgeGraphManager class contains all operations to interact with the knowledge graph
+class KnowledgeGraphManager {
+    async loadGraph() {
+        try {
+            const data = await fs.readFile(MEMORY_FILE_PATH, "utf-8");
+            const lines = data.split("\n").filter(line => line.trim() !== "");
+            return lines.reduce((graph, line) => {
+                const item = JSON.parse(line);
+                if (item.type === "entity")
+                    graph.entities.push(item);
+                if (item.type === "relation")
+                    graph.relations.push(item);
+                return graph;
+            }, { entities: [], relations: [] });
+        }
+        catch (error) {
+            if (error instanceof Error && 'code' in error && error.code === "ENOENT") {
+                return { entities: [], relations: [] };
+            }
+            throw error;
+        }
+    }
+    async saveGraph(graph) {
+        const lines = [
+            ...graph.entities.map(e => JSON.stringify({ type: "entity", ...e })),
+            ...graph.relations.map(r => JSON.stringify({ type: "relation", ...r })),
+        ];
+        await fs.writeFile(MEMORY_FILE_PATH, lines.join("\n"));
+    }
+    async createEntities(entities) {
+        const graph = await this.loadGraph();
+        const newEntities = entities.filter(e => !graph.entities.some(existingEntity => existingEntity.name === e.name));
+        graph.entities.push(...newEntities);
+        await this.saveGraph(graph);
+        return newEntities;
+    }
+    async createRelations(relations) {
+        const graph = await this.loadGraph();
+        const newRelations = relations.filter(r => !graph.relations.some(existingRelation => existingRelation.from === r.from &&
+            existingRelation.to === r.to &&
+            existingRelation.relationType === r.relationType));
+        graph.relations.push(...newRelations);
+        await this.saveGraph(graph);
+        return newRelations;
+    }
+    async addObservations(observations) {
+        const graph = await this.loadGraph();
+        const results = observations.map(o => {
+            const entity = graph.entities.find(e => e.name === o.entityName);
+            if (!entity) {
+                throw new Error(`Entity with name ${o.entityName} not found`);
+            }
+            const newObservations = o.contents.filter(content => !entity.observations.includes(content));
+            entity.observations.push(...newObservations);
+            return { entityName: o.entityName, addedObservations: newObservations };
+        });
+        await this.saveGraph(graph);
+        return results;
+    }
+    async deleteEntities(entityNames) {
+        const graph = await this.loadGraph();
+        graph.entities = graph.entities.filter(e => !entityNames.includes(e.name));
+        graph.relations = graph.relations.filter(r => !entityNames.includes(r.from) && !entityNames.includes(r.to));
+        await this.saveGraph(graph);
+    }
+    async deleteObservations(deletions) {
+        const graph = await this.loadGraph();
+        deletions.forEach(d => {
+            const entity = graph.entities.find(e => e.name === d.entityName);
+            if (entity) {
+                entity.observations = entity.observations.filter(o => !d.observations.includes(o));
+            }
+        });
+        await this.saveGraph(graph);
+    }
+    async deleteRelations(relations) {
+        const graph = await this.loadGraph();
+        graph.relations = graph.relations.filter(r => !relations.some(delRelation => r.from === delRelation.from &&
+            r.to === delRelation.to &&
+            r.relationType === delRelation.relationType));
+        await this.saveGraph(graph);
+    }
+    async readGraph() {
+        return this.loadGraph();
+    }
+    // Very basic search function
+    async searchNodes(query) {
+        const graph = await this.loadGraph();
+        // Filter entities
+        const filteredEntities = graph.entities.filter(e => e.name.toLowerCase().includes(query.toLowerCase()) ||
+            e.entityType.toLowerCase().includes(query.toLowerCase()) ||
+            e.observations.some(o => o.toLowerCase().includes(query.toLowerCase())));
+        // Create a Set of filtered entity names for quick lookup
+        const filteredEntityNames = new Set(filteredEntities.map(e => e.name));
+        // Filter relations to only include those between filtered entities
+        const filteredRelations = graph.relations.filter(r => filteredEntityNames.has(r.from) && filteredEntityNames.has(r.to));
+        const filteredGraph = {
+            entities: filteredEntities,
+            relations: filteredRelations,
+        };
+        return filteredGraph;
+    }
+    async openNodesFiltered(names) {
+        const graph = await this.loadGraph();
+        // Filter entities
+        const filteredEntities = graph.entities.filter(e => names.includes(e.name));
+        // Create a Set of filtered entity names for quick lookup
+        const filteredEntityNames = new Set(filteredEntities.map(e => e.name));
+        // Filter relations to only include those between filtered entities
+        const filteredRelations = graph.relations.filter(r => filteredEntityNames.has(r.from) && filteredEntityNames.has(r.to));
+        const filteredGraph = {
+            entities: filteredEntities,
+            relations: filteredRelations,
+        };
+        return filteredGraph;
+    }
+    async openNodes(names) {
+        const graph = await this.loadGraph();
+        // Filter entities
+        const filteredEntities = graph.entities.filter(e => names.includes(e.name));
+        // Create a Set of filtered entity names for quick lookup
+        const filteredEntityNames = new Set(filteredEntities.map(e => e.name));
+        // Filter relations to only include those between filtered entities
+        const filteredRelations = graph.relations.filter(r => filteredEntityNames.has(r.from));
+        const filteredGraph = {
+            entities: filteredEntities,
+            relations: filteredRelations,
+        };
+        return filteredGraph;
+    }
+    async getNeighbors(entityName, depth = 1) {
+        const graph = await this.loadGraph();
+        const visited = new Set();
+        const resultEntities = new Map();
+        const resultRelations = [];
+        const traverse = (currentName, currentDepth) => {
+            if (currentDepth > depth || visited.has(currentName))
+                return;
+            visited.add(currentName);
+            const entity = graph.entities.find(e => e.name === currentName);
+            if (entity) {
+                resultEntities.set(currentName, entity);
+            }
+            // Find all relations involving this entity
+            const connectedRelations = graph.relations.filter(r => r.from === currentName || r.to === currentName);
+            resultRelations.push(...connectedRelations);
+            if (currentDepth < depth) {
+                // Traverse to connected entities
+                connectedRelations.forEach(r => {
+                    const nextEntity = r.from === currentName ? r.to : r.from;
+                    traverse(nextEntity, currentDepth + 1);
+                });
+            }
+        };
+        traverse(entityName, 0);
+        return {
+            entities: Array.from(resultEntities.values()),
+            relations: resultRelations
+        };
+    }
+    async findPath(fromEntity, toEntity, maxDepth = 5) {
+        const graph = await this.loadGraph();
+        const visited = new Set();
+        const dfs = (current, target, path, depth) => {
+            if (depth > maxDepth || visited.has(current))
+                return null;
+            if (current === target)
+                return path;
+            visited.add(current);
+            const outgoingRelations = graph.relations.filter(r => r.from === current);
+            for (const relation of outgoingRelations) {
+                const result = dfs(relation.to, target, [...path, relation], depth + 1);
+                if (result)
+                    return result;
+            }
+            visited.delete(current);
+            return null;
+        };
+        return dfs(fromEntity, toEntity, [], 0) || [];
+    }
+    async getEntitiesByType(entityType) {
+        const graph = await this.loadGraph();
+        return graph.entities.filter(e => e.entityType === entityType);
+    }
+    async getEntityTypes() {
+        const graph = await this.loadGraph();
+        const types = new Set(graph.entities.map(e => e.entityType));
+        return Array.from(types).sort();
+    }
+    async getRelationTypes() {
+        const graph = await this.loadGraph();
+        const types = new Set(graph.relations.map(r => r.relationType));
+        return Array.from(types).sort();
+    }
+    async getStats() {
+        const graph = await this.loadGraph();
+        const entityTypes = new Set(graph.entities.map(e => e.entityType));
+        const relationTypes = new Set(graph.relations.map(r => r.relationType));
+        return {
+            entityCount: graph.entities.length,
+            relationCount: graph.relations.length,
+            entityTypes: entityTypes.size,
+            relationTypes: relationTypes.size
+        };
+    }
+    async getOrphanedEntities() {
+        const graph = await this.loadGraph();
+        const connectedEntityNames = new Set();
+        graph.relations.forEach(r => {
+            connectedEntityNames.add(r.from);
+            connectedEntityNames.add(r.to);
+        });
+        return graph.entities.filter(e => !connectedEntityNames.has(e.name));
+    }
+    async validateGraph() {
+        const graph = await this.loadGraph();
+        const entityNames = new Set(graph.entities.map(e => e.name));
+        const missingEntities = new Set();
+        graph.relations.forEach(r => {
+            if (!entityNames.has(r.from)) {
+                missingEntities.add(r.from);
+            }
+            if (!entityNames.has(r.to)) {
+                missingEntities.add(r.to);
+            }
+        });
+        return Array.from(missingEntities);
+    }
+}
+const knowledgeGraphManager = new KnowledgeGraphManager();
+// The server instance and tools exposed to Claude
+const server = new Server({
+    name: "memory-server",
+    version: "0.0.0",
+}, {
+    capabilities: {
+        tools: {},
+    },
+});
+server.setRequestHandler(ListToolsRequestSchema, async () => {
+    return {
+        tools: [
+            {
+                name: "create_entities",
+                description: "Create multiple new entities in the knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        entities: {
+                            type: "array",
+                            items: {
+                                type: "object",
+                                properties: {
+                                    name: { type: "string", description: "The name of the entity" },
+                                    entityType: { type: "string", description: "The type of the entity" },
+                                    observations: {
+                                        type: "array",
+                                        items: { type: "string" },
+                                        description: "An array of observation contents associated with the entity"
+                                    },
+                                },
+                                required: ["name", "entityType", "observations"],
+                            },
+                        },
+                    },
+                    required: ["entities"],
+                },
+            },
+            {
+                name: "create_relations",
+                description: "Create multiple new relations between entities in the knowledge graph. Relations should be in active voice",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        relations: {
+                            type: "array",
+                            items: {
+                                type: "object",
+                                properties: {
+                                    from: { type: "string", description: "The name of the entity where the relation starts" },
+                                    to: { type: "string", description: "The name of the entity where the relation ends" },
+                                    relationType: { type: "string", description: "The type of the relation" },
+                                },
+                                required: ["from", "to", "relationType"],
+                            },
+                        },
+                    },
+                    required: ["relations"],
+                },
+            },
+            {
+                name: "add_observations",
+                description: "Add new observations to existing entities in the knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        observations: {
+                            type: "array",
+                            items: {
+                                type: "object",
+                                properties: {
+                                    entityName: { type: "string", description: "The name of the entity to add the observations to" },
+                                    contents: {
+                                        type: "array",
+                                        items: { type: "string" },
+                                        description: "An array of observation contents to add"
+                                    },
+                                },
+                                required: ["entityName", "contents"],
+                            },
+                        },
+                    },
+                    required: ["observations"],
+                },
+            },
+            {
+                name: "delete_entities",
+                description: "Delete multiple entities and their associated relations from the knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        entityNames: {
+                            type: "array",
+                            items: { type: "string" },
+                            description: "An array of entity names to delete"
+                        },
+                    },
+                    required: ["entityNames"],
+                },
+            },
+            {
+                name: "delete_observations",
+                description: "Delete specific observations from entities in the knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        deletions: {
+                            type: "array",
+                            items: {
+                                type: "object",
+                                properties: {
+                                    entityName: { type: "string", description: "The name of the entity containing the observations" },
+                                    observations: {
+                                        type: "array",
+                                        items: { type: "string" },
+                                        description: "An array of observations to delete"
+                                    },
+                                },
+                                required: ["entityName", "observations"],
+                            },
+                        },
+                    },
+                    required: ["deletions"],
+                },
+            },
+            {
+                name: "delete_relations",
+                description: "Delete multiple relations from the knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        relations: {
+                            type: "array",
+                            items: {
+                                type: "object",
+                                properties: {
+                                    from: { type: "string", description: "The name of the entity where the relation starts" },
+                                    to: { type: "string", description: "The name of the entity where the relation ends" },
+                                    relationType: { type: "string", description: "The type of the relation" },
+                                },
+                                required: ["from", "to", "relationType"],
+                            },
+                            description: "An array of relations to delete"
+                        },
+                    },
+                    required: ["relations"],
+                },
+            },
+            {
+                name: "read_graph",
+                description: "Read the entire knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {},
+                },
+            },
+            {
+                name: "search_nodes",
+                description: "Search for nodes in the knowledge graph based on a query",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        query: { type: "string", description: "The search query to match against entity names, types, and observation content" },
+                    },
+                    required: ["query"],
+                },
+            },
+            {
+                name: "open_nodes_filtered",
+                description: "Open specific nodes in the knowledge graph by their names, filtering relations to only those between the opened nodes",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        names: {
+                            type: "array",
+                            items: { type: "string" },
+                            description: "An array of entity names to retrieve",
+                        },
+                    },
+                    required: ["names"],
+                },
+            },
+            {
+                name: "open_nodes",
+                description: "Open specific nodes in the knowledge graph by their names",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        names: {
+                            type: "array",
+                            items: { type: "string" },
+                            description: "An array of entity names to retrieve",
+                        },
+                    },
+                    required: ["names"],
+                },
+            },
+            {
+                name: "get_neighbors",
+                description: "Get neighboring entities connected to a specific entity within a given depth",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        entityName: { type: "string", description: "The name of the entity to find neighbors for" },
+                        depth: { type: "number", description: "Maximum depth to traverse (default: 1)", default: 1 },
+                    },
+                    required: ["entityName"],
+                },
+            },
+            {
+                name: "find_path",
+                description: "Find a path between two entities in the knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        fromEntity: { type: "string", description: "The name of the starting entity" },
+                        toEntity: { type: "string", description: "The name of the target entity" },
+                        maxDepth: { type: "number", description: "Maximum depth to search (default: 5)", default: 5 },
+                    },
+                    required: ["fromEntity", "toEntity"],
+                },
+            },
+            {
+                name: "get_entities_by_type",
+                description: "Get all entities of a specific type",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        entityType: { type: "string", description: "The type of entities to retrieve" },
+                    },
+                    required: ["entityType"],
+                },
+            },
+            {
+                name: "get_entity_types",
+                description: "Get all unique entity types in the knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {},
+                },
+            },
+            {
+                name: "get_relation_types",
+                description: "Get all unique relation types in the knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {},
+                },
+            },
+            {
+                name: "get_stats",
+                description: "Get statistics about the knowledge graph",
+                inputSchema: {
+                    type: "object",
+                    properties: {},
+                },
+            },
+            {
+                name: "get_orphaned_entities",
+                description: "Get entities that have no relations (orphaned entities)",
+                inputSchema: {
+                    type: "object",
+                    properties: {},
+                },
+            },
+            {
+                name: "validate_graph",
+                description: "Validate the knowledge graph and return a list of missing entities referenced in relations",
+                inputSchema: {
+                    type: "object",
+                    properties: {},
+                },
+            },
+        ],
+    };
+});
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    const { name, arguments: args } = request.params;
+    if (!args) {
+        throw new Error(`No arguments provided for tool: ${name}`);
+    }
+    switch (name) {
+        case "create_entities":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.createEntities(args.entities), null, 2) }] };
+        case "create_relations":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.createRelations(args.relations), null, 2) }] };
+        case "add_observations":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.addObservations(args.observations), null, 2) }] };
+        case "delete_entities":
+            await knowledgeGraphManager.deleteEntities(args.entityNames);
+            return { content: [{ type: "text", text: "Entities deleted successfully" }] };
+        case "delete_observations":
+            await knowledgeGraphManager.deleteObservations(args.deletions);
+            return { content: [{ type: "text", text: "Observations deleted successfully" }] };
+        case "delete_relations":
+            await knowledgeGraphManager.deleteRelations(args.relations);
+            return { content: [{ type: "text", text: "Relations deleted successfully" }] };
+        case "read_graph":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.readGraph(), null, 2) }] };
+        case "search_nodes":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.searchNodes(args.query), null, 2) }] };
+        case "open_nodes_filtered":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.openNodesFiltered(args.names), null, 2) }] };
+        case "open_nodes":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.openNodes(args.names), null, 2) }] };
+        case "get_neighbors":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.getNeighbors(args.entityName, args.depth), null, 2) }] };
+        case "find_path":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.findPath(args.fromEntity, args.toEntity, args.maxDepth), null, 2) }] };
+        case "get_entities_by_type":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.getEntitiesByType(args.entityType), null, 2) }] };
+        case "get_entity_types":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.getEntityTypes(), null, 2) }] };
+        case "get_relation_types":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.getRelationTypes(), null, 2) }] };
+        case "get_stats":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.getStats(), null, 2) }] };
+        case "get_orphaned_entities":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.getOrphanedEntities(), null, 2) }] };
+        case "validate_graph":
+            return { content: [{ type: "text", text: JSON.stringify(await knowledgeGraphManager.validateGraph(), null, 2) }] };
+        default:
+            throw new Error(`Unknown tool: ${name}`);
+    }
+});
+async function main() {
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    console.error("Knowledge Graph MCP Server running on stdio");
+}
+main().catch((error) => {
+    console.error("Fatal error in main():", error);
+    process.exit(1);
+});

package/package.json

@@ -0,0 +1,29 @@
+{
+  "name": "@levalicious/server-memory",
+  "version": "0.0.0",
+  "description": "MCP server for enabling memory for Claude through a knowledge graph",
+  "license": "MIT",
+  "author": "Levalicious",
+  "homepage": "https://modelcontextprotocol.io",
+  "bugs": "https://github.com/modelcontextprotocol/servers/issues",
+  "type": "module",
+  "bin": {
+    "mcp-server-memory": "dist/index.js"
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc && shx chmod +x dist/*.js",
+    "prepare": "npm run build",
+    "watch": "tsc --watch"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "1.0.1"
+  },
+  "devDependencies": {
+    "@types/node": "^22",
+    "shx": "^0.3.4",
+    "typescript": "^5.6.2"
+  }
+}