streaming-sortformer-node 0.1.0

package/dist/Sortformer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Sortformer.js","sourceRoot":"","sources":["../src/Sortformer.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD;;;;;;;;;;;;;GAaG;AACH,MAAM,OAAO,UAAU;IAIrB;;;OAGG;IACH,YAAoB,MAAW;QANvB,WAAM,GAAY,KAAK,CAAC;QAO9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAAiB,EAAE,OAAqB;QACxD,iBAAiB;QACjB,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;YAChD,MAAM,IAAI,SAAS,CAAC,sCAAsC,CAAC,CAAC;QAC9D,CAAC;QAED,qBAAqB;QACrB,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;QAE7B,+BAA+B;QAC/B,wCAAwC;QACxC,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,CAAC,CAAC;QAEtC,IAAI,OAAO,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QAED,2BAA2B;QAC3B,MAAM,MAAM,GAAG,IAAI,OAAO,CAAC,eAAe,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAE/D,OAAO,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED;;;;;;;;;;;;;;;;;OAiBG;IACH,KAAK,CAAC,OAAO,CAAC,KAAmB,EAAE,OAAwB;QACzD,2BAA2B;QAC3B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QAClE,CAAC;QAED,uBAAuB;QACvB,IAAI,CAAC,CAAC,KAAK,YAAY,YAAY,CAAC,EAAE,CAAC;YACrC,MAAM,IAAI,SAAS,CAAC,8BAA8B,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,mBAAmB;QACnB,IAAI,OAAO,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;YACrC,IAAI,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ,IAAI,OAAO,CAAC,SAAS,GAAG,CAAC,IAAI,OAAO,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC;gBAC5F,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAED,IAAI,OAAO,EAAE,YAAY,KAAK,SAAS,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,YAAY,GAAG,CAAC,IAAI,OAAO,CAAC,YAAY,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1G,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QAED,6CAA6C;QAC7C,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,IAAI,SAAS,CAAC;QACxC,MAAM,aAAa,GAAQ;YACzB,SAAS,EAAE,OAAO,EAAE,SAAS,IAAI,GAAG;YACpC,YAAY,EAAE,OAAO,EAAE,YAAY,IAAI,EAAE;SAC1C,CAAC;QAEF,yDAAyD;QACzD,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,IAAI,CAAC;YACzC,MAAM,YAAY,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;YAE9C,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CAAC,2BAA2B,OAAO,EAAE,CAAC,CAAC;YACxD,CAAC;YAED,aAAa,CAAC,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAC;YAC/C,aAAa,CAAC,YAAY,GAAG,YAAY,CAAC,YAAY,CAAC;YACvD,aAAa,CAAC,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;YAC7C,aAAa,CAAC,oBAAoB,GAAG,YAAY,CAAC,oBAAoB,CAAC;QACzE,CAAC;aAAM,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YAC9B,yBAAyB;YACzB,aAAa,CAAC,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC;YACjD,aAAa,CAAC,YAAY,GAAG,cAAc,CAAC,YAAY,CAAC;YACzD,aAAa,CAAC,OAAO,GAAG,cAAc,CAAC,OAAO,CAAC;YAC/C,aAAa,CAAC,oBAAoB,GAAG,cAAc,CAAC,oBAAoB,CAAC;QAC3E,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,6BAA6B,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,0BAA0B;QAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QAE/D,4BAA4B;QAC5B,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,CAAC,CAAC,MAAM,CAAC,WAAW,YAAY,YAAY,CAAC,EAAE,CAAC;YAClD,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;QAChF,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,MAAM,CAAC,UAAU,GAAG,CAAC,EAAE,CAAC;YAClE,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;QACvF,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,MAAM,CAAC,YAAY,GAAG,CAAC,IAAI,MAAM,CAAC,YAAY,GAAG,CAAC,EAAE,CAAC;YACjG,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;QACxE,CAAC;QAED,OAAO,MAAuB,CAAC;IACjC,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK;QACH,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,IAAI,IAAI,CAAC,MAAM,IAAI,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;gBAC3D,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YACtB,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,QAAQ;QACN,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;OAuBG;IACH,sBAAsB,CAAC,OAAiC;QACtD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACvE,CAAC;QAED,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,IAAI,CAAC;QAEvC,kCAAkC;QAClC,MAAM,SAAS,GAAoC;YACjD,KAAK,EAAE,CAAC,EAAG,gCAAgC;YAC3C,IAAI,EAAE,CAAC,EAAI,uBAAuB;YAClC,IAAI,EAAE,CAAC,EAAI,uBAAuB;YAClC,IAAI,EAAE,CAAC,EAAI,uBAAuB;SACnC,CAAC;QAEF,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,mBAAmB,MAAM,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,wCAAwC;QACxC,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;QAC7B,MAAM,aAAa,GAAG,IAAI,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAE3E,OAAO,IAAI,gBAAgB,CAAC,aAAa,CAAC,CAAC;IAC7C,CAAC;CACF"}

package/dist/StreamingSession.d.ts

@@ -0,0 +1,88 @@
+/**
+ * TypeScript wrapper for native StreamingSession
+ */
+import type { FeedResult } from './types.js';
+/**
+ * Streaming diarization session
+ *
+ * Maintains state across incremental audio feed calls for true real-time
+ * speaker diarization. State is kept in native C code for efficiency.
+ *
+ * @example
+ * ```typescript
+ * const session = model.createStreamingSession({ preset: '2s' });
+ *
+ * // Feed audio chunks as they arrive
+ * const result1 = session.feed(chunk1);
+ * const result2 = session.feed(chunk2);
+ *
+ * // Get total frames processed
+ * console.log(session.totalFrames);
+ *
+ * // Reset for new audio stream
+ * session.reset();
+ *
+ * // Clean up
+ * session.close();
+ * ```
+ */
+export declare class StreamingSession {
+    private native;
+    private _closed;
+    /**
+     * Create a new streaming session
+     * @param native - Native StreamingSession instance from binding
+     * @internal
+     */
+    constructor(native: any);
+    /**
+     * Feed audio samples and get predictions for this chunk
+     *
+     * @param audio - Audio samples as Float32Array (16kHz mono)
+     * @returns Predictions for the new frames in this chunk
+     * @throws Error if session is closed or audio is invalid
+     *
+     * @example
+     * ```typescript
+     * const audio = new Float32Array(48000); // 3 seconds
+     * const result = session.feed(audio);
+     * console.log(`Got ${result.frameCount} new frames`);
+     * ```
+     */
+    feed(audio: Float32Array): FeedResult;
+    /**
+     * Flush remaining buffered audio at end of stream
+     *
+     * Call this when the audio stream ends to process any remaining
+     * buffered audio that hasn't been output yet due to latency buffering.
+     *
+     * @returns Final predictions for buffered audio
+     * @throws Error if session is closed
+     */
+    flush(): FeedResult;
+    /**
+     * Reset the streaming state for a new audio stream
+     *
+     * Clears all internal buffers (spkcache, fifo, mel overlap) while
+     * keeping the model loaded. Use this when starting a new recording.
+     *
+     * @throws Error if session is closed
+     */
+    reset(): void;
+    /**
+     * Close the session and free native resources
+     *
+     * After calling close(), the session cannot be used.
+     * Calling close() multiple times is safe (idempotent).
+     */
+    close(): void;
+    /**
+     * Get total frames output so far
+     */
+    get totalFrames(): number;
+    /**
+     * Check if the session is closed
+     */
+    get isClosed(): boolean;
+}
+//# sourceMappingURL=StreamingSession.d.ts.map

