gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix typo

Browse Source
This commit is contained in:
ArnoChen
2025-02-13 17:32:05 +08:00
parent adb234d5cf
commit e4562c761c
5 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/lightrag_ollama_neo4j_milvus_mongo_demo.py
View File

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

2
examples/lightrag_openai_neo4j_milvus_redis_demo.py
View File

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

2
lightrag/api/lightrag_server.py
View File

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

4
lightrag/kg/milvus_impl.py
View File

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

2
lightrag/lightrag.py
View File

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