| #!/usr/bin/env bash |
| |
| set -e |
|
|
| INDEX_DIR="/app/pcp/data/index" |
| if [ ! -f "${INDEX_DIR}/embeddings.npy" ] && [ -n "${PCP_INDEX_DATASET}" ]; then |
| echo "[entrypoint] telechargement de l'index depuis ${PCP_INDEX_DATASET}..." |
| python - <<'PY' |
| import os |
| from huggingface_hub import snapshot_download |
| repo = os.environ["PCP_INDEX_DATASET"] |
| token = os.environ.get("HF_TOKEN") or None |
| dst = "/app/pcp/data/index" |
| os.makedirs(dst, exist_ok=True) |
| snapshot_download(repo_id=repo, repo_type="dataset", local_dir=dst, token=token, |
| allow_patterns=["chunks.json", "embeddings.npy", "meta.json", |
| "catalogue_vecs.npy", "catalogue_vecs_meta.json", |
| "example_vecs.npy", "example_blocks.json"]) |
| print("[entrypoint] index telecharge ->", os.listdir(dst)) |
| PY |
| fi |
|
|
| |
| |
| |
| KB_DIR="/app/pcp/data/kb" |
| if [ ! -f "${KB_DIR}/core_context.md" ] && [ -n "${PCP_INDEX_DATASET}" ]; then |
| echo "[entrypoint] telechargement de la KB v4 depuis ${PCP_INDEX_DATASET}..." |
| python - <<'PY' || echo "[entrypoint] AVERTISSEMENT : KB v4 indisponible (voir ci-dessus), demarrage sans" |
| import os |
| from huggingface_hub import snapshot_download |
| repo = os.environ["PCP_INDEX_DATASET"] |
| token = os.environ.get("HF_TOKEN") or None |
| snapshot_download(repo_id=repo, repo_type="dataset", local_dir="/app/pcp/data", |
| token=token, allow_patterns=["kb/**", "corpus_anon/**"]) |
| kb = "/app/pcp/data/kb" |
| if os.path.isdir(kb): |
| print("[entrypoint] kb:", len(os.listdir(kb)), "entrees") |
| else: |
| raise SystemExit("kb/ absent du dataset " + repo) |
| PY |
| fi |
|
|
| exec uvicorn pcp.server:app --host 0.0.0.0 --port 7860 |
|
|
