Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add MongoDB VectorDB Support and Optimize Existing MongoDB Implementations #781

Merged
merged 2 commits into from
Feb 16, 2025

Conversation

ArnoChenFx
Copy link
Contributor

@ArnoChenFx ArnoChenFx commented Feb 14, 2025

This Pull Request introduces MongoDB VectorDB support and optimizes existing MongoDB implementations for KV, Doc Status, and Graph storage within LightRAG.

  • MongoVectorDBStorage (New): Implements vector data storage and provides similarity search capabilities using Atlas Vector Search. The MongoDB VectorDB implementation was contributed by @ankit1063 in issue Regarding Mongo DB vector storage integration. #672.
  • MongoKVStorage (Optimized): This PR includes improvements for asynchronous operations.
  • MongoDocStatusStorage (Optimized): This PR includes improvements for asynchronous operations.
  • MongoGraphStorage (Optimized): This PR includes improvements for graph traversal and asynchronous operations. Adds the ability to retrieve all labels and query knowledge graphs with a specified label and maximum depth.

Specific Changes:

  • Added MongoVectorDBStorage to the STORAGE_IMPLEMENTATIONS and STORAGES dictionaries in lightrag.py.
  • Updated api/README.md to include MongoVectorDBStorage in the list of supported storage implementations.
  • Modified the kg/mongo_impl.py file, containing the implementations and optimizations for existing features and the vector database functionality.
  • Optimized existing KV, Doc Status, and Graph implementations to improve asynchronous operation compatibility.

@LarFii LarFii merged commit 292b653 into HKUDS:main Feb 16, 2025
1 check passed
@ArnoChenFx ArnoChenFx deleted the enhance-mongodb-backends branch February 18, 2025 06:00
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.

2 participants