gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Skip entity attribute extraction when no fields defined

Browse source
This commit is contained in:
Daniel Chalef 2025-09-23 18:48:41 -07:00
parent 36056ad141
commit 6e20b7f5e8
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
graphiti_core/utils/maintenance/node_operations.py
View file

@ -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', '')