Skip to content

fix: clean up graph search tools and results#1012

Open
wayyoungboy wants to merge 1 commit into
oceanbase:mainfrom
wayyoungboy:fix/issue-1004-graph-search-cleanups
Open

fix: clean up graph search tools and results#1012
wayyoungboy wants to merge 1 commit into
oceanbase:mainfrom
wayyoungboy:fix/issue-1004-graph-search-cleanups

Conversation

@wayyoungboy

Copy link
Copy Markdown
Member

Summary

Closes #1004.

This PR fixes the graph search/tool issues reported in #1004.

Changes:

  • aligns the structured relations tool name with the regular establish_relationships tool name
  • passes noop alongside entity extraction and relation extraction tools so the LLM can opt out
  • deduplicates graph search triples across overlapping seed-node traversals
  • includes the BM25 rerank score in returned graph search results
  • adds unit coverage for the tool name, noop tools, duplicate removal, and score exposure

Validation

  • python -m pytest tests/unit/test_oceanbase_graph.py -q
  • python -m pytest tests/unit -q
  • python -m flake8 src/powermem/storage/oceanbase/oceanbase_graph.py src/powermem/prompts/graph/graph_tools_prompts.py tests/unit/test_oceanbase_graph.py --select=F601,F821,E999
  • git diff --check

@wayyoungboy wayyoungboy marked this pull request as ready for review June 11, 2026 03:23
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.

Graph search: tool name mismatch, duplicate results, and missing noop tool

1 participant