CREATE EXTENSION IF NOT EXISTS vector;

CREATE TABLE documents (
  id SERIAL PRIMARY KEY,
  content TEXT,
  tenant_id UUID NOT NULL,
  embedding vector(1536)
);

CREATE INDEX ON documents
USING hnsw (embedding vector_cosine_ops)
WITH (m = 16, ef_construction = 64);

SELECT id, content,
       1 - (embedding <=> $1) as similarity
FROM documents
WHERE tenant_id = $2
ORDER BY embedding <=> $1
LIMIT 10;

import { neon } from '@neondatabase/serverless';

const sql = neon(process.env.DATABASE_URL!);

const similar = await sql`
  SELECT id, content,
         1 - (embedding <=> ${embedding}) as similarity
  FROM documents
  WHERE tenant_id = ${tenantId}
  ORDER BY embedding <=> ${embedding}
  LIMIT 10
`;

CREATE INDEX ON documents
USING hnsw (embedding vector_cosine_ops)
WHERE tenant_id = 'large-tenant-uuid';