audio worker fix

package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@ztimson/ai-utils",
-	"version": "0.6.7",
+	"version": "0.6.9",
 	"description": "AI Utility library",
 	"author": "Zak Timson",
 	"license": "MIT",

src/asr.ts

@@ -84,6 +84,7 @@ function combineSpeakerTranscript(chunks: any[], speakers: any[]): string {
 
 parentPort?.on('message', async ({ file, speaker, model, modelDir }) => {
 	try {
+		console.log('worker', file);
 		if(!whisperPipeline) whisperPipeline = await pipeline('automatic-speech-recognition', `Xenova/${model}`, {cache_dir: modelDir, quantized: true});
 
 		// Prepare audio file (convert to mono channel wave)

src/audio.ts

@@ -1,7 +1,8 @@
+import {fileURLToPath} from 'url';
 import {Worker} from 'worker_threads';
-import Path from 'node:path';
 import {AbortablePromise, Ai} from './ai.ts';
 import {canDiarization} from './asr.ts';
+import {dirname, join} from 'path';
 
 export class Audio {
 	constructor(private ai: Ai) {}
@@ -12,7 +13,7 @@ export class Audio {
 		const abort = () => { aborted = true; };
 
 		const p = new Promise<string | null>((resolve, reject) => {
-			const worker = new Worker(Path.join(import.meta.dirname, 'asr.js'));
+			const worker = new Worker(join(dirname(fileURLToPath(import.meta.url)), 'asr.js'));
 			const handleMessage = ({ text, warning, error }: any) => {
 				worker.terminate();
 				if(aborted) return;

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});
 });

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);