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