ElasticSearch storage impelemented

Author: hbao

env.example

@@ -187,6 +187,11 @@ EMBEDDING_BINDING_API_KEY=your_api_key
 # LIGHTRAG_GRAPH_STORAGE=MongoGraphStorage
 # LIGHTRAG_VECTOR_STORAGE=MongoVectorDBStorage
 
+### ElasticSearch
+# LIGHTRAG_KV_STORAGE=ESKVStorage
+# LIGHTRAG_VECTOR_STORAGE=ESVectorDBStorage
+# LIGHTRAG_DOC_STATUS_STORAGE=ESDocStatusStorage
+
 ### PostgreSQL Configuration
 POSTGRES_HOST=localhost
 POSTGRES_PORT=5432
@@ -240,6 +245,11 @@ REDIS_MAX_CONNECTIONS=100
 REDIS_RETRY_ATTEMPTS=3
 # REDIS_WORKSPACE=forced_workspace_name
 
+### ElasticSearch Configuration
+ES_HOST=http://localhost:9200
+# ES_USERNAME=elastic
+# ES_PASSWORD='corilead'
+
 ### Memgraph Configuration
 MEMGRAPH_URI=bolt://localhost:7687
 MEMGRAPH_USERNAME=

lightrag/kg/__init__.py

@@ -5,6 +5,7 @@
             "RedisKVStorage",
             "PGKVStorage",
             "MongoKVStorage",
+            "ESKVStorage",
             # "TiDBKVStorage",
         ],
         "required_methods": ["get_by_id", "upsert"],
@@ -30,6 +31,7 @@
             "FaissVectorDBStorage",
             "QdrantVectorDBStorage",
             "MongoVectorDBStorage",
+            "ESVectorDBStorage",
             # "ChromaVectorDBStorage",
             # "TiDBVectorDBStorage",
         ],
@@ -41,6 +43,7 @@
             "RedisDocStatusStorage",
             "PGDocStatusStorage",
             "MongoDocStatusStorage",
+            "ESDocStatusStorage",
         ],
         "required_methods": ["get_docs_by_status"],
     },
@@ -52,6 +55,7 @@
     "JsonKVStorage": [],
     "MongoKVStorage": [],
     "RedisKVStorage": ["REDIS_URI"],
+    "ESKVStorage": ["ES_HOST"],
     # "TiDBKVStorage": ["TIDB_USER", "TIDB_PASSWORD", "TIDB_DATABASE"],
     "PGKVStorage": ["POSTGRES_USER", "POSTGRES_PASSWORD", "POSTGRES_DATABASE"],
     # Graph Storage Implementations
@@ -85,6 +89,7 @@
     "RedisDocStatusStorage": ["REDIS_URI"],
     "PGDocStatusStorage": ["POSTGRES_USER", "POSTGRES_PASSWORD", "POSTGRES_DATABASE"],
     "MongoDocStatusStorage": [],
+    "ESDocStatusStorage": ["ES_HOST"],
 }
 
 # Storage implementation module mapping
@@ -114,6 +119,9 @@
     "FaissVectorDBStorage": ".kg.faiss_impl",
     "QdrantVectorDBStorage": ".kg.qdrant_impl",
     "MemgraphStorage": ".kg.memgraph_impl",
+    "ESKVStorage": ".kg.es_impl",
+    "ESDocStatusStorage": ".kg.es_impl",
+    "ESVectorDBStorage": ".kg.es_impl",
 }