fix(node): fixed generate answer node pydantic schema

Author: PeriniM

scrapegraphai/nodes/generate_answer_node.py

@@ -93,35 +93,20 @@ def execute(self, state: dict) -> dict:
 
         # Use tqdm to add progress bar
         for i, chunk in enumerate(tqdm(doc, desc="Processing chunks", disable=not self.verbose)):
-            if self.node_config.get("schema", None) is None and len(doc) == 1:
+            if len(doc) == 1:
                 prompt = PromptTemplate(
                     template=template_no_chunks,
                     input_variables=["question"],
                     partial_variables={"context": chunk.page_content,
                                        "format_instructions": format_instructions})
-            elif self.node_config.get("schema", None) is not None and len(doc) == 1:
-                 prompt = PromptTemplate(
-                    template=template_no_chunks_with_schema,
-                    input_variables=["question"],
-                    partial_variables={"context": chunk.page_content,
-                                       "format_instructions": format_instructions,
-                                       "schema": self.node_config.get("schema", None)
-                                       })
-            elif self.node_config.get("schema", None) is None and len(doc) > 1:
+                
+            else:
                 prompt = PromptTemplate(
                     template=template_chunks,
                     input_variables=["question"],
                     partial_variables={"context": chunk.page_content,
                                         "chunk_id": i + 1,
                                         "format_instructions": format_instructions})
-            elif self.node_config.get("schema", None) is not None and len(doc) > 1:
-                prompt = PromptTemplate(
-                    template=template_chunks_with_schema,
-                    input_variables=["question"],
-                    partial_variables={"context": chunk.page_content,
-                                        "chunk_id": i + 1,
-                                        "format_instructions": format_instructions,
-                                        "schema": self.node_config.get("schema", None)})
 
             # Dynamically name the chains based on their index
             chain_name = f"chunk{i+1}"
@@ -147,4 +132,4 @@ def execute(self, state: dict) -> dict:
 
         # Update the state with the generated answer
         state.update({self.output[0]: answer})
-        return state
+        return state