add feedback judgement

Author: 黑布林

src/memos/mem_feedback/feedback.py

@@ -115,7 +115,7 @@ def _retry_db_operation(self, operation):
             return operation()
         except Exception as e:
             logger.error(
-                f"[Feedback Core: _retry_db_operation] DB operation failed: {e}", exc_info=True
+                f"[1223 Feedback Core: _retry_db_operation] DB operation failed: {e}", exc_info=True
             )
             raise
 
@@ -129,7 +129,7 @@ def _batch_embed(self, texts: list[str], embed_bs: int = 5):
                 results.extend(self._embed_once(batch))
             except Exception as e:
                 logger.error(
-                    f"[Feedback Core: process_feedback_core] Embedding batch failed, Cover with all zeros: {len(batch)} entries: {e}"
+                    f"[1223 Feedback Core: process_feedback_core] Embedding batch failed, Cover with all zeros: {len(batch)} entries: {e}"
                 )
                 results.extend([[0.0] * dim for _ in range(len(batch))])
         return results
@@ -145,7 +145,7 @@ def _pure_add(self, user_name: str, feedback_content: str, feedback_time: str, i
             lambda: self.memory_manager.add(to_add_memories, user_name=user_name, use_batch=False)
         )
         logger.info(
-            f"[Feedback Core: _pure_add] Pure added {len(added_ids)} memories for user {user_name}."
+            f"[1223 Feedback Core: _pure_add] Pure added {len(added_ids)} memories for user {user_name}."
         )
         return {
             "record": {
@@ -182,7 +182,7 @@ def _keyword_replace_judgement(self, feedback_content: str) -> dict | None:
             return judge_res
         else:
             logger.warning(
-                "[Feedback Core: _feedback_judgement] feedback judgement failed, return []"
+                "[1223 Feedback Core: _feedback_judgement] feedback judgement failed, return []"
             )
             return {}
 
@@ -207,7 +207,7 @@ def _feedback_judgement(
             return judge_res
         else:
             logger.warning(
-                "[Feedback Core: _feedback_judgement] feedback judgement failed, return []"
+                "[1223 Feedback Core: _feedback_judgement] feedback judgement failed, return []"
             )
             return []
 
@@ -334,11 +334,11 @@ def _del_working_binding(self, user_name, mem_items: list[TextualMemoryItem]) ->
                 self.graph_store.delete_node(mid, user_name=user_name)
 
                 logger.info(
-                    f"[Feedback Core:_del_working_binding] Delete raw/working mem_ids: {delete_ids} for user_name: {user_name}"
+                    f"[1223 Feedback Core:_del_working_binding] Delete raw/working mem_ids: {delete_ids} for user_name: {user_name}"
                 )
             except Exception as e:
                 logger.warning(
-                    f"[Feedback Core:_del_working_binding] TreeTextMemory.delete_hard: failed to delete {mid}: {e}"
+                    f"[1223 Feedback Core:_del_working_binding] TreeTextMemory.delete_hard: failed to delete {mid}: {e}"
                 )
 
     def semantics_feedback(
@@ -407,7 +407,9 @@ def semantics_feedback(
                         ):
                             all_operations.extend(chunk_operations["operations"])
                     except Exception as e:
-                        logger.error(f"[Feedback Core: semantics_feedback] Operation failed: {e}")
+                        logger.error(
+                            f"[1223 Feedback Core: semantics_feedback] Operation failed: {e}"
+                        )
 
             standard_operations = self.standard_operations(all_operations, current_memories)
             operations = self.filter_fault_update(standard_operations)
@@ -456,7 +458,7 @@ def semantics_feedback(
                         update_results.append(result)
                 except Exception as e:
                     logger.error(
-                        f"[Feedback Core: semantics_feedback] Operation failed for {original_op}: {e}",
+                        f"[1223 Feedback Core: semantics_feedback] Operation failed for {original_op}: {e}",
                         exc_info=True,
                     )
         if update_results:
@@ -484,7 +486,7 @@ def _feedback_memory(
         ]
         if filterd_ids:
             logger.warning(
-                f"[Feedback Core: _feedback_memory] Since the tags mode is fast, no modifications are made to the following memory {filterd_ids}."
+                f"[1223 Feedback Core: _feedback_memory] Since the tags mode is fast, no modifications are made to the following memory {filterd_ids}."
             )
 
         current_memories = [
@@ -516,7 +518,7 @@ def _feedback_memory(
                         results[i] = node
                 except Exception as e:
                     logger.error(
-                        f"[Feedback Core: _feedback_memory] Error processing memory index {i}: {e}",
+                        f"[1223 Feedback Core: _feedback_memory] Error processing memory index {i}: {e}",
                         exc_info=True,
                     )
             mem_res = [r for r in results if r]
@@ -575,15 +577,15 @@ def _vec_query(self, new_memories_embedding: list[float], user_name=None):
 
         if not retrieved_ids:
             logger.info(
-                f"[Feedback Core: _vec_query] No similar memories found for embedding query for user {user_name}."
+                f"[1223 Feedback Core: _vec_query] No similar memories found for embedding query for user {user_name}."
             )
 
         filterd_ids = [
             item["id"] for item in current_memories if "mode:fast" in item["metadata"]["tags"]
         ]
         if filterd_ids:
             logger.warning(
-                f"[Feedback Core: _vec_query] Since the tags mode is fast, no modifications are made to the following memory {filterd_ids}."
+                f"[1223 Feedback Core: _vec_query] Since the tags mode is fast, no modifications are made to the following memory {filterd_ids}."
             )
         return [
             TextualMemoryItem(**item)
@@ -637,9 +639,9 @@ def filter_fault_update(self, operations: list[dict]):
                     ):
                         all_judge.extend(judge_res["operations_judgement"])
                 except Exception as e:
-                    logger.error(f"[Feedback Core: filter_fault_update] Judgement failed: {e}")
+                    logger.error(f"[1223 Feedback Core: filter_fault_update] Judgement failed: {e}")
 
-        logger.info(f"[Feedback Core: filter_fault_update] LLM judgement: {all_judge}")
+        logger.info(f"[1223 Feedback Core: filter_fault_update] LLM judgement: {all_judge}")
         id2op = {item["id"]: item for item in updated_operations}
         valid_updates = []
         for judge in all_judge:
@@ -650,7 +652,7 @@ def filter_fault_update(self, operations: list[dict]):
                 valid_updates.append(valid_update)
 
         logger.info(
-            f"[Feedback Core: filter_fault_update] {len(updated_operations)} -> {len(valid_updates)}"
+            f"[1223 Feedback Core: filter_fault_update] {len(updated_operations)} -> {len(valid_updates)}"
         )
         return valid_updates + [item for item in operations if item["operation"] != "UPDATE"]
 
@@ -682,7 +684,7 @@ def correct_item(data):
 
                 if not should_keep_update(data["text"], data["old_memory"]):
                     logger.warning(
-                        f"[Feedback Core: semantics_feedback] Due to the excessive proportion of changes, skip update: {data}"
+                        f"[1223 Feedback Core: semantics_feedback] Due to the excessive proportion of changes, skip update: {data}"
                     )
                     return None
 
@@ -702,14 +704,14 @@ def correct_item(data):
                     return data
             except Exception:
                 logger.error(
-                    f"[Feedback Core: standard_operations] Error processing operation item: {data}",
+                    f"[1223 Feedback Core: standard_operations] Error processing operation item: {data}",
                     exc_info=True,
                 )
             return None
 
         dehallu_res = [correct_item(item) for item in operations]
         dehalluded_operations = [item for item in dehallu_res if item]
-        logger.info(f"[Feedback Core: dehalluded_operations] {dehalluded_operations}")
+        logger.info(f"[1223 Feedback Core: dehalluded_operations] {dehalluded_operations}")
 
         # c add objects
         add_texts = []
@@ -723,7 +725,7 @@ def correct_item(data):
             elif item["operation"].lower() == "update":
                 llm_operations.append(item)
         logger.info(
-            f"[Feedback Core: deduplicate add] {len(dehalluded_operations)} ->  {len(llm_operations)} memories"
+            f"[1223 Feedback Core: deduplicate add] {len(dehalluded_operations)} ->  {len(llm_operations)} memories"
         )
 
         # Update takes precedence over add
@@ -737,7 +739,7 @@ def correct_item(data):
             ]
             if filtered_items:
                 logger.info(
-                    f"[Feedback Core: semantics_feedback] Due to have update objects, skip add: {filtered_items}"
+                    f"[1223 Feedback Core: semantics_feedback] Due to have update objects, skip add: {filtered_items}"
                 )
             return update_items
         else:
@@ -785,7 +787,7 @@ def _doc_filter(self, doc_scope: str, memories: list[TextualMemoryItem]):
             memid for inscope_file in inscope_docs for memid in filename2_memid[inscope_file]
         ]
         logger.info(
-            f"[Feedback Core: process_keyword_replace] These docs are in scope : {inscope_docs}, relared memids: {inscope_ids}"
+            f"[1223 Feedback Core: process_keyword_replace] These docs are in scope : {inscope_docs}, relared memids: {inscope_ids}"
         )
         filter_memories = [mem for mem in memories if mem.id in inscope_ids]
         return filter_memories
@@ -839,7 +841,7 @@ def process_keyword_replace(
             retrieved_memories = self._doc_filter(doc_scope, retrieved_memories)
 
         logger.info(
-            f"[Feedback Core: process_keyword_replace] Keywords recalled memory for user {user_name}: {len(retrieved_ids)} memories | After filtering: {len(retrieved_memories)} memories."
+            f"[1223 Feedback Core: process_keyword_replace] Keywords recalled memory for user {user_name}: {len(retrieved_ids)} memories | After filtering: {len(retrieved_memories)} memories."
         )
 
         if not retrieved_memories:
@@ -924,7 +926,7 @@ def check_validity(item):
                 info.update({"user_id": user_id, "user_name": user_name, "session_id": session_id})
 
             logger.info(
-                f"[Feedback Core: process_feedback_core] Starting memory feedback process for user {user_name}"
+                f"[1223 Feedback Core: process_feedback_core] Starting memory feedback process for user {user_name}"
             )
             # feedback keywords update
             kwp_judge = self._keyword_replace_judgement(feedback_content)
@@ -957,7 +959,7 @@ def check_validity(item):
 
                 if not valid_feedback:
                     logger.warning(
-                        f"[Feedback Core: process_feedback_core] No valid judgements for user {user_name}: {raw_judge}."
+                        f"[1223 Feedback Core: process_feedback_core] No valid judgements for user {user_name}: {raw_judge}."
                     )
                     return {"record": {"add": [], "update": []}}
 
@@ -1005,12 +1007,14 @@ def check_validity(item):
                 add_memories = mem_record["record"]["add"]
                 update_memories = mem_record["record"]["update"]
                 logger.info(
-                    f"[Feedback Core: process_feedback_core] Processed {len(feedback_memories)} feedback | add {len(add_memories)} memories | update {len(update_memories)} memories for user {user_name}."
+                    f"[1223 Feedback Core: process_feedback_core] Processed {len(feedback_memories)} feedback | add {len(add_memories)} memories | update {len(update_memories)} memories for user {user_name}."
                 )
                 return mem_record
 
         except Exception as e:
-            logger.error(f"[Feedback Core: process_feedback_core] Error for user {user_name}: {e}")
+            logger.error(
+                f"[1223 Feedback Core: process_feedback_core] Error for user {user_name}: {e}"
+            )
             return {"record": {"add": [], "update": []}}
 
     def process_feedback(