refactoring of node names

ScrapeGraphAI · Sep 22, 2024 · c5a3f89 · c5a3f89
1 parent 2ffeb36
commit c5a3f89
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 6 deletions.
diff --git a/scrapegraphai/nodes/conditional_node.py b/scrapegraphai/nodes/conditional_node.py
@@ -32,7 +32,6 @@ def __init__(self):
         """
         Initializes an empty ConditionalNode.
         """
-
         #super().__init__(node_name, "node", input, output, 2, node_config)
         pass
 

diff --git a/scrapegraphai/nodes/fetch_screen_node.py b/scrapegraphai/nodes/fetch_screen_node.py
@@ -16,7 +16,7 @@ def __init__(
         input: str,
         output: List[str],
         node_config: Optional[dict] = None,
-        node_name: str = "FetchScreenNode",
+        node_name: str = "FetchScreen",
     ):
         super().__init__(node_name, "node", input, output, 2, node_config)
         self.url = node_config.get("link")

diff --git a/scrapegraphai/nodes/parse_node.py b/scrapegraphai/nodes/parse_node.py
@@ -33,7 +33,7 @@ def __init__(
         input: str,
         output: List[str],
         node_config: Optional[dict] = None,
-        node_name: str = "Parse",
+        node_name: str = "ParseNode",
     ):
         super().__init__(node_name, "node", input, output, 1, node_config)
 

diff --git a/scrapegraphai/nodes/search_link_node.py b/scrapegraphai/nodes/search_link_node.py
@@ -36,7 +36,7 @@ def __init__(
         input: str,
         output: List[str],
         node_config: Optional[dict] = None,
-        node_name: str = "GenerateLinks",
+        node_name: str = "SearchLinks",
     ):
         super().__init__(node_name, "node", input, output, 1, node_config)
 

diff --git a/scrapegraphai/nodes/search_node_with_context.py b/scrapegraphai/nodes/search_node_with_context.py
@@ -23,15 +23,15 @@ class SearchLinksWithContext(BaseNode):
         input (str): Boolean expression defining the input keys needed from the state.
         output (List[str]): List of output keys to be updated in the state.
         node_config (dict): Additional configuration for the node.
-        node_name (str): The unique identifier name for the node, defaulting to "GenerateAnswer".
+        node_name (str): The unique identifier name for the node, defaulting to "SearchLinksWithContext".
     """
 
     def __init__(
         self,
         input: str,
         output: List[str],
         node_config: Optional[dict] = None,
-        node_name: str = "GenerateAnswer",
+        node_name: str = "SearchLinksWithContext",
     ):
         super().__init__(node_name, "node", input, output, 2, node_config)
         self.llm_model = node_config["llm_model"]