refac: only create space admin user when configured

This commit is contained in:
Jun Siang Cheah 2024-05-21 22:09:25 +01:00
parent 889a2fd77e
commit b20ec0586e

View File

@ -39,6 +39,8 @@ if [ -n "$SPACE_ID" ]; then
echo "Copying litellm_config.yaml to the desired location with specified ownership..." echo "Copying litellm_config.yaml to the desired location with specified ownership..."
cp -f ./space/litellm_config.yaml ./data/litellm/config.yaml cp -f ./space/litellm_config.yaml ./data/litellm/config.yaml
if [ -n "$ADMIN_USER_EMAIL" ] && [ -n "$ADMIN_USER_PASSWORD" ]; then
echo "Admin user configured, creating"
WEBUI_SECRET_KEY="$WEBUI_SECRET_KEY" uvicorn main:app --host "$HOST" --port "$PORT" --forwarded-allow-ips '*' & WEBUI_SECRET_KEY="$WEBUI_SECRET_KEY" uvicorn main:app --host "$HOST" --port "$PORT" --forwarded-allow-ips '*' &
webui_pid=$! webui_pid=$!
echo "Waiting for webui to start..." echo "Waiting for webui to start..."
@ -53,6 +55,8 @@ if [ -n "$SPACE_ID" ]; then
-d "{ \"email\": \"${ADMIN_USER_EMAIL}\", \"password\": \"${ADMIN_USER_PASSWORD}\", \"name\": \"Admin\" }" -d "{ \"email\": \"${ADMIN_USER_EMAIL}\", \"password\": \"${ADMIN_USER_PASSWORD}\", \"name\": \"Admin\" }"
echo "Shutting down webui..." echo "Shutting down webui..."
kill $webui_pid kill $webui_pid
fi
export WEBUI_URL=${SPACE_HOST} export WEBUI_URL=${SPACE_HOST}
fi fi