fix typo

examples/lightrag_ollama_neo4j_milvus_mongo_demo.py

@@ -44,7 +44,7 @@ rag = LightRAG(
     ),
     kv_storage="MongoKVStorage",
     graph_storage="Neo4JStorage",
-    vector_storage="MilvusVectorDBStorge",
+    vector_storage="MilvusVectorDBStorage",
 )
 
 file = "./book.txt"

examples/lightrag_openai_neo4j_milvus_redis_demo.py

@@ -58,7 +58,7 @@ rag = LightRAG(
     chunk_overlap_token_size=256,
     kv_storage="RedisKVStorage",
     graph_storage="Neo4JStorage",
-    vector_storage="MilvusVectorDBStorge",
+    vector_storage="MilvusVectorDBStorage",
     doc_status_storage="RedisKVStorage",
 )
 

lightrag/api/lightrag_server.py

@@ -111,7 +111,7 @@ if milvus_uri:
     os.environ["MILVUS_USER"] = milvus_user
     os.environ["MILVUS_PASSWORD"] = milvus_password
     os.environ["MILVUS_DB_NAME"] = milvus_db_name
-    rag_storage_config.VECTOR_STORAGE = "MilvusVectorDBStorge"
+    rag_storage_config.VECTOR_STORAGE = "MilvusVectorDBStorage"
 
 # Qdrant config
 qdrant_uri = config.get("qdrant", "uri", fallback=None)

lightrag/kg/milvus_impl.py

@@ -14,7 +14,7 @@ from pymilvus import MilvusClient
 
 
 @dataclass
-class MilvusVectorDBStorge(BaseVectorStorage):
+class MilvusVectorDBStorage(BaseVectorStorage):
     @staticmethod
     def create_collection_if_not_exist(
         client: MilvusClient, collection_name: str, **kwargs
@@ -37,7 +37,7 @@ class MilvusVectorDBStorge(BaseVectorStorage):
             db_name=os.environ.get("MILVUS_DB_NAME", ""),
         )
         self._max_batch_size = self.global_config["embedding_batch_num"]
-        MilvusVectorDBStorge.create_collection_if_not_exist(
+        MilvusVectorDBStorage.create_collection_if_not_exist(
             self._client,
             self.namespace,
             dimension=self.embedding_func.embedding_dim,

lightrag/lightrag.py

@@ -44,7 +44,7 @@ STORAGES = {
     "OracleKVStorage": ".kg.oracle_impl",
     "OracleGraphStorage": ".kg.oracle_impl",
     "OracleVectorDBStorage": ".kg.oracle_impl",
-    "MilvusVectorDBStorge": ".kg.milvus_impl",
+    "MilvusVectorDBStorage": ".kg.milvus_impl",
     "MongoKVStorage": ".kg.mongo_impl",
     "MongoDocStatusStorage": ".kg.mongo_impl",
     "MongoGraphStorage": ".kg.mongo_impl",