package main

import (
	"database/sql"
	"log"
	"os"

	_ "github.com/jackc/pgx/v5/stdlib"
)

func main() {
	db, err := sql.Open("pgx", os.Getenv("DATABASE_URL"))
	if err != nil { log.Fatal(err) }
	defer db.Close()

	// Enable pgvector + tiny demo table
	_, _ = db.Exec(`CREATE EXTENSION IF NOT EXISTS vector;
	                CREATE TABLE docs (id bigserial PRIMARY KEY, t text, e vector(3));`)

	// Insert one demo embedding (pretend it came from your model)
	_, _ = db.Exec(`INSERT INTO docs (t, e) VALUES ($1, $2::vector)`,
		"reset password", "[0.12,0.05,0.88]")

	// Query: nearest doc to our query embedding
	var match string
	if err := db.QueryRow(`SELECT t FROM docs ORDER BY e <-> $1::vector LIMIT 1`,
		"[0.10,0.07,0.80]").Scan(&match); err != nil {
		log.Fatal(err)
	}
	log.Println("Nearest match:", match