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

Update generate_answer_pdf_node.py

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-06-12 15:47:04 +02:00
parent e45f159a31
commit dc1340e302
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scrapegraphai/nodes/generate_answer_pdf_node.py
View File

@ -113,7 +113,7 @@ class GenerateAnswerPDFNode(BaseNode):
template=template_no_chunks_pdf,
input_variables=["question"],
partial_variables={
"context":chunk,
"context":chunk.page_content,
"format_instructions": format_instructions,
},
)
@ -150,5 +150,5 @@ class GenerateAnswerPDFNode(BaseNode):
answer = merge_chain.invoke({"context": answer, "question": user_prompt})
# Update the state with the generated answer
state.update({self.output[0]: answer})
state.update({self.output[0]: answer.get("Response", {})})
return state