fix pre_commit

Author: yuan.wang

src/memos/memories/textual/prefer_text_memory/adder.py

@@ -125,15 +125,17 @@ def _update_memory_op_trace(
                 "context_summary": new_memory.memory,
                 "preference": new_memory.payload["preference"],
             }
-            for new_memory in new_vec_db_items if new_memory.payload.get("preference", None)
+            for new_memory in new_vec_db_items
+            if new_memory.payload.get("preference", None)
         ]
         retrieved_mem_inputs = [
             {
                 "id": mem.id,
                 "context_summary": mem.memory,
                 "preference": mem.payload["preference"],
             }
-            for mem in retrieved_memories if mem.payload.get("preference", None)
+            for mem in retrieved_memories
+            if mem.payload.get("preference", None)
         ]
 
         rsp = self._judge_update_or_add_trace_op(
@@ -209,17 +211,15 @@ def _update_memory_fine(
             payload=payload,
         )
 
-        new_mem_input = {
-            "memory": new_memory.memory,
-            "preference": new_memory.metadata.preference
-        }
+        new_mem_input = {"memory": new_memory.memory, "preference": new_memory.metadata.preference}
         retrieved_mem_inputs = [
             {
                 "id": mem.id,
                 "memory": mem.memory,
                 "preference": mem.payload["preference"],
             }
-            for mem in retrieved_memories if mem.payload.get("preference", None)
+            for mem in retrieved_memories
+            if mem.payload.get("preference", None)
         ]
         rsp = self._judge_update_or_add_fine(
             new_mem=json.dumps(new_mem_input),
@@ -295,9 +295,7 @@ def _update_memory(
         if update_mode == "fast":
             return self._update_memory_fast(new_memory, retrieved_memories, collection_name)
         elif update_mode == "fine":
-            return self._update_memory_fine(
-                new_memory, retrieved_memories, collection_name
-            )
+            return self._update_memory_fine(new_memory, retrieved_memories, collection_name)
         else:
             raise ValueError(f"Invalid update mode: {update_mode}")
 
@@ -376,10 +374,10 @@ def process_memory_batch(self, memories: list[TextualMemoryItem], *args, **kwarg
                 implicit_recalls,
                 pref_type_collection_map["implicit_preference"],
             )
-            
+
             explicit_added_ids = explicit_future.result()
             implicit_added_ids = implicit_future.result()
-        
+
         return explicit_added_ids + implicit_added_ids
 
     def process_memory_single(

src/memos/memories/textual/prefer_text_memory/extractor.py

@@ -68,7 +68,7 @@ def extract_explicit_preference(self, qa_pair: MessageList | str) -> dict[str, A
             response = response.strip().replace("```json", "").replace("```", "").strip()
             result = json.loads(response)
             for d in result:
-                d['preference'] = d.pop('explicit_preference')
+                d["preference"] = d.pop("explicit_preference")
             return result
         except Exception as e:
             logger.error(f"Error extracting explicit preference: {e}, return None")
@@ -90,7 +90,7 @@ def extract_implicit_preference(self, qa_pair: MessageList | str) -> dict[str, A
             response = self.llm_provider.generate([{"role": "user", "content": prompt}])
             response = response.strip().replace("```json", "").replace("```", "").strip()
             result = json.loads(response)
-            result['preference'] = result.pop('implicit_preference')
+            result["preference"] = result.pop("implicit_preference")
             return result
         except Exception as e:
             logger.error(f"Error extracting implicit preferences: {e}, return None")