fix: fix lint errors

Author: ChenZiHong-Gavin

graphgen/common/init_storage.py

@@ -8,11 +8,11 @@
 class KVStorageActor:
     def __init__(self, backend: str, working_dir: str, namespace: str):
         if backend == "json_kv":
-            from graphgen.models import JsonKVStorage
+            from graphgen.storage import JsonKVStorage
 
             self.kv = JsonKVStorage(working_dir, namespace)
         elif backend == "rocksdb":
-            from graphgen.models import RocksDBKVStorage
+            from graphgen.storage import RocksDBKVStorage
 
             self.kv = RocksDBKVStorage(working_dir, namespace)
         else:
@@ -61,11 +61,11 @@ def ready(self) -> bool:
 class GraphStorageActor:
     def __init__(self, backend: str, working_dir: str, namespace: str):
         if backend == "networkx":
-            from graphgen.models import NetworkXStorage
+            from graphgen.storage import NetworkXStorage
 
             self.graph = NetworkXStorage(working_dir, namespace)
         elif backend == "kuzu":
-            from graphgen.models import KuzuStorage
+            from graphgen.storage import KuzuStorage
 
             self.graph = KuzuStorage(working_dir, namespace)
         else:

graphgen/models/__init__.py

@@ -43,5 +43,4 @@
 from .searcher.web.bing_search import BingSearch
 from .searcher.web.google_search import GoogleSearch
 from .splitter import ChineseRecursiveTextSplitter, RecursiveCharacterSplitter
-from .storage import JsonKVStorage, KuzuStorage, NetworkXStorage, RocksDBKVStorage
 from .tokenizer import Tokenizer

graphgen/models/storage/__init__.py

@@ -0,0 +1,4 @@
+from .graph.kuzu_storage import KuzuStorage
+from .graph.networkx_storage import NetworkXStorage
+from .kv.json_storage import JsonKVStorage
+from .kv.rocksdb_storage import RocksDBKVStorage