SemanticSearchPOC / tools /dev /restart_dev_environment.sh
KevinStephenson
Adding in weaviate code
b110593
raw
history blame
3.54 kB
#!/bin/bash
set -e
# Jump to root directory
cd "$( dirname "${BASH_SOURCE[0]}" )"/../.. || exit 1
DOCKER_COMPOSE_FILE=docker-compose.yml
ADDITIONAL_SERVICES=()
if [[ "$*" == *--keycloak* ]]; then
ADDITIONAL_SERVICES+=('keycloak')
fi
if [[ "$*" == *--transformers-passage-query* ]]; then
ADDITIONAL_SERVICES+=('t2v-transformers-passage')
ADDITIONAL_SERVICES+=('t2v-transformers-query')
elif [[ "$*" == *--transformers* ]]; then
ADDITIONAL_SERVICES+=('t2v-transformers')
else
ADDITIONAL_SERVICES+=('contextionary')
fi
if [[ "$*" == *--contextionary* ]]; then
ADDITIONAL_SERVICES+=('contextionary')
fi
if [[ "$*" == *--qna* ]]; then
ADDITIONAL_SERVICES+=('qna-transformers')
fi
if [[ "$*" == *--sum* ]]; then
ADDITIONAL_SERVICES+=('sum-transformers')
fi
if [[ "$*" == *--image* ]]; then
ADDITIONAL_SERVICES+=('i2v-neural')
fi
if [[ "$*" == *--ner* ]]; then
ADDITIONAL_SERVICES+=('ner-transformers')
fi
if [[ "$*" == *--spellcheck* ]]; then
ADDITIONAL_SERVICES+=('text-spellcheck')
fi
if [[ "$*" == *--clip* ]]; then
ADDITIONAL_SERVICES+=('multi2vec-clip')
fi
if [[ "$*" == *--bind* ]]; then
ADDITIONAL_SERVICES+=('multi2vec-bind')
fi
if [[ "$*" == *--reranker* ]]; then
ADDITIONAL_SERVICES+=('reranker-transformers')
fi
if [[ "$*" == *--gpt4all* ]]; then
ADDITIONAL_SERVICES+=('t2v-gpt4all')
fi
if [[ "$*" == *--prometheus* ]]; then
ADDITIONAL_SERVICES+=('prometheus')
ADDITIONAL_SERVICES+=('grafana')
fi
if [[ "$*" == *--s3* ]]; then
ADDITIONAL_SERVICES+=('backup-s3')
fi
if [[ "$*" == *--gcs* ]]; then
ADDITIONAL_SERVICES+=('backup-gcs')
fi
if [[ "$*" == *--azure* ]]; then
ADDITIONAL_SERVICES+=('backup-azure')
fi
docker compose -f $DOCKER_COMPOSE_FILE down --remove-orphans
rm -rf data data-node2 connector_state.json schema_state.json
docker compose -f $DOCKER_COMPOSE_FILE up -d "${ADDITIONAL_SERVICES[@]}"
if [[ "$*" == *--keycloak* ]]; then
echo "Since you have specified the --keycloak option, we must now wait for"
echo "keycloak to spin up, so that we can add some demo users"
echo -n "Waiting "
until curl --fail -s localhost:9090/auth; do
echo -n .
sleep 1
done
echo
echo Keycloak is ready, now importing some users.
./tools/dev/keycloak/import_users.sh
fi
if [[ "$*" == *--transformers-passage-query* ]]; then
echo "You have specified the --transformers-passage-query option. Starting up"
echo "the t2v-transformers-passage and t2v-transformers-query model containers"
elif [[ "$*" == *--transformers* ]]; then
echo "You have specified the --transformers option. Starting up"
echo "the text2vec-transformers model container"
fi
if [[ "$*" == *--image* ]]; then
echo "You have specified the --image option. Starting up"
echo "the text2vec-contextionary model container with img2vec-image module"
fi
if [[ "$*" == *--s3* ]]; then
echo "You have specified the --s3 option. Starting up"
echo "the text2vec-contextionary model container with backup-s3 module"
fi
if [[ "$*" == *--gcs* ]]; then
echo "You have specified the --gcs option. Starting up"
echo "the text2vec-contextionary model container with backup-gcs module"
fi
if [[ "$*" == *--azure* ]]; then
echo "You have specified the --azure option. Starting up"
echo "the text2vec-contextionary model container with backup-azure module"
fi
echo "You can now run the dev version with: ./tools/dev/run_dev_server.sh or ./tools/dev/run_dev_server_no_network.sh"