diff --git a/scrapegraphai/graphs/csv_scraper_graph.py b/scrapegraphai/graphs/csv_scraper_graph.py index 9ae3e605..9a5eb931 100644 --- a/scrapegraphai/graphs/csv_scraper_graph.py +++ b/scrapegraphai/graphs/csv_scraper_graph.py @@ -6,7 +6,7 @@ from ..nodes import ( FetchNode, ParseNode, RAGNode, - GenerateAnswerNodeCsv + GenerateAnswerCSVNode ) from .abstract_graph import AbstractGraph @@ -54,7 +54,7 @@ class CSVScraperGraph(AbstractGraph): "verbose": self.verbose } ) - generate_answer_node = GenerateAnswerNodeCsv( + generate_answer_node = GenerateAnswerCSVNode( input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], node_config={ diff --git a/scrapegraphai/nodes/__init__.py b/scrapegraphai/nodes/__init__.py index 85218bc7..150df0df 100644 --- a/scrapegraphai/nodes/__init__.py +++ b/scrapegraphai/nodes/__init__.py @@ -14,4 +14,4 @@ from .search_internet_node import SearchInternetNode from .generate_scraper_node import GenerateScraperNode from .search_link_node import SearchLinkNode from .robots_node import RobotsNode -from .generate_answer_node_csv import GenerateAnswerNodeCsv +from .generate_answer_node_csv import GenerateAnswerCSVNode diff --git a/scrapegraphai/nodes/generate_answer_node_csv.py b/scrapegraphai/nodes/generate_answer_node_csv.py index 5a3e74f0..ac861816 100644 --- a/scrapegraphai/nodes/generate_answer_node_csv.py +++ b/scrapegraphai/nodes/generate_answer_node_csv.py @@ -14,7 +14,7 @@ from langchain_core.runnables import RunnableParallel from .base_node import BaseNode -class GenerateAnswerNodeCsv(BaseNode): +class GenerateAnswerCSVNode(BaseNode): """ A node that generates an answer using a language model (LLM) based on the user's input and the content extracted from a webpage. It constructs a prompt from the user's input