Feat: feedback add strict info filter (#685)

* 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

---------

Co-authored-by: 黑布林 <11641432+heiheiyouyou@user.noreply.gitee.com>
Co-authored-by: CaralHsi <caralhsi@gmail.com>
Co-authored-by: chunyu li <78344051+fridayL@users.noreply.github.com>

Author: 4 people

src/memos/api/handlers/add_handler.py

@@ -89,6 +89,7 @@ def _check_messages(messages: MessageList) -> None:
                     feedback_content=feedback_content,
                     writable_cube_ids=add_req.writable_cube_ids,
                     async_mode=add_req.async_mode,
+                    info=add_req.info,
                 )
                 process_record = cube_view.feedback_memories(feedback_req)
 

src/memos/api/handlers/component_init.py

@@ -304,6 +304,7 @@ def init_server() -> dict[str, Any]:
         memory_manager=memory_manager,
         mem_reader=mem_reader,
         searcher=searcher,
+        reranker=reranker,
     )
 
     # Initialize Scheduler

src/memos/api/handlers/feedback_handler.py

@@ -28,7 +28,7 @@ def __init__(self, dependencies: HandlerDependencies):
             dependencies: HandlerDependencies instance
         """
         super().__init__(dependencies)
-        self._validate_dependencies("mem_reader", "mem_scheduler", "searcher")
+        self._validate_dependencies("mem_reader", "mem_scheduler", "searcher", "reranker")
 
     def handle_feedback_memories(self, feedback_req: APIFeedbackRequest) -> MemoryResponse:
         """

src/memos/api/product_models.py

@@ -684,6 +684,19 @@ class APIFeedbackRequest(BaseRequest):
         "async", description="feedback mode: sync or async"
     )
     corrected_answer: bool = Field(False, description="Whether need return corrected answer")
+    info: dict[str, Any] | None = Field(
+        None,
+        description=(
+            "Additional metadata for the add request. "
+            "All keys can be used as filters in search. "
+            "Example: "
+            "{'agent_id': 'xxxxxx', "
+            "'app_id': 'xxxx', "
+            "'source_type': 'web', "
+            "'source_url': 'https://www.baidu.com', "
+            "'source_content': 'West Lake is the most famous scenic spot in Hangzhou'}."
+        ),
+    )
     # ==== mem_cube_id is NOT enabled====
     mem_cube_id: str | None = Field(
         None,

src/memos/graph_dbs/polardb.py

@@ -1639,12 +1639,9 @@ def seach_by_keywords_like(
             """
 
         params = (query_word,)
-        logger.info(
-            f"[seach_by_keywords_LIKE start:]  user_name: {user_name}, query: {query}, params: {params}"
-        )
-        conn = None
+        logger.info(f"[seach_by_keywords_LIKE start:]  user_name: {user_name}, params: {params}")
+        conn = self._get_connection()
         try:
-            conn = self._get_connection()
             with conn.cursor() as cursor:
                 cursor.execute(query, params)
                 results = cursor.fetchall()
@@ -1654,7 +1651,7 @@ def seach_by_keywords_like(
                     id_val = str(oldid)
                     output.append({"id": id_val})
                 logger.info(
-                    f"[seach_by_keywords_LIKE end:] user_name: {user_name}, query: {query}, params: {params} recalled: {output}"
+                    f"[seach_by_keywords_LIKE end:] user_name: {user_name}, params: {params} recalled: {output}"
                 )
                 return output
         finally:
@@ -1739,9 +1736,8 @@ def seach_by_keywords_tfidf(
         logger.info(
             f"[seach_by_keywords_TFIDF start:] user_name: {user_name}, query: {query}, params: {params}"
         )
-        conn = None
+        conn = self._get_connection()
         try:
-            conn = self._get_connection()
             with conn.cursor() as cursor:
                 cursor.execute(query, params)
                 results = cursor.fetchall()
@@ -1751,6 +1747,9 @@ def seach_by_keywords_tfidf(
                     id_val = str(oldid)
                     output.append({"id": id_val})
 
+                logger.info(
+                    f"[seach_by_keywords_TFIDF end:] user_name: {user_name}, query: {query}, params: {params} recalled: {output}"
+                )
                 logger.info(
                     f"[seach_by_keywords_TFIDF end:] user_name: {user_name}, query: {query}, params: {params} recalled: {output}"
                 )