fix: change language detection to use detect_main_language directly in quiz_generator

CHERRY-ui8 · CHERRY-ui8 · commit c0faf3b0fe69 · 2025-11-20T20:23:05.000+08:00
diff --git a/graphgen/models/generator/quiz_generator.py b/graphgen/models/generator/quiz_generator.py
@@ -43,7 +43,7 @@ def build_prompt_for_description(description: str, template_type: str = "TEMPLAT
         :param template_type: Either "TEMPLATE" (same meaning) or "ANTI_TEMPLATE" (opposite meaning)
         :return: Prompt string
         """
-        language = "English" if detect_main_language(description) == "en" else "Chinese"
+        language = detect_main_language(description)
         prompt = DESCRIPTION_REPHRASING_PROMPT[language][template_type].format(
             input_sentence=description
         )

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@ def build_prompt_for_description(description: str, template_type: str = "TEMPLAT`
`43`	`43`	`:param template_type: Either "TEMPLATE" (same meaning) or "ANTI_TEMPLATE" (opposite meaning)`
`44`	`44`	`:return: Prompt string`
`45`	`45`	`"""`
`46`		`- language = "English" if detect_main_language(description) == "en" else "Chinese"`
	`46`	`+ language = detect_main_language(description)`
`47`	`47`	`prompt = DESCRIPTION_REPHRASING_PROMPT[language][template_type].format(`
`48`	`48`	`input_sentence=description`
`49`	`49`	`)`