diff --git a/src/memos/api/handlers/chat_handler.py b/src/memos/api/handlers/chat_handler.py index 9b0048ed..f6023e5c 100644 --- a/src/memos/api/handlers/chat_handler.py +++ b/src/memos/api/handlers/chat_handler.py @@ -116,7 +116,7 @@ def handle_chat_complete(self, chat_req: APIChatCompleteRequest) -> dict[str, An query=chat_req.query, top_k=chat_req.top_k or 10, session_id=chat_req.session_id, - mode=SearchMode.FINE, + mode=SearchMode.FAST, internet_search=chat_req.internet_search, moscube=chat_req.moscube, chat_history=chat_req.history, @@ -213,8 +213,8 @@ def generate_chat_response() -> Generator[str, None, None]: query=chat_req.query, top_k=20, session_id=chat_req.session_id, - mode=SearchMode.FINE, - internet_search=chat_req.internet_search, + mode=SearchMode.FINE if chat_req.internet_search else SearchMode.FAST, + internet_search=chat_req.internet_search, # TODO this param is not worked at fine mode moscube=chat_req.moscube, chat_history=chat_req.history, ) diff --git a/src/memos/api/handlers/search_handler.py b/src/memos/api/handlers/search_handler.py index 9fc8a5b2..e8e4e07d 100644 --- a/src/memos/api/handlers/search_handler.py +++ b/src/memos/api/handlers/search_handler.py @@ -245,7 +245,7 @@ def _fine_search( query=search_req.query, user_name=user_context.mem_cube_id, top_k=search_req.top_k, - mode=SearchMode.FAST, + mode=SearchMode.FINE, manual_close_internet=not search_req.internet_search, moscube=search_req.moscube, search_filter=search_filter, diff --git a/src/memos/memories/textual/tree_text_memory/retrieve/searcher.py b/src/memos/memories/textual/tree_text_memory/retrieve/searcher.py index f196c556..14ea8e2c 100644 --- a/src/memos/memories/textual/tree_text_memory/retrieve/searcher.py +++ b/src/memos/memories/textual/tree_text_memory/retrieve/searcher.py @@ -182,7 +182,7 @@ def _parse_task( query_embedding = None # fine mode will trigger initial embedding search - if mode == "fine": + if mode == "fine_old": logger.info("[SEARCH] Fine mode: embedding search") query_embedding = self.embedder.embed([query])[0]