File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
src/memos/memories/textual/tree_text_memory/retrieve Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments