Fix summarize_text recursion calls (#4876)

`summarize_text` is currently broken, because it calls itself with the
wrong args (missing `config`)
This commit is contained in:
James
2023-07-04 21:28:02 +08:00
committed by GitHub
parent 80151dd6b2
commit 5318535d0d

View File

@@ -131,13 +131,12 @@ def summarize_text(
logger.info(
f"Summarizing chunk {i + 1} / {len(chunks)} of length {chunk_length} tokens"
)
summary, _ = summarize_text(chunk, instruction)
summary, _ = summarize_text(chunk, config, instruction)
summaries.append(summary)
logger.info(f"Summarized {len(chunks)} chunks")
summary, _ = summarize_text("\n\n".join(summaries))
summary, _ = summarize_text("\n\n".join(summaries), config, instruction)
return summary.strip(), [
(summaries[i], chunks[i][0]) for i in range(0, len(chunks))
]