From 729d5d75975cdce0474d544cf91010e107955447 Mon Sep 17 00:00:00 2001 From: Eric Page Date: Sun, 5 May 2024 09:36:11 +0200 Subject: [PATCH] Changed node_config["llm"] to node_config["llm_model"] --- scrapegraphai/nodes/generate_answer_csv_node.py | 8 ++++---- scrapegraphai/nodes/generate_answer_node.py | 2 +- scrapegraphai/nodes/generate_answer_node_csv.py | 8 ++++---- scrapegraphai/nodes/generate_scraper_node.py | 2 +- scrapegraphai/nodes/rag_node.py | 2 +- scrapegraphai/nodes/robots_node.py | 2 +- scrapegraphai/nodes/search_internet_node.py | 2 +- scrapegraphai/nodes/search_link_node.py | 2 +- 8 files changed, 14 insertions(+), 14 deletions(-) diff --git a/scrapegraphai/nodes/generate_answer_csv_node.py b/scrapegraphai/nodes/generate_answer_csv_node.py index ac861816..6d2b84fc 100644 --- a/scrapegraphai/nodes/generate_answer_csv_node.py +++ b/scrapegraphai/nodes/generate_answer_csv_node.py @@ -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) diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index e9b4dd40..df3078ef 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -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: diff --git a/scrapegraphai/nodes/generate_answer_node_csv.py b/scrapegraphai/nodes/generate_answer_node_csv.py index ac861816..6d2b84fc 100644 --- a/scrapegraphai/nodes/generate_answer_node_csv.py +++ b/scrapegraphai/nodes/generate_answer_node_csv.py @@ -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) diff --git a/scrapegraphai/nodes/generate_scraper_node.py b/scrapegraphai/nodes/generate_scraper_node.py index 9c80fc19..2e1f959e 100644 --- a/scrapegraphai/nodes/generate_scraper_node.py +++ b/scrapegraphai/nodes/generate_scraper_node.py @@ -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 diff --git a/scrapegraphai/nodes/rag_node.py b/scrapegraphai/nodes/rag_node.py index 92e7011f..f25f871d 100644 --- a/scrapegraphai/nodes/rag_node.py +++ b/scrapegraphai/nodes/rag_node.py @@ -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) diff --git a/scrapegraphai/nodes/robots_node.py b/scrapegraphai/nodes/robots_node.py index 001de62d..8c341183 100644 --- a/scrapegraphai/nodes/robots_node.py +++ b/scrapegraphai/nodes/robots_node.py @@ -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) diff --git a/scrapegraphai/nodes/search_internet_node.py b/scrapegraphai/nodes/search_internet_node.py index 00cf9211..01095ef8 100644 --- a/scrapegraphai/nodes/search_internet_node.py +++ b/scrapegraphai/nodes/search_internet_node.py @@ -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: diff --git a/scrapegraphai/nodes/search_link_node.py b/scrapegraphai/nodes/search_link_node.py index 7f766b5b..037b862e 100644 --- a/scrapegraphai/nodes/search_link_node.py +++ b/scrapegraphai/nodes/search_link_node.py @@ -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: