Skip to content

Conversation

CaralHsi
Copy link
Collaborator

@CaralHsi CaralHsi commented Aug 14, 2025

Description

Summary:

  • Added cube_name parameter to get_nodes and search_by_embedding in Neo4j, Neo4jCommunity, and NebulaGraph backends to filter by cube.
  • Implemented retrieve_from_cube in recall.py to perform hybrid graph & vector retrieval from a specific cube.
  • Added _retrieve_from_memcubes method in searcher.py to integrate cube-based retrieval into search flow.
  • Updated TextualMemoryMetadata.source to include "system" value.
  • Extended system prompt (mos_prompts.py) with REJECT_PROMPT for unsafe content filtering.
  • Adjusted unit tests (test_tree_searcher.py) to reflect new retrieval path and reranker return format.

Fix: #(issue)

Docs Issue/PR: (docs-issue-or-pr-link)

Reviewer: @(reviewer)

Checklist:

  • I have performed a self-review of my own code | 我已自行检查了自己的代码
  • I have commented my code in hard-to-understand areas | 我已在难以理解的地方对代码进行了注释
  • I have added tests that prove my fix is effective or that my feature works | 我已添加测试以证明我的修复有效或功能正常
  • I have created related documentation issue/PR in MemOS-Docs (if applicable) | 我已在 MemOS-Docs 中创建了相关的文档 issue/PR(如果适用)
  • I have linked the issue to this PR (if applicable) | 我已将 issue 链接到此 PR(如果适用)
  • I have mentioned the person who will review this PR | 我已提及将审查此 PR 的人

@CaralHsi CaralHsi changed the base branch from main to dev August 14, 2025 14:41
@CaralHsi CaralHsi changed the title feat: search from memos_cube feat: support retrieval from specified memos_cube Aug 15, 2025
@CaralHsi CaralHsi marked this pull request as ready for review August 15, 2025 02:55
@CaralHsi CaralHsi merged commit 9824ed2 into MemTensor:dev Aug 15, 2025
20 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant