Skip to content

Commit f09a8e9

Browse files
committed
Fixed CI
1 parent 0d254b0 commit f09a8e9

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

‎tests/pg-native.test.js‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
11
import assert from 'node:assert';
22
import test from 'node:test';
3+
// @ts-ignore
34
import Client from 'pg-native';
45
import pgvector from 'pgvector/pg';
56
import { SparseVector } from 'pgvector';
67

78
function isDeno() {
9+
// @ts-ignore
810
return typeof Deno !== 'undefined';
911
}
1012

@@ -25,11 +27,11 @@ test('pg-native example', {skip: isDeno()}, () => {
2527
client.querySync('INSERT INTO pg_native_items (embedding, half_embedding, binary_embedding, sparse_embedding) VALUES ($1, $2, $3, $4), ($5, $6, $7, $8), ($9, $10, $11, $12), ($13, $14, $15, $16)', params);
2628

2729
const rows = client.querySync('SELECT * FROM pg_native_items ORDER BY embedding <-> $1 LIMIT 5', [pgvector.toSql([1, 1, 1])]);
28-
assert.deepEqual(rows.map(v => v.id), [1, 3, 2, 4]);
30+
assert.deepEqual(rows.map(/** @type {function(any): number} */ v => v.id), [1, 3, 2, 4]);
2931
assert.deepEqual(pgvector.fromSql(rows[0].embedding), [1, 1, 1]);
3032
assert.deepEqual(pgvector.fromSql(rows[0].half_embedding), [1, 1, 1]);
3133
assert.deepEqual(rows[0].binary_embedding, '000');
32-
assert.deepEqual(pgvector.fromSql(rows[0].sparse_embedding).toArray(), [1, 1, 1]);
34+
assert.deepEqual(pgvector.fromSql(rows[0].sparse_embedding), new SparseVector([1, 1, 1]));
3335

3436
client.querySync('CREATE INDEX ON pg_native_items USING hnsw (embedding vector_l2_ops)');
3537

0 commit comments

Comments
 (0)