From daea974d68edd0ce36d2673992ecbf47e9512fd3 Mon Sep 17 00:00:00 2001
From: cce <51567+cce@users.noreply.github.com>
Date: Fri, 6 Dec 2024 23:28:30 -0500
Subject: [PATCH] Pass --context_paragraph_limit option from CLI (#437)

---
 book_maker/cli.py                | 1 +
 book_maker/loader/epub_loader.py | 2 +-
 book_maker/loader/srt_loader.py  | 1 +
 book_maker/loader/txt_loader.py  | 1 +
 4 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/book_maker/cli.py b/book_maker/cli.py
index f768440..ee91108 100644
--- a/book_maker/cli.py
+++ b/book_maker/cli.py
@@ -440,6 +440,7 @@ So you are close to reaching the limit. You have to choose your own value, there
         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 c2b1437..cfed728 100644
--- a/book_maker/loader/epub_loader.py
+++ b/book_maker/loader/epub_loader.py
@@ -33,8 +33,8 @@ class EPUBBookLoader(BaseBookLoader):
         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 4328d98..8fd2733 100644
--- a/book_maker/loader/srt_loader.py
+++ b/book_maker/loader/srt_loader.py
@@ -25,6 +25,7 @@ class SRTBookLoader(BaseBookLoader):
         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 8139685..c34357e 100644
--- a/book_maker/loader/txt_loader.py
+++ b/book_maker/loader/txt_loader.py
@@ -20,6 +20,7 @@ class TXTBookLoader(BaseBookLoader):
         prompt_config=None,
         single_translate=False,
         context_flag=False,
+        context_paragraph_limit=0,
         temperature=1.0,
     ) -> None:
         self.txt_name = txt_name