Add pgvector (#15)

Author: dqii

Dockerfile

@@ -20,7 +20,13 @@ RUN git clone https://github.com/citusdata/pg_cron.git /tmp/pg_cron && \
     make -j && \
     make install
 
-# Install Lantern
+    # Install pgvector
+RUN git clone --branch narek/array-to-sparsevec-casts https://github.com/Ngalstyan4/pgvector.git /tmp/pgvector && \
+    cd /tmp/pgvector && \
+    make && \
+    make install
+
+    # Install Lantern
 RUN cd /tmp && \
     wget https://github.com/lanterndata/lantern/releases/download/v${LANTERN_VERSION}/lantern-${LANTERN_VERSION}.tar -O lantern.tar && \
     tar xf lantern.tar && \

entrypoint/create-extensions.sql

@@ -1,4 +1,5 @@
 create extension pg_stat_statements;
 create extension pg_cron;
+create extension vector;
 create extension lantern;
 create extension lantern_extras;

test/extension-count.sh

@@ -1,8 +1,8 @@
-count=$(PGPASSWORD=postgres psql -h lantern -U postgres -t -A -c "SELECT COUNT(*) FROM pg_available_extensions WHERE name IN ('lantern', 'lantern_extras', 'pg_cron', 'pg_stat_statements');")
+count=$(PGPASSWORD=postgres psql -h lantern -U postgres -t -A -c "SELECT COUNT(*) FROM pg_available_extensions WHERE name IN ('lantern', 'lantern_extras', 'pg_cron', 'pg_stat_statements', 'vector');")
 echo "Count: $count"
 if echo "$count" | grep -q "ERROR"; then
   echo "Failed to retrieve extension count" && exit 1
-elif [ "$count" -ne 4 ]; then
+elif [ "$count" -ne 5 ]; then
   echo "Extension check failed" && exit 1
 else
   echo "Extension check passed"