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",
graph_storage="Neo4JStorage",
vector_storage="MilvusVectorDBStorge",
vector_storage="MilvusVectorDBStorage",
)
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,
kv_storage="RedisKVStorage",
graph_storage="Neo4JStorage",
vector_storage="MilvusVectorDBStorge",
vector_storage="MilvusVectorDBStorage",
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_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)

4
lightrag/kg/milvus_impl.py
View File

@@ -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,

2
lightrag/lightrag.py
View File

@@ -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",