mirror of
https://github.com/HKUDS/LightRAG.git
synced 2025-06-26 22:00:19 +00:00
82 lines
2.8 KiB
Bash
Executable File
82 lines
2.8 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
NAMESPACE=rag
|
|
|
|
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
|
|
|
|
check_dependencies(){
|
|
echo "Checking dependencies..."
|
|
command -v kubectl >/dev/null 2>&1 || { echo "Error: kubectl command not found"; exit 1; }
|
|
command -v helm >/dev/null 2>&1 || { echo "Error: helm command not found"; exit 1; }
|
|
|
|
# Check if Kubernetes is available
|
|
echo "Checking if Kubernetes is available..."
|
|
kubectl cluster-info &>/dev/null
|
|
if [ $? -ne 0 ]; then
|
|
echo "Error: Kubernetes cluster is not accessible. Please ensure you have proper access to a Kubernetes cluster."
|
|
exit 1
|
|
fi
|
|
echo "Kubernetes cluster is accessible."
|
|
}
|
|
|
|
check_dependencies
|
|
|
|
if [ -z "$OPENAI_API_KEY" ]; then
|
|
echo "OPENAI_API_KEY environment variable is not set"
|
|
read -s -p "Enter your OpenAI API key: " OPENAI_API_KEY
|
|
if [ -z "$OPENAI_API_KEY" ]; then
|
|
echo "Error: OPENAI_API_KEY must be provided"
|
|
exit 1
|
|
fi
|
|
export OPENAI_API_KEY=$OPENAI_API_KEY
|
|
fi
|
|
|
|
if [ -z "$OPENAI_API_BASE" ]; then
|
|
echo "OPENAI_API_BASE environment variable is not set, will use default value"
|
|
read -p "Enter OpenAI API base URL (press Enter to skip if not needed): " OPENAI_API_BASE
|
|
export OPENAI_API_BASE=$OPENAI_API_BASE
|
|
fi
|
|
|
|
required_env_vars=("OPENAI_API_BASE" "OPENAI_API_KEY")
|
|
|
|
for var in "${required_env_vars[@]}"; do
|
|
if [ -z "${!var}" ]; then
|
|
echo "Error: $var environment variable is not set"
|
|
exit 1
|
|
fi
|
|
done
|
|
|
|
if ! kubectl get namespace rag &> /dev/null; then
|
|
echo "creating namespace 'rag'..."
|
|
kubectl create namespace rag
|
|
fi
|
|
|
|
helm upgrade --install lightrag-dev $SCRIPT_DIR/lightrag \
|
|
--namespace rag \
|
|
--set-string env.LIGHTRAG_KV_STORAGE=JsonKVStorage \
|
|
--set-string env.LIGHTRAG_VECTOR_STORAGE=NanoVectorDBStorage \
|
|
--set-string env.LIGHTRAG_GRAPH_STORAGE=NetworkXStorage \
|
|
--set-string env.LIGHTRAG_DOC_STATUS_STORAGE=JsonDocStatusStorage \
|
|
--set-string env.LLM_BINDING=openai \
|
|
--set-string env.LLM_MODEL=gpt-4o-mini \
|
|
--set-string env.LLM_BINDING_HOST=$OPENAI_API_BASE \
|
|
--set-string env.LLM_BINDING_API_KEY=$OPENAI_API_KEY \
|
|
--set-string env.EMBEDDING_BINDING=openai \
|
|
--set-string env.EMBEDDING_MODEL=text-embedding-ada-002 \
|
|
--set-string env.EMBEDDING_DIM=1536 \
|
|
--set-string env.EMBEDDING_BINDING_API_KEY=$OPENAI_API_KEY
|
|
|
|
# Wait for LightRAG pod to be ready
|
|
echo ""
|
|
echo "Waiting for lightrag-dev pod to be ready..."
|
|
kubectl wait --for=condition=ready pod -l app.kubernetes.io/instance=lightrag-dev --timeout=300s -n rag
|
|
echo "lightrag-dev pod is ready"
|
|
echo ""
|
|
echo "Running Port-Forward:"
|
|
echo " kubectl --namespace rag port-forward svc/lightrag-dev 9621:9621"
|
|
echo "==========================================="
|
|
echo ""
|
|
echo "✅ You can visit LightRAG at: http://localhost:9621"
|
|
echo ""
|
|
kubectl --namespace rag port-forward svc/lightrag-dev 9621:9621
|