Add NebulaGraph Enterprise 5.2.1 Support for Episodic Memory by wenhaocs · Pull Request #1054 · MemMachine/MemMachine

wenhaocs · 2026-02-05T01:12:03Z

Purpose of the change

This PR adds NebulaGraph Enterprise as an alternative vector graph store backend for episodic memory, alongside the existing Neo4j support. Users can now choose between Neo4j and NebulaGraph based on their deployment and scalability needs.

Description

Motivation

NebulaGraph has the follow features:

High performance dual read and write, with high QPS and low latency
Separete compute and storage architecture allows for large dataset and high QPS scenarios with indepdent scaling
GQL (ISO/IEC 76120) - standardized graph query language
Horizontal scalability for large-scale deployments
Enterprise-grade features including high availability, 0-downtime update & upgrade, LDAP, backup/restore, monitoring, etc.

Core Implementation

Add src/memmachine/common/vector_graph_store/nebula_graph_vector_graph_store.py which implements vector_graph_store.py
Added NebulaGraphConf in src/memmachine/common/configuration/database_conf.py
Added NebulaGraph client management to src/memmachine/common/resource_manager/database_manager.py
Added nebula = ["nebula5-python>=5.0.0"] optional dependency

Fixes/Closes

Fixes #(issue number)

Type of change

[Please delete options that are not relevant.]

Bug fix (non-breaking change which fixes an issue)
New feature (non-breaking change which adds functionality)
Breaking change (fix or feature that would cause existing functionality to not work as expected)
Refactor (does not change functionality, e.g., code style improvements, linting)
Documentation update
Project Maintenance (updates to build scripts, CI, etc., that do not affect the main project)
Security (improves security without changing functionality)

How Has This Been Tested?

All Nebula related tests are base on a local Nebula cluster.
Note that the CI environment does not have Nebula cluster set yet. These tests will be skipped until the environment is ready.

New Test Files:

tests/memmachine/common/vector_graph_store/test_nebula_graph_vector_graph_store.py
- 12 integration tests

Enhanced Test Files:

tests/memmachine/episodic_memory/declarative_memory/test_declarative_memory.py
tests/memmachine/episodic_memory/long_term_memory/test_long_term_memory.py
Added parameterized fixtures for both Neo4j and NebulaGraph

[Please delete options that are not relevant.]

Test Results: [Attach logs, screenshots, or relevant output]

Checklist

[Please delete options that are not relevant.]

Maintainer Checklist

Confirmed all checks passed
Contributor has signed the commit(s)
Reviewed the code
Run, Tested, and Verified the change(s) work as expected

Screenshots/Gifs

[If applicable, add screenshots or GIFs that show the changes in action. This is especially helpful for API responses. Otherwise, delete this section or type "N/A".]

Further comments

[Add any other relevant information here, such as potential side effects, future considerations, or any specific questions for the reviewer. Otherwise, type "None".]

Copilot

Pull request overview

Adds NebulaGraph Enterprise 5.x as an alternative vector graph store backend for episodic memory, alongside existing Neo4j support.

Changes:

Introduces NebulaGraphVectorGraphStore implementation and wiring via DatabaseManager + new NebulaGraphConf.
Adds NebulaGraph optional dependency extras and new/updated integration tests to exercise both backends.
Adds documentation for NebulaGraph setup and updates install guide to reflect optional graph DB choice.

Reviewed changes

Copilot reviewed 10 out of 11 changed files in this pull request and generated 8 comments.

Show a summary per file

File	Description
tests/memmachine/episodic_memory/long_term_memory/test_long_term_memory.py	Adds NebulaGraph fixtures and parametrizes tests to run on Neo4j and NebulaGraph.
tests/memmachine/episodic_memory/declarative_memory/test_declarative_memory.py	Same parametrization approach for declarative memory tests across both backends.
tests/memmachine/common/vector_graph_store/test_nebula_graph_vector_graph_store.py	Adds NebulaGraph vector graph store integration test suite.
src/memmachine/common/vector_graph_store/nebula_graph_vector_graph_store.py	Implements NebulaGraph-backed `VectorGraphStore` including vector search, schema/index management, CRUD.
src/memmachine/common/resource_manager/database_manager.py	Adds optional NebulaGraph client lifecycle management and backend auto-detection in `get_vector_graph_store`.
src/memmachine/common/configuration/database_conf.py	Adds `NebulaGraphConf` and extends supported providers + parsing/serialization.
pyproject.toml	Adds `nebula` optional dependency group.
packages/server/pyproject.toml	Adds `nebula` optional dependency group for the server package.
docs/nebula_graph_integration_guide.mdx	New integration guide for NebulaGraph setup and tuning.
docs/install_guide/install_guide.mdx	Updates install instructions to allow choosing Neo4j or NebulaGraph Enterprise.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

