Updated test to use Vector class

pgvector · Sep 10, 2024 · a0ae84d · a0ae84d
1 parent d0e0f27
commit a0ae84d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 0.1.1 (unreleased)
+
+- Added `Vector` class
+
 ## 0.1.0 (2023-10-17)
 
 - First release
diff --git a/lib/pgvector.dart b/lib/pgvector.dart
@@ -2,6 +2,7 @@ import 'dart:convert';
 import 'dart:typed_data';
 
 import 'vector.dart';
+export 'vector.dart' show Vector;
 
 class Pgvector {
   const Pgvector();

diff --git a/test/postgres_test.dart b/test/postgres_test.dart
@@ -22,20 +22,20 @@ void main() {
     await connection.execute(
         Sql.named("INSERT INTO items (embedding) VALUES (@a), (@b), (@c)"),
         parameters: {
-          "a": pgvector.encode([1, 1, 1]),
-          "b": pgvector.encode([2, 2, 2]),
-          "c": pgvector.encode([1, 1, 2])
+          "a": Vector([1, 1, 1]).toString(),
+          "b": Vector([2, 2, 2]).toString(),
+          "c": Vector([1, 1, 2]).toString()
         });
 
     List<List<dynamic>> results = await connection.execute(
         Sql.named(
             "SELECT id, embedding FROM items ORDER BY embedding <-> @embedding LIMIT 5"),
         parameters: {
-          "embedding": pgvector.encode([1, 1, 1])
+          "embedding": Vector([1, 1, 1]).toString()
         });
     for (final row in results) {
       print(row[0]);
-      print(pgvector.decode(row[1].bytes));
+      print(Vector.fromBinary(row[1].bytes));
     }
 
     await connection