@samyok/annoy 1.0.0 → 1.1.0

package/dist/index.d.mts

@@ -0,0 +1,29 @@
+type Metric = "angular" | "euclidean" | "manhattan" | "dot";
+interface NearestNeighborsResult {
+    neighbors: number[];
+    distances: number[];
+}
+declare class AnnoyIndex {
+    private index;
+    constructor(f: number, metric?: Metric);
+    addItem(index: number, vector: number[]): void;
+    build(numTrees: number, numThreads?: number): void;
+    unbuild(): void;
+    save(filename: string, prefault?: boolean): void;
+    load(filename: string, prefault?: boolean): void;
+    unload(): void;
+    getDistance(i: number, j: number): number;
+    getNnsByItem(item: number, n: number, searchK: number, includeDistances: true): NearestNeighborsResult;
+    getNnsByItem(item: number, n: number, searchK?: number, includeDistances?: false): number[];
+    getNnsByVector(vector: number[], n: number, searchK: number, includeDistances: true): NearestNeighborsResult;
+    getNnsByVector(vector: number[], n: number, searchK?: number, includeDistances?: false): number[];
+    getNItems(): number;
+    getNTrees(): number;
+    getItem(item: number): number[];
+    setSeed(seed: number): void;
+    onDiskBuild(filename: string): void;
+    verbose(v: boolean): void;
+    getF(): number;
+}
+
+export { AnnoyIndex, type Metric, type NearestNeighborsResult, AnnoyIndex as default };

package/dist/index.d.ts

@@ -1,9 +1,9 @@
-export type Metric = "angular" | "euclidean" | "manhattan" | "dot";
-export interface NearestNeighborsResult {
+type Metric = "angular" | "euclidean" | "manhattan" | "dot";
+interface NearestNeighborsResult {
     neighbors: number[];
     distances: number[];
 }
-export declare class AnnoyIndex {
+declare class AnnoyIndex {
     private index;
     constructor(f: number, metric?: Metric);
     addItem(index: number, vector: number[]): void;
@@ -25,4 +25,5 @@ export declare class AnnoyIndex {
     verbose(v: boolean): void;
     getF(): number;
 }
-export default AnnoyIndex;
+
+export { AnnoyIndex, type Metric, type NearestNeighborsResult, AnnoyIndex as default };

package/dist/index.js

@@ -1,103 +1,122 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  AnnoyIndex: () => AnnoyIndex,
+  default: () => index_default
 });
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
+module.exports = __toCommonJS(index_exports);
+var import_module = require("module");
+var path = __toESM(require("path"));
+var _require = (0, import_module.createRequire)(__filename);
+var addon = _require(
+  path.join(__dirname, "..", "build", "Release", "annoy.node")
+);
+var AnnoyIndex = class {
+  constructor(f, metric = "angular") {
+    const constructors = {
+      angular: addon.AnnoyIndexAngular,
+      euclidean: addon.AnnoyIndexEuclidean,
+      manhattan: addon.AnnoyIndexManhattan,
+      dot: addon.AnnoyIndexDotProduct
     };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AnnoyIndex = void 0;
-const path = __importStar(require("path"));
-const addon = require(path.join(__dirname, "..", "build", "Release", "annoy.node"));
-class AnnoyIndex {
-    constructor(f, metric = "angular") {
-        const constructors = {
-            angular: addon.AnnoyIndexAngular,
-            euclidean: addon.AnnoyIndexEuclidean,
-            manhattan: addon.AnnoyIndexManhattan,
-            dot: addon.AnnoyIndexDotProduct,
-        };
-        const Ctor = constructors[metric];
-        if (!Ctor) {
-            throw new Error(`Unknown metric "${metric}". Use: angular, euclidean, manhattan, dot`);
-        }
-        this.index = new Ctor(f);
-    }
-    addItem(index, vector) {
-        this.index.addItem(index, vector);
-    }
-    build(numTrees, numThreads = -1) {
-        this.index.build(numTrees, numThreads);
-    }
-    unbuild() {
-        this.index.unbuild();
-    }
-    save(filename, prefault = false) {
-        this.index.save(filename, prefault);
-    }
-    load(filename, prefault = false) {
-        this.index.load(filename, prefault);
-    }
-    unload() {
-        this.index.unload();
-    }
-    getDistance(i, j) {
-        return this.index.getDistance(i, j);
-    }
-    getNnsByItem(item, n, searchK = -1, includeDistances = false) {
-        return this.index.getNnsByItem(item, n, searchK, includeDistances);
-    }
-    getNnsByVector(vector, n, searchK = -1, includeDistances = false) {
-        return this.index.getNnsByVector(vector, n, searchK, includeDistances);
-    }
-    getNItems() {
-        return this.index.getNItems();
-    }
-    getNTrees() {
-        return this.index.getNTrees();
-    }
-    getItem(item) {
-        return this.index.getItem(item);
-    }
-    setSeed(seed) {
-        this.index.setSeed(seed);
-    }
-    onDiskBuild(filename) {
-        this.index.onDiskBuild(filename);
-    }
-    verbose(v) {
-        this.index.verbose(v);
-    }
-    getF() {
-        return this.index.getF();
-    }
-}
-exports.AnnoyIndex = AnnoyIndex;
-exports.default = AnnoyIndex;
+    const Ctor = constructors[metric];
+    if (!Ctor) {
+      throw new Error(
+        `Unknown metric "${metric}". Use: angular, euclidean, manhattan, dot`
+      );
+    }
+    this.index = new Ctor(f);
+  }
+  addItem(index, vector) {
+    this.index.addItem(index, vector);
+  }
+  build(numTrees, numThreads = -1) {
+    this.index.build(numTrees, numThreads);
+  }
+  unbuild() {
+    this.index.unbuild();
+  }
+  save(filename, prefault = false) {
+    this.index.save(filename, prefault);
+  }
+  load(filename, prefault = false) {
+    this.index.load(filename, prefault);
+  }
+  unload() {
+    this.index.unload();
+  }
+  getDistance(i, j) {
+    return this.index.getDistance(i, j);
+  }
+  getNnsByItem(item, n, searchK = -1, includeDistances = false) {
+    return this.index.getNnsByItem(
+      item,
+      n,
+      searchK,
+      includeDistances
+    );
+  }
+  getNnsByVector(vector, n, searchK = -1, includeDistances = false) {
+    return this.index.getNnsByVector(
+      vector,
+      n,
+      searchK,
+      includeDistances
+    );
+  }
+  getNItems() {
+    return this.index.getNItems();
+  }
+  getNTrees() {
+    return this.index.getNTrees();
+  }
+  getItem(item) {
+    return this.index.getItem(item);
+  }
+  setSeed(seed) {
+    this.index.setSeed(seed);
+  }
+  onDiskBuild(filename) {
+    this.index.onDiskBuild(filename);
+  }
+  verbose(v) {
+    this.index.verbose(v);
+  }
+  getF() {
+    return this.index.getF();
+  }
+};
+var index_default = AnnoyIndex;
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AnnoyIndex
+});

package/dist/index.mjs

