add get functions on the dictionary

Author: VinciGit00

scrapegraphai/nodes/generate_answer_csv_node.py

@@ -95,8 +95,8 @@ def execute(self, state):
         doc = input_data[1]
 
         # Initialize the output parser
-        if self.node_config["schema"] is not None:
-            output_parser = PydanticOutputParser(pydantic_object=self.node_config["schema"])
+        if self.node_config.get("schema", None) is not None:
+            output_parser = PydanticOutputParser(pydantic_object=self.node_config.get("schema", None))
         else:
             output_parser = JsonOutputParser()
 

scrapegraphai/nodes/generate_answer_node.py

@@ -81,8 +81,8 @@ def execute(self, state: dict) -> dict:
         doc = input_data[1]
 
         # Initialize the output parser
-        if self.node_config["schema"] is not None:
-            output_parser = PydanticOutputParser(pydantic_object=self.node_config["schema"])
+        if self.node_config.get("schema",None) is not None:
+            output_parser = PydanticOutputParser(pydantic_object=self.node_config.get("schema", None))
         else:
             output_parser = JsonOutputParser()
 

scrapegraphai/nodes/generate_answer_omni_node.py

@@ -81,8 +81,8 @@ def execute(self, state: dict) -> dict:
         imag_desc = input_data[2]
 
         # Initialize the output parser
-        if self.node_config["schema"] is not None:
-            output_parser = PydanticOutputParser(pydantic_object=self.node_config["schema"])
+        if self.node_config.get("schema", None) is not None:
+            output_parser = PydanticOutputParser(pydantic_object=self.node_config.get("schema", None))
         else:
             output_parser = JsonOutputParser()
 

scrapegraphai/nodes/generate_answer_pdf_node.py

@@ -96,8 +96,8 @@ def execute(self, state):
         doc = input_data[1]
 
         # Initialize the output parser
-        if self.node_config["schema"] is not None:
-            output_parser = PydanticOutputParser(pydantic_object=self.node_config["schema"])
+        if self.node_config.get("schema",None) is not None:
+            output_parser = PydanticOutputParser(pydantic_object=self.node_config.get("schema", None))
         else:
             output_parser = JsonOutputParser()