Added support for objects to SparseVector

pgvector · Jun 27, 2024 · 7142864 · 7142864
1 parent b7d93e9
commit 7142864
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/utils/sparse-vector.js b/src/utils/sparse-vector.js
@@ -62,7 +62,8 @@ class SparseVector {
  this.indices = [];
  this.values = [];
 
- for (const [i, v] of map.entries()) {
+ const entries = map instanceof Map ? map.entries() : Object.entries(map);
+ for (const [i, v] of entries) {
  if (v != 0) {
  this.indices.push(Number(i));
  this.values.push(Number(v));

diff --git a/tests/utils/sparse-vector.test.mjs b/tests/utils/sparse-vector.test.mjs
@@ -28,6 +28,14 @@ test('fromMap', () => {
  expect(vec.values).toStrictEqual([2, 3, 1]);
 });
 
+test('fromMapObject', () => {
+ const map = {2: 2, 4: 3, 0: 1, 3: 0};
+ const vec = new SparseVector(map, 6);
+ expect(vec.dimensions).toStrictEqual(6);
+ expect(vec.indices).toStrictEqual([0, 2, 4]);
+ expect(vec.values).toStrictEqual([1, 2, 3]);
+});
+
 test('toPostgres', () => {
  const vec = new SparseVector([1.23456789]);
  expect(vec.toPostgres()).toStrictEqual('{1:1.23456789}/1');