feat(high_level): set rate limit

awwaawwa · Dec 23, 2024 · 394a06e · 394a06e
1 parent 0bf5002
commit 394a06e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pdf2zh/high_level.py b/pdf2zh/high_level.py
@@ -25,6 +25,7 @@
 from pdf2zh.converter import TranslateConverter
 from pdf2zh.doclayout import DocLayoutModel
 from pdf2zh.pdfinterp import PDFPageInterpreterEx
+from pdf2zh.translator import set_translate_rate_limiter
 
 model = DocLayoutModel.load_available()
 
@@ -322,6 +323,7 @@ def translate(
     cancellation_event: asyncio.Event = None,
     **kwarg: Any,
 ):
+    set_translate_rate_limiter(thread)
     if not files:
         raise PDFValueError("No files to process.")