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

add integration for the RAG for ParseHTMLNode

Browse Source
This commit is contained in:
VinciGit00 2024-02-25 11:24:44 +01:00
parent fc0f0957e1
commit 043ab48466
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/graph_examples/custom_graph_example.py
View File

@ -22,7 +22,7 @@ model = OpenAI(llm_config)
# define the nodes for the graph
fetch_html_node = FetchHTMLNode("fetch_html")
parse_document_node = ParseHTMLNode("parse_document")
parse_document_node = ParseHTMLNode(model, "parse_document")
generate_answer_node = GenerateAnswerNode(model, "generate_answer")
# create the graph

2
scrapegraphai/nodes/generate_answer_node.py
View File

@ -36,7 +36,7 @@ class GenerateAnswerNode(BaseNode):
"""
Initializes the GenerateAnswerNode with a language model client and a node name.
"""
super().__init__(node_name, "node")
super().__init__(node_name, "generateAnswer")
self.llm = llm
def execute(self, state: dict) -> dict:

2
scrapegraphai/nodes/parse_html_node.py
View File

@ -41,7 +41,7 @@ class ParseHTMLNode(BaseNode):
"""
Initializes the ParseHTMLNode with a node name.
"""
super().__init__(node_name, "node")
super().__init__(node_name, "parseHTML")
self.llm = llm
def execute(self, state):