From aafee39a4dcfa6bd26f6b77d9300e6905ef661ca Mon Sep 17 00:00:00 2001 From: Alok Saboo Date: Sun, 26 May 2024 08:45:31 -0400 Subject: [PATCH] feat: fix typo in GenerateAnswerPDFNode's llm_model attribute --- 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 2c0d5388..47e98992 100644 --- a/scrapegraphai/nodes/generate_answer_pdf_node.py +++ b/scrapegraphai/nodes/generate_answer_pdf_node.py @@ -57,7 +57,7 @@ def __init__( node_name (str): name of the node """ super().__init__(node_name, "node", input, output, 2, node_config) - self.llm_model = node_config["llm"] + self.llm_model = node_config["llm_model"] self.verbose = ( False if node_config is None else node_config.get("verbose", False) ) @@ -95,7 +95,7 @@ def execute(self, state): output_parser = JsonOutputParser() format_instructions = output_parser.get_format_instructions() - + chains_dict = {} # Use tqdm to add progress bar