From 2d8a262ac37104a27ecc7778498f2dd9c698509e Mon Sep 17 00:00:00 2001 From: yangdx Date: Fri, 21 Feb 2025 16:26:56 +0800 Subject: [PATCH] Improve entity extraction logging with cleaner summary and verbose debug output MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit • Add count-based summary log message • Move detailed data to verbose debug logs --- lightrag/operate.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/lightrag/operate.py b/lightrag/operate.py index a79192ac..4e235327 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -23,6 +23,7 @@ from .utils import ( CacheData, statistic_data, get_conversation_turns, + verbose_debug, ) from .base import ( BaseGraphStorage, @@ -532,7 +533,13 @@ async def extract_entities( logger.info("Didn't extract any relationships") logger.info( - f"New entities or relationships extracted, entities:{all_entities_data}, relationships:{all_relationships_data}" + f"Extracted {len(all_entities_data)} entities and {len(all_relationships_data)} relationships" + ) + verbose_debug( + f"New entities:{all_entities_data}, relationships:{all_relationships_data}" + ) + verbose_debug( + f"New relationships:{all_relationships_data}" ) if entity_vdb is not None: