diff --git a/lightrag/kg/__init__.py b/lightrag/kg/__init__.py index b917bfa0..45115c11 100644 --- a/lightrag/kg/__init__.py +++ b/lightrag/kg/__init__.py @@ -2,10 +2,10 @@ STORAGE_IMPLEMENTATIONS = { "KV_STORAGE": { "implementations": [ "JsonKVStorage", - "MongoKVStorage", "RedisKVStorage", - # "TiDBKVStorage", "PGKVStorage", + "MongoKVStorage", + # "TiDBKVStorage", ], "required_methods": ["get_by_id", "upsert"], }, @@ -13,11 +13,11 @@ STORAGE_IMPLEMENTATIONS = { "implementations": [ "NetworkXStorage", "Neo4JStorage", + "PGGraphStorage", + "AGEStorage", # "MongoGraphStorage", # "TiDBGraphStorage", - "AGEStorage", # "GremlinStorage", - "PGGraphStorage", ], "required_methods": ["upsert_node", "upsert_edge"], }, @@ -26,11 +26,11 @@ STORAGE_IMPLEMENTATIONS = { "NanoVectorDBStorage", "MilvusVectorDBStorage", "ChromaVectorDBStorage", - # "TiDBVectorDBStorage", "PGVectorStorage", "FaissVectorDBStorage", "QdrantVectorDBStorage", "MongoVectorDBStorage", + # "TiDBVectorDBStorage", ], "required_methods": ["query", "upsert"], },