update
This commit is contained in:
parent
7e6d93fa32
commit
469703e180
2 changed files with 62 additions and 60 deletions
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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" },
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue