From 2af8a93dc71a97701379ac5db93c34d3c4f9f87f Mon Sep 17 00:00:00 2001
From: yangdx <gzdaniel@me.com>
Date: Thu, 31 Jul 2025 02:16:56 +0800
Subject: [PATCH] fix: resolve _sort_key error in Redis get_docs_paginated
 function

---
 lightrag/kg/redis_impl.py | 20 ++++++++------------
 lightrag/utils.py         |  3 ++-
 2 files changed, 10 insertions(+), 13 deletions(-)

diff --git a/lightrag/kg/redis_impl.py b/lightrag/kg/redis_impl.py
index 796d8410..ae18242f 100644
--- a/lightrag/kg/redis_impl.py
+++ b/lightrag/kg/redis_impl.py
@@ -999,14 +999,14 @@ class RedisDocStatusStorage(DocStatusStorage):
                                     if "error_msg" not in data:
                                         data["error_msg"] = None
 
-                                    # Add sort key for sorting
+                                    # Calculate sort key for sorting (but don't add to data)
                                     if sort_field == "id":
-                                        data["_sort_key"] = doc_id
+                                        sort_key = doc_id
                                     else:
-                                        data["_sort_key"] = data.get(sort_field, "")
+                                        sort_key = data.get(sort_field, "")
 
                                     doc_status = DocProcessingStatus(**data)
-                                    all_docs.append((doc_id, doc_status))
+                                    all_docs.append((doc_id, doc_status, sort_key))
 
                                 except (json.JSONDecodeError, KeyError) as e:
                                     logger.error(
@@ -1021,16 +1021,12 @@ class RedisDocStatusStorage(DocStatusStorage):
                 logger.error(f"Error getting paginated docs: {e}")
                 return [], 0
 
-        # Sort documents
+        # Sort documents using the separate sort key
         reverse_sort = sort_direction.lower() == "desc"
-        all_docs.sort(
-            key=lambda x: getattr(x[1], "_sort_key", ""), reverse=reverse_sort
-        )
+        all_docs.sort(key=lambda x: x[2], reverse=reverse_sort)
 
-        # Remove sort key from documents
-        for doc_id, doc in all_docs:
-            if hasattr(doc, "_sort_key"):
-                delattr(doc, "_sort_key")
+        # Remove sort key from tuples and keep only (doc_id, doc_status)
+        all_docs = [(doc_id, doc_status) for doc_id, doc_status, _ in all_docs]
 
         total_count = len(all_docs)
 
diff --git a/lightrag/utils.py b/lightrag/utils.py
index b809de0c..599fcc9c 100644
--- a/lightrag/utils.py
+++ b/lightrag/utils.py
@@ -145,7 +145,8 @@ class LightragPathFilter(logging.Filter):
 
             # Filter out successful GET requests to filtered paths
             if (
-                method == "GET" or method == "POST"
+                method == "GET"
+                or method == "POST"
                 and (status == 200 or status == 304)
                 and path in self.filtered_paths
             ):