Add Gremlin graph storage

2024-12-19 17:47:42 +01:00
parent a19ffbaea7
commit 6f71293c83
4 changed files with 506 additions and 0 deletions
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -81,6 +81,7 @@ 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")
+GremlinStorage = lazy_external_import(".kg.gremlin_impl", "GremlinStorage")


 def always_get_an_event_loop() -> asyncio.AbstractEventLoop:
@@ -284,6 +285,7 @@ class LightRAG:
            "OracleGraphStorage": OracleGraphStorage,
            "AGEStorage": AGEStorage,
            "TiDBGraphStorage": TiDBGraphStorage,
+            "GremlinStorage": GremlinStorage,
            # "ArangoDBStorage": ArangoDBStorage
        }