make value on str enum

lightrag/kg/json_doc_status_impl.py

@@ -49,10 +49,10 @@ class JsonDocStatusStorage(DocStatusStorage):
     ) -> dict[str, DocProcessingStatus]:
         """Get all documents with a specific status"""
         return {
-                    k: DocProcessingStatus(**v)
-                    for k, v in self._data.items()
-                    if v["status"] == status.value
-                }
+            k: DocProcessingStatus(**v)
+            for k, v in self._data.items()
+            if v["status"] == status.value
+        }
 
     async def index_done_callback(self) -> None:
         write_json(self._data, self._file_name)

lightrag/kg/redis_impl.py

@@ -18,6 +18,7 @@ import json
 config = configparser.ConfigParser()
 config.read("config.ini", "utf-8")
 
+
 @final
 @dataclass
 class RedisKVStorage(BaseKVStorage):

lightrag/lightrag.py

@@ -674,7 +674,7 @@ class LightRAG:
                 "content": content,
                 "content_summary": self._get_content_summary(content),
                 "content_length": len(content),
-                "status": DocStatus.PENDING,
+                "status": DocStatus.PENDING.value,
                 "created_at": datetime.now().isoformat(),
                 "updated_at": datetime.now().isoformat(),
             }
@@ -745,7 +745,7 @@ class LightRAG:
                 await self.doc_status.upsert(
                     {
                         doc_status_id: {
-                            "status": DocStatus.PROCESSING,
+                            "status": DocStatus.PROCESSING.value,
                             "updated_at": datetime.now().isoformat(),
                             "content": status_doc.content,
                             "content_summary": status_doc.content_summary,
@@ -782,7 +782,7 @@ class LightRAG:
                     await self.doc_status.upsert(
                         {
                             doc_status_id: {
-                                "status": DocStatus.PROCESSED,
+                                "status": DocStatus.PROCESSED.value,
                                 "chunks_count": len(chunks),
                                 "content": status_doc.content,
                                 "content_summary": status_doc.content_summary,
@@ -799,7 +799,7 @@ class LightRAG:
                     await self.doc_status.upsert(
                         {
                             doc_status_id: {
-                                "status": DocStatus.FAILED,
+                                "status": DocStatus.FAILED.value,
                                 "error": str(e),
                                 "content": status_doc.content,
                                 "content_summary": status_doc.content_summary,