Skip to content

Commit 801994b

Browse files
authored
Feat: remove dup func name and add agentic search (#537)
* feat: update memos headers * feat: headers add * feat: update search agent * feat: upadte mem story * feat: update mem scehduler * feat: update deepsearch mem code * feat: update deepsearch agent * feat: update test code * fix: remove dup config * feat: dock search pipeline * fix: code test * feat: add test scripts * feat: add test * feat: update need_raw process * fix: add initter * fix: change agent search func name
1 parent f696b41 commit 801994b

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/memos/multi_mem_cube/single_cube.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ def _deep_search(
203203
formatted_memories = [format_memory_item(data) for data in enhanced_memories]
204204
return formatted_memories
205205

206-
def _deep_search(
206+
def _agentic_search(
207207
self, search_req: APISearchRequest, user_context: UserContext, max_thinking_depth: int
208208
) -> list:
209209
deepsearch_results = self.deepsearch_agent.run(
@@ -229,6 +229,8 @@ def _fine_search(
229229
"""
230230
if FINE_STRATEGY == FineStrategy.DEEP_SEARCH:
231231
return self._deep_search(search_req=search_req, user_context=user_context)
232+
elif FINE_STRATEGY == FineStrategy.AGENTIC_SEARCH:
233+
return self._agentic_search(search_req=search_req, user_context=user_context)
232234

233235
target_session_id = search_req.session_id or "default_session"
234236
search_filter = {"session_id": search_req.session_id} if search_req.session_id else None

src/memos/types/general_types.py

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

106107

107108
# algorithm strategies

0 commit comments

Comments
 (0)