diff --git a/graphiti_core/utils/bulk_utils.py b/graphiti_core/utils/bulk_utils.py index 5f253d30..ed3fd00e 100644 --- a/graphiti_core/utils/bulk_utils.py +++ b/graphiti_core/utils/bulk_utils.py @@ -139,7 +139,7 @@ async def add_nodes_and_edges_bulk_tx( for edge in entity_edges: if edge.fact_embedding is None: - await edge.generate_name_fact(embedder) + await edge.generate_embedding(embedder) await tx.run(EPISODIC_NODE_SAVE_BULK, episodes=episodes) await tx.run(ENTITY_NODE_SAVE_BULK, nodes=nodes) diff --git a/graphiti_core/utils/maintenance/edge_operations.py b/graphiti_core/utils/maintenance/edge_operations.py index fcebfb7a..58d3a33b 100644 --- a/graphiti_core/utils/maintenance/edge_operations.py +++ b/graphiti_core/utils/maintenance/edge_operations.py @@ -280,7 +280,7 @@ async def resolve_extracted_edges( logger.debug(f'Resolved edges: {[(e.name, e.uuid) for e in resolved_edges]}') await semaphore_gather( - create_entity_edge_embeddings(embedder, extracted_edges), + create_entity_edge_embeddings(embedder, resolved_edges), create_entity_edge_embeddings(embedder, invalidated_edges), ) diff --git a/pyproject.toml b/pyproject.toml index 7ae136c7..a04be18a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "graphiti-core" description = "A temporal graph building library" -version = "0.11.6pre3" +version = "0.11.6pre4" authors = [ { "name" = "Paul Paliychuk", "email" = "paul@getzep.com" }, { "name" = "Preston Rasmussen", "email" = "preston@getzep.com" },