klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Renamed unknown node type (#2649)

Browse Source
This commit is contained in:
Prateek Chhikara
2025-05-07 23:19:58 -07:00
committed by GitHub
parent 2b58775c17
commit 0e7c34f541
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mem0/memory/graph_memory.py
View File

@@ -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)

4
mem0/memory/memgraph_memory.py
View File

@@ -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)