@postnesia/mcp 0.1.7 → 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')
@@ -232,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 }) => {
@@ -310,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.7",
+  "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",
-    "zod": "^4.3.6",
-    "@postnesia/db": "0.1.7"
+    "@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"
   }
-}
+}