diff --git a/src/memos/mem_scheduler/memory_manage_modules/retriever.py b/src/memos/mem_scheduler/memory_manage_modules/retriever.py index 2278abc2a..fdd8a8cfe 100644 --- a/src/memos/mem_scheduler/memory_manage_modules/retriever.py +++ b/src/memos/mem_scheduler/memory_manage_modules/retriever.py @@ -134,7 +134,10 @@ def _process_enhancement_batch( for new_mem in processed_text_memories: enhanced_memories.append( TextualMemoryItem( - memory=new_mem, metadata=TextualMemoryMetadata(user_id=user_id) + memory=new_mem, + metadata=TextualMemoryMetadata( + user_id=user_id, memory_type="LongTermMemory" + ), # TODO add memory_type ) ) elif FINE_STRATEGY == FineStrategy.REWRITE: diff --git a/src/memos/memories/textual/tree.py b/src/memos/memories/textual/tree.py index 813142826..b4b1c0f23 100644 --- a/src/memos/memories/textual/tree.py +++ b/src/memos/memories/textual/tree.py @@ -138,7 +138,7 @@ def get_searcher( self.embedder, self.reranker, bm25_retriever=self.bm25_retriever, - internet_retriever=self.internet_retriever, + internet_retriever=None, search_strategy=self.search_strategy, manual_close_internet=manual_close_internet, process_llm=process_llm, diff --git a/src/memos/memories/textual/tree_text_memory/retrieve/recall.py b/src/memos/memories/textual/tree_text_memory/retrieve/recall.py index dea83887e..0b86b4ab2 100644 --- a/src/memos/memories/textual/tree_text_memory/retrieve/recall.py +++ b/src/memos/memories/textual/tree_text_memory/retrieve/recall.py @@ -227,7 +227,7 @@ def process_node(node): {"field": "key", "op": "in", "value": parsed_goal.keys}, {"field": "memory_type", "op": "=", "value": memory_scope}, ] - key_ids = self.graph_store.get_by_metadata(key_filters) + key_ids = self.graph_store.get_by_metadata(key_filters, user_name=user_name) candidate_ids.update(key_ids) # 2) tag-based OR branch @@ -236,7 +236,7 @@ def process_node(node): {"field": "tags", "op": "contains", "value": parsed_goal.tags}, {"field": "memory_type", "op": "=", "value": memory_scope}, ] - tag_ids = self.graph_store.get_by_metadata(tag_filters) + tag_ids = self.graph_store.get_by_metadata(tag_filters, user_name=user_name) candidate_ids.update(tag_ids) # No matches → return empty diff --git a/src/memos/types/general_types.py b/src/memos/types/general_types.py index 3706b49da..44c75ec02 100644 --- a/src/memos/types/general_types.py +++ b/src/memos/types/general_types.py @@ -101,6 +101,7 @@ class FineStrategy(str, Enum): REWRITE = "rewrite" RECREATE = "recreate" DEEP_SEARCH = "deep_search" + AGENTIC_SEARCH = "agentic_search" # algorithm strategies