Merge pull request #393 from partoneplay/main

Add Milvus as vector storage
2024-12-05 12:05:30 +08:00
parent c90327298c 052322b213
commit 6d274019dd
4 changed files with 143 additions and 0 deletions
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -44,6 +44,8 @@ from .kg.neo4j_impl import Neo4JStorage

 from .kg.oracle_impl import OracleKVStorage, OracleGraphStorage, OracleVectorDBStorage

+from .kg.milvus_impl import MilvusVectorDBStorge
+
 # future KG integrations

 # from .kg.ArangoDB_impl import (
@@ -228,6 +230,7 @@ class LightRAG:
            # vector storage
            "NanoVectorDBStorage": NanoVectorDBStorage,
            "OracleVectorDBStorage": OracleVectorDBStorage,
+            "MilvusVectorDBStorge": MilvusVectorDBStorge,
            # graph storage
            "NetworkXStorage": NetworkXStorage,
            "Neo4JStorage": Neo4JStorage,