From abad9f235c8d18ed440bf9d45d860a0600540f35 Mon Sep 17 00:00:00 2001
From: Yannick Stephan <stephan.yannick@me.com>
Date: Sun, 16 Feb 2025 15:54:54 +0100
Subject: [PATCH] make value on str enum

---
 lightrag/kg/json_doc_status_impl.py | 8 ++++----
 lightrag/kg/redis_impl.py           | 1 +
 lightrag/lightrag.py                | 8 ++++----
 3 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/lightrag/kg/json_doc_status_impl.py b/lightrag/kg/json_doc_status_impl.py
index 33df6d43..6c667891 100644
--- a/lightrag/kg/json_doc_status_impl.py
+++ b/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)
diff --git a/lightrag/kg/redis_impl.py b/lightrag/kg/redis_impl.py
index 98258741..8dae1e77 100644
--- a/lightrag/kg/redis_impl.py
+++ b/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):
diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index 82209504..3ad94da0 100644
--- a/lightrag/lightrag.py
+++ b/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,