diff --git a/lightrag/kg/json_doc_status_impl.py b/lightrag/kg/json_doc_status_impl.py index 15fbfcde..7fccf3c3 100644 --- a/lightrag/kg/json_doc_status_impl.py +++ b/lightrag/kg/json_doc_status_impl.py @@ -39,7 +39,7 @@ class JsonDocStatusStorage(DocStatusStorage): async def get_status_counts(self) -> dict[str, int]: """Get counts of documents in each status""" - counts = {status: 0 for status in DocStatus} + counts = {status.value: 0 for status in DocStatus} for doc in self._data.values(): counts[doc["status"]] += 1 return counts