smk/graph-of-thoughts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor Document Merging

Browse Source
This commit is contained in:
Nils Blach 2023-10-18 10:31:50 +09:00 committed by Nils Blach
parent 7441be41ba
commit 14e05abbe1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/doc_merge/doc_merge.py
View File

@ -219,7 +219,9 @@ NDA <S{num}>:
prompt += self.improve_summary_prompt_end.format(summary=current) prompt += self.improve_summary_prompt_end.format(summary=current)
return prompt return prompt
elif method.startswith("got"): 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 == "": if current is None or current == "":
prompt += self.merge_doc_prompt_start.format(num=len(parts)) prompt += self.merge_doc_prompt_start.format(num=len(parts))
for i, part in enumerate(sorted(list(parts))): for i, part in enumerate(sorted(list(parts))):