refactor: improve content extraction in translator response handling

Byaidu · Feb 25, 2025 · a88bf06 · a88bf06
1 parent 8d1c75e
commit a88bf06
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pdf2zh/translator.py b/pdf2zh/translator.py
@@ -433,7 +433,9 @@ def do_translate(self, text) -> str:
         if not response.choices:
             if hasattr(response, "error"):
                 raise ValueError("Error response from Service", response.error)
-        return response.choices[0].message.content.strip()
+        content = response.choices[0].message.content.strip()
+        content = re.sub(r"^<think>.+?</think>", "", content, flags=re.DOTALL).strip()
+        return content
 
     def get_formular_placeholder(self, id: int):
         return "{{v" + str(id) + "}}"