Improve entity extraction logging with cleaner summary and verbose debug output

• Add count-based summary log message
• Move detailed data to verbose debug logs
This commit is contained in:
yangdx
2025-02-21 16:26:56 +08:00
parent 8164c57b7e
commit 2d8a262ac3

View File

@@ -23,6 +23,7 @@ from .utils import (
CacheData, CacheData,
statistic_data, statistic_data,
get_conversation_turns, get_conversation_turns,
verbose_debug,
) )
from .base import ( from .base import (
BaseGraphStorage, BaseGraphStorage,
@@ -532,7 +533,13 @@ async def extract_entities(
logger.info("Didn't extract any relationships") logger.info("Didn't extract any relationships")
logger.info( 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: if entity_vdb is not None: