fix: fix dataclass decorator for generators

Author: ChenZiHong-Gavin

graphgen/models/generator/aggregated_generator.py

@@ -1,12 +1,10 @@
-from dataclasses import dataclass
 from typing import Any
 
-from graphgen.bases import BaseGenerator, BaseLLMClient
+from graphgen.bases import BaseGenerator
 from graphgen.templates import AGGREGATED_GENERATION_PROMPT
 from graphgen.utils import compute_content_hash, detect_main_language, logger
 
 
-@dataclass
 class AggregatedGenerator(BaseGenerator):
     """
     Aggregated Generator follows a TWO-STEP process:
@@ -15,8 +13,6 @@ class AggregatedGenerator(BaseGenerator):
     2. question generation: Generate relevant questions based on the rephrased text.
     """
 
-    llm_client: BaseLLMClient = None
-
     @staticmethod
     def build_prompt(
         batch: tuple[list[tuple[str, dict]], list[tuple[Any, Any, dict]]]

graphgen/models/generator/atomic_generator.py

@@ -1,15 +1,11 @@
-from dataclasses import dataclass
 from typing import Any
 
-from graphgen.bases import BaseGenerator, BaseLLMClient
+from graphgen.bases import BaseGenerator
 from graphgen.templates import ATOMIC_GENERATION_PROMPT
 from graphgen.utils import compute_content_hash, detect_main_language, logger
 
 
-@dataclass
 class AtomicGenerator(BaseGenerator):
-    llm_client: BaseLLMClient = None
-
     @staticmethod
     def build_prompt(
         batch: tuple[list[tuple[str, dict]], list[tuple[Any, Any, dict]]]

graphgen/models/generator/cot_generator.py

@@ -1,15 +1,11 @@
-from dataclasses import dataclass
 from typing import Any
 
-from graphgen.bases import BaseGenerator, BaseLLMClient
+from graphgen.bases import BaseGenerator
 from graphgen.templates import COT_GENERATION_PROMPT
 from graphgen.utils import compute_content_hash, detect_main_language, logger
 
 
-@dataclass
 class CoTGenerator(BaseGenerator):
-    llm_client: BaseLLMClient = None
-
     @staticmethod
     def build_prompt(
         batch: tuple[list[tuple[str, dict]], list[tuple[Any, Any, dict]]]

graphgen/models/generator/multi_hop_generator.py

@@ -1,15 +1,11 @@
-from dataclasses import dataclass
 from typing import Any
 
-from graphgen.bases import BaseGenerator, BaseLLMClient
+from graphgen.bases import BaseGenerator
 from graphgen.templates import MULTI_HOP_GENERATION_PROMPT
 from graphgen.utils import compute_content_hash, detect_main_language, logger
 
 
-@dataclass
 class MultiHopGenerator(BaseGenerator):
-    llm_client: BaseLLMClient = None
-
     @staticmethod
     def build_prompt(
         batch: tuple[list[tuple[str, dict]], list[tuple[Any, Any, dict]]]

graphgen/models/generator/vqa_generator.py

@@ -1,15 +1,11 @@
-from dataclasses import dataclass
 from typing import Any
 
-from graphgen.bases import BaseGenerator, BaseLLMClient
+from graphgen.bases import BaseGenerator
 from graphgen.templates import VQA_GENERATION_PROMPT
 from graphgen.utils import compute_content_hash, detect_main_language, logger
 
 
-@dataclass
 class VQAGenerator(BaseGenerator):
-    llm_client: BaseLLMClient = None
-
     @staticmethod
     def build_prompt(
         batch: tuple[list[tuple[str, dict]], list[tuple[Any, Any, dict]]]