From 0e7c34f5416c8c2a710915e88003f97e4986dbd2 Mon Sep 17 00:00:00 2001 From: Prateek Chhikara <46902268+prateekchhikara@users.noreply.github.com> Date: Wed, 7 May 2025 23:19:58 -0700 Subject: [PATCH] Renamed unknown node type (#2649) --- mem0/memory/graph_memory.py | 4 ++-- mem0/memory/memgraph_memory.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mem0/memory/graph_memory.py b/mem0/memory/graph_memory.py index 760d864a..685862a1 100644 --- a/mem0/memory/graph_memory.py +++ b/mem0/memory/graph_memory.py @@ -327,8 +327,8 @@ class MemoryGraph: relationship = item["relationship"] # types - source_type = entity_type_map.get(source, "unknown") - destination_type = entity_type_map.get(destination, "unknown") + source_type = entity_type_map.get(source, "__User__") + destination_type = entity_type_map.get(destination, "__User__") # embeddings source_embedding = self.embedding_model.embed(source) diff --git a/mem0/memory/memgraph_memory.py b/mem0/memory/memgraph_memory.py index 3ebc4254..8071c1c0 100644 --- a/mem0/memory/memgraph_memory.py +++ b/mem0/memory/memgraph_memory.py @@ -369,8 +369,8 @@ class MemoryGraph: relationship = item["relationship"] # types - source_type = entity_type_map.get(source, "unknown") - destination_type = entity_type_map.get(destination, "unknown") + source_type = entity_type_map.get(source, "__User__") + destination_type = entity_type_map.get(destination, "__User__") # embeddings source_embedding = self.embedding_model.embed(source)