diff --git a/book_maker/cli.py b/book_maker/cli.py index f7684401..ee91108e 100644 --- a/book_maker/cli.py +++ b/book_maker/cli.py @@ -440,6 +440,7 @@ def main(): prompt_config=parse_prompt_arg(options.prompt_arg), single_translate=options.single_translate, context_flag=options.context_flag, + context_paragraph_limit=options.context_paragraph_limit, temperature=options.temperature, ) # other options diff --git a/book_maker/loader/epub_loader.py b/book_maker/loader/epub_loader.py index c2b14376..cfed728d 100644 --- a/book_maker/loader/epub_loader.py +++ b/book_maker/loader/epub_loader.py @@ -33,8 +33,8 @@ def __init__( prompt_config=None, single_translate=False, context_flag=False, - temperature=1.0, context_paragraph_limit=0, + temperature=1.0, ): self.epub_name = epub_name self.new_epub = epub.EpubBook() diff --git a/book_maker/loader/srt_loader.py b/book_maker/loader/srt_loader.py index 4328d988..8fd2733a 100644 --- a/book_maker/loader/srt_loader.py +++ b/book_maker/loader/srt_loader.py @@ -25,6 +25,7 @@ def __init__( prompt_config=None, single_translate=False, context_flag=False, + context_paragraph_limit=0, temperature=1.0, ) -> None: self.srt_name = srt_name diff --git a/book_maker/loader/txt_loader.py b/book_maker/loader/txt_loader.py index 81396850..c34357e2 100644 --- a/book_maker/loader/txt_loader.py +++ b/book_maker/loader/txt_loader.py @@ -20,6 +20,7 @@ def __init__( prompt_config=None, single_translate=False, context_flag=False, + context_paragraph_limit=0, temperature=1.0, ) -> None: self.txt_name = txt_name