CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-01 21:00:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: return urls in searchgraph

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-09-07 11:43:10 +02:00
parent 9e9c77551f
commit afb6eb7e47
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
scrapegraphai/nodes/merge_answers_node.py
View File

@ -101,6 +101,7 @@ class MergeAnswersNode(BaseNode):
merge_chain = prompt_template | self.llm_model | output_parser merge_chain = prompt_template | self.llm_model | output_parser
answer = merge_chain.invoke({"user_prompt": user_prompt}) answer = merge_chain.invoke({"user_prompt": user_prompt})
answer["sources"] = state.get("urls")
state.update({self.output[0]: answer}) state.update({self.output[0]: answer})
return state return state