File tree Expand file tree Collapse file tree 2 files changed +14
-2
lines changed
sqldelight/griffio/migrations Expand file tree Collapse file tree 2 files changed +14
-2
lines changed Original file line number Diff line number Diff line change 1
1
package griffio
2
2
3
+ import app.cash.sqldelight.ColumnAdapter
3
4
import app.cash.sqldelight.driver.jdbc.asJdbcDriver
5
+ import com.pgvector.PGvector
6
+ import griffio.migrations.Items
4
7
import griffio.queries.Sample
5
8
import org.postgresql.ds.PGSimpleDataSource
6
9
@@ -11,9 +14,16 @@ private fun getSqlDriver() = PGSimpleDataSource().apply {
11
14
password = " "
12
15
}.asJdbcDriver()
13
16
17
+ val vectorAdapter = object : ColumnAdapter <PGvector , String > {
18
+ override fun decode (databaseValue : String ): PGvector = PGvector (databaseValue)
19
+ override fun encode (value : PGvector ): String = value.toString()
20
+ }
21
+
22
+ val adapters = Items .Adapter (vectorAdapter)
23
+
14
24
fun main () {
15
25
val driver = getSqlDriver()
16
- val sample = Sample (driver)
26
+ val sample = Sample (driver, adapters )
17
27
sample.vectorQueries.insert()
18
28
sample.vectorQueries.select().executeAsList().forEach(::println)
19
29
println (" embeddings" )
Original file line number Diff line number Diff line change
1
+ import com.pgvector.PGvector;
2
+
1
3
CREATE EXTENSION IF NOT EXISTS vector;
2
4
3
5
CREATE TABLE items (
4
6
id BIGSERIAL PRIMARY KEY,
5
- embedding VECTOR(3)
7
+ embedding VECTOR(3) AS PGvector
6
8
);
7
9
8
10
CREATE INDEX idx_embedding_hnsw ON items USING hnsw (embedding vector_l2_ops);
You can’t perform that action at this time.
0 commit comments