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): """