Improved GORM example

pgvector · Jan 19, 2024 · 0dbe778 · 0dbe778
1 parent 804b307
commit 0dbe778
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -268,6 +268,16 @@ db.Clauses(clause.OrderBy{
 }).Limit(5).Find(&items)
 ```
 
+Add an approximate index
+
+```go
+db.Exec("CREATE INDEX ON items USING ivfflat (embedding vector_l2_ops) WITH (lists = 100)")
+// or
+db.Exec("CREATE INDEX ON items USING hnsw (embedding vector_l2_ops)")
+```
+
+Use `vector_ip_ops` for inner product and `vector_cosine_ops` for cosine distance
+
 See a [full example](gorm_test.go)
 
 ## History

diff --git a/gorm_test.go b/gorm_test.go
@@ -41,6 +41,8 @@ func TestGorm(t *testing.T) {
 
 	db.AutoMigrate(&GormItem{})
 
+	db.Exec("CREATE INDEX ON gorm_items USING hnsw (embedding vector_l2_ops)")
+
 	CreateGormItems(db)
 
 	var items []GormItem