fix(SmartScraper): pass llm_model to ParseNode

f-aguzzi · f-aguzzi · commit 524216657590 · 2024-09-02T15:04:54.000+02:00
diff --git a/scrapegraphai/graphs/smart_scraper_graph.py b/scrapegraphai/graphs/smart_scraper_graph.py
@@ -74,6 +74,7 @@ def _create_graph(self) -> BaseGraph:
             input="doc",
             output=["parsed_doc"],
             node_config={
+                "llm_model": self.llm_model,
                 "chunk_size": self.model_token
             }
         )

Original file line number	Diff line number	Diff line change
`@@ -74,6 +74,7 @@ def _create_graph(self) -> BaseGraph:`
`74`	`74`	`input="doc",`
`75`	`75`	`output=["parsed_doc"],`
`76`	`76`	`node_config={`
	`77`	`+ "llm_model": self.llm_model,`
`77`	`78`	`"chunk_size": self.model_token`
`78`	`79`	`}`
`79`	`80`	`)`