Replace delete() with drop_cache_by_modes() method to implement cache clearing operations

This commit is contained in:
yangdx
2025-03-31 23:23:31 +08:00
parent 95a8ee27ed
commit 0981ee1af9

View File

@@ -1894,12 +1894,18 @@ class LightRAG:
try:
# Reset the cache storage for specified mode
if modes:
await self.llm_response_cache.delete(modes)
success = await self.llm_response_cache.drop_cache_by_modes(modes)
if success:
logger.info(f"Cleared cache for modes: {modes}")
else:
logger.warning(f"Failed to clear cache for modes: {modes}")
else:
# Clear all modes
await self.llm_response_cache.delete(valid_modes)
success = await self.llm_response_cache.drop_cache_by_modes(valid_modes)
if success:
logger.info("Cleared all cache")
else:
logger.warning("Failed to clear all cache")
await self.llm_response_cache.index_done_callback()