diff --git a/utils/__init__.py b/utils/__init__.py index f5d94d7..8b13789 100644 --- a/utils/__init__.py +++ b/utils/__init__.py @@ -1 +1 @@ -# Utils package for the synthetic-docs project \ No newline at end of file + diff --git a/utils/file.py b/utils/file.py index 1db4baf..33802ff 100644 --- a/utils/file.py +++ b/utils/file.py @@ -5,7 +5,6 @@ import subprocess from rich.console import Console -# Console for output console = Console() def load_prompt_template(filename): diff --git a/utils/generation.py b/utils/generation.py index 0b2a46b..6e1cfd6 100644 --- a/utils/generation.py +++ b/utils/generation.py @@ -391,7 +391,12 @@ def generate_sections(self): recap_model = self.config["recap_model"] extract_recap_prompt = extract_recap_template.format(section_content=section_content) - recap = self._call_llm(recap_model, extract_recap_prompt, StepIdentifier.RECAP_EXTRACTION, f"Recap Extraction: {chapter_title}") + recap, _ = self._call_llm( + recap_model, + extract_recap_prompt, + StepIdentifier.RECAP_EXTRACTION, + f"Recap Extraction: {chapter_title}", + ) # --- Update Structure --- current_chapter["content"] = section_content