From 0b5e5f4d089ffe22074ed67f9b93008da467dff0 Mon Sep 17 00:00:00 2001
From: wenping <angenpn@gmail.com>
Date: Wed, 6 Nov 2024 10:12:30 +0800
Subject: [PATCH] fix: Fix parameter mismatch in
 EPUBBookLoaderHelper.translate_with_backoff

- Fix TypeError when calling translate_with_backoff with multiple arguments
- Add proper parameter handling in the decorated method
- Add jitter=None to prevent extra parameters from backoff decorator
- Improve code readability and error handling
---
 book_maker/loader/helper.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/book_maker/loader/helper.py b/book_maker/loader/helper.py
index b628f4f..b473b21 100644
--- a/book_maker/loader/helper.py
+++ b/book_maker/loader/helper.py
@@ -1,7 +1,7 @@
 import re
-from copy import copy
 import backoff
 import logging
+from copy import copy
 
 logging.basicConfig(level=logging.WARNING)
 logger = logging.getLogger(__name__)
@@ -37,9 +37,10 @@ class EPUBBookLoaderHelper:
         Exception,
         on_backoff=lambda details: logger.warning(f"retry backoff: {details}"),
         on_giveup=lambda details: logger.warning(f"retry abort: {details}"),
+        jitter=None
     )
-    def translate_with_backoff(self, **kwargs):
-        return self.translate_model.translate(**kwargs)
+    def translate_with_backoff(self, text, context_flag=False):
+        return self.translate_model.translate(text, context_flag)
 
     def deal_new(self, p, wait_p_list, single_translate=False):
         self.deal_old(wait_p_list, single_translate, self.context_flag)
@@ -128,4 +129,4 @@ def not_trans(s):
             is_text_digit_and_space(s),
             is_text_isbn(s),
         ]
-    )
+    )
\ No newline at end of file