diff --git a/src/memos/api/handlers/memory_handler.py b/src/memos/api/handlers/memory_handler.py index 7110fae0..a4f50056 100644 --- a/src/memos/api/handlers/memory_handler.py +++ b/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 = { diff --git a/src/memos/api/routers/server_router.py b/src/memos/api/routers/server_router.py index 8371c41b..86b75d73 100644 --- a/src/memos/api/routers/server_router.py +++ b/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, )