Skip to content
Merged
10 changes: 10 additions & 0 deletions src/memos/api/handlers/search_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 (
Expand Down Expand Up @@ -81,6 +82,15 @@ def handle_search_memories(self, search_req: APISearchRequest) -> SearchResponse
results = self._mmr_dedup_text_memories(results, search_req.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=search_req_local.top_k,
file_mem_proportion=0.5,
)

self.logger.info(
f"[SearchHandler] Final search results: count={len(results)} results={results}"
)
Expand Down