support TiDBGraphStorage

2024-12-17 15:24:38 +08:00
parent 874f3b392b
commit 344d8f277b
3 changed files with 248 additions and 43 deletions
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -79,6 +79,7 @@ MongoKVStorage = lazy_external_import(".kg.mongo_impl", "MongoKVStorage")
 ChromaVectorDBStorage = lazy_external_import(".kg.chroma_impl", "ChromaVectorDBStorage")
 TiDBKVStorage = lazy_external_import(".kg.tidb_impl", "TiDBKVStorage")
 TiDBVectorDBStorage = lazy_external_import(".kg.tidb_impl", "TiDBVectorDBStorage")
+TiDBGraphStorage = lazy_external_import(".kg.tidb_impl", "TiDBGraphStorage")
 AGEStorage = lazy_external_import(".kg.age_impl", "AGEStorage")


@@ -282,6 +283,7 @@ class LightRAG:
            "Neo4JStorage": Neo4JStorage,
            "OracleGraphStorage": OracleGraphStorage,
            "AGEStorage": AGEStorage,
+            "TiDBGraphStorage": TiDBGraphStorage,
            # "ArangoDBStorage": ArangoDBStorage
        }