Fix dataclass decorator issues in new VQA and MM modules

Co-authored-by: ChenZiHong-Gavin <[email protected]>

Author: Copilot

graphgen/models/generator/vqa_generator.py

@@ -1,13 +1,15 @@
 from dataclasses import dataclass
 from typing import Any
 
-from graphgen.bases import BaseGenerator
+from graphgen.bases import BaseGenerator, BaseLLMClient
 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]]]

graphgen/models/kg_builder/mm_kg_builder.py

@@ -1,5 +1,6 @@
 import re
 from collections import defaultdict
+from dataclasses import dataclass
 from typing import Dict, List, Tuple
 
 from graphgen.bases import BaseLLMClient, Chunk
@@ -15,8 +16,10 @@
 from .light_rag_kg_builder import LightRAGKGBuilder
 
 
+@dataclass
 class MMKGBuilder(LightRAGKGBuilder):
     llm_client: BaseLLMClient = None
+    max_loop: int = 3
 
     async def extract(
         self, chunk: Chunk