Feat/fix palyground bug

src/memos/api/handlers/memory_handler.py

@@ -23,10 +23,6 @@
     remove_embedding_recursive,
     sort_children_by_memory_type,
 )
-from memos.memories.textual.tree_text_memory.retrieve.retrieve_utils import (
-    cosine_similarity_matrix,
-    find_best_unrelated_subgroup,
-)
 
 
 if TYPE_CHECKING:
@@ -41,7 +37,6 @@ def handle_get_all_memories(
     mem_cube_id: str,
     memory_type: Literal["text_mem", "act_mem", "param_mem", "para_mem"],
     naive_mem_cube: Any,
-    embedder: Any,
 ) -> MemoryResponse:
     """
     Main handler for getting all memories.
@@ -64,14 +59,6 @@ def handle_get_all_memories(
             # Get all text memories from the graph database
             memories = naive_mem_cube.text_mem.get_all(user_name=mem_cube_id)
 
-            mems = [mem.get("memory", "") for mem in memories.get("nodes", [])]
-            embeddings = embedder.embed(mems)
-            similarity_matrix = cosine_similarity_matrix(embeddings)
-            selected_indices, _ = find_best_unrelated_subgroup(
-                embeddings, similarity_matrix, bar=0.9
-            )
-            memories["nodes"] = [memories["nodes"][i] for i in selected_indices]
-
             # Format and convert to tree structure
             memories_cleaned = remove_embedding_recursive(memories)
             custom_type_ratios = {

src/memos/api/routers/server_router.py

@@ -88,7 +88,6 @@
 naive_mem_cube = components["naive_mem_cube"]
 redis_client = components["redis_client"]
 status_tracker = TaskStatusTracker(redis_client=redis_client)
-embedder = components["embedder"]
 graph_db = components["graph_db"]
 vector_db = components["vector_db"]
 
@@ -302,7 +301,6 @@ def get_all_memories(memory_req: GetMemoryPlaygroundRequest):
             ),
             memory_type=memory_req.memory_type or "text_mem",
             naive_mem_cube=naive_mem_cube,
-            embedder=embedder,
         )