feat: return urls in searchgraph

2026-06-28 21:01:55 +08:00 · 2024-09-07 11:43:10 +02:00 · 2024-09-07 11:43:10 +02:00 · afb6eb7e47
commit afb6eb7e47
parent 9e9c77551f
1 changed files with 1 additions and 0 deletions
--- 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