From afb6eb7e4796ab208a050ad04ad96a83406f7fa1 Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sat, 7 Sep 2024 11:43:10 +0200 Subject: [PATCH] feat: return urls in searchgraph --- scrapegraphai/nodes/merge_answers_node.py | 1 + 1 file changed, 1 insertion(+) 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