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

Merge 2110dbe564 into 422558d06c

Browse source
This commit is contained in:
Sarthak Agarwal 2025-11-27 16:44:57 +01:00 • committed by GitHub
commit 3f603a794a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
graphiti_core/utils/ontology_utils/entity_types_utils.py
View file

@ -26,12 +26,17 @@ def validate_entity_types(
if entity_types is None:
return True
entity_node_field_names = EntityNode.model_fields.keys()
# Iterate through the provided entity types
for entity_type_name, entity_type_model in entity_types.items():
entity_type_field_names = entity_type_model.model_fields.keys()
for entity_type_field_name in entity_type_field_names:
if entity_type_field_name in entity_node_field_names:
raise EntityTypeValidationError(entity_type_name, entity_type_field_name)
# Convert model fields to set for fast intersection
entity_type_field_names = set(entity_type_model.model_fields.keys())
# Intersect to find any clashing field
conflict_fields = _ENTITY_NODE_FIELD_NAMES & entity_type_field_names
if conflict_fields:
# Only raise for the first conflict found, as per original behavior
raise EntityTypeValidationError(entity_type_name, next(iter(conflict_fields)))
return True
return True # Preserve existing comment
_ENTITY_NODE_FIELD_NAMES = set(EntityNode.model_fields.keys())