Added test for pg-native

Author: ankane

package.json

@@ -70,6 +70,7 @@
     "kysely": "^0.29.2",
     "objection": "^3.1.3",
     "pg": "^8.6.0",
+    "pg-native": "^3.8.0",
     "pg-promise": "^12.3.0",
     "postgres": "^3.3.4",
     "prisma": "^7.1.0",

src/pg/index.js

@@ -1,5 +1,5 @@
 import { deprecate } from 'node:util';
-import { toSql } from '../index.js';
+import { fromSql, toSql } from '../index.js';
 import { vectorFromSql, halfvecFromSql, sparsevecFromSql } from '../utils.js';
 
 /** @import { ClientBase } from 'pg' */
@@ -38,6 +38,6 @@ async function registerTypes(client) {
 
 const registerType = deprecate(registerTypes, 'registerType() is deprecated. Use registerTypes() instead.');
 
-export { registerType, registerTypes, toSql };
+export { registerType, registerTypes, fromSql, toSql };
 
-export default { registerType, registerTypes, toSql };
+export default { registerType, registerTypes, fromSql, toSql };

tests/pg-native.test.js

@@ -0,0 +1,37 @@
+import assert from 'node:assert';
+import test from 'node:test';
+import Client from 'pg-native';
+import pgvector from 'pgvector/pg';
+import { SparseVector } from 'pgvector';
+
+function isDeno() {
+  return typeof Deno !== 'undefined';
+}
+
+test('pg-native example', {skip: isDeno()}, () => {
+  const client = new Client();
+  client.connectSync('postgres://localhost/pgvector_node_test');
+
+  client.querySync('CREATE EXTENSION IF NOT EXISTS vector');
+  client.querySync('DROP TABLE IF EXISTS pg_native_items');
+  client.querySync('CREATE TABLE pg_native_items (id serial PRIMARY KEY, embedding vector(3), half_embedding halfvec(3), binary_embedding bit(3), sparse_embedding sparsevec(3))');
+
+  const params = [
+    pgvector.toSql([1, 1, 1]), pgvector.toSql([1, 1, 1]), '000', new SparseVector([1, 1, 1]),
+    pgvector.toSql([2, 2, 2]), pgvector.toSql([2, 2, 2]), '101', new SparseVector([2, 2, 2]),
+    pgvector.toSql([1, 1, 2]), pgvector.toSql([1, 1, 2]), '111', new SparseVector([1, 1, 2]),
+    null, null, null, null
+  ];
+  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);
+
+  const rows = client.querySync('SELECT * FROM pg_native_items ORDER BY embedding <-> $1 LIMIT 5', [pgvector.toSql([1, 1, 1])]);
+  assert.deepEqual(rows.map(v => v.id), [1, 3, 2, 4]);
+  assert.deepEqual(pgvector.fromSql(rows[0].embedding), [1, 1, 1]);
+  assert.deepEqual(pgvector.fromSql(rows[0].half_embedding), [1, 1, 1]);
+  assert.deepEqual(rows[0].binary_embedding, '000');
+  assert.deepEqual(pgvector.fromSql(rows[0].sparse_embedding).toArray(), [1, 1, 1]);
+
+  client.querySync('CREATE INDEX ON pg_native_items USING hnsw (embedding vector_l2_ops)');
+
+  client.end();
+});

types/pg/index.d.ts

@@ -1,11 +1,13 @@
 declare namespace _default {
     export { registerType };
     export { registerTypes };
+    export { fromSql };
     export { toSql };
 }
 export default _default;
 export function registerType(client: ClientBase): Promise<void>;
 export function registerTypes(client: ClientBase): Promise<void>;
+import { fromSql } from '../index.js';
 import { toSql } from '../index.js';
 import type { ClientBase } from 'pg';
-export { toSql };
+export { fromSql, toSql };