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

Merge pull request #641 from ScrapeGraphAI/urls_search_graph

Browse Source 
feat: return urls in searchgraph
This commit is contained in:
Federico Aguzzi 2024-09-09 12:03:36 +02:00 committed by GitHub
commit 8a0d46b714
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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

@ -100,6 +100,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