fix: update regex for think filter to handle edge cases

Byaidu · Feb 25, 2025 · a0e6c28 · a0e6c28
1 parent f56a7d1
commit a0e6c28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pdf2zh/translator.py b/pdf2zh/translator.py
@@ -423,7 +423,7 @@ def __init__(
         self.prompttext = prompt
         self.add_cache_impact_parameters("temperature", self.options["temperature"])
         self.add_cache_impact_parameters("prompt", self.prompt("", self.prompttext))
-        think_filter_regex = r"^<think>.+?\n*(</think>|\n)+(</think>)\n*"
+        think_filter_regex = r"^<think>.+?\n*(</think>|\n)*(</think>)\n*"
         self.add_cache_impact_parameters('think_filter_regex', think_filter_regex)
         self.think_filter_regex = re.compile(think_filter_regex, flags=re.DOTALL)