From 6e20b7f5e8f591d44222df4179d3ba1cd81ad0d5 Mon Sep 17 00:00:00 2001 From: Daniel Chalef <131175+danielchalef@users.noreply.github.com> Date: Tue, 23 Sep 2025 18:48:41 -0700 Subject: [PATCH] Skip entity attribute extraction when no fields defined --- graphiti_core/utils/maintenance/node_operations.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/graphiti_core/utils/maintenance/node_operations.py b/graphiti_core/utils/maintenance/node_operations.py index f17706ab..55e79b9c 100644 --- a/graphiti_core/utils/maintenance/node_operations.py +++ b/graphiti_core/utils/maintenance/node_operations.py @@ -362,6 +362,10 @@ async def extract_attributes_from_node( 'ensure_ascii': ensure_ascii, } + has_entity_attributes: bool = bool( + entity_type is not None and len(entity_type.model_fields) != 0 + ) + llm_response = ( ( await llm_client.generate_response( @@ -370,7 +374,7 @@ async def extract_attributes_from_node( model_size=ModelSize.small, ) ) - if entity_type is not None + if has_entity_attributes else {} ) @@ -380,7 +384,7 @@ async def extract_attributes_from_node( model_size=ModelSize.small, ) - if entity_type is not None: + if has_entity_attributes and entity_type is not None: entity_type(**llm_response) node.summary = summary_response.get('summary', '')