@@ -0,0 +1,95 @@
+// node_modules/tsup/assets/esm_shims.js
+import path from "path";
+import { fileURLToPath } from "url";
+var getFilename = () => fileURLToPath(import.meta.url);
+var getDirname = () => path.dirname(getFilename());
+var __dirname = /* @__PURE__ */ getDirname();
+var __filename = /* @__PURE__ */ getFilename();
+
+// src/index.ts
+import { createRequire } from "module";
+import * as path2 from "path";
+var _require = createRequire(__filename);
+var addon = _require(
+  path2.join(__dirname, "..", "build", "Release", "annoy.node")
+);
+var AnnoyIndex = class {
+  constructor(f, metric = "angular") {
+    const constructors = {
+      angular: addon.AnnoyIndexAngular,
+      euclidean: addon.AnnoyIndexEuclidean,
+      manhattan: addon.AnnoyIndexManhattan,
+      dot: addon.AnnoyIndexDotProduct
+    };
+    const Ctor = constructors[metric];
+    if (!Ctor) {
+      throw new Error(
+        `Unknown metric "${metric}". Use: angular, euclidean, manhattan, dot`
+      );
+    }
+    this.index = new Ctor(f);
+  }
+  addItem(index, vector) {
+    this.index.addItem(index, vector);
+  }
+  build(numTrees, numThreads = -1) {
+    this.index.build(numTrees, numThreads);
+  }
+  unbuild() {
+    this.index.unbuild();
+  }
+  save(filename, prefault = false) {
+    this.index.save(filename, prefault);
+  }
+  load(filename, prefault = false) {
+    this.index.load(filename, prefault);
+  }
+  unload() {
+    this.index.unload();
+  }
+  getDistance(i, j) {
+    return this.index.getDistance(i, j);
+  }
+  getNnsByItem(item, n, searchK = -1, includeDistances = false) {
+    return this.index.getNnsByItem(
+      item,
+      n,
+      searchK,
+      includeDistances
+    );
+  }
+  getNnsByVector(vector, n, searchK = -1, includeDistances = false) {
+    return this.index.getNnsByVector(
+      vector,
+      n,
+      searchK,
+      includeDistances
+    );
+  }
+  getNItems() {
+    return this.index.getNItems();
+  }
+  getNTrees() {
+    return this.index.getNTrees();
+  }
+  getItem(item) {
+    return this.index.getItem(item);
+  }
+  setSeed(seed) {
+    this.index.setSeed(seed);
+  }
+  onDiskBuild(filename) {
+    this.index.onDiskBuild(filename);
+  }
+  verbose(v) {
+    this.index.verbose(v);
+  }
+  getF() {
+    return this.index.getF();
+  }
+};
+var index_default = AnnoyIndex;
+export {
+  AnnoyIndex,
+  index_default as default
+};

package/dist/test.d.mts

@@ -0,0 +1,2 @@
+
+export {  }

package/dist/test.d.ts

@@ -1 +1,2 @@
-export {};
+
+export {  }

package/dist/test.js

