yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-12-02 09:57:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Fix linting"

Browse Source 
This reverts commit abd9de2a63b39021320855fab9f5c3b332dee162.
This commit is contained in:
yangdx 2025-06-29 22:35:26 +08:00
parent 75ea0bc38d
commit 10cd9c90e7
1 changed files with 8 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
lightrag/kg/redis_impl.py
View File

@ -86,39 +86,32 @@ class RedisKVStorage(BaseKVStorage):
result = {}
pattern = f"{self.namespace}:*"
cursor = 0
while True:
cursor, keys = await redis.scan(
cursor, match=pattern, count=100
)
cursor, keys = await redis.scan(cursor, match=pattern, count=100)
if keys:
# Batch get values for these keys
pipe = redis.pipeline()
for key in keys:
pipe.get(key)
values = await pipe.execute()
# Check each value for cache_type == "extract"
for key, value in zip(keys, values):
if value:
try:
data = json.loads(value)
if (
isinstance(data, dict)
and data.get("cache_type") == "extract"
):
if isinstance(data, dict) and data.get("cache_type") == "extract":
# Extract cache key (remove namespace prefix)
cache_key = key.replace(
f"{self.namespace}:", ""
)
cache_key = key.replace(f"{self.namespace}:", "")
result[cache_key] = data
except json.JSONDecodeError:
continue
if cursor == 0:
break
return result if result else None
except Exception as e:
logger.error(f"Error scanning Redis for extract cache entries: {e}")