From 3b30b7985b60b7def817999e026c8ee03822fb85 Mon Sep 17 00:00:00 2001
From: Yannick Stephan <stephan.yannick@me.com>
Date: Mon, 17 Feb 2025 18:26:07 +0100
Subject: [PATCH] fixed str enum

---
 lightrag/base.py     | 4 ++--
 lightrag/lightrag.py | 8 ++++----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/lightrag/base.py b/lightrag/base.py
index 5de3d423..2dc7b035 100644
--- a/lightrag/base.py
+++ b/lightrag/base.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from enum import StrEnum
+from enum import Enum
 import os
 from dotenv import load_dotenv
 from dataclasses import dataclass, field
@@ -205,7 +205,7 @@ class BaseGraphStorage(StorageNameSpace, ABC):
         """Retrieve a subgraph of the knowledge graph starting from a given node."""
 
 
-class DocStatus(StrEnum):
+class DocStatus(str, Enum):
     """Document processing status"""
 
     PENDING = "pending"
diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index bf1c02d2..09a8df3f 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.value,
+                "status": DocStatus.PENDING,
                 "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.value,
+                            "status": DocStatus.PROCESSING,
                             "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.value,
+                                "status": DocStatus.PROCESSED,
                                 "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.value,
+                                "status": DocStatus.FAILED,
                                 "error": str(e),
                                 "content": status_doc.content,
                                 "content_summary": status_doc.content_summary,