CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-06-23 21:00:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Changed node_config["llm"] to node_config["llm_model"]

Browse Source
This commit is contained in:
Eric Page 2024-05-05 09:36:11 +02:00
parent c2c61625b3
commit 729d5d7597
8 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scrapegraphai/nodes/generate_answer_csv_node.py
View File

@ -22,14 +22,14 @@ class GenerateAnswerCSVNode(BaseNode):
an answer.
Attributes:
llm: An instance of a language model client, configured for generating answers.
llm_model: An instance of a language model client, configured for generating answers.
node_name (str): The unique identifier name for the node, defaulting
to "GenerateAnswerNodeCsv".
node_type (str): The type of the node, set to "node" indicating a
standard operational node.
Args:
llm: An instance of the language model client (e.g., ChatOpenAI) used
llm_model: An instance of the language model client (e.g., ChatOpenAI) used
for generating answers.
node_name (str, optional): The unique identifier name for the node.
Defaults to "GenerateAnswerNodeCsv".
@ -44,11 +44,11 @@ class GenerateAnswerCSVNode(BaseNode):
"""
Initializes the GenerateAnswerNodeCsv with a language model client and a node name.
Args:
llm: An instance of the OpenAIImageToText class.
llm_model: An instance of the OpenAIImageToText class.
node_name (str): name of the node
"""
super().__init__(node_name, "node", input, output, 2, node_config)
self.llm_model = node_config["llm"]
self.llm_model = node_config["llm_model"]
self.verbose = True if node_config is None else node_config.get(
"verbose", False)

2
scrapegraphai/nodes/generate_answer_node.py
View File

@ -37,7 +37,7 @@ class GenerateAnswerNode(BaseNode):
node_name: str = "GenerateAnswer"):
super().__init__(node_name, "node", input, output, 2, node_config)
self.llm_model = node_config["llm"]
self.llm_model = node_config["llm_model"]
self.verbose = True if node_config is None else node_config.get("verbose", False)
def execute(self, state: dict) -> dict:

8
scrapegraphai/nodes/generate_answer_node_csv.py
View File

@ -22,14 +22,14 @@ class GenerateAnswerCSVNode(BaseNode):
an answer.
Attributes:
llm: An instance of a language model client, configured for generating answers.
llm_model: An instance of a language model client, configured for generating answers.
node_name (str): The unique identifier name for the node, defaulting
to "GenerateAnswerNodeCsv".
node_type (str): The type of the node, set to "node" indicating a
standard operational node.
Args:
llm: An instance of the language model client (e.g., ChatOpenAI) used
llm_model: An instance of the language model client (e.g., ChatOpenAI) used
for generating answers.
node_name (str, optional): The unique identifier name for the node.
Defaults to "GenerateAnswerNodeCsv".
@ -44,11 +44,11 @@ class GenerateAnswerCSVNode(BaseNode):
"""
Initializes the GenerateAnswerNodeCsv with a language model client and a node name.
Args:
llm: An instance of the OpenAIImageToText class.
llm_model: An instance of the OpenAIImageToText class.
node_name (str): name of the node
"""
super().__init__(node_name, "node", input, output, 2, node_config)
self.llm_model = node_config["llm"]
self.llm_model = node_config["llm_model"]
self.verbose = True if node_config is None else node_config.get(
"verbose", False)

2
scrapegraphai/nodes/generate_scraper_node.py
View File

@ -40,7 +40,7 @@ class GenerateScraperNode(BaseNode):
library: str, website: str, node_name: str = "GenerateAnswer"):
super().__init__(node_name, "node", input, output, 2, node_config)
self.llm_model = node_config["llm"]
self.llm_model = node_config["llm_model"]
self.library = library
self.source = website

2
scrapegraphai/nodes/rag_node.py
View File

@ -39,7 +39,7 @@ class RAGNode(BaseNode):
def __init__(self, input: str, output: List[str], node_config: dict, node_name: str = "RAG"):
super().__init__(node_name, "node", input, output, 2, node_config)
self.llm_model = node_config["llm"]
self.llm_model = node_config["llm_model"]
self.embedder_model = node_config.get("embedder_model", None)
self.verbose = True if node_config is None else node_config.get(
"verbose", False)

2
scrapegraphai/nodes/robots_node.py
View File

@ -38,7 +38,7 @@ class RobotsNode(BaseNode):
node_name: str = "Robots"):
super().__init__(node_name, "node", input, output, 1)
self.llm_model = node_config["llm"]
self.llm_model = node_config["llm_model"]
self.force_scraping = force_scraping
self.verbose = True if node_config is None else node_config.get("verbose", False)

2
scrapegraphai/nodes/search_internet_node.py
View File

@ -31,7 +31,7 @@ class SearchInternetNode(BaseNode):
node_name: str = "SearchInternet"):
super().__init__(node_name, "node", input, output, 1, node_config)
self.llm_model = node_config["llm"]
self.llm_model = node_config["llm_model"]
self.verbose = True if node_config is None else node_config.get("verbose", False)
def execute(self, state: dict) -> dict:

2
scrapegraphai/nodes/search_link_node.py
View File

@ -37,7 +37,7 @@ class SearchLinkNode(BaseNode):
node_name: str = "GenerateLinks"):
super().__init__(node_name, "node", input, output, 1, node_config)
self.llm_model = node_config["llm"]
self.llm_model = node_config["llm_model"]
self.verbose = True if node_config is None else node_config.get("verbose", False)
def execute(self, state: dict) -> dict: