From 2178485b3d1ef06ff87e2ba52621169214307101 Mon Sep 17 00:00:00 2001 From: Eric Page Date: Sun, 5 May 2024 09:47:13 +0200 Subject: [PATCH] Adjusted graphs to reflect node_config change --- scrapegraphai/graphs/csv_scraper_graph.py | 4 ++-- scrapegraphai/graphs/json_scraper_graph.py | 4 ++-- scrapegraphai/graphs/script_creator_graph.py | 4 ++-- scrapegraphai/graphs/search_graph.py | 6 +++--- scrapegraphai/graphs/smart_scraper_graph.py | 4 ++-- scrapegraphai/graphs/speech_graph.py | 4 ++-- scrapegraphai/graphs/xml_scraper_graph.py | 4 ++-- 7 files changed, 15 insertions(+), 15 deletions(-) diff --git a/scrapegraphai/graphs/csv_scraper_graph.py b/scrapegraphai/graphs/csv_scraper_graph.py index 9a5eb931..cf6c9e28 100644 --- a/scrapegraphai/graphs/csv_scraper_graph.py +++ b/scrapegraphai/graphs/csv_scraper_graph.py @@ -49,7 +49,7 @@ class CSVScraperGraph(AbstractGraph): input="user_prompt & (parsed_doc | doc)", output=["relevant_chunks"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "embedder_model": self.embedder_model, "verbose": self.verbose } @@ -58,7 +58,7 @@ class CSVScraperGraph(AbstractGraph): input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "verbose": self.verbose } ) diff --git a/scrapegraphai/graphs/json_scraper_graph.py b/scrapegraphai/graphs/json_scraper_graph.py index f7392212..e99d82f4 100644 --- a/scrapegraphai/graphs/json_scraper_graph.py +++ b/scrapegraphai/graphs/json_scraper_graph.py @@ -73,7 +73,7 @@ class JSONScraperGraph(AbstractGraph): input="user_prompt & (parsed_doc | doc)", output=["relevant_chunks"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "embedder_model": self.embedder_model, "verbose": self.verbose } @@ -82,7 +82,7 @@ class JSONScraperGraph(AbstractGraph): input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "verbose": self.verbose } ) diff --git a/scrapegraphai/graphs/script_creator_graph.py b/scrapegraphai/graphs/script_creator_graph.py index 105048db..b34bdf91 100644 --- a/scrapegraphai/graphs/script_creator_graph.py +++ b/scrapegraphai/graphs/script_creator_graph.py @@ -77,7 +77,7 @@ class ScriptCreatorGraph(AbstractGraph): input="user_prompt & (parsed_doc | doc)", output=["relevant_chunks"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "embedder_model": self.embedder_model, "verbose": self.verbose } @@ -85,7 +85,7 @@ class ScriptCreatorGraph(AbstractGraph): generate_scraper_node = GenerateScraperNode( input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], - node_config={"llm": self.llm_model, + node_config={"llm_model": self.llm_model, "verbose": self.verbose}, library=self.library, website=self.source diff --git a/scrapegraphai/graphs/search_graph.py b/scrapegraphai/graphs/search_graph.py index 41548a77..09d05a99 100644 --- a/scrapegraphai/graphs/search_graph.py +++ b/scrapegraphai/graphs/search_graph.py @@ -50,7 +50,7 @@ class SearchGraph(AbstractGraph): input="user_prompt", output=["url"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "verbose": self.verbose } ) @@ -74,7 +74,7 @@ class SearchGraph(AbstractGraph): input="user_prompt & (parsed_doc | doc)", output=["relevant_chunks"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "embedder_model": self.embedder_model, "verbose": self.verbose } @@ -83,7 +83,7 @@ class SearchGraph(AbstractGraph): input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "verbose": self.verbose } ) diff --git a/scrapegraphai/graphs/smart_scraper_graph.py b/scrapegraphai/graphs/smart_scraper_graph.py index 4d6b0e93..d9077d49 100644 --- a/scrapegraphai/graphs/smart_scraper_graph.py +++ b/scrapegraphai/graphs/smart_scraper_graph.py @@ -75,7 +75,7 @@ class SmartScraperGraph(AbstractGraph): input="user_prompt & (parsed_doc | doc)", output=["relevant_chunks"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "embedder_model": self.embedder_model, "verbose": self.verbose } @@ -84,7 +84,7 @@ class SmartScraperGraph(AbstractGraph): input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "verbose": self.verbose } ) diff --git a/scrapegraphai/graphs/speech_graph.py b/scrapegraphai/graphs/speech_graph.py index 3edadfd0..d4a2b669 100644 --- a/scrapegraphai/graphs/speech_graph.py +++ b/scrapegraphai/graphs/speech_graph.py @@ -74,7 +74,7 @@ class SpeechGraph(AbstractGraph): input="user_prompt & (parsed_doc | doc)", output=["relevant_chunks"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "embedder_model": self.embedder_model, "verbose": self.verbose } @@ -83,7 +83,7 @@ class SpeechGraph(AbstractGraph): input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "verbose": self.verbose } ) diff --git a/scrapegraphai/graphs/xml_scraper_graph.py b/scrapegraphai/graphs/xml_scraper_graph.py index c84e1506..168cb4a0 100644 --- a/scrapegraphai/graphs/xml_scraper_graph.py +++ b/scrapegraphai/graphs/xml_scraper_graph.py @@ -75,7 +75,7 @@ class XMLScraperGraph(AbstractGraph): input="user_prompt & (parsed_doc | doc)", output=["relevant_chunks"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "embedder_model": self.embedder_model, "verbose": self.verbose } @@ -84,7 +84,7 @@ class XMLScraperGraph(AbstractGraph): input="user_prompt & (relevant_chunks | parsed_doc | doc)", output=["answer"], node_config={ - "llm": self.llm_model, + "llm_model": self.llm_model, "verbose": self.verbose } )