Skip to content

Commit 312223c

Browse files
authored
fix: not include embedding (#330)
* feat: add default processing in mem-reader * feat: not include embedding
1 parent b25f68b commit 312223c

File tree

1 file changed

+3
-3
lines changed
  • src/memos/memories/textual/tree_text_memory/retrieve

1 file changed

+3
-3
lines changed

src/memos/memories/textual/tree_text_memory/retrieve/recall.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def retrieve(
5353
if memory_scope == "WorkingMemory":
5454
# For working memory, retrieve all entries (no filtering)
5555
working_memories = self.graph_store.get_all_memory_items(
56-
scope="WorkingMemory", include_embedding=True
56+
scope="WorkingMemory", include_embedding=False
5757
)
5858
return [TextualMemoryItem.from_dict(record) for record in working_memories]
5959

@@ -165,7 +165,7 @@ def _graph_recall(
165165
return []
166166

167167
# Load nodes and post-filter
168-
node_dicts = self.graph_store.get_nodes(list(candidate_ids), include_embedding=True)
168+
node_dicts = self.graph_store.get_nodes(list(candidate_ids), include_embedding=False)
169169

170170
final_nodes = []
171171
for node in node_dicts:
@@ -240,7 +240,7 @@ def search_single(vec, filt=None):
240240
unique_ids = {r["id"] for r in all_hits if r.get("id")}
241241
node_dicts = (
242242
self.graph_store.get_nodes(
243-
list(unique_ids), include_embedding=True, cube_name=cube_name
243+
list(unique_ids), include_embedding=False, cube_name=cube_name
244244
)
245245
or []
246246
)

0 commit comments

Comments
 (0)