sscargal

@wenhaocs Please review and address the items highlighted by CoPilot. Thanks.

Copilot

Pull request overview

Copilot reviewed 11 out of 12 changed files in this pull request and generated 5 comments.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

sscargal · 2026-02-11T00:15:43Z

Thanks for the updates.

Copilot

Pull request overview

Copilot reviewed 11 out of 12 changed files in this pull request and generated 1 comment.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copilot

Pull request overview

Copilot reviewed 11 out of 12 changed files in this pull request and generated 4 comments.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copilot

Pull request overview

Copilot reviewed 11 out of 12 changed files in this pull request and generated 6 comments.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copilot

Pull request overview

Copilot reviewed 12 out of 13 changed files in this pull request and generated 9 comments.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copilot

Pull request overview

Copilot reviewed 12 out of 13 changed files in this pull request and generated 4 comments.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copilot

Pull request overview

Copilot reviewed 13 out of 14 changed files in this pull request and generated 3 comments.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copilot

Pull request overview

Copilot reviewed 13 out of 14 changed files in this pull request and generated 5 comments.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copilot

Pull request overview

Copilot reviewed 13 out of 14 changed files in this pull request and generated 2 comments.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copilot

Pull request overview

Copilot reviewed 13 out of 14 changed files in this pull request and generated 1 comment.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

fix test fix todo update config Now nebula testing is optional until the ci environment is set. Add an integration doc. 5.2.0 address comments add conftest address comment of fixed credentials address comments Update docs/nebula_graph_integration_guide.mdx Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Signed-off-by: haowen <19355821+wenhaocs@users.noreply.github.com> exclude bool check in int check create range index when the number of nodes reach the threshold create embedding for edges and add range index for future extendibility handle different metric type fix doc string fix doc string address comments add more test cases and fix bugs remove backticks and unify version in the doc string sanitize fix comment fix comment Signed-off-by: wenhaocs <19355821+wenhaocs@users.noreply.github.com>

- Change memmachine.common.* to memmachine_server.common.* - Use OrderedValue instead of OrderedPropertyValue - Update nebula-related imports for new package structure - All 28 tests passing Signed-off-by: wenhaocs <19355821+wenhaocs@users.noreply.github.com>

Signed-off-by: wenhaocs <19355821+wenhaocs@users.noreply.github.com>

sscargal

LGTM

wenhaocs force-pushed the haowen/nebula5.0 branch from 94ad1d8 to 36e4fb2 Compare February 5, 2026 18:11

wenhaocs marked this pull request as ready for review February 6, 2026 00:49

sscargal requested review from Copilot, jealous, malatewang and o-love February 6, 2026 18:54

Copilot AI reviewed Feb 6, 2026

View reviewed changes

o-love requested a review from edwinyyyu February 6, 2026 22:20

sscargal requested changes Feb 9, 2026

View reviewed changes

wenhaocs requested review from Copilot and sscargal February 10, 2026 19:03

Copilot started reviewing on behalf of wenhaocs February 10, 2026 19:04 View session

Copilot AI reviewed Feb 10, 2026

View reviewed changes

wenhaocs requested a review from Copilot February 12, 2026 19:50

Copilot started reviewing on behalf of wenhaocs February 12, 2026 19:52 View session

Copilot AI reviewed Feb 12, 2026

View reviewed changes

Comment thread docs/nebula_graph_integration_guide.mdx Outdated

wenhaocs requested a review from Copilot February 13, 2026 00:39

Copilot started reviewing on behalf of wenhaocs February 13, 2026 00:39 View session

Copilot AI reviewed Feb 13, 2026

View reviewed changes

wenhaocs requested a review from Copilot February 13, 2026 01:37

Copilot started reviewing on behalf of wenhaocs February 13, 2026 01:37 View session

Copilot AI reviewed Feb 13, 2026

View reviewed changes

