feat: add pgvector to postgres stack

Author: diogocasteluber

README.md

@@ -17,14 +17,16 @@ bash -c "$(curl -fsSL https://raw.githubusercontent.com/opencodeco/stack/main/in
 | `stack mysql`| MySQL & phpMyAdmin (http://localhost:8031) |
 | `stack redis` | Redis & RedisInsight (http://localhost:8032) |
 | `stack mongo` | MongoDB & Mongo Express (http://localhost:8033) |
-| `stack postgres` | PostgreSQL & pgAdmin (http://localhost:8039) |
+| `stack postgres` | PostgreSQL with pgvector & pgAdmin (http://localhost:8039) |
 | `stack kafka` | Kafka and UI for Apache Kafka (http://localhost:8037) |
 | `stack rabbitmq` | RabbitMQ & Management Plugin (http://localhost:8038) |
 | `stack aws` | AWS services via LocalStack _(legacy, [see details below](#aws-localstack-vs-ministack))_ (http://localhost:4566) |
 | `stack aws-ministack` | AWS services via MiniStack (http://localhost:4567) |
 | `stack hyperdx` | HyperDX local (http://localhost:8080) | 
 | `stack o11y` | OpenTelemetry Collector, Jaeger UI, Prometheus & Grafana (see below) |
 
+`stack postgres` includes [pgvector](https://github.com/pgvector/pgvector). Enable it in each database where it is needed with `CREATE EXTENSION vector;`.
+
 ### Observability (o11y)
 
 | Component | Description | Port |

postgres/Dockerfile

@@ -0,0 +1,6 @@
+FROM pgvector/pgvector:0.8.5-pg17-bookworm AS pgvector
+
+FROM postgres:17.4
+
+COPY --from=pgvector /usr/lib/postgresql/17/lib/vector.so /usr/lib/postgresql/17/lib/
+COPY --from=pgvector /usr/share/postgresql/17/extension/vector* /usr/share/postgresql/17/extension/

postgres/docker-compose.yml

@@ -1,7 +1,8 @@
 services:
   opencodeco-postgres:
     container_name: opencodeco-postgres
-    image: postgres:17.4
+    build: .
+    image: opencodeco-postgres:17.4-pgvector
     restart: always
     ports:
       - ${POSTGRES_PORT}:5432

tests/test-postgres.sh

@@ -29,4 +29,13 @@ assert_contains "create table, insert, and select" "hello" "$result"
 db_list=$(docker exec opencodeco-postgres psql -U postgres -tAc "SELECT datname FROM pg_database;" 2>&1)
 assert_contains "default postgres database exists" "postgres" "$db_list"
 
+vector_result=$(docker exec opencodeco-postgres psql -v ON_ERROR_STOP=1 -U postgres -tAc "
+  CREATE EXTENSION IF NOT EXISTS vector;
+  CREATE TABLE stack_vector_test (id SERIAL PRIMARY KEY, embedding vector(3));
+  INSERT INTO stack_vector_test (embedding) VALUES ('[1,2,3]'), ('[4,5,6]');
+  SELECT id FROM stack_vector_test ORDER BY embedding <-> '[1,2,3]' LIMIT 1;
+  DROP TABLE stack_vector_test;
+" 2>&1)
+assert_eq "pgvector nearest-neighbor query" "1" "$vector_result"
+
 report