fix: fix lint problem

Author: ChenZiHong-Gavin

graphgen/bases/base_storage.py

@@ -1,12 +1,13 @@
+from dataclasses import dataclass
 from typing import Generic, TypeVar, Union
 
 T = TypeVar("T")
 
 
+@dataclass
 class StorageNameSpace:
-    def __init__(self, working_dir: str = None, namespace: str = None):
-        self.working_dir = working_dir
-        self.namespace = namespace
+    working_dir: str = None
+    namespace: str = None
 
     async def index_done_callback(self):
         """commit the storage operations after indexing"""

graphgen/models/kg_builder/mm_kg_builder.py

@@ -2,7 +2,7 @@
 from collections import defaultdict
 from typing import Dict, List, Tuple
 
-from graphgen.bases import BaseLLMClient, Chunk
+from graphgen.bases import Chunk
 from graphgen.templates import MMKG_EXTRACTION_PROMPT
 from graphgen.utils import (
     detect_main_language,