Skip to content

Commit df1ecc0

Browse files
committed
Update generate_answer_node.py
1 parent cf27343 commit df1ecc0

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

scrapegraphai/nodes/generate_answer_node.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -107,8 +107,6 @@ def execute(self, state: dict) -> dict:
107107
template_chunks_prompt = self.additional_info + template_chunks_prompt
108108
template_merge_prompt = self.additional_info + template_merge_prompt
109109

110-
chains_dict = {}
111-
112110
if len(doc) == 1:
113111
prompt = PromptTemplate(
114112
template=template_no_chunks_prompt,
@@ -121,7 +119,7 @@ def execute(self, state: dict) -> dict:
121119
state.update({self.output[0]: answer})
122120
return state
123121

124-
# Use tqdm to add progress bar
122+
chains_dict = {}
125123
for i, chunk in enumerate(tqdm(doc, desc="Processing chunks", disable=not self.verbose)):
126124

127125
prompt = PromptTemplate(

0 commit comments

Comments
 (0)