test fixes

getzep · Feb 12, 2025 · 273a669 · 273a669
1 parent 1929eed
commit 273a669
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/graphiti_core/graphiti.py b/graphiti_core/graphiti.py
@@ -733,14 +733,16 @@ async def add_triplet(self, source_node: EntityNode, edge: EntityEdge, target_no
             ],
         )
 
+        updated_edge = resolve_edge_pointers([edge], uuid_map)[0]
+
         related_edges = await get_relevant_edges(
             self.driver,
-            [edge],
+            [updated_edge],
             source_node_uuid=resolved_nodes[0].uuid,
             target_node_uuid=resolved_nodes[1].uuid,
         )
 
-        resolved_edge = await dedupe_extracted_edge(self.llm_client, edge, related_edges)
+        resolved_edge = await dedupe_extracted_edge(self.llm_client, updated_edge, related_edges)
 
         contradicting_edges = await get_edge_contradictions(self.llm_client, edge, related_edges)
         invalidated_edges = resolve_edge_contradictions(resolved_edge, contradicting_edges)

diff --git a/graphiti_core/prompts/summarize_nodes.py b/graphiti_core/prompts/summarize_nodes.py
@@ -80,7 +80,7 @@ def summarize_context(context: dict[str, Any]) -> list[Message]:
         </MESSAGES>
         
         Given the above MESSAGES and the following ENTITY name and ENTITY CONTEXT, create a summary for the ENTITY. Your summary must only use
-        information from the provided MESSAGES. Your summary should also only contain information relevant to the
+        information from the provided MESSAGES and from the ENTITY CONTEXT. Your summary should also only contain information relevant to the
         provided ENTITY.
         
         Summaries must be under 500 words.