Skip to content

Commit 178cb09

Browse files
authored
feat: reverted code and add memory_type (#627)
* feat: update agenticx searcg * feat: add memtype for recreate * fix: code format * feat: add use_name for not use_fast
1 parent 5a396b6 commit 178cb09

File tree

4 files changed

+8
-4
lines changed

4 files changed

+8
-4
lines changed

src/memos/mem_scheduler/memory_manage_modules/retriever.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,10 @@ def _process_enhancement_batch(
134134
for new_mem in processed_text_memories:
135135
enhanced_memories.append(
136136
TextualMemoryItem(
137-
memory=new_mem, metadata=TextualMemoryMetadata(user_id=user_id)
137+
memory=new_mem,
138+
metadata=TextualMemoryMetadata(
139+
user_id=user_id, memory_type="LongTermMemory"
140+
), # TODO add memory_type
138141
)
139142
)
140143
elif FINE_STRATEGY == FineStrategy.REWRITE:

src/memos/memories/textual/tree.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ def get_searcher(
138138
self.embedder,
139139
self.reranker,
140140
bm25_retriever=self.bm25_retriever,
141-
internet_retriever=self.internet_retriever,
141+
internet_retriever=None,
142142
search_strategy=self.search_strategy,
143143
manual_close_internet=manual_close_internet,
144144
process_llm=process_llm,

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ def process_node(node):
227227
{"field": "key", "op": "in", "value": parsed_goal.keys},
228228
{"field": "memory_type", "op": "=", "value": memory_scope},
229229
]
230-
key_ids = self.graph_store.get_by_metadata(key_filters)
230+
key_ids = self.graph_store.get_by_metadata(key_filters, user_name=user_name)
231231
candidate_ids.update(key_ids)
232232

233233
# 2) tag-based OR branch
@@ -236,7 +236,7 @@ def process_node(node):
236236
{"field": "tags", "op": "contains", "value": parsed_goal.tags},
237237
{"field": "memory_type", "op": "=", "value": memory_scope},
238238
]
239-
tag_ids = self.graph_store.get_by_metadata(tag_filters)
239+
tag_ids = self.graph_store.get_by_metadata(tag_filters, user_name=user_name)
240240
candidate_ids.update(tag_ids)
241241

242242
# No matches → return empty

src/memos/types/general_types.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,7 @@ class FineStrategy(str, Enum):
101101
REWRITE = "rewrite"
102102
RECREATE = "recreate"
103103
DEEP_SEARCH = "deep_search"
104+
AGENTIC_SEARCH = "agentic_search"
104105

105106

106107
# algorithm strategies

0 commit comments

Comments
 (0)