AI Agent commited on
Commit Β·
5ed03b8
1
Parent(s): fbc58fa
Fix Kuzu graph write lock error and suppress pip/transformer warnings
Browse files- pipeline/graph_store.py +3 -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 |
-
|
| 90 |
-
|
|
|
|
| 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 |
-
|
|
|
|
| 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)..."
|