Update graphgen/bases/base_generator.py

ChenZiHong-Gavin · gemini-code-assist[bot] · web-flow · commit 64578e99479b · 2026-01-15T18:46:18.000+08:00
Co-authored-by: gemini-code-assist[bot] &lt;176961590+gemini-code-assist[bot]@users.noreply.github.com&gt;
diff --git a/graphgen/bases/base_generator.py b/graphgen/bases/base_generator.py
@@ -77,14 +77,15 @@ def format_generation_results(
                         }
                     )
                 elif output_data_format == "ChatML":
-                    results.append(
+                    flat_results.append(
                         {
                             "messages": [
                                 {"role": "user", "content": question},
                                 {"role": "assistant", "content": answer},
                             ]
                         }
                     )
+                    )
                 else:
                     raise ValueError(
                         f"Unknown output data format: {output_data_format}"

Original file line number	Diff line number	Diff line change
`@@ -77,14 +77,15 @@ def format_generation_results(`
`77`	`77`	`}`
`78`	`78`	`)`
`79`	`79`	`elif output_data_format == "ChatML":`
`80`		`- results.append(`
	`80`	`+ flat_results.append(`
`81`	`81`	`{`
`82`	`82`	`"messages": [`
`83`	`83`	`{"role": "user", "content": question},`
`84`	`84`	`{"role": "assistant", "content": answer},`
`85`	`85`	`]`
`86`	`86`	`}`
`87`	`87`	`)`
	`88`	`+ )`
`88`	`89`	`else:`
`89`	`90`	`raise ValueError(`
`90`	`91`	`f"Unknown output data format: {output_data_format}"`