refactor cond node structure to fit with the new implementation

Author: ekinsenler

scrapegraphai/graphs/base_graph.py

@@ -95,7 +95,10 @@ def _set_conditional_node_edges(self):
                     raise ValueError(f"ConditionalNode '{node.node_name}' must have exactly two outgoing edges.")
                 # Assign true_node_name and false_node_name
                 node.true_node_name = outgoing_edges[0][1].node_name
-                node.false_node_name = outgoing_edges[1][1].node_name
+                try:
+                    node.false_node_name = outgoing_edges[1][1].node_name
+                except:
+                    node.false_node_name = None
 
     def _execute_standard(self, initial_state: dict) -> Tuple[dict, list]:
         """
@@ -221,6 +224,8 @@ def _execute_standard(self, initial_state: dict) -> Tuple[dict, list]:
                 node_names = {node.node_name for node in self.nodes}
                 if result in node_names:
                     current_node_name = result
+                elif result is None:
+                    current_node_name = None
                 else:
                     raise ValueError(f"Conditional Node returned a node name '{result}' that does not exist in the graph")
 

scrapegraphai/graphs/smart_scraper_graph.py

@@ -94,16 +94,16 @@ def _create_graph(self) -> BaseGraph:
         regen_node = None
         if self.config.get("reattempt") is True:
             cond_node = ConditionalNode(
-                input="results",
-                output=["results"],
+                input="answer",
+                output=["answer"],
                 node_name="ConditionalNode",
                 node_config={
-                    "key_name": "results",
-                    "condition": 'results and results!="NA"',
+                    "key_name": "answer",
+                    "condition": 'not answer or answer=="NA"',
                 }
             )
             regen_node = GenerateAnswerNode(
-                input="user_prompt & results",
+                input="user_prompt & answer",
                 output=["answer"],
                 node_config={
                     "llm_model": self.llm_model,

scrapegraphai/nodes/conditional_node.py

@@ -61,7 +61,7 @@ def execute(self, state: dict) -> dict:
             str: The name of the next node to execute based on the presence of the key.
         """
 
-        if self.true_node_name is None or self.false_node_name is None:
+        if self.true_node_name is None:
             raise ValueError("ConditionalNode's next nodes are not set properly.")
 
         if self.condition: