fix: fix multi_hop & cot generation

Author: ChenZiHong-Gavin

graphgen/generate.py

@@ -75,7 +75,7 @@ def main():
     # Use pipeline according to the output data type
     if mode in ["atomic", "aggregated", "multi_hop"]:
         logger.info("Generation mode set to '%s'. Start generation.", mode)
-        if "quiz_and_judge" in config:
+        if "quiz_and_judge" in config and config["quiz_and_judge"]["enabled"]:
             graph_gen.quiz_and_judge(quiz_and_judge_config=config["quiz_and_judge"])
         else:
             logger.warning(

graphgen/graphgen.py

@@ -265,11 +265,10 @@ async def generate(self, partition_config: Dict, generate_config: Dict):
                 self.progress_bar,
             )
         elif mode == "cot":
-            method_params = generate_config.get("method_params", {})
             results = await generate_cot(
                 self.graph_storage,
                 self.synthesizer_llm_client,
-                method_params=method_params,
+                method_params=partition_config["leiden_params"],
             )
         else:
             raise ValueError(f"Unknown generation mode: {mode}")