refatoring

Author: VinciGit00

scrapegraphai/graphs/script_creator_multi_graph.py

@@ -113,4 +113,4 @@ def run(self) -> str:
         """
         inputs = {"user_prompt": self.prompt, "urls": self.source}
         self.final_state, self.execution_info = self.graph.execute(inputs)
-        return self.final_state.get("merged_script", "Failed to generate the script.")
+        return self.final_state.get("merged_script", "Failed to generate the script.")

scrapegraphai/graphs/search_link_graph.py

@@ -101,4 +101,4 @@ def run(self) -> str:
         inputs = {"user_prompt": self.prompt, self.input_key: self.source}
         self.final_state, self.execution_info = self.graph.execute(inputs)
 
-        return self.final_state.get("parsed_doc", "No answer found.")
+        return self.final_state.get("parsed_doc", "No answer found.")

scrapegraphai/nodes/conditional_node.py

@@ -1,10 +1,9 @@
 """ 
 Module for implementing the conditional node
 """
-
+from typing import Optional, List
 from .base_node import BaseNode
 
-
 class ConditionalNode(BaseNode):
     """
     A node that determines the next step in the graph's execution flow based on 
@@ -29,16 +28,21 @@ class ConditionalNode(BaseNode):
 
     """
 
-    def __init__(self, key_name: str, node_name="ConditionalNode"):
+    def __init__(self,
+        input: str,
+        output: List[str],
+        node_config: Optional[dict] = None,
+        node_name: str = "GenerateAnswerCSV",
+    ):
         """
         Initializes the node with the key to check and the next node names based on the condition.
 
         Args:
             key_name (str): The name of the key to check in the state.
         """
 
-        super().__init__(node_name, "conditional_node")
-        self.key_name = key_name
+       super().__init__(node_name, "node", input, output, 2, node_config)
+
 
     def execute(self, state: dict) -> dict:
         """

scrapegraphai/nodes/generate_answer_csv_node.py

@@ -58,7 +58,7 @@ def __init__(
         self.verbose = (
             False if node_config is None else node_config.get("verbose", False)
         )
-        
+
         self.additional_info = node_config.get("additional_info")
 
     def execute(self, state):