add if for checking if API_KEY and pass have been set

Dockerfile

@@ -22,8 +22,6 @@ ARG API_KEY=$(cat /run/secrets/API_KEY)
 RUN --mount=type=secret,id=PASSWORD,mode=0444,required=false
 ARG PASSWORD=$(cat /run/secrets/PASSWORD)
 
-RUN echo $PASSWORD
-
 # Install Python deps
 RUN pip3 install -U -r requirements.txt
 

start.sh

@@ -1,4 +1,16 @@
 
 ES_JAVA_OPTS=-'Xms512m -Xmx512m' elasticsearch -E node.name=elasticsearch -E cluster.name=es-local -E discovery.type=single-node -E cluster.routing.allocation.disk.threshold_enabled=false &
 ./wait-for-it.sh -t 30 localhost:9200
-python3 -m argilla
+if [ -z "$API_KEY" ] && [ -z "$PASSWORD" ]; then
+  python -m argilla
+elif [ -z "$API_KEY" ] && [ -n "$PASSWORD" ]; then
+  export ARGILLA_LOCAL_AUTH_DEFAULT_PASSWORD=$PASSWORD
+  python -m argilla
+elif [ -n "$API_KEY" ] && [ -z "$PASSWORD" ]; then
+  export ARGILLA_LOCAL_AUTH_DEFAULT_APIKEY=$API_KEY
+  python -m argilla
+else
+  export ARGILLA_LOCAL_AUTH_DEFAULT_APIKEY=$API_KEY
+  export ARGILLA_LOCAL_AUTH_DEFAULT_PASSWORD=$PASSWORD
+  python -m argilla
+fi