@@ -1,171 +1,244 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+
+// src/index.ts
+var import_module = require("module");
+var path = __toESM(require("path"));
+var _require = (0, import_module.createRequire)(__filename);
+var addon = _require(
+  path.join(__dirname, "..", "build", "Release", "annoy.node")
+);
+var AnnoyIndex = class {
+  constructor(f, metric = "angular") {
+    const constructors = {
+      angular: addon.AnnoyIndexAngular,
+      euclidean: addon.AnnoyIndexEuclidean,
+      manhattan: addon.AnnoyIndexManhattan,
+      dot: addon.AnnoyIndexDotProduct
     };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_1 = require("./index");
-const fs = __importStar(require("fs"));
-const path = __importStar(require("path"));
-const DIMS = 40;
-const NUM_ITEMS = 1000;
-const NUM_TREES = 10;
-const tmpFile = path.join(__dirname, "..", "test_index.ann");
-let passed = 0;
-let failed = 0;
-function assert(condition, msg) {
-    if (condition) {
-        passed++;
-        console.log(`  PASS: ${msg}`);
-    }
-    else {
-        failed++;
-        console.error(`  FAIL: ${msg}`);
+    const Ctor = constructors[metric];
+    if (!Ctor) {
+      throw new Error(
+        `Unknown metric "${metric}". Use: angular, euclidean, manhattan, dot`
+      );
     }
+    this.index = new Ctor(f);
+  }
+  addItem(index, vector) {
+    this.index.addItem(index, vector);
+  }
+  build(numTrees, numThreads = -1) {
+    this.index.build(numTrees, numThreads);
+  }
+  unbuild() {
+    this.index.unbuild();
+  }
+  save(filename, prefault = false) {
+    this.index.save(filename, prefault);
+  }
+  load(filename, prefault = false) {
+    this.index.load(filename, prefault);
+  }
+  unload() {
+    this.index.unload();
+  }
+  getDistance(i, j) {
+    return this.index.getDistance(i, j);
+  }
+  getNnsByItem(item, n, searchK = -1, includeDistances = false) {
+    return this.index.getNnsByItem(
+      item,
+      n,
+      searchK,
+      includeDistances
+    );
+  }
+  getNnsByVector(vector, n, searchK = -1, includeDistances = false) {
+    return this.index.getNnsByVector(
+      vector,
+      n,
+      searchK,
+      includeDistances
+    );
+  }
+  getNItems() {
+    return this.index.getNItems();
+  }
+  getNTrees() {
+    return this.index.getNTrees();
+  }
+  getItem(item) {
+    return this.index.getItem(item);
+  }
+  setSeed(seed) {
+    this.index.setSeed(seed);
+  }
+  onDiskBuild(filename) {
+    this.index.onDiskBuild(filename);
+  }
+  verbose(v) {
+    this.index.verbose(v);
+  }
+  getF() {
+    return this.index.getF();
+  }
+};
+
+// src/test.ts
+var fs = __toESM(require("fs"));
+var path2 = __toESM(require("path"));
+var DIMS = 40;
+var NUM_ITEMS = 1e3;
+var NUM_TREES = 10;
+var tmpFile = path2.join(__dirname, "..", "test_index.ann");
+var passed = 0;
+var failed = 0;
+function assert(condition, msg) {
+  if (condition) {
+    passed++;
+    console.log(`  PASS: ${msg}`);
+  } else {
+    failed++;
+    console.error(`  FAIL: ${msg}`);
+  }
 }
 function randomVector(dims) {
-    return Array.from({ length: dims }, () => Math.random() * 2 - 1);
+  return Array.from({ length: dims }, () => Math.random() * 2 - 1);
 }
 function cleanup() {
-    try {
-        fs.unlinkSync(tmpFile);
-    }
-    catch { }
+  try {
+    fs.unlinkSync(tmpFile);
+  } catch {
+  }
 }
 console.log("=== annoy-node test suite ===\n");
-// Test 1: Basic build and query (Angular)
 console.log("Test 1: Build and query (Angular)");
 {
-    const index = new index_1.AnnoyIndex(DIMS, "angular");
-    for (let i = 0; i < NUM_ITEMS; i++) {
-        index.addItem(i, randomVector(DIMS));
-    }
-    index.build(NUM_TREES);
-    assert(index.getNItems() === NUM_ITEMS, `getNItems() === ${NUM_ITEMS}`);
-    assert(index.getNTrees() === NUM_TREES, `getNTrees() === ${NUM_TREES}`);
-    assert(index.getF() === DIMS, `getF() === ${DIMS}`);
-    const neighbors = index.getNnsByItem(0, 10);
-    assert(Array.isArray(neighbors), "getNnsByItem returns array");
-    assert(neighbors.length === 10, "returns 10 neighbors");
-    assert(neighbors[0] === 0, "nearest neighbor of item 0 is itself");
+  const index = new AnnoyIndex(DIMS, "angular");
+  for (let i = 0; i < NUM_ITEMS; i++) {
+    index.addItem(i, randomVector(DIMS));
+  }
+  index.build(NUM_TREES);
+  assert(index.getNItems() === NUM_ITEMS, `getNItems() === ${NUM_ITEMS}`);
+  assert(index.getNTrees() === NUM_TREES, `getNTrees() === ${NUM_TREES}`);
+  assert(index.getF() === DIMS, `getF() === ${DIMS}`);
+  const neighbors = index.getNnsByItem(0, 10);
+  assert(Array.isArray(neighbors), "getNnsByItem returns array");
+  assert(neighbors.length === 10, "returns 10 neighbors");
+  assert(neighbors[0] === 0, "nearest neighbor of item 0 is itself");
 }
-// Test 2: Include distances
 console.log("\nTest 2: Include distances");
 {
-    const index = new index_1.AnnoyIndex(DIMS, "angular");
-    for (let i = 0; i < 100; i++) {
-        index.addItem(i, randomVector(DIMS));
-    }
-    index.build(NUM_TREES);
-    const result = index.getNnsByItem(0, 5, -1, true);
-    assert("neighbors" in result && "distances" in result, "result has neighbors and distances");
-    assert(result.neighbors.length === 5, "5 neighbors returned");
-    assert(result.distances.length === 5, "5 distances returned");
-    assert(result.distances[0] === 0, "distance to self is 0");
+  const index = new AnnoyIndex(DIMS, "angular");
+  for (let i = 0; i < 100; i++) {
+    index.addItem(i, randomVector(DIMS));
+  }
+  index.build(NUM_TREES);
+  const result = index.getNnsByItem(0, 5, -1, true);
+  assert(
+    "neighbors" in result && "distances" in result,
+    "result has neighbors and distances"
+  );
+  assert(result.neighbors.length === 5, "5 neighbors returned");
+  assert(result.distances.length === 5, "5 distances returned");
+  assert(result.distances[0] === 0, "distance to self is 0");
 }
-// Test 3: getNnsByVector
 console.log("\nTest 3: getNnsByVector");
 {
-    const index = new index_1.AnnoyIndex(DIMS, "euclidean");
-    const vectors = [];
-    for (let i = 0; i < 100; i++) {
-        const v = randomVector(DIMS);
-        vectors.push(v);
-        index.addItem(i, v);
-    }
-    index.build(NUM_TREES);
-    const result = index.getNnsByVector(vectors[0], 5);
-    assert(Array.isArray(result), "getNnsByVector returns array");
-    assert(result[0] === 0, "closest to vector[0] is item 0");
+  const index = new AnnoyIndex(DIMS, "euclidean");
+  const vectors = [];
+  for (let i = 0; i < 100; i++) {
+    const v = randomVector(DIMS);
+    vectors.push(v);
+    index.addItem(i, v);
+  }
+  index.build(NUM_TREES);
+  const result = index.getNnsByVector(vectors[0], 5);
+  assert(Array.isArray(result), "getNnsByVector returns array");
+  assert(result[0] === 0, "closest to vector[0] is item 0");
 }
-// Test 4: Save and load
 console.log("\nTest 4: Save and load");
 {
-    cleanup();
-    const index = new index_1.AnnoyIndex(DIMS, "angular");
-    for (let i = 0; i < 100; i++) {
-        index.addItem(i, randomVector(DIMS));
-    }
-    index.build(NUM_TREES);
-    index.save(tmpFile);
-    const loaded = new index_1.AnnoyIndex(DIMS, "angular");
-    loaded.load(tmpFile);
-    assert(loaded.getNItems() === 100, "loaded index has 100 items");
-    assert(loaded.getNTrees() === NUM_TREES, `loaded index has ${NUM_TREES} trees`);
-    const item = loaded.getItem(0);
-    assert(item.length === DIMS, `getItem returns vector of length ${DIMS}`);
-    cleanup();
+  cleanup();
+  const index = new AnnoyIndex(DIMS, "angular");
+  for (let i = 0; i < 100; i++) {
+    index.addItem(i, randomVector(DIMS));
+  }
+  index.build(NUM_TREES);
+  index.save(tmpFile);
+  const loaded = new AnnoyIndex(DIMS, "angular");
+  loaded.load(tmpFile);
+  assert(loaded.getNItems() === 100, "loaded index has 100 items");
+  assert(loaded.getNTrees() === NUM_TREES, `loaded index has ${NUM_TREES} trees`);
+  const item = loaded.getItem(0);
+  assert(item.length === DIMS, `getItem returns vector of length ${DIMS}`);
+  cleanup();
 }
-// Test 5: getDistance
 console.log("\nTest 5: getDistance");
 {
-    const index = new index_1.AnnoyIndex(DIMS, "euclidean");
-    const v = randomVector(DIMS);
-    index.addItem(0, v);
-    index.addItem(1, v);
-    index.build(1);
-    const dist = index.getDistance(0, 1);
-    assert(Math.abs(dist) < 1e-6, `distance between identical vectors is ~0 (got ${dist})`);
+  const index = new AnnoyIndex(DIMS, "euclidean");
+  const v = randomVector(DIMS);
+  index.addItem(0, v);
+  index.addItem(1, v);
+  index.build(1);
+  const dist = index.getDistance(0, 1);
+  assert(Math.abs(dist) < 1e-6, `distance between identical vectors is ~0 (got ${dist})`);
 }
-// Test 6: All metrics
 console.log("\nTest 6: All metrics");
 {
-    const metrics = ["angular", "euclidean", "manhattan", "dot"];
-    for (const metric of metrics) {
-        const index = new index_1.AnnoyIndex(DIMS, metric);
-        for (let i = 0; i < 50; i++) {
-            index.addItem(i, randomVector(DIMS));
-        }
-        index.build(5);
-        const nn = index.getNnsByItem(0, 5);
-        assert(nn.length === 5, `${metric}: returns 5 neighbors`);
+  const metrics = ["angular", "euclidean", "manhattan", "dot"];
+  for (const metric of metrics) {
+    const index = new AnnoyIndex(DIMS, metric);
+    for (let i = 0; i < 50; i++) {
+      index.addItem(i, randomVector(DIMS));
     }
+    index.build(5);
+    const nn = index.getNnsByItem(0, 5);
+    assert(nn.length === 5, `${metric}: returns 5 neighbors`);
+  }
 }
-// Test 7: setSeed reproducibility
 console.log("\nTest 7: setSeed reproducibility");
 {
-    const vectors = Array.from({ length: 100 }, () => randomVector(DIMS));
-    function buildWithSeed(seed) {
-        const idx = new index_1.AnnoyIndex(DIMS, "angular");
-        idx.setSeed(seed);
-        for (let i = 0; i < 100; i++) {
-            idx.addItem(i, vectors[i]);
-        }
-        idx.build(5);
-        return idx.getNnsByItem(0, 10);
+  let buildWithSeed = function(seed) {
+    const idx = new AnnoyIndex(DIMS, "angular");
+    idx.setSeed(seed);
+    for (let i = 0; i < 100; i++) {
+      idx.addItem(i, vectors[i]);
     }
-    const r1 = buildWithSeed(42);
-    const r2 = buildWithSeed(42);
-    assert(JSON.stringify(r1) === JSON.stringify(r2), "same seed produces same results");
+    idx.build(5);
+    return idx.getNnsByItem(0, 10);
+  };
+  buildWithSeed2 = buildWithSeed;
+  const vectors = Array.from({ length: 100 }, () => randomVector(DIMS));
+  const r1 = buildWithSeed(42);
+  const r2 = buildWithSeed(42);
+  assert(
+    JSON.stringify(r1) === JSON.stringify(r2),
+    "same seed produces same results"
+  );
 }
-console.log(`\n=== Results: ${passed} passed, ${failed} failed ===`);
+var buildWithSeed2;
+console.log(`
+=== Results: ${passed} passed, ${failed} failed ===`);
 process.exit(failed > 0 ? 1 : 0);

package/dist/test.mjs

@@ -0,0 +1,228 @@
+// node_modules/tsup/assets/esm_shims.js
+import path from "path";
+import { fileURLToPath } from "url";
+var getFilename = () => fileURLToPath(import.meta.url);
+var getDirname = () => path.dirname(getFilename());
+var __dirname = /* @__PURE__ */ getDirname();
+var __filename = /* @__PURE__ */ getFilename();
+
+// src/index.ts
+import { createRequire } from "module";
+import * as path2 from "path";
+var _require = createRequire(__filename);
+var addon = _require(
+  path2.join(__dirname, "..", "build", "Release", "annoy.node")
+);
+var AnnoyIndex = class {
+  constructor(f, metric = "angular") {
+    const constructors = {
+      angular: addon.AnnoyIndexAngular,
+      euclidean: addon.AnnoyIndexEuclidean,
+      manhattan: addon.AnnoyIndexManhattan,
+      dot: addon.AnnoyIndexDotProduct
+    };
+    const Ctor = constructors[metric];
+    if (!Ctor) {
+      throw new Error(
+        `Unknown metric "${metric}". Use: angular, euclidean, manhattan, dot`
+      );
+    }
+    this.index = new Ctor(f);
+  }
+  addItem(index, vector) {
+    this.index.addItem(index, vector);
+  }
+  build(numTrees, numThreads = -1) {
+    this.index.build(numTrees, numThreads);
+  }
+  unbuild() {
+    this.index.unbuild();
+  }
+  save(filename, prefault = false) {
+    this.index.save(filename, prefault);
+  }
+  load(filename, prefault = false) {
+    this.index.load(filename, prefault);
+  }
+  unload() {
+    this.index.unload();
+  }
+  getDistance(i, j) {
+    return this.index.getDistance(i, j);
+  }
+  getNnsByItem(item, n, searchK = -1, includeDistances = false) {
+    return this.index.getNnsByItem(
+      item,
+      n,
+      searchK,
+      includeDistances
+    );
+  }
+  getNnsByVector(vector, n, searchK = -1, includeDistances = false) {
+    return this.index.getNnsByVector(
+      vector,
+      n,
+      searchK,
+      includeDistances
+    );
+  }
+  getNItems() {
+    return this.index.getNItems();
+  }
+  getNTrees() {
+    return this.index.getNTrees();
+  }
+  getItem(item) {
+    return this.index.getItem(item);
+  }
+  setSeed(seed) {
+    this.index.setSeed(seed);
+  }
+  onDiskBuild(filename) {
+    this.index.onDiskBuild(filename);
+  }
+  verbose(v) {
+    this.index.verbose(v);
+  }
+  getF() {
+    return this.index.getF();
+  }
+};
+
+// src/test.ts
+import * as fs from "fs";
+import * as path3 from "path";
+var DIMS = 40;
+var NUM_ITEMS = 1e3;
+var NUM_TREES = 10;
+var tmpFile = path3.join(__dirname, "..", "test_index.ann");
+var passed = 0;
+var failed = 0;
+function assert(condition, msg) {
+  if (condition) {
+    passed++;
+    console.log(`  PASS: ${msg}`);
+  } else {
+    failed++;
+    console.error(`  FAIL: ${msg}`);
+  }
+}
+function randomVector(dims) {
+  return Array.from({ length: dims }, () => Math.random() * 2 - 1);
+}
+function cleanup() {
+  try {
+    fs.unlinkSync(tmpFile);
+  } catch {
+  }
+}
+console.log("=== annoy-node test suite ===\n");
+console.log("Test 1: Build and query (Angular)");
+{
+  const index = new AnnoyIndex(DIMS, "angular");
+  for (let i = 0; i < NUM_ITEMS; i++) {
+    index.addItem(i, randomVector(DIMS));
+  }
+  index.build(NUM_TREES);
+  assert(index.getNItems() === NUM_ITEMS, `getNItems() === ${NUM_ITEMS}`);
+  assert(index.getNTrees() === NUM_TREES, `getNTrees() === ${NUM_TREES}`);
+  assert(index.getF() === DIMS, `getF() === ${DIMS}`);
+  const neighbors = index.getNnsByItem(0, 10);
+  assert(Array.isArray(neighbors), "getNnsByItem returns array");
+  assert(neighbors.length === 10, "returns 10 neighbors");
+  assert(neighbors[0] === 0, "nearest neighbor of item 0 is itself");
+}
+console.log("\nTest 2: Include distances");
+{
+  const index = new AnnoyIndex(DIMS, "angular");
+  for (let i = 0; i < 100; i++) {
+    index.addItem(i, randomVector(DIMS));
+  }
+  index.build(NUM_TREES);
+  const result = index.getNnsByItem(0, 5, -1, true);
+  assert(
+    "neighbors" in result && "distances" in result,
+    "result has neighbors and distances"
+  );
+  assert(result.neighbors.length === 5, "5 neighbors returned");
+  assert(result.distances.length === 5, "5 distances returned");
+  assert(result.distances[0] === 0, "distance to self is 0");
+}
+console.log("\nTest 3: getNnsByVector");
+{
+  const index = new AnnoyIndex(DIMS, "euclidean");
+  const vectors = [];
+  for (let i = 0; i < 100; i++) {
+    const v = randomVector(DIMS);
+    vectors.push(v);
+    index.addItem(i, v);
+  }
+  index.build(NUM_TREES);
+  const result = index.getNnsByVector(vectors[0], 5);
+  assert(Array.isArray(result), "getNnsByVector returns array");
+  assert(result[0] === 0, "closest to vector[0] is item 0");
+}
+console.log("\nTest 4: Save and load");
+{
+  cleanup();
+  const index = new AnnoyIndex(DIMS, "angular");
+  for (let i = 0; i < 100; i++) {
+    index.addItem(i, randomVector(DIMS));
+  }
+  index.build(NUM_TREES);
+  index.save(tmpFile);
+  const loaded = new AnnoyIndex(DIMS, "angular");
+  loaded.load(tmpFile);
+  assert(loaded.getNItems() === 100, "loaded index has 100 items");
+  assert(loaded.getNTrees() === NUM_TREES, `loaded index has ${NUM_TREES} trees`);
+  const item = loaded.getItem(0);
+  assert(item.length === DIMS, `getItem returns vector of length ${DIMS}`);
+  cleanup();
+}
+console.log("\nTest 5: getDistance");
+{
+  const index = new AnnoyIndex(DIMS, "euclidean");
+  const v = randomVector(DIMS);
+  index.addItem(0, v);
+  index.addItem(1, v);
+  index.build(1);
+  const dist = index.getDistance(0, 1);
+  assert(Math.abs(dist) < 1e-6, `distance between identical vectors is ~0 (got ${dist})`);
+}
+console.log("\nTest 6: All metrics");
+{
+  const metrics = ["angular", "euclidean", "manhattan", "dot"];
+  for (const metric of metrics) {
+    const index = new AnnoyIndex(DIMS, metric);
+    for (let i = 0; i < 50; i++) {
+      index.addItem(i, randomVector(DIMS));
+    }
+    index.build(5);
+    const nn = index.getNnsByItem(0, 5);
+    assert(nn.length === 5, `${metric}: returns 5 neighbors`);
+  }
+}
+console.log("\nTest 7: setSeed reproducibility");
+{
+  let buildWithSeed = function(seed) {
+    const idx = new AnnoyIndex(DIMS, "angular");
+    idx.setSeed(seed);
+    for (let i = 0; i < 100; i++) {
+      idx.addItem(i, vectors[i]);
+    }
+    idx.build(5);
+    return idx.getNnsByItem(0, 10);
+  };
+  buildWithSeed2 = buildWithSeed;
+  const vectors = Array.from({ length: 100 }, () => randomVector(DIMS));
+  const r1 = buildWithSeed(42);
+  const r2 = buildWithSeed(42);
+  assert(
+    JSON.stringify(r1) === JSON.stringify(r2),
+    "same seed produces same results"
+  );
+}
+var buildWithSeed2;
+console.log(`
+=== Results: ${passed} passed, ${failed} failed ===`);
+process.exit(failed > 0 ? 1 : 0);

package/package.json

@@ -1,15 +1,28 @@
 {
   "name": "@samyok/annoy",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "Native Node.js bindings for Spotify's Annoy — fast approximate nearest neighbor search",
   "main": "dist/index.js",
+  "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.mts",
+        "default": "./dist/index.mjs"
+      },
+      "require": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      }
+    }
+  },
   "scripts": {
     "install": "node-gyp rebuild",
     "build:native": "node-gyp rebuild",
-    "build:ts": "tsc",
+    "build:ts": "tsup",
     "build": "npm run build:native && npm run build:ts",
-    "test": "npm run build:ts && node dist/test.js",
+    "test": "npm run build:ts && node dist/test.js && node --input-type=module -e \"import('./dist/test.mjs')\"",
     "prepublishOnly": "npm run build:ts"
   },
   "keywords": [
@@ -42,11 +55,11 @@
   "devDependencies": {
     "@types/node": "^25.7.0",
     "node-gyp": "^11.2.0",
+    "tsup": "^8.5.1",
     "typescript": "^5.8.3"
   },
   "files": [
-    "dist/**/*.js",
-    "dist/**/*.d.ts",
+    "dist",
     "src/addon.cc",
     "deps/annoy/src/annoylib.h",
     "deps/annoy/src/kissrandom.h",