diff --git a/scrapegraphai/graphs/csv_scraper_multi_graph.py b/scrapegraphai/graphs/csv_scraper_multi_graph.py index 808549aa..59e84783 100644 --- a/scrapegraphai/graphs/csv_scraper_multi_graph.py +++ b/scrapegraphai/graphs/csv_scraper_multi_graph.py @@ -62,7 +62,7 @@ class CSVScraperMultiGraph(AbstractGraph): """ # ************************************************ - # Create a SmartScraperGraph instance + # Create a CSVScraperGraph instance # ************************************************ smart_scraper_instance = CSVScraperGraph( diff --git a/scrapegraphai/graphs/deep_scraper_graph.py b/scrapegraphai/graphs/deep_scraper_graph.py index ca617d19..d07a5276 100644 --- a/scrapegraphai/graphs/deep_scraper_graph.py +++ b/scrapegraphai/graphs/deep_scraper_graph.py @@ -10,7 +10,6 @@ from ..nodes import ( FetchNode, SearchLinkNode, ParseNode, - RAGNode, GenerateAnswerNode, GraphIteratorNode, MergeAnswersNode @@ -79,13 +78,7 @@ class DeepScraperGraph(AbstractGraph): "chunk_size": self.model_token } ) - rag_node = RAGNode( - input="user_prompt & (parsed_doc | doc)", - output=["relevant_chunks"], - node_config={ - "llm_model": self.llm_model, - } - ) + generate_answer_node = GenerateAnswerNode( input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], @@ -123,7 +116,6 @@ class DeepScraperGraph(AbstractGraph): nodes=[ fetch_node, parse_node, - rag_node, generate_answer_node, search_node, graph_iterator_node, @@ -131,9 +123,6 @@ class DeepScraperGraph(AbstractGraph): ], edges=[ (fetch_node, parse_node), - (parse_node, rag_node), - (rag_node, generate_answer_node), - (rag_node, search_node), (search_node, graph_iterator_node), (graph_iterator_node, merge_answers_node) ], diff --git a/scrapegraphai/graphs/json_scraper_graph.py b/scrapegraphai/graphs/json_scraper_graph.py index a23c1f38..288b8ee1 100644 --- a/scrapegraphai/graphs/json_scraper_graph.py +++ b/scrapegraphai/graphs/json_scraper_graph.py @@ -58,7 +58,7 @@ class JSONScraperGraph(AbstractGraph): input="json | json_dir", output=["doc", "link_urls", "img_urls"], ) - + generate_answer_node = GenerateAnswerNode( input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], diff --git a/scrapegraphai/graphs/json_scraper_multi_graph.py b/scrapegraphai/graphs/json_scraper_multi_graph.py index da7f33ba..42d2232e 100644 --- a/scrapegraphai/graphs/json_scraper_multi_graph.py +++ b/scrapegraphai/graphs/json_scraper_multi_graph.py @@ -63,7 +63,7 @@ class JSONScraperMultiGraph(AbstractGraph): """ # ************************************************ - # Create a SmartScraperGraph instance + # Create a JSONScraperGraph instance # ************************************************ smart_scraper_instance = JSONScraperGraph( diff --git a/scrapegraphai/graphs/markdown_scraper_multi_graph.py b/scrapegraphai/graphs/markdown_scraper_multi_graph.py index e59f6e5a..9796c11a 100644 --- a/scrapegraphai/graphs/markdown_scraper_multi_graph.py +++ b/scrapegraphai/graphs/markdown_scraper_multi_graph.py @@ -58,7 +58,6 @@ class MDScraperMultiGraph(AbstractGraph): Returns: BaseGraph: A graph instance representing the web scraping and searching workflow. """ - # Create a SmartScraperGraph instance smart_scraper_instance = MDScraperGraph( prompt="", source="", diff --git a/scrapegraphai/graphs/omni_scraper_graph.py b/scrapegraphai/graphs/omni_scraper_graph.py index 6849ee12..8b5f7fc9 100644 --- a/scrapegraphai/graphs/omni_scraper_graph.py +++ b/scrapegraphai/graphs/omni_scraper_graph.py @@ -85,7 +85,7 @@ class OmniScraperGraph(AbstractGraph): "max_images": self.max_images } ) - + generate_answer_omni_node = GenerateAnswerOmniNode( input="user_prompt & (relevant_chunks | parsed_doc | doc) & img_desc", output=["answer"], diff --git a/scrapegraphai/graphs/xml_scraper_graph.py b/scrapegraphai/graphs/xml_scraper_graph.py index e0a149eb..f5806f56 100644 --- a/scrapegraphai/graphs/xml_scraper_graph.py +++ b/scrapegraphai/graphs/xml_scraper_graph.py @@ -62,7 +62,7 @@ class XMLScraperGraph(AbstractGraph): input="xml | xml_dir", output=["doc", "link_urls", "img_urls"] ) - + generate_answer_node = GenerateAnswerNode( input="user_prompt & (relevant_chunks | doc)", output=["answer"], diff --git a/scrapegraphai/graphs/xml_scraper_multi_graph.py b/scrapegraphai/graphs/xml_scraper_multi_graph.py index 648db500..36831580 100644 --- a/scrapegraphai/graphs/xml_scraper_multi_graph.py +++ b/scrapegraphai/graphs/xml_scraper_multi_graph.py @@ -63,7 +63,7 @@ class XMLScraperMultiGraph(AbstractGraph): """ # ************************************************ - # Create a SmartScraperGraph instance + # Create a XMLScraperGraph instance # ************************************************ smart_scraper_instance = XMLScraperGraph(