Support build image for ARM64

Dockerfile

@@ -16,6 +16,10 @@ WORKDIR /code
 
 COPY --from=requirements-stage /tmp/requirements.txt /code/requirements.txt
 
+ENV PATH="${PATH}:/root/.cargo/bin"
+RUN curl --proto '=https' --tlsv1.3 -o rust.sh -sSf https://sh.rustup.rs && \
+    chmod +x rust.sh;./rust.sh -y
+
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . /code/

datastore/providers/chroma_datastore.py

@@ -46,7 +46,7 @@ def __init__(
         if client:
             self._client = client
         else:
-            if in_memory:
+            if in_memory == "True":
                 settings = (
                     chromadb.config.Settings(
                         chroma_db_impl="duckdb+parquet",

docker-compose.yml

@@ -0,0 +1,13 @@
+version: '3.9'
+
+services:
+  server:
+    image: chatgpt-retrieval-plugin:latest
+    container_name: chatgpt-api
+    restart: always
+    env_file:
+      - .env
+    ports:
+      - 8080:8080
+    volumes:
+      - ./:/code/

env.sample

@@ -0,0 +1,8 @@
+# Update this file and change copy to `.env`
+CHROMA_COLLECTION=
+CHROMA_IN_MEMORY=
+CHROMA_HOST=
+BEARER_TOKEN=
+OPENAI_API_KEY=
+CHROMA_PORT=
+DATASTORE=chroma