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 @@
-