diff --git a/scrapegraphai/nodes/merge_answers_node.py b/scrapegraphai/nodes/merge_answers_node.py index 1e25cccb..b2e1bd22 100644 --- a/scrapegraphai/nodes/merge_answers_node.py +++ b/scrapegraphai/nodes/merge_answers_node.py @@ -101,6 +101,7 @@ class MergeAnswersNode(BaseNode): merge_chain = prompt_template | self.llm_model | output_parser answer = merge_chain.invoke({"user_prompt": user_prompt}) + answer["sources"] = state.get("urls") state.update({self.output[0]: answer}) return state