Adjusted graphs to reflect node_config change

epage480 · epage480 · commit 2178485b3d1e · 2024-05-05T09:47:13.000+02:00
diff --git a/scrapegraphai/graphs/csv_scraper_graph.py b/scrapegraphai/graphs/csv_scraper_graph.py
@@ -49,7 +49,7 @@ def _create_graph(self):
             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 @@ def _create_graph(self):
             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
@@ -73,7 +73,7 @@ def _create_graph(self) -> BaseGraph:
             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 @@ def _create_graph(self) -> BaseGraph:
             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
@@ -77,15 +77,15 @@ def _create_graph(self) -> BaseGraph:
             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
             }
         )
         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
@@ -50,7 +50,7 @@ def _create_graph(self) -> BaseGraph:
             input="user_prompt",
             output=["url"],
             node_config={
-                "llm": self.llm_model,
+                "llm_model": self.llm_model,
                 "verbose": self.verbose
             }
         )
@@ -74,7 +74,7 @@ def _create_graph(self) -> BaseGraph:
             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 @@ def _create_graph(self) -> BaseGraph:
             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
@@ -75,7 +75,7 @@ def _create_graph(self) -> BaseGraph:
             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 @@ def _create_graph(self) -> BaseGraph:
             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
@@ -74,7 +74,7 @@ def _create_graph(self) -> BaseGraph:
             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 @@ def _create_graph(self) -> BaseGraph:
             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
@@ -75,7 +75,7 @@ def _create_graph(self) -> BaseGraph:
             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 @@ def _create_graph(self) -> BaseGraph:
             input="user_prompt & (relevant_chunks | parsed_doc | doc)",
             output=["answer"],
             node_config={
-                "llm": self.llm_model,
+                "llm_model": self.llm_model,
                 "verbose": self.verbose
             }
         )

Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,7 @@ def _create_graph(self):`
`49`	`49`	`input="user_prompt & (parsed_doc \| doc)",`
`50`	`50`	`output=["relevant_chunks"],`
`51`	`51`	`node_config={`
`52`		`- "llm": self.llm_model,`
	`52`	`+ "llm_model": self.llm_model,`
`53`	`53`	`"embedder_model": self.embedder_model,`
`54`	`54`	`"verbose": self.verbose`
`55`	`55`	`}`
`@@ -58,7 +58,7 @@ def _create_graph(self):`
`58`	`58`	`input="user_prompt & (relevant_chunks \| parsed_doc \| doc)",`
`59`	`59`	`output=["answer"],`
`60`	`60`	`node_config={`
`61`		`- "llm": self.llm_model,`
	`61`	`+ "llm_model": self.llm_model,`
`62`	`62`	`"verbose": self.verbose`
`63`	`63`	`}`
`64`	`64`	`)`
Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ def _create_graph(self) -> BaseGraph:`
`73`	`73`	`input="user_prompt & (parsed_doc \| doc)",`
`74`	`74`	`output=["relevant_chunks"],`
`75`	`75`	`node_config={`
`76`		`- "llm": self.llm_model,`
	`76`	`+ "llm_model": self.llm_model,`
`77`	`77`	`"embedder_model": self.embedder_model,`
`78`	`78`	`"verbose": self.verbose`
`79`	`79`	`}`
`@@ -82,7 +82,7 @@ def _create_graph(self) -> BaseGraph:`
`82`	`82`	`input="user_prompt & (relevant_chunks \| parsed_doc \| doc)",`
`83`	`83`	`output=["answer"],`
`84`	`84`	`node_config={`
`85`		`- "llm": self.llm_model,`
	`85`	`+ "llm_model": self.llm_model,`
`86`	`86`	`"verbose": self.verbose`
`87`	`87`	`}`
`88`	`88`	`)`
Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ def _create_graph(self) -> BaseGraph:`
`50`	`50`	`input="user_prompt",`
`51`	`51`	`output=["url"],`
`52`	`52`	`node_config={`
`53`		`- "llm": self.llm_model,`
	`53`	`+ "llm_model": self.llm_model,`
`54`	`54`	`"verbose": self.verbose`
`55`	`55`	`}`
`56`	`56`	`)`
`@@ -74,7 +74,7 @@ def _create_graph(self) -> BaseGraph:`
`74`	`74`	`input="user_prompt & (parsed_doc \| doc)",`
`75`	`75`	`output=["relevant_chunks"],`
`76`	`76`	`node_config={`
`77`		`- "llm": self.llm_model,`
	`77`	`+ "llm_model": self.llm_model,`
`78`	`78`	`"embedder_model": self.embedder_model,`
`79`	`79`	`"verbose": self.verbose`
`80`	`80`	`}`
`@@ -83,7 +83,7 @@ def _create_graph(self) -> BaseGraph:`
`83`	`83`	`input="user_prompt & (relevant_chunks \| parsed_doc \| doc)",`
`84`	`84`	`output=["answer"],`
`85`	`85`	`node_config={`
`86`		`- "llm": self.llm_model,`
	`86`	`+ "llm_model": self.llm_model,`
`87`	`87`	`"verbose": self.verbose`
`88`	`88`	`}`
`89`	`89`	`)`
Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ def _create_graph(self) -> BaseGraph:`
`75`	`75`	`input="user_prompt & (parsed_doc \| doc)",`
`76`	`76`	`output=["relevant_chunks"],`
`77`	`77`	`node_config={`
`78`		`- "llm": self.llm_model,`
	`78`	`+ "llm_model": self.llm_model,`
`79`	`79`	`"embedder_model": self.embedder_model,`
`80`	`80`	`"verbose": self.verbose`
`81`	`81`	`}`
`@@ -84,7 +84,7 @@ def _create_graph(self) -> BaseGraph:`
`84`	`84`	`input="user_prompt & (relevant_chunks \| parsed_doc \| doc)",`
`85`	`85`	`output=["answer"],`
`86`	`86`	`node_config={`
`87`		`- "llm": self.llm_model,`
	`87`	`+ "llm_model": self.llm_model,`
`88`	`88`	`"verbose": self.verbose`
`89`	`89`	`}`
`90`	`90`	`)`