fix: add reranker to init components (#739)

* update reader and search strategy

* set strategy reader and search config

* fix install problem

* fix

* fix test

* turn off graph recall

* turn off graph recall

* turn off graph recall

* fix Searcher input bug

* fix Searcher

* fix Search

* fix bug

* adjust strategy reader

* adjust strategy reader

* adjust search config input

* reformat code

* re pr

* format repair

* fix time issue

* develop feedback process

* feedback handler configuration

* upgrade feedback using

* add threshold

* update prompt

* update prompt

* fix handler

* add feedback scheduler

* add handler change node update

* add handler change node update

* add handler change node update

* add handler change node update

* fix interface input

* add chunk and ratio filter

* update stopwords

* fix messages queue

* add seach_by_keywords_LIKE

* add doc filter

* add retrieve query

* add retrieve queies

* patch info filter

* add log and make embedding safety net

* add log and make embedding safety net

* deduplicate add objects

* use _add_memories_parallel

* delete Special characters

* delete Special characters

* delete Special characters

* delete Special characters

* add source_doc_id

* add source_doc_id

* add reranker in init com..

* fix circle import

---------

Co-authored-by: 黑布林 <[email protected]>
Co-authored-by: CaralHsi <[email protected]>
Co-authored-by: chunyu li <[email protected]>

Author: 4 people

src/memos/mem_scheduler/general_modules/init_components_for_scheduler.py

@@ -160,6 +160,16 @@ def build_reranker_config() -> dict[str, Any]:
     return RerankerConfigFactory.model_validate(APIConfig.get_reranker_config())
 
 
+def build_feedback_reranker_config() -> dict[str, Any]:
+    """
+    Build reranker configuration.
+
+    Returns:
+        Validated reranker configuration dictionary
+    """
+    return RerankerConfigFactory.model_validate(APIConfig.get_feedback_reranker_config())
+
+
 def build_internet_retriever_config() -> dict[str, Any]:
     """
     Build internet retriever configuration.
@@ -277,6 +287,7 @@ def init_components() -> dict[str, Any]:
     embedder_config = build_embedder_config()
     mem_reader_config = build_mem_reader_config()
     reranker_config = build_reranker_config()
+    feedback_reranker_config = build_feedback_reranker_config()
     internet_retriever_config = build_internet_retriever_config()
     vector_db_config = build_vec_db_config()
     pref_extractor_config = build_pref_extractor_config()
@@ -296,6 +307,7 @@ def init_components() -> dict[str, Any]:
     embedder = EmbedderFactory.from_config(embedder_config)
     mem_reader = MemReaderFactory.from_config(mem_reader_config)
     reranker = RerankerFactory.from_config(reranker_config)
+    feedback_reranker = RerankerFactory.from_config(feedback_reranker_config)
     internet_retriever = InternetRetrieverFactory.from_config(
         internet_retriever_config, embedder=embedder
     )
@@ -359,7 +371,7 @@ def init_components() -> dict[str, Any]:
             config_factory=pref_retriever_config,
             llm_provider=llm,
             embedder=embedder,
-            reranker=reranker,
+            reranker=feedback_reranker,
             vector_db=vector_db,
         )
         if os.getenv("ENABLE_PREFERENCE_MEMORY", "false") == "true"
@@ -374,7 +386,7 @@ def init_components() -> dict[str, Any]:
             extractor_llm=llm,
             vector_db=vector_db,
             embedder=embedder,
-            reranker=reranker,
+            reranker=feedback_reranker,
             extractor=pref_extractor,
             adder=pref_adder,
             retriever=pref_retriever,
@@ -405,6 +417,7 @@ def init_components() -> dict[str, Any]:
         memory_manager=memory_manager,
         mem_reader=mem_reader,
         searcher=searcher,
+        reranker=feedback_reranker,
     )
     # Return all components as a dictionary for easy access and extension
     return {"naive_mem_cube": naive_mem_cube, "feedback_server": feedback_server}