Merge pull request #695 from ShanGor/main

Fix the bug from main stream that using doc['status'] and improve Apache AGE performance
2025-02-02 18:27:11 +08:00
parent fade69a2ae 02ac96ff8e
commit c07b5522fe
3 changed files with 52 additions and 6 deletions
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -373,7 +373,7 @@ class LightRAG:
            doc_id
            for doc_id in new_docs.keys()
            if (current_doc := await self.doc_status.get_by_id(doc_id)) is None
-            or current_doc["status"] == DocStatus.FAILED
+            or current_doc.status == DocStatus.FAILED
        }
        new_docs = {k: v for k, v in new_docs.items() if k in _add_doc_keys}