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

Skip entity attribute extraction when no fields defined (#924)

Browse source
This commit is contained in:
Daniel Chalef 2025-09-24 10:23:37 -07:00 • committed by GitHub
parent 36056ad141
commit 7cf5ee6288
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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, 'ensure_ascii': ensure_ascii,
} }
has_entity_attributes: bool = bool(
entity_type is not None and len(entity_type.model_fields) != 0
)
llm_response = ( llm_response = (
( (
await llm_client.generate_response( await llm_client.generate_response(
@ -370,7 +374,7 @@ async def extract_attributes_from_node(
model_size=ModelSize.small, model_size=ModelSize.small,
) )
) )
if entity_type is not None if has_entity_attributes
else {} else {}
) )
@ -380,7 +384,7 @@ async def extract_attributes_from_node(
model_size=ModelSize.small, model_size=ModelSize.small,
) )
if entity_type is not None: if has_entity_attributes and entity_type is not None:
entity_type(**llm_response) entity_type(**llm_response)
node.summary = summary_response.get('summary', '') node.summary = summary_response.get('summary', '')