ruvector/npm/packages/diskann/package.json
rUv 8fbe768629 feat(diskann): Vamana ANN + PQ + NAPI bindings — 14 tests, 1.0 recall, 90µs search (#334)
* feat(ruvector): implement missing capabilities (ADR-143)

- speculativeEmbed: real FNV-1a hash embedding (128-dim) from file content
- ragRetrieve: cosine similarity on embeddings + TF-IDF keyword fallback
- contextRank: TF-IDF weighted scoring instead of raw keyword matching
- Remove false DiskANN claim (will implement as Rust crate next)

Co-Authored-By: claude-flow <ruv@ruv.net>

* feat(diskann): Vamana graph + PQ — SSD-friendly billion-scale ANN (ADR-143)

New Rust crate: ruvector-diskann

Core algorithm (NeurIPS 2019 DiskANN paper):
- Vamana graph with α-robust pruning (bounded out-degree R)
- k-means++ seeded Product Quantization (M subspaces, 256 centroids)
- Asymmetric PQ distance tables for fast candidate filtering
- Two-phase search: PQ-filtered beam search → exact re-ranking
- Memory-mapped persistence (mmap vectors + binary graph)

Performance characteristics:
- L2-squared distance with 8-wide loop unrolling (auto-vectorized)
- Greedy beam search with bounded visited set
- Save/load with flat binary format (mmap-friendly)

9 tests passing: distance, PQ train/encode, Vamana build/search,
bounded degree, full index CRUD, PQ-accelerated search, save/load.

Co-Authored-By: claude-flow <ruv@ruv.net>

* feat(diskann): NAPI-RS bindings + npm package + 14 tests passing

Rust core (ruvector-diskann):
- 4-accumulator L2 distance for ILP optimization
- Recall@10 = 1.000 on 2K vectors
- Search latency: 90µs (5K vectors, 128d, k=10)
- 14 tests: distance, PQ, Vamana, recall, scale, edge cases

NAPI-RS bindings (ruvector-diskann-node):
- Sync + async build/search
- Batch insert (flat Float32Array)
- Save/load, delete, count
- Thread-safe via parking_lot::RwLock

npm package (@ruvector/diskann):
- Platform-specific loader (linux/darwin/win)
- TypeScript declarations
- Node.js test passing

Co-Authored-By: claude-flow <ruv@ruv.net>

* ci(diskann): add cross-platform build + publish workflow

5 targets: linux-x64, linux-arm64, darwin-x64, darwin-arm64, win32-x64

Co-Authored-By: claude-flow <ruv@ruv.net>

* perf(diskann): FlatVectors + VisitedSet + ILP + optional SIMD/GPU

Optimizations applied:
- FlatVectors: contiguous f32 slab (eliminates Vec<Vec> indirection)
- VisitedSet: O(1) clear via generation counter (replaces HashSet)
- 4-accumulator ILP for L2 distance (auto-vectorized)
- Flat PQ distance table (cache-line friendly)
- Parallel medoid finding via rayon
- Zero-copy save (write flat slab directly)
- Optional simsimd feature for hardware NEON/AVX2/AVX-512
- Optional gpu feature with Metal/CUDA/Vulkan dispatch stubs

Results (5K vectors, 128d):
- Search: 90µs → 55µs (1.6x faster)
- Build: 6.9s → 6.2s (10% faster)
- Recall@10: 0.998 (maintained)
- 17 tests passing

Co-Authored-By: claude-flow <ruv@ruv.net>

---------

Co-authored-by: Reuven <cohen@ruv-mac-mini.local>
2026-04-06 17:55:06 -04:00

52 lines
1.2 KiB
JSON

{
"name": "@ruvector/diskann",
"version": "0.1.0",
"description": "DiskANN/Vamana — SSD-friendly billion-scale approximate nearest neighbor search with product quantization",
"main": "index.js",
"types": "index.d.ts",
"author": "ruv.io Team <info@ruv.io> (https://ruv.io)",
"homepage": "https://ruv.io",
"repository": {
"type": "git",
"url": "https://github.com/ruvnet/ruvector.git",
"directory": "npm/packages/diskann"
},
"license": "MIT",
"engines": {
"node": ">=18.0.0"
},
"files": [
"index.js",
"index.d.ts",
"README.md"
],
"scripts": {
"test": "node test.js"
},
"devDependencies": {
"@napi-rs/cli": "^2.18.0"
},
"optionalDependencies": {
"@ruvector/diskann-linux-x64-gnu": "0.1.0",
"@ruvector/diskann-linux-arm64-gnu": "0.1.0",
"@ruvector/diskann-darwin-x64": "0.1.0",
"@ruvector/diskann-darwin-arm64": "0.1.0",
"@ruvector/diskann-win32-x64-msvc": "0.1.0"
},
"publishConfig": {
"access": "public"
},
"keywords": [
"diskann",
"vamana",
"ann",
"approximate-nearest-neighbor",
"vector-search",
"product-quantization",
"ssd",
"billion-scale",
"rust",
"napi",
"ruvector"
]
}