snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

internal/endtoend/testdata/pg_vector/postgresql/pgx/go/db.go

@@ -0,0 +1,32 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgconn"
+)
+
+type DBTX interface {
+	Exec(context.Context, string, ...any) (pgconn.CommandTag, error)
+	Query(context.Context, string, ...any) (pgx.Rows, error)
+	QueryRow(context.Context, string, ...any) pgx.Row
+}
+
+func New(db DBTX) *Queries {
+	return &Queries{db: db}
+}
+
+type Queries struct {
+	db DBTX
+}
+
+func (q *Queries) WithTx(tx pgx.Tx) *Queries {
+	return &Queries{
+		db: tx,
+	}
+}

internal/endtoend/testdata/pg_vector/postgresql/pgx/go/models.go

@@ -0,0 +1,14 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"github.com/pgvector/pgvector-go"
+)
+
+type Item struct {
+	ID        int64
+	Embedding pgvector.Vector
+}

internal/endtoend/testdata/pg_vector/postgresql/pgx/go/query.sql.go

@@ -0,0 +1,48 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+
+	"github.com/pgvector/pgvector-go"
+)
+
+const insertVector = `-- name: InsertVector :exec
+INSERT INTO items (embedding) VALUES ($1)
+`
+
+func (q *Queries) InsertVector(ctx context.Context, embedding pgvector.Vector) error {
+	_, err := q.db.Exec(ctx, insertVector, embedding)
+	return err
+}
+
+const nearestNeighbor = `-- name: NearestNeighbor :many
+SELECT id, embedding
+FROM items
+ORDER BY embedding <-> $1
+LIMIT 5
+`
+
+func (q *Queries) NearestNeighbor(ctx context.Context, embedding pgvector.Vector) ([]Item, error) {
+	rows, err := q.db.Query(ctx, nearestNeighbor, embedding)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []Item
+	for rows.Next() {
+		var i Item
+		if err := rows.Scan(&i.ID, &i.Embedding); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}