diff --git a/lightrag/kg/redis_impl.py b/lightrag/kg/redis_impl.py
index 38a12ab5..db343dee 100644
--- a/lightrag/kg/redis_impl.py
+++ b/lightrag/kg/redis_impl.py
@@ -9,9 +9,9 @@ if not pm.is_installed("redis"):
     pm.install("redis")
 
 # aioredis is a depricated library, replaced with redis
-from redis.asyncio import Redis, ConnectionPool
-from redis.exceptions import RedisError, ConnectionError
-from lightrag.utils import logger, compute_mdhash_id
+from redis.asyncio import Redis, ConnectionPool # type: ignore
+from redis.exceptions import RedisError, ConnectionError # type: ignore
+from lightrag.utils import logger
 
 from lightrag.base import BaseKVStorage
 import json
diff --git a/lightrag/operate.py b/lightrag/operate.py
index 97eccca0..1e023004 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -1395,7 +1395,7 @@ async def _find_most_related_text_unit_from_entities(
                 tasks.append((c_id, index, this_edges))
 
     # Process in batches of 25 tasks at a time to avoid overwhelming resources
-    batch_size = 25
+    batch_size = 5
     results = []
     
     for i in range(0, len(tasks), batch_size):