Remove delete_entity and delete_entity_relation from RediskKVStorage
This commit is contained in:
@@ -8,8 +8,8 @@ if not pm.is_installed("redis"):
|
|||||||
pm.install("redis")
|
pm.install("redis")
|
||||||
|
|
||||||
# aioredis is a depricated library, replaced with redis
|
# aioredis is a depricated library, replaced with redis
|
||||||
from redis.asyncio import Redis
|
from redis.asyncio import Redis # type: ignore
|
||||||
from lightrag.utils import logger, compute_mdhash_id
|
from lightrag.utils import logger
|
||||||
from lightrag.base import BaseKVStorage
|
from lightrag.base import BaseKVStorage
|
||||||
import json
|
import json
|
||||||
|
|
||||||
@@ -83,67 +83,3 @@ class RedisKVStorage(BaseKVStorage):
|
|||||||
logger.info(
|
logger.info(
|
||||||
f"Deleted {deleted_count} of {len(ids)} entries from {self.namespace}"
|
f"Deleted {deleted_count} of {len(ids)} entries from {self.namespace}"
|
||||||
)
|
)
|
||||||
|
|
||||||
async def delete_entity(self, entity_name: str) -> None:
|
|
||||||
"""Delete an entity by name
|
|
||||||
|
|
||||||
Args:
|
|
||||||
entity_name: Name of the entity to delete
|
|
||||||
"""
|
|
||||||
|
|
||||||
try:
|
|
||||||
entity_id = compute_mdhash_id(entity_name, prefix="ent-")
|
|
||||||
logger.debug(
|
|
||||||
f"Attempting to delete entity {entity_name} with ID {entity_id}"
|
|
||||||
)
|
|
||||||
|
|
||||||
# Delete the entity
|
|
||||||
result = await self._redis.delete(f"{self.namespace}:{entity_id}")
|
|
||||||
|
|
||||||
if result:
|
|
||||||
logger.debug(f"Successfully deleted entity {entity_name}")
|
|
||||||
else:
|
|
||||||
logger.debug(f"Entity {entity_name} not found in storage")
|
|
||||||
except Exception as e:
|
|
||||||
logger.error(f"Error deleting entity {entity_name}: {e}")
|
|
||||||
|
|
||||||
async def delete_entity_relation(self, entity_name: str) -> None:
|
|
||||||
"""Delete all relations associated with an entity
|
|
||||||
|
|
||||||
Args:
|
|
||||||
entity_name: Name of the entity whose relations should be deleted
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
# Get all keys in this namespace
|
|
||||||
cursor = 0
|
|
||||||
relation_keys = []
|
|
||||||
pattern = f"{self.namespace}:*"
|
|
||||||
|
|
||||||
while True:
|
|
||||||
cursor, keys = await self._redis.scan(cursor, match=pattern)
|
|
||||||
|
|
||||||
# For each key, get the value and check if it's related to entity_name
|
|
||||||
for key in keys:
|
|
||||||
value = await self._redis.get(key)
|
|
||||||
if value:
|
|
||||||
data = json.loads(value)
|
|
||||||
# Check if this is a relation involving the entity
|
|
||||||
if (
|
|
||||||
data.get("src_id") == entity_name
|
|
||||||
or data.get("tgt_id") == entity_name
|
|
||||||
):
|
|
||||||
relation_keys.append(key)
|
|
||||||
|
|
||||||
# Exit loop when cursor returns to 0
|
|
||||||
if cursor == 0:
|
|
||||||
break
|
|
||||||
|
|
||||||
# Delete the relation keys
|
|
||||||
if relation_keys:
|
|
||||||
deleted = await self._redis.delete(*relation_keys)
|
|
||||||
logger.debug(f"Deleted {deleted} relations for {entity_name}")
|
|
||||||
else:
|
|
||||||
logger.debug(f"No relations found for entity {entity_name}")
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
logger.error(f"Error deleting relations for {entity_name}: {e}")
|
|
||||||
|
Reference in New Issue
Block a user