Skip to content

Commit 729d5d7

Browse files
committed
Changed node_config["llm"] to node_config["llm_model"]
1 parent c2c6162 commit 729d5d7

File tree

8 files changed

+14
-14
lines changed

8 files changed

+14
-14
lines changed

scrapegraphai/nodes/generate_answer_csv_node.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@ class GenerateAnswerCSVNode(BaseNode):
2222
an answer.
2323
2424
Attributes:
25-
llm: An instance of a language model client, configured for generating answers.
25+
llm_model: An instance of a language model client, configured for generating answers.
2626
node_name (str): The unique identifier name for the node, defaulting
2727
to "GenerateAnswerNodeCsv".
2828
node_type (str): The type of the node, set to "node" indicating a
2929
standard operational node.
3030
3131
Args:
32-
llm: An instance of the language model client (e.g., ChatOpenAI) used
32+
llm_model: An instance of the language model client (e.g., ChatOpenAI) used
3333
for generating answers.
3434
node_name (str, optional): The unique identifier name for the node.
3535
Defaults to "GenerateAnswerNodeCsv".
@@ -44,11 +44,11 @@ def __init__(self, input: str, output: List[str], node_config: dict,
4444
"""
4545
Initializes the GenerateAnswerNodeCsv with a language model client and a node name.
4646
Args:
47-
llm: An instance of the OpenAIImageToText class.
47+
llm_model: An instance of the OpenAIImageToText class.
4848
node_name (str): name of the node
4949
"""
5050
super().__init__(node_name, "node", input, output, 2, node_config)
51-
self.llm_model = node_config["llm"]
51+
self.llm_model = node_config["llm_model"]
5252
self.verbose = True if node_config is None else node_config.get(
5353
"verbose", False)
5454

scrapegraphai/nodes/generate_answer_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def __init__(self, input: str, output: List[str], node_config: dict,
3737
node_name: str = "GenerateAnswer"):
3838
super().__init__(node_name, "node", input, output, 2, node_config)
3939

40-
self.llm_model = node_config["llm"]
40+
self.llm_model = node_config["llm_model"]
4141
self.verbose = True if node_config is None else node_config.get("verbose", False)
4242

4343
def execute(self, state: dict) -> dict:

scrapegraphai/nodes/generate_answer_node_csv.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@ class GenerateAnswerCSVNode(BaseNode):
2222
an answer.
2323
2424
Attributes:
25-
llm: An instance of a language model client, configured for generating answers.
25+
llm_model: An instance of a language model client, configured for generating answers.
2626
node_name (str): The unique identifier name for the node, defaulting
2727
to "GenerateAnswerNodeCsv".
2828
node_type (str): The type of the node, set to "node" indicating a
2929
standard operational node.
3030
3131
Args:
32-
llm: An instance of the language model client (e.g., ChatOpenAI) used
32+
llm_model: An instance of the language model client (e.g., ChatOpenAI) used
3333
for generating answers.
3434
node_name (str, optional): The unique identifier name for the node.
3535
Defaults to "GenerateAnswerNodeCsv".
@@ -44,11 +44,11 @@ def __init__(self, input: str, output: List[str], node_config: dict,
4444
"""
4545
Initializes the GenerateAnswerNodeCsv with a language model client and a node name.
4646
Args:
47-
llm: An instance of the OpenAIImageToText class.
47+
llm_model: An instance of the OpenAIImageToText class.
4848
node_name (str): name of the node
4949
"""
5050
super().__init__(node_name, "node", input, output, 2, node_config)
51-
self.llm_model = node_config["llm"]
51+
self.llm_model = node_config["llm_model"]
5252
self.verbose = True if node_config is None else node_config.get(
5353
"verbose", False)
5454

scrapegraphai/nodes/generate_scraper_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def __init__(self, input: str, output: List[str], node_config: dict,
4040
library: str, website: str, node_name: str = "GenerateAnswer"):
4141
super().__init__(node_name, "node", input, output, 2, node_config)
4242

43-
self.llm_model = node_config["llm"]
43+
self.llm_model = node_config["llm_model"]
4444
self.library = library
4545
self.source = website
4646

scrapegraphai/nodes/rag_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ class RAGNode(BaseNode):
3939
def __init__(self, input: str, output: List[str], node_config: dict, node_name: str = "RAG"):
4040
super().__init__(node_name, "node", input, output, 2, node_config)
4141

42-
self.llm_model = node_config["llm"]
42+
self.llm_model = node_config["llm_model"]
4343
self.embedder_model = node_config.get("embedder_model", None)
4444
self.verbose = True if node_config is None else node_config.get(
4545
"verbose", False)

scrapegraphai/nodes/robots_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __init__(self, input: str, output: List[str], node_config: dict, force_scra
3838
node_name: str = "Robots"):
3939
super().__init__(node_name, "node", input, output, 1)
4040

41-
self.llm_model = node_config["llm"]
41+
self.llm_model = node_config["llm_model"]
4242
self.force_scraping = force_scraping
4343
self.verbose = True if node_config is None else node_config.get("verbose", False)
4444

scrapegraphai/nodes/search_internet_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def __init__(self, input: str, output: List[str], node_config: dict,
3131
node_name: str = "SearchInternet"):
3232
super().__init__(node_name, "node", input, output, 1, node_config)
3333

34-
self.llm_model = node_config["llm"]
34+
self.llm_model = node_config["llm_model"]
3535
self.verbose = True if node_config is None else node_config.get("verbose", False)
3636

3737
def execute(self, state: dict) -> dict:

scrapegraphai/nodes/search_link_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def __init__(self, input: str, output: List[str], node_config: dict,
3737
node_name: str = "GenerateLinks"):
3838
super().__init__(node_name, "node", input, output, 1, node_config)
3939

40-
self.llm_model = node_config["llm"]
40+
self.llm_model = node_config["llm_model"]
4141
self.verbose = True if node_config is None else node_config.get("verbose", False)
4242

4343
def execute(self, state: dict) -> dict:

0 commit comments

Comments
 (0)