feat: add pgvector and database dump

Dockerfile

@@ -3,11 +3,14 @@ FROM postgres:14.3
 WORKDIR /app
 
 RUN apt update && \
-    apt install -y \
+    apt install -y --no-install-recommends \
+    build-essential \
     python3 \
     python3-pip \
+    postgresql-server-dev-14 \
     libpq-dev \
-    htop
+    htop \
+    git
 
 COPY ./ /app/
 
@@ -15,12 +18,26 @@ RUN pip3 install -r ./requirements.txt
 
 EXPOSE 7860
 
+ENV POSTGRES_USER=postgres
+ENV POSTGRES_PASSWORD=pwd
+ENV POSTGRES_PASSWORD=sorbobot
+
+# Install PGVector
+WORKDIR /tmp
+RUN git clone --branch v0.5.1 https://github.com/pgvector/pgvector.git
+WORKDIR /tmp/pgvector
+RUN make
+RUN make install # may need sudo
+WORKDIR /app
+
 COPY "execution.sh" "/usr/local/bin/"
 
-ENTRYPOINT ["execution.sh"]
+COPY ./docker-entrypoint-initdb.d/ /docker-entrypoint-initdb.d/
 
-HEALTHCHECK CMD curl --fail http://localhost:7860/_stcore/health
+ENTRYPOINT ["execution.sh"]
 
 STOPSIGNAL SIGINT
 
+HEALTHCHECK CMD curl --fail http://localhost:7860/_stcore/health
+
 CMD ["postgres"]

docker-entrypoint-initdb.d/01-restore.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+
+file="/docker-entrypoint-initdb.d/dump.pgdata"
+dbname=sorbobot
+
+echo "Restoring DB using $file"
+pg_restore -U postgres --dbname=$dbname --verbose --single-transaction < "$file" || exit 1