29 lines
855 B
Bash
29 lines
855 B
Bash
#!/usr/bin/env bash
|
|
|
|
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
pushd "../../../../../" > /dev/null
|
|
source .set_env.sh
|
|
popd > /dev/null
|
|
|
|
export host_ip=$(hostname -I | awk '{print $1}')
|
|
|
|
if [ -z "${HUGGINGFACEHUB_API_TOKEN}" ]; then
|
|
echo "Error: HUGGINGFACEHUB_API_TOKEN is not set. Please set HUGGINGFACEHUB_API_TOKEN."
|
|
fi
|
|
|
|
if [ -z "${host_ip}" ]; then
|
|
echo "Error: host_ip is not set. Please set host_ip first."
|
|
fi
|
|
|
|
export HUGGINGFACEHUB_API_TOKEN=${HUGGINGFACEHUB_API_TOKEN}
|
|
export EMBEDDING_MODEL_ID="BAAI/bge-base-en-v1.5"
|
|
export RERANK_MODEL_ID="BAAI/bge-reranker-base"
|
|
export INDEX_NAME="rag-redis"
|
|
export OLLAMA_HOST=${host_ip}
|
|
export OLLAMA_MODEL="llama3.2"
|
|
# Set it as a non-null string, such as true, if you want to enable logging facility,
|
|
# otherwise, keep it as "" to disable it.
|
|
export LOGFLAG=""
|