AI Agent commited on
Commit
5ed03b8
Β·
1 Parent(s): fbc58fa

Fix Kuzu graph write lock error and suppress pip/transformer warnings

Browse files
Files changed (2) hide show
  1. pipeline/graph_store.py +3 -2
  2. start.sh +2 -1
pipeline/graph_store.py CHANGED
@@ -86,8 +86,9 @@ def store_entities(entities: list[dict], source: str, tier: str = "extended", se
86
  if not conn:
87
  return
88
 
89
- # Store nodes
90
- for ent in entities:
 
91
  if not isinstance(ent, dict):
92
  continue
93
  ent_name = ent.get("name") or ent.get("entity") or ent.get("id")
 
86
  if not conn:
87
  return
88
 
89
+ with _db_lock:
90
+ # Store nodes
91
+ for ent in entities:
92
  if not isinstance(ent, dict):
93
  continue
94
  ent_name = ent.get("name") or ent.get("entity") or ent.get("id")
start.sh CHANGED
@@ -98,7 +98,8 @@ fi
98
 
99
  # ── Uninstall conflicting torchaudio to prevent OSError ─────────────────────────
100
  echo "[pre-flight] Removing torchaudio to prevent transformers loading issues on HF..."
101
- python -m pip uninstall -y torchaudio || true
 
102
 
103
  # ── Start embed_llm on port 8003 ──────────────────────────────────────────────
104
  echo "[1/3] Starting embed_llm (port 8003)..."
 
98
 
99
  # ── Uninstall conflicting torchaudio to prevent OSError ─────────────────────────
100
  echo "[pre-flight] Removing torchaudio to prevent transformers loading issues on HF..."
101
+ export TRANSFORMERS_NO_ADVISORY_WARNINGS=1
102
+ python -m pip uninstall -y torchaudio --root-user-action=ignore || true
103
 
104
  # ── Start embed_llm on port 8003 ──────────────────────────────────────────────
105
  echo "[1/3] Starting embed_llm (port 8003)..."