Fix reindex_embeddings crash: add networkx for Unstructured xlsx.
CI / test (pull_request) Successful in 10s
Unit Tests / test (pull_request) Successful in 10s

unstructured.partition.xlsx imports networkx; without it reindex dies
mid-run after clearing Chroma. Also skip per-document ingest failures
so one bad file cannot abort the whole rebuild.
This commit is contained in:
2026-08-02 13:54:58 -05:00
parent d8f5b8ebf2
commit 8ff3ad83d1
4 changed files with 33 additions and 1 deletions
@@ -46,7 +46,18 @@ class Command(BaseCommand):
self.stdout.write("Clearing Chroma collection…") self.stdout.write("Clearing Chroma collection…")
rag.clear_vector_store() rag.clear_vector_store()
self.stdout.write("Re-ingesting documents…") self.stdout.write("Re-ingesting documents…")
try:
rag.ingest_documents() rag.ingest_documents()
except Exception as exc:
# Partial ingest may still have written some chunks; report and re-raise.
count = rag.vector_store._collection.count()
self.stderr.write(
self.style.ERROR(
f"Reindex failed after clearing Chroma ({exc}). "
f"Vector chunks currently: {count}. Fix the error and re-run."
)
)
raise
count = rag.vector_store._collection.count() count = rag.vector_store._collection.count()
self.stdout.write( self.stdout.write(
self.style.SUCCESS(f"Reindex complete. Vector chunks now: {count}") self.style.SUCCESS(f"Reindex complete. Vector chunks now: {count}")
@@ -161,6 +161,13 @@ class RAGService(BaseService):
) )
if chunks: if chunks:
self.vector_store.add_documents(chunks) self.vector_store.add_documents(chunks)
except Exception:
# Keep reindex/ingest moving; one bad file must not wipe progress.
logger.exception(
"Failed to ingest document_id=%s file=%s",
doc.id,
doc.file.name,
)
finally: finally:
if os.path.exists(tmp_path): if os.path.exists(tmp_path):
os.unlink(tmp_path) os.unlink(tmp_path)
+1
View File
@@ -47,6 +47,7 @@ dependencies = [
"python-dateutil==2.9.0.post0", "python-dateutil==2.9.0.post0",
"pytz==2025.2", "pytz==2025.2",
"stripe>=12.0.0,<14.0.0", "stripe>=12.0.0,<14.0.0",
"networkx>=3.0,<4",
] ]
[dependency-groups] [dependency-groups]
Generated
+13
View File
@@ -651,6 +651,7 @@ dependencies = [
{ name = "langchain-text-splitters" }, { name = "langchain-text-splitters" },
{ name = "langgraph" }, { name = "langgraph" },
{ name = "matplotlib" }, { name = "matplotlib" },
{ name = "networkx" },
{ name = "numpy" }, { name = "numpy" },
{ name = "ollama" }, { name = "ollama" },
{ name = "openpyxl" }, { name = "openpyxl" },
@@ -661,6 +662,7 @@ dependencies = [
{ name = "python-dateutil" }, { name = "python-dateutil" },
{ name = "python-docx" }, { name = "python-docx" },
{ name = "pytz" }, { name = "pytz" },
{ name = "requests" },
{ name = "stripe" }, { name = "stripe" },
{ name = "unstructured" }, { name = "unstructured" },
{ name = "uvicorn" }, { name = "uvicorn" },
@@ -698,6 +700,7 @@ requires-dist = [
{ name = "langchain-text-splitters", specifier = "==1.1.2" }, { name = "langchain-text-splitters", specifier = "==1.1.2" },
{ name = "langgraph", specifier = ">=1.2.5,<1.3.0" }, { name = "langgraph", specifier = ">=1.2.5,<1.3.0" },
{ name = "matplotlib", specifier = "==3.10.7" }, { name = "matplotlib", specifier = "==3.10.7" },
{ name = "networkx", specifier = ">=3.0,<4" },
{ name = "numpy", specifier = "==2.2.6" }, { name = "numpy", specifier = "==2.2.6" },
{ name = "ollama", specifier = "==0.6.1" }, { name = "ollama", specifier = "==0.6.1" },
{ name = "openpyxl", specifier = "==3.1.5" }, { name = "openpyxl", specifier = "==3.1.5" },
@@ -708,6 +711,7 @@ requires-dist = [
{ name = "python-dateutil", specifier = "==2.9.0.post0" }, { name = "python-dateutil", specifier = "==2.9.0.post0" },
{ name = "python-docx", specifier = "==1.2.0" }, { name = "python-docx", specifier = "==1.2.0" },
{ name = "pytz", specifier = "==2025.2" }, { name = "pytz", specifier = "==2025.2" },
{ name = "requests", specifier = ">=2.32,<3" },
{ name = "stripe", specifier = ">=12.0.0,<14.0.0" }, { name = "stripe", specifier = ">=12.0.0,<14.0.0" },
{ name = "unstructured", specifier = "==0.18.21" }, { name = "unstructured", specifier = "==0.18.21" },
{ name = "uvicorn", specifier = "==0.38.0" }, { name = "uvicorn", specifier = "==0.38.0" },
@@ -2394,6 +2398,15 @@ wheels = [
{ url = "https://files.pythonhosted.org/packages/79/7b/2c79738432f5c924bef5071f933bcc9efd0473bac3b4aa584a6f7c1c8df8/mypy_extensions-1.1.0-py3-none-any.whl", hash = "sha256:1be4cccdb0f2482337c4743e60421de3a356cd97508abadd57d47403e94f5505", size = 4963, upload-time = "2025-04-22T14:54:22.983Z" }, { url = "https://files.pythonhosted.org/packages/79/7b/2c79738432f5c924bef5071f933bcc9efd0473bac3b4aa584a6f7c1c8df8/mypy_extensions-1.1.0-py3-none-any.whl", hash = "sha256:1be4cccdb0f2482337c4743e60421de3a356cd97508abadd57d47403e94f5505", size = 4963, upload-time = "2025-04-22T14:54:22.983Z" },
] ]
[[package]]
name = "networkx"
version = "3.6.1"
source = { registry = "https://pypi.org/simple" }
sdist = { url = "https://files.pythonhosted.org/packages/6a/51/63fe664f3908c97be9d2e4f1158eb633317598cfa6e1fc14af5383f17512/networkx-3.6.1.tar.gz", hash = "sha256:26b7c357accc0c8cde558ad486283728b65b6a95d85ee1cd66bafab4c8168509", size = 2517025, upload-time = "2025-12-08T17:02:39.908Z" }
wheels = [
{ url = "https://files.pythonhosted.org/packages/9e/c9/b2622292ea83fbb4ec318f5b9ab867d0a28ab43c5717bb85b0a5f6b3b0a4/networkx-3.6.1-py3-none-any.whl", hash = "sha256:d47fbf302e7d9cbbb9e2555a0d267983d2aa476bac30e90dfbe5669bd57f3762", size = 2068504, upload-time = "2025-12-08T17:02:38.159Z" },
]
[[package]] [[package]]
name = "nltk" name = "nltk"
version = "3.10.0" version = "3.10.0"