@postnesia/mcp 0.1.6 → 0.1.8

package/dist/index.js

@@ -6,7 +6,8 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { z } from "zod";
-import { getDb, queries, createMemory } from "@postnesia/db";
+import { getDb, closeDb, createMemory } from "@postnesia/db";
+import queries from "@postnesia/db/queries";
 import { embed } from "@postnesia/db/embeddings";
 import { logAccess } from "@postnesia/db/access";
 import { runConsolidation } from "@postnesia/db/importance";
@@ -25,7 +26,10 @@ server.registerTool("memory_search", {
         limit: z.number().optional().describe("Maximum results to return (default: 10)"),
     },
 }, async ({ query, limit = 10 }) => {
-    const embedding = await embed(query);
+    const [embedding] = await embed(query);
+    if (!embedding) {
+        throw new ReferenceError('Unable to create query embedding');
+    }
     const results = queries.vectorSearch(db).all(Buffer.from(embedding.buffer), limit);
     for (const result of results) {
         logAccess(result.id, "search");
@@ -47,7 +51,10 @@ server.registerTool("memory_add", {
     },
 }, async ({ content, contentL1, type, importance, tags, context, core }) => {
     const content_l1 = contentL1 || content.slice(0, 200);
-    const embedding = await embed(content);
+    const [embedding] = await embed(content);
+    if (!embedding) {
+        throw new ReferenceError('Unable to create query embedding');
+    }
     const id = createMemory(db, {
         timestamp: new Date().toISOString(),
         content,
@@ -81,7 +88,10 @@ server.registerTool("memory_update_core", {
         throw new Error(`Memory #${memoryId} not found`);
     if (!existing.core)
         throw new Error(`Memory #${memoryId} is not a core memory. Use memory_add with supersedes_id for regular memories.`);
-    const embedding = await embed(content);
+    const [embedding] = await embed(content);
+    if (!embedding) {
+        throw new ReferenceError('Unable to create query embedding');
+    }
     db.prepare(`
       UPDATE memory
       SET content = ?, content_l1 = ?, updated_at = datetime('now')
@@ -154,6 +164,45 @@ server.registerTool("memory_relationships", {
         content: [{ type: "text", text: JSON.stringify(relationships, null, 2) }],
     };
 });
+server.registerTool("memory_link", {
+    description: "Explicitly create a typed relationship between two memories",
+    inputSchema: {
+        fromId: z.number().describe("Source memory ID"),
+        toId: z.number().describe("Target memory ID"),
+        type: z.enum(["related", "supersedes", "supports", "contradicts", "derives_from"])
+            .describe("Relationship type"),
+    },
+}, async ({ fromId, toId, type }) => {
+    const from = db.prepare("SELECT id FROM memory WHERE id = ?").get(fromId);
+    if (!from)
+        throw new Error(`Memory #${fromId} not found`);
+    const to = db.prepare("SELECT id FROM memory WHERE id = ?").get(toId);
+    if (!to)
+        throw new Error(`Memory #${toId} not found`);
+    const existing = queries.findRelationshipBetween(db).get(fromId, toId, type);
+    if (existing) {
+        return { content: [{ type: "text", text: `Relationship already exists (id: ${existing.id})` }] };
+    }
+    const result = queries.insertRelationship(db).run(fromId, toId, type);
+    const id = Number(result.lastInsertRowid);
+    return {
+        content: [{ type: "text", text: `Created relationship #${id}: #${fromId} -[${type}]-> #${toId}` }],
+    };
+});
+server.registerTool("memory_unlink", {
+    description: "Remove a relationship between two memories by relationship ID",
+    inputSchema: {
+        relationshipId: z.number().describe("Relationship ID to delete"),
+    },
+}, async ({ relationshipId }) => {
+    const existing = db.prepare("SELECT id FROM relationship WHERE id = ?").get(relationshipId);
+    if (!existing)
+        throw new Error(`Relationship #${relationshipId} not found`);
+    queries.deleteRelationship(db).run(relationshipId);
+    return {
+        content: [{ type: "text", text: `Deleted relationship #${relationshipId}` }],
+    };
+});
 // ---------------------------------------------------------------------------
 // journal tools
 // ---------------------------------------------------------------------------
@@ -193,7 +242,7 @@ server.registerTool("task_create", {
     inputSchema: {
         title: z.string().describe("Short task title"),
         description: z.string().optional().describe("Detailed description of what needs to be done"),
-        session_id: z.string().optional().describe("Project or feature label to group related tasks (e.g. 'openmind-mcp', 'auth-refactor')"),
+        session_id: z.string().optional().describe("Project or feature label to group related tasks (e.g. 'postnesia-mcp', 'auth-refactor')"),
         memory_id: z.number().optional().describe("Optional ID of a related memory"),
     },
 }, async ({ title, description, session_id, memory_id }) => {
@@ -271,5 +320,8 @@ async function main() {
 }
 main().catch((error) => {
     console.error("Fatal error:", error);
+    closeDb();
     process.exit(1);
 });
+process.on('SIGINT', () => { closeDb(); process.exit(0); });
+process.on('SIGTERM', () => { closeDb(); process.exit(0); });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@postnesia/mcp",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "An MCP server to interact with the Postnesia database",
   "type": "module",
   "private": false,
@@ -8,21 +8,21 @@
     "dist"
   ],
   "bin": {
-    "postnesia-mcp": "./dist/index.js"
+    "pn-mcp": "./dist/index.js"
+  },
+  "scripts": {
+    "build": "pnpm clean && tsc -p tsconfig.json",
+    "clean": "rm -rf ./dist",
+    "start": "tsx src/index.ts"
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.26.0",
-    "@postnesia/db": "^0.1.5",
+    "@postnesia/db": "workspace:*",
     "zod": "^4.3.6"
   },
   "devDependencies": {
     "@types/node": "^22.10.5",
     "tsx": "^4.19.2",
     "typescript": "^5.7.3"
-  },
-  "scripts": {
-    "build": "pnpm clean && tsc -p tsconfig.json",
-    "clean": "rm -rf ./dist",
-    "start": "tsx src/index.ts"
   }
-}
+}