Skip to content
Merged
Show file tree
Hide file tree
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion src/memos/mem_scheduler/memory_manage_modules/retriever.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
2 changes: 1 addition & 1 deletion src/memos/memories/textual/tree.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
1 change: 1 addition & 0 deletions src/memos/types/general_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,7 @@ class FineStrategy(str, Enum):
REWRITE = "rewrite"
RECREATE = "recreate"
DEEP_SEARCH = "deep_search"
AGENTIC_SEARCH = "agentic_search"


# algorithm strategies
Expand Down
Loading