From df1ecc00192a48abc6bbbe16444507c4bdf6362c Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Tue, 23 Jul 2024 13:06:59 +0200 Subject: [PATCH] Update generate_answer_node.py --- scrapegraphai/nodes/generate_answer_node.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index d864124e..81812598 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -107,8 +107,6 @@ class GenerateAnswerNode(BaseNode): template_chunks_prompt = self.additional_info + template_chunks_prompt template_merge_prompt = self.additional_info + template_merge_prompt - chains_dict = {} - if len(doc) == 1: prompt = PromptTemplate( template=template_no_chunks_prompt, @@ -121,7 +119,7 @@ class GenerateAnswerNode(BaseNode): state.update({self.output[0]: answer}) return state - # Use tqdm to add progress bar + chains_dict = {} for i, chunk in enumerate(tqdm(doc, desc="Processing chunks", disable=not self.verbose)): prompt = PromptTemplate(