From 043ab48466bed5bebd5a259a4b6175e8cee032c9 Mon Sep 17 00:00:00 2001 From: VinciGit00 Date: Sun, 25 Feb 2024 11:24:44 +0100 Subject: [PATCH] add integration for the RAG for ParseHTMLNode --- examples/graph_examples/custom_graph_example.py | 2 +- scrapegraphai/nodes/generate_answer_node.py | 2 +- scrapegraphai/nodes/parse_html_node.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/graph_examples/custom_graph_example.py b/examples/graph_examples/custom_graph_example.py index db1f69dd..840bc7f6 100644 --- a/examples/graph_examples/custom_graph_example.py +++ b/examples/graph_examples/custom_graph_example.py @@ -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 diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index 6bb90772..9eeebd12 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -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: diff --git a/scrapegraphai/nodes/parse_html_node.py b/scrapegraphai/nodes/parse_html_node.py index 9dbe5c67..d8359d7c 100644 --- a/scrapegraphai/nodes/parse_html_node.py +++ b/scrapegraphai/nodes/parse_html_node.py @@ -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):