iframe-proxy | Sunbelt Computer Software

24 lines (18 loc) · 726 Bytes

import { SQL } from 'bun';
import pgvector from 'pgvector';
const sql = new SQL({database: 'pgvector_example'});
await sql`CREATE EXTENSION IF NOT EXISTS vector`;
await sql`DROP TABLE IF EXISTS items`;
await sql`CREATE TABLE items (id bigserial PRIMARY KEY, embedding vector(3))`;
const item = {
  embedding: pgvector.toSql([1, 2, 3])
await sql`INSERT INTO items ${sql(item)}`;
const items = [
  {embedding: pgvector.toSql([4, 5, 6])},
  {embedding: pgvector.toSql([7, 8, 9])}
await sql`INSERT INTO items ${sql(items)}`;
await sql`CREATE INDEX ON items USING hnsw (embedding vector_l2_ops)`;
const rows = await sql`SELECT * FROM items ORDER BY embedding <-> ${item.embedding} LIMIT 5`.values();
console.log(rows);

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

example.js

Latest commit

History

example.js

File metadata and controls