From 2fe343f53235fcadc88e1577c95138d7f9116c57 Mon Sep 17 00:00:00 2001 From: Dubberman <48425266+whipser030@users.noreply.github.com> Date: Thu, 5 Feb 2026 16:34:38 +0800 Subject: [PATCH] hotfix: Code restoration, with the original content of the knowledge base replaced by the sources. (#1020) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit use knowledge source mem Co-authored-by: 黑布林 <11641432+heiheiyouyou@user.noreply.gitee.com> --- src/memos/api/handlers/search_handler.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/memos/api/handlers/search_handler.py b/src/memos/api/handlers/search_handler.py index 6eda1e2aa..f8ab90470 100644 --- a/src/memos/api/handlers/search_handler.py +++ b/src/memos/api/handlers/search_handler.py @@ -11,6 +11,7 @@ from typing import Any from memos.api.handlers.base_handler import BaseHandler, HandlerDependencies +from memos.api.handlers.formatters_handler import rerank_knowledge_mem from memos.api.product_models import APISearchRequest, SearchResponse from memos.log import get_logger from memos.memories.textual.tree_text_memory.retrieve.retrieve_utils import ( @@ -93,6 +94,15 @@ def handle_search_memories(self, search_req: APISearchRequest) -> SearchResponse results = self._mmr_dedup_text_memories(results, original_top_k, pref_top_k) self._strip_embeddings(results) + text_mem = results["text_mem"] + results["text_mem"] = rerank_knowledge_mem( + self.reranker, + query=search_req.query, + text_mem=text_mem, + top_k=original_top_k, + file_mem_proportion=0.5, + ) + self.logger.info( f"[SearchHandler] Final search results: count={len(results)} results={results}" )