From dc1340e302117a6bb5e5b12e6f51d097ff79cb47 Mon Sep 17 00:00:00 2001
From: Marco Vinciguerra <mvincig11@gmail.com>
Date: Wed, 12 Jun 2024 15:47:04 +0200
Subject: [PATCH] Update generate_answer_pdf_node.py

---
 scrapegraphai/nodes/generate_answer_pdf_node.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scrapegraphai/nodes/generate_answer_pdf_node.py b/scrapegraphai/nodes/generate_answer_pdf_node.py
index bf003411..e58ae35e 100644
--- a/scrapegraphai/nodes/generate_answer_pdf_node.py
+++ b/scrapegraphai/nodes/generate_answer_pdf_node.py
@@ -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