Adjust storage position in STORAGE_IMPLEMENTATIONS

2025-04-02 16:40:25 +08:00
parent c2f007199f
commit 454e21a868
1 changed files with 5 additions and 5 deletions
--- 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"],
    },