This commit is contained in:
prestonrasmussen 2025-09-08 14:59:27 -04:00
parent 7e6d93fa32
commit 469703e180
2 changed files with 62 additions and 60 deletions

View file

@ -1015,7 +1015,8 @@ class Graphiti:
return SearchResults(edges=edges, nodes=nodes) return SearchResults(edges=edges, nodes=nodes)
async def add_triplet(self, source_node: EntityNode, edge: EntityEdge, target_node: EntityNode): async def add_triplet(self, source_node: EntityNode, edge: EntityEdge,
target_node: EntityNode) -> AddEpisodeResults:
if source_node.name_embedding is None: if source_node.name_embedding is None:
await source_node.generate_name_embedding(self.embedder) await source_node.generate_name_embedding(self.embedder)
if target_node.name_embedding is None: if target_node.name_embedding is None:
@ -1059,6 +1060,7 @@ class Graphiti:
await create_entity_node_embeddings(self.embedder, nodes) await create_entity_node_embeddings(self.embedder, nodes)
await add_nodes_and_edges_bulk(self.driver, [], [], nodes, edges, self.embedder) await add_nodes_and_edges_bulk(self.driver, [], [], nodes, edges, self.embedder)
return AddEpisodeResults(edges=edges, nodes=nodes)
async def remove_episode(self, episode_uuid: str): async def remove_episode(self, episode_uuid: str):
# Find the episode to be deleted # Find the episode to be deleted

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.20.3" version = "0.20.4"
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" },