Skip to content

Commit c64ce88

Browse files
committed
refactoting of imports
1 parent ed8e173 commit c64ce88

File tree

4 files changed

+12
-4
lines changed

4 files changed

+12
-4
lines changed

scrapegraphai/nodes/concat_answers_node.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,8 @@
88

99
class ConcatAnswersNode(BaseNode):
1010
"""
11-
A node responsible for concatenating the answers from multiple graph instances into a single answer.
11+
A node responsible for concatenating the answers from multiple
12+
graph instances into a single answer.
1213
1314
Attributes:
1415
verbose (bool): A flag indicating whether to show print statements during execution.

scrapegraphai/nodes/generate_answer_node.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,10 @@
1111
from langchain_community.chat_models import ChatOllama
1212
from tqdm import tqdm
1313
from .base_node import BaseNode
14-
from ..prompts import TEMPLATE_CHUNKS, TEMPLATE_NO_CHUNKS, TEMPLATE_MERGE, TEMPLATE_CHUNKS_MD, TEMPLATE_NO_CHUNKS_MD, TEMPLATE_MERGE_MD
14+
from ..prompts import (TEMPLATE_CHUNKS,
15+
TEMPLATE_NO_CHUNKS, TEMPLATE_MERGE,
16+
TEMPLATE_CHUNKS_MD, TEMPLATE_NO_CHUNKS_MD,
17+
TEMPLATE_MERGE_MD)
1518

1619
class GenerateAnswerNode(BaseNode):
1720
"""

scrapegraphai/nodes/generate_answer_omni_node.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,9 @@
1111
from tqdm import tqdm
1212
from langchain_community.chat_models import ChatOllama
1313
from .base_node import BaseNode
14-
from ..prompts.generate_answer_node_omni_prompts import TEMPLATE_NO_CHUNKS_OMNI, TEMPLATE_CHUNKS_OMNI, TEMPLATE_MERGE_OMNI
14+
from ..prompts.generate_answer_node_omni_prompts import (TEMPLATE_NO_CHUNKS_OMNI,
15+
TEMPLATE_CHUNKS_OMNI,
16+
TEMPLATE_MERGE_OMNI)
1517

1618
class GenerateAnswerOmniNode(BaseNode):
1719
"""

scrapegraphai/nodes/generate_answer_pdf_node.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@
1212
from langchain_community.chat_models import ChatOllama
1313
from ..utils.logging import get_logger
1414
from .base_node import BaseNode
15-
from ..prompts.generate_answer_node_pdf_prompts import TEMPLATE_CHUNKS_PDF, TEMPLATE_NO_CHUNKS_PDF, TEMPLATE_MERGE_PDF
15+
from ..prompts.generate_answer_node_pdf_prompts import (TEMPLATE_CHUNKS_PDF,
16+
TEMPLATE_NO_CHUNKS_PDF,
17+
TEMPLATE_MERGE_PDF)
1618

1719
class GenerateAnswerPDFNode(BaseNode):
1820
"""

0 commit comments

Comments
 (0)