First commit

Author: ankane

.github/workflows/build.yml

@@ -0,0 +1,19 @@
+name: build
+on: [push, pull_request]
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - run: sudo apt update && sudo apt install dub ldc
+      - uses: ankane/setup-postgres@v1
+        with:
+          database: pgvector_d_test
+          dev-files: true
+      - run: |
+          cd /tmp
+          git clone --branch v0.8.0 https://github.com/pgvector/pgvector.git
+          cd pgvector
+          make
+          sudo make install
+      - run: dub run

.gitignore

@@ -0,0 +1,2 @@
+/dub.selections.json
+/pgvector

LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2024 Andrew Kane
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

README.md

@@ -0,0 +1,85 @@
+# pgvector-d
+
+[pgvector](https://github.com/pgvector/pgvector) examples for D
+
+Supports [dpq2](https://github.com/denizzzka/dpq2)
+
+[![Build Status](https://github.com/pgvector/pgvector-d/actions/workflows/build.yml/badge.svg)](https://github.com/pgvector/pgvector-d/actions)
+
+## Getting Started
+
+Follow the instructions for your database library:
+
+- [dpq2](#dpq2)
+
+## dpq2
+
+Enable the extension
+
+```d
+conn.exec("CREATE EXTENSION IF NOT EXISTS vector");
+```
+
+Create a table
+
+```d
+conn.exec("CREATE TABLE items (id bigserial PRIMARY KEY, embedding vector(3))");
+```
+
+Insert vectors
+
+```d
+QueryParams p;
+p.sqlCommand = "INSERT INTO items (embedding) VALUES ($1::vector), ($2::vector)";
+p.argsVariadic("[1,2,3]", "[4,5,6]");
+conn.execParams(p);
+```
+
+Get the nearest neighbors
+
+```d
+QueryParams p;
+p.sqlCommand = "SELECT * FROM items ORDER BY embedding <-> $1::vector LIMIT 5";
+p.argsVariadic("[3,1,2]");
+p.resultFormat = ValueFormat.TEXT;
+auto r = conn.execParams(p);
+foreach (row; rangify(r)) {
+    writeln(row);
+}
+```
+
+Add an approximate index
+
+```d
+conn.exec("CREATE INDEX ON items USING hnsw (embedding vector_l2_ops)");
+// or
+conn.exec("CREATE INDEX ON items USING ivfflat (embedding vector_l2_ops) WITH (lists = 100)");
+```
+
+Use `vector_ip_ops` for inner product and `vector_cosine_ops` for cosine distance
+
+See a [full example](source/app.d)
+
+## Contributing
+
+Everyone is encouraged to help improve this project. Here are a few ways you can help:
+
+- [Report bugs](https://github.com/pgvector/pgvector-d/issues)
+- Fix bugs and [submit pull requests](https://github.com/pgvector/pgvector-d/pulls)
+- Write, clarify, or fix documentation
+- Suggest or add new features
+
+To get started with development:
+
+```sh
+git clone https://github.com/pgvector/pgvector-d.git
+cd pgvector-d
+createdb pgvector_d_test
+dub run
+```
+
+Specify the path to libpq if needed:
+
+```sh
+DFLAGS="-L-L/opt/homebrew/opt/libpq/lib" dub run
+```

dub.json

@@ -0,0 +1,6 @@
+{
+	"name": "pgvector",
+	"dependencies": {
+		"dpq2": "~>1.1.7"
+	}
+}

source/app.d

@@ -0,0 +1,28 @@
+import dpq2;
+import std.stdio;
+
+void main()
+{
+	Connection conn = new Connection("postgres://localhost/pgvector_d_test");
+
+	conn.exec("CREATE EXTENSION IF NOT EXISTS vector");
+	conn.exec("DROP TABLE IF EXISTS items");
+	conn.exec("CREATE TABLE items (id bigserial PRIMARY KEY, embedding vector(3))");
+
+	QueryParams p;
+	p.sqlCommand = "INSERT INTO items (embedding) VALUES ($1::vector), ($2::vector), ($3::vector)";
+	p.argsVariadic("[1,1,1]", "[2,2,2]", "[1,1,2]");
+	conn.execParams(p);
+
+	p.sqlCommand = "SELECT * FROM items ORDER BY embedding <-> $1::vector LIMIT 5";
+	p.argsVariadic("[1,1,1]");
+	p.resultFormat = ValueFormat.TEXT;
+	auto r = conn.execParams(p);
+	foreach (row; rangify(r)) {
+		writeln(row);
+	}
+
+	conn.exec("CREATE INDEX ON items USING hnsw (embedding vector_l2_ops)");
+
+	conn.destroy();
+}