package/dist/StreamingSession.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"StreamingSession.d.ts","sourceRoot":"","sources":["../src/StreamingSession.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAmB,MAAM,YAAY,CAAC;AAE9D;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAM;IACpB,OAAO,CAAC,OAAO,CAAkB;IAEjC;;;;OAIG;gBACS,MAAM,EAAE,GAAG;IAIvB;;;;;;;;;;;;;OAaG;IACH,IAAI,CAAC,KAAK,EAAE,YAAY,GAAG,UAAU;IAiBrC;;;;;;;;OAQG;IACH,KAAK,IAAI,UAAU;IAanB;;;;;;;OAOG;IACH,KAAK,IAAI,IAAI;IAOb;;;;;OAKG;IACH,KAAK,IAAI,IAAI;IASb;;OAEG;IACH,IAAI,WAAW,IAAI,MAAM,CAKxB;IAED;;OAEG;IACH,IAAI,QAAQ,IAAI,OAAO,CAEtB;CACF"}

package/dist/StreamingSession.js

@@ -0,0 +1,128 @@
+/**
+ * TypeScript wrapper for native StreamingSession
+ */
+/**
+ * Streaming diarization session
+ *
+ * Maintains state across incremental audio feed calls for true real-time
+ * speaker diarization. State is kept in native C code for efficiency.
+ *
+ * @example
+ * ```typescript
+ * const session = model.createStreamingSession({ preset: '2s' });
+ *
+ * // Feed audio chunks as they arrive
+ * const result1 = session.feed(chunk1);
+ * const result2 = session.feed(chunk2);
+ *
+ * // Get total frames processed
+ * console.log(session.totalFrames);
+ *
+ * // Reset for new audio stream
+ * session.reset();
+ *
+ * // Clean up
+ * session.close();
+ * ```
+ */
+export class StreamingSession {
+    /**
+     * Create a new streaming session
+     * @param native - Native StreamingSession instance from binding
+     * @internal
+     */
+    constructor(native) {
+        this._closed = false;
+        this.native = native;
+    }
+    /**
+     * Feed audio samples and get predictions for this chunk
+     *
+     * @param audio - Audio samples as Float32Array (16kHz mono)
+     * @returns Predictions for the new frames in this chunk
+     * @throws Error if session is closed or audio is invalid
+     *
+     * @example
+     * ```typescript
+     * const audio = new Float32Array(48000); // 3 seconds
+     * const result = session.feed(audio);
+     * console.log(`Got ${result.frameCount} new frames`);
+     * ```
+     */
+    feed(audio) {
+        if (this._closed) {
+            throw new Error('Session is closed');
+        }
+        if (!(audio instanceof Float32Array)) {
+            throw new TypeError('audio must be a Float32Array');
+        }
+        const result = this.native.feed(audio);
+        return {
+            predictions: result.predictions,
+            frameCount: result.frameCount,
+        };
+    }
+    /**
+     * Flush remaining buffered audio at end of stream
+     *
+     * Call this when the audio stream ends to process any remaining
+     * buffered audio that hasn't been output yet due to latency buffering.
+     *
+     * @returns Final predictions for buffered audio
+     * @throws Error if session is closed
+     */
+    flush() {
+        if (this._closed) {
+            throw new Error('Session is closed');
+        }
+        const result = this.native.flush();
+        return {
+            predictions: result.predictions,
+            frameCount: result.frameCount,
+        };
+    }
+    /**
+     * Reset the streaming state for a new audio stream
+     *
+     * Clears all internal buffers (spkcache, fifo, mel overlap) while
+     * keeping the model loaded. Use this when starting a new recording.
+     *
+     * @throws Error if session is closed
+     */
+    reset() {
+        if (this._closed) {
+            throw new Error('Session is closed');
+        }
+        this.native.reset();
+    }
+    /**
+     * Close the session and free native resources
+     *
+     * After calling close(), the session cannot be used.
+     * Calling close() multiple times is safe (idempotent).
+     */
+    close() {
+        if (!this._closed) {
+            if (this.native && typeof this.native.close === 'function') {
+                this.native.close();
+            }
+            this._closed = true;
+        }
+    }
+    /**
+     * Get total frames output so far
+     */
+    get totalFrames() {
+        if (this._closed) {
+            return 0;
+        }
+        return this.native.getTotalFrames();
+    }
+    /**
+     * Check if the session is closed
+     */
+    get isClosed() {
+        return this._closed;
+    }
+}
+//# sourceMappingURL=StreamingSession.js.map

