diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 7e8a3bb7..9a849921 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -38,10 +38,10 @@ from .base import ( from .prompt import GRAPH_FIELD_SEP STORAGES = { - "JsonKVStorage": ".storage", - "NanoVectorDBStorage": ".storage", - "NetworkXStorage": ".storage", - "JsonDocStatusStorage": ".storage", + "NetworkXStorage": ".storage.networkx_storage", + "JsonKVStorage": ".storage.json_kv_storage", + "NanoVectorDBStorage": ".storage.nano_vector_db", + "JsonDocStatusStorage": ".storage.jsondocstatus_storage", "Neo4JStorage": ".kg.neo4j_impl", "OracleKVStorage": ".kg.oracle_impl", "OracleGraphStorage": ".kg.oracle_impl",