diff --git a/graphiti_core/graphiti.py b/graphiti_core/graphiti.py index af7d0b344..f69b149ba 100644 --- a/graphiti_core/graphiti.py +++ b/graphiti_core/graphiti.py @@ -1189,24 +1189,18 @@ async def add_triplet( if edge.fact_embedding is None: await edge.generate_embedding(self.embedder) - if source_node.uuid is not None: - try: - resolved_source = await EntityNode.get_by_uuid(self.driver, source_node.uuid) - except NodeNotFoundError: - raise ValueError(f'Node with UUID {source_node.uuid} not found') from None - else: + try: + resolved_source = await EntityNode.get_by_uuid(self.driver, source_node.uuid) + except NodeNotFoundError: resolved_source_nodes, _, _ = await resolve_extracted_nodes( self.clients, [source_node], ) resolved_source = resolved_source_nodes[0] - if target_node.uuid is not None: - try: - resolved_target = await EntityNode.get_by_uuid(self.driver, target_node.uuid) - except NodeNotFoundError: - raise ValueError(f'Node with UUID {target_node.uuid} not found') from None - else: + try: + resolved_target = await EntityNode.get_by_uuid(self.driver, target_node.uuid) + except NodeNotFoundError: resolved_target_nodes, _, _ = await resolve_extracted_nodes( self.clients, [target_node], diff --git a/pyproject.toml b/pyproject.toml index b089b452c..eeefd4434 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "graphiti-core" description = "A temporal graph building library" -version = "0.25.0" +version = "0.25.1" authors = [ { name = "Paul Paliychuk", email = "paul@getzep.com" }, { name = "Preston Rasmussen", email = "preston@getzep.com" },