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

make egg_operations more robust (#737)

Browse source 
update
This commit is contained in:
Preston Rasmussen 2025-07-16 17:12:20 -04:00 • committed by GitHub
parent 748464dfa5
commit 0ac2541b35
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
graphiti_core/utils/maintenance/edge_operations.py
View file

@ -274,8 +274,14 @@ async def resolve_extracted_edges(
# Determine which edge types are relevant for each edge # Determine which edge types are relevant for each edge
edge_types_lst: list[dict[str, BaseModel]] = [] edge_types_lst: list[dict[str, BaseModel]] = []
for extracted_edge in extracted_edges: for extracted_edge in extracted_edges:
source_node_labels = uuid_entity_map[extracted_edge.source_node_uuid].labels + ['Entity'] source_node = uuid_entity_map.get(extracted_edge.source_node_uuid)
target_node_labels = uuid_entity_map[extracted_edge.target_node_uuid].labels + ['Entity'] target_node = uuid_entity_map.get(extracted_edge.target_node_uuid)
source_node_labels = (
source_node.labels + ['Entity'] if source_node is not None else ['Entity']
)
target_node_labels = (
target_node.labels + ['Entity'] if target_node is not None else ['Entity']
)
label_tuples = [ label_tuples = [
(source_label, target_label) (source_label, target_label)
for source_label in source_node_labels for source_label in source_node_labels

2
pyproject.toml
View file

@ -1,7 +1,7 @@
[project] [project]
name = "graphiti-core" name = "graphiti-core"
description = "A temporal graph building library" description = "A temporal graph building library"
version = "0.17.5" version = "0.17.6"
authors = [ authors = [
{ "name" = "Paul Paliychuk", "email" = "paul@getzep.com" }, { "name" = "Paul Paliychuk", "email" = "paul@getzep.com" },
{ "name" = "Preston Rasmussen", "email" = "preston@getzep.com" }, { "name" = "Preston Rasmussen", "email" = "preston@getzep.com" },