From 8f89f5e3cf7c73a20ef7a99f3c3ddcb472c89ad1 Mon Sep 17 00:00:00 2001
From: ztimson <zaktimson@gmail.com>
Date: Thu, 12 Feb 2026 20:18:56 -0500
Subject: [PATCH] embedding worker fix

---
 package.json    | 2 +-
 src/embedder.ts | 4 ++--
 src/llm.ts      | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/package.json b/package.json
index 62efdd5..d249597 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
 	"name": "@ztimson/ai-utils",
-	"version": "0.6.7-rc.1",
+	"version": "0.6.8",
 	"description": "AI Utility library",
 	"author": "Zak Timson",
 	"license": "MIT",
diff --git a/src/embedder.ts b/src/embedder.ts
index 49f5e7e..492f43a 100644
--- a/src/embedder.ts
+++ b/src/embedder.ts
@@ -3,9 +3,9 @@ import { parentPort } from 'worker_threads';
 
 let embedder: any;
 
-parentPort?.on('message', async ({ id, text, model, modelDir }) => {
+parentPort?.on('message', async ({text, model, modelDir }) => {
 	if(!embedder) embedder = await pipeline('feature-extraction', 'Xenova/' + model, {quantized: true, cache_dir: modelDir});
 	const output = await embedder(text, { pooling: 'mean', normalize: true });
 	const embedding = Array.from(output.data);
-	parentPort?.postMessage({ id, embedding });
+	parentPort?.postMessage({embedding});
 });
diff --git a/src/llm.ts b/src/llm.ts
index fdc7e5c..fc0c2d0 100644
--- a/src/llm.ts
+++ b/src/llm.ts
@@ -271,7 +271,7 @@ class LLM {
 				worker.on('exit', (code) => {
 					if(code !== 0) reject(new Error(`Worker exited with code ${code}`));
 				});
-				worker.postMessage({text, model: this.ai.options?.embedder || 'bge-small-en-v1.5', path: this.ai.options.path});
+				worker.postMessage({text, model: this.ai.options?.embedder || 'bge-small-en-v1.5', modelDir: this.ai.options.path});
 			});
 		};
 		const chunks = this.chunk(target, maxTokens, overlapTokens);