package/dist/StreamingSession.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"StreamingSession.js","sourceRoot":"","sources":["../src/StreamingSession.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,OAAO,gBAAgB;IAI3B;;;;OAIG;IACH,YAAY,MAAW;QAPf,YAAO,GAAY,KAAK,CAAC;QAQ/B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,IAAI,CAAC,KAAmB;QACtB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,CAAC,KAAK,YAAY,YAAY,CAAC,EAAE,CAAC;YACrC,MAAM,IAAI,SAAS,CAAC,8BAA8B,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEvC,OAAO;YACL,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,UAAU,EAAE,MAAM,CAAC,UAAU;SAC9B,CAAC;IACJ,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK;QACH,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAEnC,OAAO;YACL,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,UAAU,EAAE,MAAM,CAAC,UAAU;SAC9B,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACH,KAAK;QACH,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAED;;;;;OAKG;IACH,KAAK;QACH,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,IAAI,CAAC,MAAM,IAAI,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;gBAC3D,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YACtB,CAAC;YACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,IAAI,WAAW;QACb,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO,CAAC,CAAC;QACX,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;CACF"}

package/dist/binding.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Get the native binding for the current platform
+ * Detects platform and architecture, loads the appropriate platform-specific package
+ * @returns The native module binding
+ * @throws Error if platform is not supported or binding cannot be loaded
+ */
+export declare function getBinding(): any;
+//# sourceMappingURL=binding.d.ts.map

package/dist/binding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"binding.d.ts","sourceRoot":"","sources":["../src/binding.ts"],"names":[],"mappings":"AAMA;;;;;GAKG;AACH,wBAAgB,UAAU,IAAI,GAAG,CA4BhC"}

package/dist/binding.js

@@ -0,0 +1,35 @@
+import { createRequire } from 'module';
+const require = createRequire(import.meta.url);
+let cachedBinding = null;
+/**
+ * Get the native binding for the current platform
+ * Detects platform and architecture, loads the appropriate platform-specific package
+ * @returns The native module binding
+ * @throws Error if platform is not supported or binding cannot be loaded
+ */
+export function getBinding() {
+    if (cachedBinding)
+        return cachedBinding;
+    const platform = process.platform;
+    const arch = process.arch;
+    let packageName;
+    if (platform === 'darwin' && arch === 'arm64') {
+        packageName = '@streaming-sortformer-node/darwin-arm64';
+    }
+    else if (platform === 'darwin' && arch === 'x64') {
+        packageName = '@streaming-sortformer-node/darwin-x64';
+    }
+    else {
+        throw new Error(`Unsupported platform: ${platform}-${arch}. ` +
+            `streaming-sortformer-node currently supports: darwin-arm64, darwin-x64`);
+    }
+    try {
+        cachedBinding = require(packageName);
+        return cachedBinding;
+    }
+    catch (e) {
+        throw new Error(`Failed to load native binding from ${packageName}. ` +
+            `Make sure the package is installed: npm install ${packageName}`);
+    }
+}
+//# sourceMappingURL=binding.js.map

