update dockerfile

Dockerfile

@@ -1,6 +1,11 @@
 FROM docker.elastic.co/elasticsearch/elasticsearch:8.5.3
 
 # Environment variable
+ENV TEAM_PASSWORD=1234
+ENV ARGILLA_PASSWORD=1234
+ENV TEAM_API_KEY=team.apikey
+ENV ARGILLA_API_KEY=argilla.apikey
+ENV LOAD_DATA_ENABLE=true
 ENV UVICORN_PORT=6900
 ENV xpack.security.enabled=false
 ENV cluster.routing.allocation.disk.threshold_enabled=false
@@ -16,13 +21,15 @@ RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
 RUN python3.9 get-pip.py
 RUN pip3 install datasets
 
-
 COPY start.sh /
 RUN chmod +x /start.sh
-
 COPY *.whl /packages/
 
 # Install argilla
 RUN for wheel in /packages/*.whl; do pip install "$wheel"[server]; done
 
+USER elasticsearch
+RUN touch $HOME/users.yml
+RUN chown -R elasticsearch:elasticsearch $HOME/users.yml
+
 CMD ["/start.sh"]

start.sh

@@ -4,10 +4,30 @@ set -e
 
 # Start Elasticsearch
 echo "Starting Elasticsearch"
-elasticsearch &
+elasticsearch 1>/dev/null 2>/dev/null &
+whoami
+
+# Create users.yml file
+echo "Creating users schema"
+cat >"$HOME"/users.yml <<EOF
+- username: "team"
+  api_key: TEAM_API_KEY
+  full_name: Team
+  email: [email protected]
+  hashed_password: TEAM_PASSWORD
+  workspaces: []
+
+- username: "argilla"
+  api_key: ARGILLA_API_KEY
+  full_name: Argilla
+  email: [email protected]
+  hashed_password: ARGILLA_PASSWORD
+  workspaces: ["team"]
+EOF
 
 echo "Waiting for elasticsearch to start"
 sleep 15
 
 # Start Argilla
+echo "Starting Argilla"
 uvicorn argilla:app --host "0.0.0.0"