From 14e05abbe1cb6709a5bdfd0c0ef7acb0aa40603b Mon Sep 17 00:00:00 2001 From: Nils Blach Date: Wed, 18 Oct 2023 10:31:50 +0900 Subject: [PATCH] Refactor Document Merging --- examples/doc_merge/doc_merge.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/doc_merge/doc_merge.py b/examples/doc_merge/doc_merge.py index 451bbbb..a0c73b9 100644 --- a/examples/doc_merge/doc_merge.py +++ b/examples/doc_merge/doc_merge.py @@ -219,7 +219,9 @@ NDA : prompt += self.improve_summary_prompt_end.format(summary=current) return prompt elif method.startswith("got"): - parts = sorted(list(parts)) if len(parts) > 0 else list(range(len(documents))) + parts = ( + sorted(list(parts)) if len(parts) > 0 else list(range(len(documents))) + ) if current is None or current == "": prompt += self.merge_doc_prompt_start.format(num=len(parts)) for i, part in enumerate(sorted(list(parts))):