From c64ce88db80d60f652bf6a33cb7fe27ddcf1ffdb Mon Sep 17 00:00:00 2001
From: Marco Vinciguerra <mvincig11@gmail.com>
Date: Thu, 12 Sep 2024 10:16:15 +0200
Subject: [PATCH] refactoting of imports

---
 scrapegraphai/nodes/concat_answers_node.py       | 3 ++-
 scrapegraphai/nodes/generate_answer_node.py      | 5 ++++-
 scrapegraphai/nodes/generate_answer_omni_node.py | 4 +++-
 scrapegraphai/nodes/generate_answer_pdf_node.py  | 4 +++-
 4 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/scrapegraphai/nodes/concat_answers_node.py b/scrapegraphai/nodes/concat_answers_node.py
index add8c9f6..5af81702 100644
--- a/scrapegraphai/nodes/concat_answers_node.py
+++ b/scrapegraphai/nodes/concat_answers_node.py
@@ -8,7 +8,8 @@ from .base_node import BaseNode
 
 class ConcatAnswersNode(BaseNode):
     """
-    A node responsible for concatenating the answers from multiple graph instances into a single answer.
+    A node responsible for concatenating the answers from multiple 
+    graph instances into a single answer.
 
     Attributes:
         verbose (bool): A flag indicating whether to show print statements during execution.
diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py
index 42586b60..be97b832 100644
--- a/scrapegraphai/nodes/generate_answer_node.py
+++ b/scrapegraphai/nodes/generate_answer_node.py
@@ -11,7 +11,10 @@ from langchain_mistralai import ChatMistralAI
 from langchain_community.chat_models import ChatOllama
 from tqdm import tqdm
 from .base_node import BaseNode
-from ..prompts import TEMPLATE_CHUNKS, TEMPLATE_NO_CHUNKS, TEMPLATE_MERGE, TEMPLATE_CHUNKS_MD, TEMPLATE_NO_CHUNKS_MD, TEMPLATE_MERGE_MD
+from ..prompts import (TEMPLATE_CHUNKS,
+                       TEMPLATE_NO_CHUNKS, TEMPLATE_MERGE,
+                       TEMPLATE_CHUNKS_MD, TEMPLATE_NO_CHUNKS_MD,
+                       TEMPLATE_MERGE_MD)
 
 class GenerateAnswerNode(BaseNode):
     """
diff --git a/scrapegraphai/nodes/generate_answer_omni_node.py b/scrapegraphai/nodes/generate_answer_omni_node.py
index aabebce4..32dfbff6 100644
--- a/scrapegraphai/nodes/generate_answer_omni_node.py
+++ b/scrapegraphai/nodes/generate_answer_omni_node.py
@@ -11,7 +11,9 @@ from langchain_mistralai import ChatMistralAI
 from tqdm import tqdm
 from langchain_community.chat_models import ChatOllama
 from .base_node import BaseNode
-from ..prompts.generate_answer_node_omni_prompts import TEMPLATE_NO_CHUNKS_OMNI, TEMPLATE_CHUNKS_OMNI, TEMPLATE_MERGE_OMNI
+from ..prompts.generate_answer_node_omni_prompts import (TEMPLATE_NO_CHUNKS_OMNI, 
+                                                        TEMPLATE_CHUNKS_OMNI,
+                                                        TEMPLATE_MERGE_OMNI)
 
 class GenerateAnswerOmniNode(BaseNode):
     """
diff --git a/scrapegraphai/nodes/generate_answer_pdf_node.py b/scrapegraphai/nodes/generate_answer_pdf_node.py
index 2a25cc20..3f7daf73 100644
--- a/scrapegraphai/nodes/generate_answer_pdf_node.py
+++ b/scrapegraphai/nodes/generate_answer_pdf_node.py
@@ -12,7 +12,9 @@ from tqdm import tqdm
 from langchain_community.chat_models import ChatOllama
 from ..utils.logging import get_logger
 from .base_node import BaseNode
-from ..prompts.generate_answer_node_pdf_prompts import TEMPLATE_CHUNKS_PDF, TEMPLATE_NO_CHUNKS_PDF, TEMPLATE_MERGE_PDF
+from ..prompts.generate_answer_node_pdf_prompts import (TEMPLATE_CHUNKS_PDF,
+                                                        TEMPLATE_NO_CHUNKS_PDF,
+                                                        TEMPLATE_MERGE_PDF)
 
 class GenerateAnswerPDFNode(BaseNode):
     """