Spaces:
Running
Running
File size: 1,317 Bytes
31a2688 478bf35 31a2688 478bf35 31a2688 478bf35 31a2688 478bf35 31a2688 478bf35 31a2688 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | #!/bin/bash
set -e
QDRANT_URL="${QDRANT_URL:-}"
COLLECTION="${COLLECTION_NAME:-ku_documents}"
if [ -n "${QDRANT_URL}" ]; then
# ---- Remote Qdrant: wait for service ------------------------------------
echo "Waiting for Qdrant at ${QDRANT_URL} ..."
until curl -sf "${QDRANT_URL}/healthz" > /dev/null 2>&1; do
sleep 2
done
echo "Qdrant is ready."
# ---- Check if collection already has data -------------------------------
POINT_COUNT=$(curl -sf "${QDRANT_URL}/collections/${COLLECTION}" 2>/dev/null \
| python -c "import sys,json; print(json.load(sys.stdin).get('result',{}).get('points_count',0))" 2>/dev/null \
|| echo "0")
if [ "${POINT_COUNT}" = "0" ] || [ -z "${POINT_COUNT}" ]; then
echo "Collection '${COLLECTION}' is empty or missing — running ingestion ..."
python -m scripts.ingest
echo "Ingestion complete."
else
echo "Collection '${COLLECTION}' already has ${POINT_COUNT} points — skipping ingestion."
fi
else
# ---- Local Qdrant: file-based storage -----------------------------------
echo "Using local Qdrant storage at ${QDRANT_PATH:-./qdrant_data}"
fi
# ---- Start the API server --------------------------------------------------
exec uvicorn src.api.main:app --host 0.0.0.0 --port 8000
|