@deepsweet/mdn 0.1.1 → 0.1.3

package/README.md

@@ -22,7 +22,7 @@ See [dataset repo](https://huggingface.co/datasets/deepsweet/mdn) on HuggigFace
 ### 1. Download dataset and embedding model
 
 ```sh
-npx @deepsweet/mdn download
+npx -y @deepsweet/mdn download
 ```
 
 Both [dataset](https://huggingface.co/datasets/deepsweet/mdn) (\~260 MB) and the [embedding model GGUF file](https://huggingface.co/deepsweet/bge-m3-GGUF-Q4_K_M) (\~438 MB) will be downloaded directly from HugginFace and stored in its default cache location (typically `~/.cache/huggingface/`), just like the `hf download` command does.

package/dist/index.js

@@ -95,16 +95,19 @@ import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js"
 import { z as z2 } from "zod";
 
 // src/llama.ts
-import { getLlama } from "node-llama-cpp";
+import { getLlama, LlamaLogLevel } from "node-llama-cpp";
 var MAX_TOKENS = 8192;
 var getLlamaContext = async (modelPath) => {
-  const llama = await getLlama();
+  const llama = await getLlama({ logLevel: LlamaLogLevel.error });
   const model = await llama.loadModel({ modelPath });
   const context = await model.createEmbeddingContext({
     contextSize: MAX_TOKENS,
     batchSize: MAX_TOKENS,
     threads: 0
   });
+  context.onDispose.createOnceListener(() => {
+    model.dispose().then(() => llama.dispose()).catch(console.error);
+  });
   return context;
 };
 
@@ -131,7 +134,7 @@ var createReranker = async () => {
 
 // package.json
 var name = "@deepsweet/mdn";
-var version = "0.1.1";
+var version = "0.1.3";
 
 // src/server.ts
 var startMcpServer = async () => {
@@ -171,6 +174,22 @@ var startMcpServer = async () => {
     };
   });
   const transport = new StdioServerTransport;
+  const dispose = async () => {
+    if (llamaTimeout !== null) {
+      clearTimeout(llamaTimeout);
+    }
+    if (llamaContext !== null) {
+      await llamaContext.dispose();
+    }
+    db.close();
+    await server.close();
+  };
+  transport.onclose = () => {
+    dispose().catch(console.error);
+  };
+  transport.onerror = (err) => {
+    console.log(err);
+  };
   await server.connect(transport);
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@deepsweet/mdn",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "publishConfig": {
     "access": "public"
   },
@@ -14,26 +14,21 @@
   ],
   "dependencies": {
     "@huggingface/hub": "^2.11.0",
-    "@lancedb/lancedb": "0.27.2-beta.1",
+    "@lancedb/lancedb": "^0.27.2",
     "@modelcontextprotocol/sdk": "^1.29.0",
     "apache-arrow": "18.1.0",
     "node-llama-cpp": "^3.18.1",
     "zod": "^4.3.6"
   },
   "devDependencies": {
-    "@huggingface/tokenizers": "^0.1.3",
-    "@stylistic/eslint-plugin": "^5.10.0",
+    "@deepsweet/eslint-config": "^0.1.0",
     "@types/bun": "^1.3.11",
     "dedent": "^1.7.2",
     "eslint": "^10.1.0",
-    "eslint-plugin-import": "^2.32.0",
-    "eslint-plugin-perfectionist": "^5.7.0",
-    "globals": "^17.4.0",
     "gray-matter": "^4.0.3",
     "marked": "^17.0.5",
     "rimraf": "^6.1.3",
-    "typescript": "^6.0.2",
-    "typescript-eslint": "^8.58.0"
+    "typescript": "^6.0.2"
   },
   "scripts": {
     "chunk": "bun scripts/chunk.ts",