CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-04 21:00:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update generate_answer_node.py
Some checks are pending
/ build (push) Waiting to run
Details

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-07-23 13:06:59 +02:00
parent cf2734392c
commit df1ecc0019
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scrapegraphai/nodes/generate_answer_node.py
View File

@ -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(