Improved example

ankane · ankane · commit 3bb2bf5b92e2 · 2025-06-01T21:43:52.000-07:00
diff --git a/test/example.lua b/test/example.lua
@@ -12,15 +12,15 @@ assert(pg:query("DROP TABLE IF EXISTS items"))
 
 assert(pg:query("CREATE TABLE items (id bigserial PRIMARY KEY, embedding vector(3))"))
 
-embedding1 = pgvector.new({1, 1, 1})
-embedding2 = pgvector.new({2, 2, 2})
-embedding3 = pgvector.new({1, 1, 2})
+local embedding1 = pgvector.new({1, 1, 1})
+local embedding2 = pgvector.new({2, 2, 2})
+local embedding3 = pgvector.new({1, 1, 2})
 assert(pg:query("INSERT INTO items (embedding) VALUES ($1), ($2), ($3)", embedding1, embedding2, embedding3))
 
 -- optional: automatically convert vector type to table
 -- pgvector.setup_vector(pg)
 
-embedding = pgvector.new({1, 1, 1})
+local embedding = pgvector.new({1, 1, 1})
 local res = assert(pg:query("SELECT * FROM items ORDER BY embedding <-> $1 LIMIT 5", embedding))
 for i, row in ipairs(res) do
   for k, v in pairs(row) do
