From e4562c761c8de5abeeb5b2512b462f73d761b0e2 Mon Sep 17 00:00:00 2001 From: ArnoChen Date: Thu, 13 Feb 2025 17:32:05 +0800 Subject: [PATCH] fix typo --- examples/lightrag_ollama_neo4j_milvus_mongo_demo.py | 2 +- examples/lightrag_openai_neo4j_milvus_redis_demo.py | 2 +- lightrag/api/lightrag_server.py | 2 +- lightrag/kg/milvus_impl.py | 4 ++-- lightrag/lightrag.py | 2 +- 5 files changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/lightrag_ollama_neo4j_milvus_mongo_demo.py b/examples/lightrag_ollama_neo4j_milvus_mongo_demo.py index b71489c7..7e2d1026 100644 --- a/examples/lightrag_ollama_neo4j_milvus_mongo_demo.py +++ b/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" diff --git a/examples/lightrag_openai_neo4j_milvus_redis_demo.py b/examples/lightrag_openai_neo4j_milvus_redis_demo.py index 75e110aa..da5c5a8f 100644 --- a/examples/lightrag_openai_neo4j_milvus_redis_demo.py +++ b/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", ) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index 42a597c2..3e77cc59 100644 --- a/lightrag/api/lightrag_server.py +++ b/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) diff --git a/lightrag/kg/milvus_impl.py b/lightrag/kg/milvus_impl.py index 2995fd9b..ce5bf246 100644 --- a/lightrag/kg/milvus_impl.py +++ b/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, diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 726e7512..157c6ef2 100644 --- a/lightrag/lightrag.py +++ b/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",