From 56e9c9f4d5e8c4672d472e8393226376bd612729 Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Mon, 27 Jan 2025 09:59:26 +0100
Subject: [PATCH] Moved the storages to kg folder

---
 .../{storage/json_kv_storage.py => kg/json_kv_impl.py}    | 0
 .../jsondocstatus_storage.py => kg/jsondocstatus_impl.py} | 0
 .../nano_vector_db.py => kg/nano_vector_db_impl.py}       | 0
 .../{storage/networkx_storage.py => kg/networkx_impl.py}  | 0
 lightrag/lightrag.py                                      | 8 ++++----
 lightrag/storage/__init__.py                              | 1 -
 6 files changed, 4 insertions(+), 5 deletions(-)
 rename lightrag/{storage/json_kv_storage.py => kg/json_kv_impl.py} (100%)
 rename lightrag/{storage/jsondocstatus_storage.py => kg/jsondocstatus_impl.py} (100%)
 rename lightrag/{storage/nano_vector_db.py => kg/nano_vector_db_impl.py} (100%)
 rename lightrag/{storage/networkx_storage.py => kg/networkx_impl.py} (100%)
 delete mode 100644 lightrag/storage/__init__.py

diff --git a/lightrag/storage/json_kv_storage.py b/lightrag/kg/json_kv_impl.py
similarity index 100%
rename from lightrag/storage/json_kv_storage.py
rename to lightrag/kg/json_kv_impl.py
diff --git a/lightrag/storage/jsondocstatus_storage.py b/lightrag/kg/jsondocstatus_impl.py
similarity index 100%
rename from lightrag/storage/jsondocstatus_storage.py
rename to lightrag/kg/jsondocstatus_impl.py
diff --git a/lightrag/storage/nano_vector_db.py b/lightrag/kg/nano_vector_db_impl.py
similarity index 100%
rename from lightrag/storage/nano_vector_db.py
rename to lightrag/kg/nano_vector_db_impl.py
diff --git a/lightrag/storage/networkx_storage.py b/lightrag/kg/networkx_impl.py
similarity index 100%
rename from lightrag/storage/networkx_storage.py
rename to lightrag/kg/networkx_impl.py
diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index 9a849921..b40eecaa 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -38,10 +38,10 @@ from .base import (
 from .prompt import GRAPH_FIELD_SEP
 
 STORAGES = {
-    "NetworkXStorage": ".storage.networkx_storage",
-    "JsonKVStorage": ".storage.json_kv_storage",
-    "NanoVectorDBStorage": ".storage.nano_vector_db",
-    "JsonDocStatusStorage": ".storage.jsondocstatus_storage",
+    "NetworkXStorage": ".kg.networkx_impl",
+    "JsonKVStorage": ".kg.json_kv_impl",
+    "NanoVectorDBStorage": ".kg.nano_vector_db_impl",
+    "JsonDocStatusStorage": ".kg.jsondocstatus_impl",
     "Neo4JStorage": ".kg.neo4j_impl",
     "OracleKVStorage": ".kg.oracle_impl",
     "OracleGraphStorage": ".kg.oracle_impl",
diff --git a/lightrag/storage/__init__.py b/lightrag/storage/__init__.py
deleted file mode 100644
index 8b137891..00000000
--- a/lightrag/storage/__init__.py
+++ /dev/null
@@ -1 +0,0 @@
-