From ebe0b0d56e365d2cad28a6c512850847b34efb9c Mon Sep 17 00:00:00 2001 From: bezineb5 Date: Sat, 9 Nov 2024 10:18:21 +0100 Subject: [PATCH] fix --- scrapegraphai/nodes/generate_answer_node.py | 16 ++-------------- 1 file changed, 2 insertions(+), 14 deletions(-) diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index 56d57d09..8e89ff39 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -122,22 +122,10 @@ def execute(self, state: dict) -> dict: partial_variables={"context": doc, "format_instructions": format_instructions} ) chain = prompt | self.llm_model - raw_response = chain.invoke({"question": user_prompt}) - if output_parser: - try: - answer = output_parser.parse(raw_response.content) - except JSONDecodeError: - lines = raw_response.split('\n') - if lines[0].strip().startswith('```'): - lines = lines[1:] - if lines[-1].strip().endswith('```'): - lines = lines[:-1] - cleaned_response = '\n'.join(lines) - answer = output_parser.parse(cleaned_response) - else: - answer = raw_response.content + chain = chain | output_parser + answer = chain.invoke({"question": user_prompt}) state.update({self.output[0]: answer}) return state