wenhaocs changed the title ~~Add NebulaGraph Enterprise 5.0 Support for Episodic Memory~~ Add NebulaGraph Enterprise 5.2.1 Support for Episodic Memory Feb 13, 2026

wenhaocs requested a review from Copilot February 17, 2026 20:24

Copilot started reviewing on behalf of wenhaocs February 17, 2026 20:24 View session

Copilot AI reviewed Feb 17, 2026

View reviewed changes

wenhaocs requested a review from Copilot February 18, 2026 00:38

Copilot started reviewing on behalf of wenhaocs February 18, 2026 00:38 View session

Copilot AI reviewed Feb 18, 2026

View reviewed changes

wenhaocs force-pushed the haowen/nebula5.0 branch from 7ca4aff to 1cc503c Compare February 18, 2026 22:34

wenhaocs requested a review from Copilot February 18, 2026 22:35

Copilot started reviewing on behalf of wenhaocs February 18, 2026 22:35 View session

Copilot AI reviewed Feb 18, 2026

View reviewed changes

Comment thread src/memmachine/common/vector_graph_store/nebula_graph_vector_graph_store.py Outdated

Comment thread tests/memmachine/common/vector_graph_store/test_nebula_graph_vector_graph_store.py Outdated

Comment thread sample_configs/episodic_memory_config.nebula.sample Outdated

wenhaocs requested a review from Copilot February 23, 2026 04:33

Copilot started reviewing on behalf of wenhaocs February 23, 2026 04:33 View session

Copilot AI reviewed Feb 23, 2026

View reviewed changes

wenhaocs requested a review from Copilot February 24, 2026 22:50

Copilot started reviewing on behalf of wenhaocs February 24, 2026 22:50 View session

Copilot AI reviewed Feb 24, 2026

View reviewed changes

Comment thread ...es/server/src/memmachine_server/common/vector_graph_store/nebula_graph_vector_graph_store.py

Comment thread src/memmachine/common/configuration/database_conf.py Outdated

wenhaocs requested a review from Copilot February 25, 2026 00:35

Copilot started reviewing on behalf of wenhaocs February 25, 2026 00:36 View session

Copilot AI reviewed Feb 25, 2026

View reviewed changes

Comment thread src/memmachine/common/resource_manager/database_manager.py Outdated

wenhaocs force-pushed the haowen/nebula5.0 branch from a1a2544 to e30c004 Compare March 6, 2026 22:44

wenhaocs added 3 commits March 6, 2026 15:13

uv.lock

4f6dd87

Signed-off-by: wenhaocs <19355821+wenhaocs@users.noreply.github.com>

wenhaocs force-pushed the haowen/nebula5.0 branch from b88c9c7 to 4f6dd87 Compare March 6, 2026 23:14

sscargal added this to the v0.3.1 milestone Mar 11, 2026

wenhaocs added 4 commits March 11, 2026 15:02

lint, format, and ut

20d6183

Signed-off-by: wenhaocs <19355821+wenhaocs@users.noreply.github.com>

suppress C901 and move doc file location

fa196e5

Signed-off-by: wenhaocs <19355821+wenhaocs@users.noreply.github.com>

add docs/install_guide/integrate/nebula_graph_integration_guide.mdx

afceb00

Signed-off-by: wenhaocs <19355821+wenhaocs@users.noreply.github.com>

fix doc link

9e18277

Signed-off-by: wenhaocs <19355821+wenhaocs@users.noreply.github.com>

wenhaocs force-pushed the haowen/nebula5.0 branch from 742db51 to 9e18277 Compare March 11, 2026 22:03

malatewang approved these changes Mar 12, 2026

View reviewed changes

Merge branch 'main' into haowen/nebula5.0

f2c285c

sscargal approved these changes Mar 13, 2026

View reviewed changes

sscargal merged commit 09a837e into MemMachine:main Mar 13, 2026
45 of 47 checks passed

Conversation

wenhaocs commented Feb 5, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Purpose of the change

Description

Motivation

Core Implementation

Fixes/Closes

Type of change

How Has This Been Tested?

Checklist

Maintainer Checklist

Screenshots/Gifs

Further comments

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

sscargal left a comment

Choose a reason for hiding this comment

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

sscargal commented Feb 11, 2026

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

wenhaocs commented Feb 5, 2026 •

edited

Loading