package/dist/binding.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"binding.js","sourceRoot":"","sources":["../src/binding.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AAEvC,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAE/C,IAAI,aAAa,GAAQ,IAAI,CAAC;AAE9B;;;;;GAKG;AACH,MAAM,UAAU,UAAU;IACxB,IAAI,aAAa;QAAE,OAAO,aAAa,CAAC;IAExC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IAClC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAE1B,IAAI,WAAmB,CAAC;IAExB,IAAI,QAAQ,KAAK,QAAQ,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;QAC9C,WAAW,GAAG,yCAAyC,CAAC;IAC1D,CAAC;SAAM,IAAI,QAAQ,KAAK,QAAQ,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;QACnD,WAAW,GAAG,uCAAuC,CAAC;IACxD,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CACb,yBAAyB,QAAQ,IAAI,IAAI,IAAI;YAC7C,wEAAwE,CACzE,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,aAAa,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;QACrC,OAAO,aAAa,CAAC;IACvB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CACb,sCAAsC,WAAW,IAAI;YACrD,mDAAmD,WAAW,EAAE,CACjE,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,5 @@
+export { Sortformer } from './Sortformer.js';
+export { StreamingSession } from './StreamingSession.js';
+export type { LoadOptions, DiarizeOptions, DiarizeResult, LatencyPreset, DiarizeMode, StreamingPreset, StreamingSessionOptions, FeedResult, } from './types.js';
+export { LATENCY_PRESETS, OFFLINE_PARAMS } from './presets.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,YAAY,EACV,WAAW,EACX,cAAc,EACd,aAAa,EACb,aAAa,EACb,WAAW,EACX,eAAe,EACf,uBAAuB,EACvB,UAAU,GACX,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC"}

package/dist/index.js

@@ -0,0 +1,4 @@
+export { Sortformer } from './Sortformer.js';
+export { StreamingSession } from './StreamingSession.js';
+export { LATENCY_PRESETS, OFFLINE_PARAMS } from './presets.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAWzD,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC"}

package/dist/presets.d.ts

@@ -0,0 +1,45 @@
+/**
+ * Latency presets for streaming diarization
+ * Maps preset names to their corresponding parameter configurations
+ */
+import type { LatencyPreset } from './types';
+/**
+ * Streaming latency preset parameters
+ * Each preset controls chunk processing, buffering, and speaker cache update behavior
+ */
+export interface PresetParams {
+    /** Chunk length in frames (16kHz, hop=160) */
+    chunkLen: number;
+    /** Right context frames for conformer processing */
+    rightContext: number;
+    /** FIFO buffer length in frames */
+    fifoLen: number;
+    /** Speaker cache update period in frames */
+    spkcacheUpdatePeriod: number;
+}
+/**
+ * Streaming latency presets
+ * - 'low': ~188ms latency, minimal buffering
+ * - '2s': ~2 second latency
+ * - '3s': ~3 second latency
+ * - '5s': ~5 second latency
+ */
+export declare const LATENCY_PRESETS: Record<LatencyPreset, PresetParams>;
+/**
+ * Offline mode parameters
+ * Used when mode='offline' to process entire audio at once
+ */
+export declare const OFFLINE_PARAMS: PresetParams;
+/**
+ * Get preset parameters by name
+ * @param preset - Latency preset name
+ * @returns Preset parameters
+ * @throws Error if preset is not found
+ */
+export declare function getPresetParams(preset: LatencyPreset): PresetParams;
+/**
+ * Get default preset parameters for offline mode
+ * @returns Offline mode parameters
+ */
+export declare function getOfflineParams(): PresetParams;
+//# sourceMappingURL=presets.d.ts.map

package/dist/presets.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"presets.d.ts","sourceRoot":"","sources":["../src/presets.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAE7C;;;GAGG;AACH,MAAM,WAAW,YAAY;IAC3B,8CAA8C;IAC9C,QAAQ,EAAE,MAAM,CAAC;IACjB,oDAAoD;IACpD,YAAY,EAAE,MAAM,CAAC;IACrB,mCAAmC;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,4CAA4C;IAC5C,oBAAoB,EAAE,MAAM,CAAC;CAC9B;AAED;;;;;;GAMG;AACH,eAAO,MAAM,eAAe,EAAE,MAAM,CAAC,aAAa,EAAE,YAAY,CAyBtD,CAAC;AAEX;;;GAGG;AACH,eAAO,MAAM,cAAc,EAAE,YAKnB,CAAC;AAEX;;;;;GAKG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,aAAa,GAAG,YAAY,CAMnE;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,IAAI,YAAY,CAE/C"}

package/dist/presets.js

@@ -0,0 +1,68 @@
+/**
+ * Latency presets for streaming diarization
+ * Maps preset names to their corresponding parameter configurations
+ */
+/**
+ * Streaming latency presets
+ * - 'low': ~188ms latency, minimal buffering
+ * - '2s': ~2 second latency
+ * - '3s': ~3 second latency
+ * - '5s': ~5 second latency
+ */
+export const LATENCY_PRESETS = {
+    'low': {
+        chunkLen: 6,
+        rightContext: 7,
+        fifoLen: 188,
+        spkcacheUpdatePeriod: 144,
+    },
+    '2s': {
+        chunkLen: 15,
+        rightContext: 10,
+        fifoLen: 100,
+        spkcacheUpdatePeriod: 144,
+    },
+    '3s': {
+        chunkLen: 30,
+        rightContext: 7,
+        fifoLen: 100,
+        spkcacheUpdatePeriod: 100,
+    },
+    '5s': {
+        chunkLen: 55,
+        rightContext: 7,
+        fifoLen: 100,
+        spkcacheUpdatePeriod: 100,
+    },
+};
+/**
+ * Offline mode parameters
+ * Used when mode='offline' to process entire audio at once
+ */
+export const OFFLINE_PARAMS = {
+    chunkLen: 188,
+    rightContext: 1,
+    fifoLen: 0,
+    spkcacheUpdatePeriod: 188,
+};
+/**
+ * Get preset parameters by name
+ * @param preset - Latency preset name
+ * @returns Preset parameters
+ * @throws Error if preset is not found
+ */
+export function getPresetParams(preset) {
+    const params = LATENCY_PRESETS[preset];
+    if (!params) {
+        throw new Error(`Unknown latency preset: ${preset}`);
+    }
+    return params;
+}
+/**
+ * Get default preset parameters for offline mode
+ * @returns Offline mode parameters
+ */
+export function getOfflineParams() {
+    return OFFLINE_PARAMS;
+}
+//# sourceMappingURL=presets.js.map

package/dist/presets.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"presets.js","sourceRoot":"","sources":["../src/presets.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAmBH;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,eAAe,GAAwC;IAClE,KAAK,EAAE;QACL,QAAQ,EAAE,CAAC;QACX,YAAY,EAAE,CAAC;QACf,OAAO,EAAE,GAAG;QACZ,oBAAoB,EAAE,GAAG;KAC1B;IACD,IAAI,EAAE;QACJ,QAAQ,EAAE,EAAE;QACZ,YAAY,EAAE,EAAE;QAChB,OAAO,EAAE,GAAG;QACZ,oBAAoB,EAAE,GAAG;KAC1B;IACD,IAAI,EAAE;QACJ,QAAQ,EAAE,EAAE;QACZ,YAAY,EAAE,CAAC;QACf,OAAO,EAAE,GAAG;QACZ,oBAAoB,EAAE,GAAG;KAC1B;IACD,IAAI,EAAE;QACJ,QAAQ,EAAE,EAAE;QACZ,YAAY,EAAE,CAAC;QACf,OAAO,EAAE,GAAG;QACZ,oBAAoB,EAAE,GAAG;KAC1B;CACO,CAAC;AAEX;;;GAGG;AACH,MAAM,CAAC,MAAM,cAAc,GAAiB;IAC1C,QAAQ,EAAE,GAAG;IACb,YAAY,EAAE,CAAC;IACf,OAAO,EAAE,CAAC;IACV,oBAAoB,EAAE,GAAG;CACjB,CAAC;AAEX;;;;;GAKG;AACH,MAAM,UAAU,eAAe,CAAC,MAAqB;IACnD,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IACvC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,2BAA2B,MAAM,EAAE,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,gBAAgB;IAC9B,OAAO,cAAc,CAAC;AACxB,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,107 @@
+/**
+ * TypeScript type definitions for streaming-sortformer-node
+ */
+/**
+ * Diarization mode: offline processes entire audio at once,
+ * streaming processes audio in chunks with latency control
+ */
+export type DiarizeMode = 'offline' | 'streaming';
+/**
+ * Latency preset for streaming mode
+ * - 'low': ~188ms latency, minimal buffering
+ * - '2s': ~2 second latency
+ * - '3s': ~3 second latency
+ * - '5s': ~5 second latency
+ */
+export type LatencyPreset = 'low' | '2s' | '3s' | '5s';
+/**
+ * Options for loading a SortFormer model
+ */
+export interface LoadOptions {
+    /**
+     * Number of CPU threads to use for inference
+     * @default auto-detected based on CPU cores
+     */
+    threads?: number;
+}
+/**
+ * Options for diarization inference
+ */
+export interface DiarizeOptions {
+    /**
+     * Diarization mode: 'offline' or 'streaming'
+     * @default 'offline'
+     */
+    mode?: DiarizeMode;
+    /**
+     * Latency preset for streaming mode
+     * Only used when mode='streaming'
+     * @default '2s'
+     */
+    latency?: LatencyPreset;
+    /**
+     * Speaker activity threshold (0.0 to 1.0)
+     * Frames with prediction >= threshold are considered active
+     * @default 0.5
+     */
+    threshold?: number;
+    /**
+     * Median filter window size for smoothing predictions
+     * Must be odd number >= 1
+     * @default 11
+     */
+    medianFilter?: number;
+}
+/**
+ * Result of diarization inference
+ */
+export interface DiarizeResult {
+    /**
+     * RTTM format output (speaker diarization segments)
+     * Format: SPEAKER <filename> <channel> <start> <duration> <conf> <spk_type> <spk_id> <score>
+     */
+    rttm: string;
+    /**
+     * Raw per-frame speaker activity predictions
+     * Shape: [frameCount, 4] (4 speakers max)
+     * Values: 0.0 to 1.0 (probability of speaker activity)
+     */
+    predictions: Float32Array;
+    /**
+     * Number of frames in the output
+     */
+    frameCount: number;
+    /**
+     * Number of speakers detected (1-4)
+     */
+    speakerCount: number;
+}
+/**
+ * Streaming preset type
+ */
+export type StreamingPreset = 'low' | '2s' | '3s' | '5s';
+/**
+ * Options for creating a streaming session
+ */
+export interface StreamingSessionOptions {
+    /**
+     * Latency preset
+     * @default '2s'
+     */
+    preset?: StreamingPreset;
+}
+/**
+ * Result from feeding audio to streaming session
+ */
+export interface FeedResult {
+    /**
+     * Per-frame speaker predictions for this chunk
+     * Shape: [frameCount, 4]
+     */
+    predictions: Float32Array;
+    /**
+     * Number of new frames in this result
+     */
+    frameCount: number;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH;;;GAGG;AACH,MAAM,MAAM,WAAW,GAAG,SAAS,GAAG,WAAW,CAAC;AAElD;;;;;;GAMG;AACH,MAAM,MAAM,aAAa,GAAG,KAAK,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;AAEvD;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B;;;OAGG;IACH,IAAI,CAAC,EAAE,WAAW,CAAC;IAEnB;;;;OAIG;IACH,OAAO,CAAC,EAAE,aAAa,CAAC;IAExB;;;;OAIG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;;;OAIG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;;OAGG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;;;OAIG;IACH,WAAW,EAAE,YAAY,CAAC;IAE1B;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IAEnB;;OAEG;IACH,YAAY,EAAE,MAAM,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,KAAK,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;AAEzD;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC;;;OAGG;IACH,MAAM,CAAC,EAAE,eAAe,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB;;;OAGG;IACH,WAAW,EAAE,YAAY,CAAC;IAE1B;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;CACpB"}

package/dist/types.js

@@ -0,0 +1,5 @@
+/**
+ * TypeScript type definitions for streaming-sortformer-node
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "streaming-sortformer-node",
+  "version": "0.1.0",
+  "description": "Node.js bindings for SortFormer streaming speaker diarization",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist",
+    "src"
+  ],
+  "dependencies": {
+    "node-addon-api": "^7.0.0"
+  },
+  "devDependencies": {
+    "typescript": "^5.0.0",
+    "cmake-js": "^7.0.0"
+  },
+  "optionalDependencies": {
+    "@streaming-sortformer-node/darwin-x64": "0.1.0",
+    "@streaming-sortformer-node/darwin-arm64": "0.1.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "echo 'No tests yet'",
+    "clean": "rm -rf dist"
+  }
+}