chore: update entrypoint

.docker/entrypoint.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+# Ensure correct ownership of the local_data directory
+chown -R worker /home/worker/app/local_data
+
+# Ensure correct ownership of the models directory if it exists
+if [ -d "/home/worker/app/models" ]; then
+    chown -R worker /home/worker/app/models
+fi
+
+# Execute the command passed to the entrypoint
+exec "$@"

Dockerfile.llamacpp-cpu

@@ -60,5 +60,11 @@ COPY --chown=worker private_gpt/ private_gpt
 COPY --chown=worker *.yaml .
 COPY --chown=worker scripts/ scripts
 
+# Copy the entrypoint script and set correct permissions
+COPY ./.docker/entrypoint.sh /entrypoint.sh
+RUN chown worker /entrypoint.sh && \
+    chmod +x /entrypoint.sh
+
 USER worker
-ENTRYPOINT python -m private_gpt
+ENTRYPOINT ["/entrypoint.sh"]
+CMD ["python", "-m", "private_gpt"]

Dockerfile.ollama

@@ -47,5 +47,11 @@ COPY --chown=worker private_gpt/ private_gpt
 COPY --chown=worker *.yaml .
 COPY --chown=worker scripts/ scripts
 
+# Copy the entrypoint script and set correct permissions
+COPY ./.docker/entrypoint.sh /entrypoint.sh
+RUN chown worker /entrypoint.sh && \
+    chmod +x /entrypoint.sh
+
 USER worker
-ENTRYPOINT python -m private_gpt
+ENTRYPOINT ["/entrypoint.sh"]
+CMD ["python", "-m", "private_gpt"]