Add Bit Vector

Queries for bit vector operator

Author: griffio

src/main/kotlin/griffio/Main.kt

@@ -2,6 +2,7 @@ package griffio
 
 import app.cash.sqldelight.ColumnAdapter
 import app.cash.sqldelight.driver.jdbc.asJdbcDriver
+import com.pgvector.PGbit
 import com.pgvector.PGvector
 import griffio.migrations.Items
 import griffio.queries.Sample
@@ -14,12 +15,16 @@ private fun getSqlDriver() = PGSimpleDataSource().apply {
     password = ""
 }.asJdbcDriver()
 
+val bitVectorAdapter = object: ColumnAdapter<PGbit, String> {
+    override fun decode(databaseValue: String): PGbit = PGbit(databaseValue)
+    override fun encode(value: PGbit): String = value.toString()
+}
 val vectorAdapter = object: ColumnAdapter<PGvector, String> {
     override fun decode(databaseValue: String): PGvector = PGvector(databaseValue)
     override fun encode(value: PGvector): String = value.toString()
 }
 
-val adapters = Items.Adapter(vectorAdapter)
+val adapters = Items.Adapter(vectorAdapter, bitVectorAdapter)
 
 fun main() {
     val driver = getSqlDriver()
@@ -52,4 +57,8 @@ fun main() {
     sample.vectorQueries.selectMultiply().executeAsList().forEach(::println)
     println("selectNegativeDistanceOperator")
     sample.vectorQueries.selectNegativeDistanceOperator().executeAsList().forEach(::println)
+    println("selectHammingDistanceOperator")
+    sample.vectorQueries.selectHammingDistanceOperator().executeAsList().forEach(::println)
+    println("selectJaccardDistanceOperator")
+    sample.vectorQueries.selectJaccardDistanceOperator().executeAsList().forEach(::println)
 }

src/main/sqldelight/griffio/migrations/V1__Initial_version.sqm

@@ -1,12 +1,15 @@
+import com.pgvector.PGbit;
 import com.pgvector.PGvector;
 
 CREATE EXTENSION IF NOT EXISTS vector;
 
 CREATE TABLE items (
     id BIGSERIAL PRIMARY KEY,
-    embedding VECTOR(3) AS PGvector
+    embedding VECTOR(3) AS PGvector,
+    bits BIT(3) AS PGbit
 );
 
+
 CREATE INDEX idx_embedding_hnsw ON items USING hnsw (embedding vector_l2_ops);
 
 CREATE INDEX idx_embedding_ivfflat ON items USING ivfflat (embedding vector_l2_ops) WITH (lists = 100);

src/main/sqldelight/griffio/queries/vector.sq

@@ -1,5 +1,5 @@
 insert:
-INSERT INTO items (embedding) VALUES ('[1,2,3]'), ('[4,5,6]');
+INSERT INTO items (embedding, bits) VALUES ('[1,2,3]', '000'), ('[4,5,6]', '111');
 
 select:
 SELECT *
@@ -52,3 +52,9 @@ SELECT avg(embedding) FROM items;
 
 selectNegativeDistanceOperator:
 SELECT embedding <#> embedding FROM items;
+
+selectHammingDistanceOperator :
+SELECT bits <~> bits FROM items;
+
+selectJaccardDistanceOperator :
+SELECT bits <%> bits FROM items;