change: rerank_source for reranking

Author: fridayL

src/memos/reranker/concat.py

@@ -28,7 +28,8 @@ def process_source(
                 if "assistant:" in content:
                     continue
                 concat_data.append(content)
-    concat_data = [memory] + concat_data
+    if memory is not None:
+        concat_data = [memory, *concat_data]
     return "\n".join(concat_data)
 
 

src/memos/reranker/factory.py

@@ -29,6 +29,7 @@ def from_config(cfg: RerankerConfigFactory | None) -> BaseReranker | None:
                 model=c.get("model", "bge-reranker-v2-m3"),
                 timeout=int(c.get("timeout", 10)),
                 headers_extra=c.get("headers_extra"),
+                rerank_source=c.get("rerank_source"),
             )
 
         if backend in {"cosine_local", "cosine"}:

src/memos/reranker/http_bge.py

@@ -34,7 +34,7 @@ def __init__(
         model: str = "bge-reranker-v2-m3",
         timeout: int = 10,
         headers_extra: dict | None = None,
-        concat_source: list[str] | None = None,
+        rerank_source: list[str] | None = None,
     ):
         if not reranker_url:
             raise ValueError("reranker_url must not be empty")
@@ -43,9 +43,7 @@ def __init__(
         self.model = model
         self.timeout = timeout
         self.headers_extra = headers_extra or {}
-        # self.concat_source = concat_source or ["sources"]
-        self.concat_source = []
-        
+        self.concat_source = rerank_source
 
     def rerank(
         self,
@@ -59,7 +57,7 @@ def rerank(
 
         documents = []
         if self.concat_source:
-            documents = concat_original_source(graph_results)
+            documents = concat_original_source(graph_results, self.concat_source)
         else:
             documents = [
                 (_TAG1.sub("", m) if isinstance((m := getattr(item, "memory", None)), str) else m)