From 9fa3df2390ddc1a93ca1bc30d728fbdc3f334bde Mon Sep 17 00:00:00 2001 From: Matteo Vedovati Date: Sun, 11 Aug 2024 09:33:21 +0200 Subject: [PATCH] Changed prompts import from helpers to prompts --- scrapegraphai/nodes/generate_answer_csv_node.py | 2 +- scrapegraphai/nodes/generate_answer_node.py | 2 +- scrapegraphai/nodes/generate_answer_omni_node.py | 2 +- scrapegraphai/nodes/generate_answer_pdf_node.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/scrapegraphai/nodes/generate_answer_csv_node.py b/scrapegraphai/nodes/generate_answer_csv_node.py index a91dae3f..0adf266d 100644 --- a/scrapegraphai/nodes/generate_answer_csv_node.py +++ b/scrapegraphai/nodes/generate_answer_csv_node.py @@ -10,7 +10,7 @@ from langchain_core.runnables import RunnableParallel from tqdm import tqdm from ..utils.logging import get_logger from .base_node import BaseNode -from ..helpers.generate_answer_node_csv_prompts import template_chunks_csv, template_no_chunks_csv, template_merge_csv +from ..prompts.generate_answer_node_csv_prompts import template_chunks_csv, template_no_chunks_csv, template_merge_csv class GenerateAnswerCSVNode(BaseNode): diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index 9c530688..d01b50d2 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -10,7 +10,7 @@ from langchain_community.chat_models import ChatOllama from tqdm import tqdm from ..utils.logging import get_logger from .base_node import BaseNode -from ..helpers 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 93e96f4e..77aa38ab 100644 --- a/scrapegraphai/nodes/generate_answer_omni_node.py +++ b/scrapegraphai/nodes/generate_answer_omni_node.py @@ -13,7 +13,7 @@ from tqdm import tqdm from langchain_community.chat_models import ChatOllama # Imports from the library from .base_node import BaseNode -from ..helpers.generate_answer_node_omni_prompts import template_no_chunk_omni, template_chunks_omni, template_merge_omni +from ..prompts.generate_answer_node_omni_prompts import template_no_chunk_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 4cef7ae9..bddb936d 100644 --- a/scrapegraphai/nodes/generate_answer_pdf_node.py +++ b/scrapegraphai/nodes/generate_answer_pdf_node.py @@ -10,7 +10,7 @@ from tqdm import tqdm from langchain_community.chat_models import ChatOllama from ..utils.logging import get_logger from .base_node import BaseNode -from ..helpers.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):