CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-06-23 21:00:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: modified node name

Browse Source
This commit is contained in:
Marco Perini 2024-05-01 21:33:25 +02:00
parent 276ac8b40c
commit 02d1af006c
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scrapegraphai/graphs/csv_scraper_graph.py
View File

@ -6,7 +6,7 @@ from ..nodes import (
FetchNode,
ParseNode,
RAGNode,
GenerateAnswerNodeCsv
GenerateAnswerCSVNode
)
from .abstract_graph import AbstractGraph
@ -54,7 +54,7 @@ class CSVScraperGraph(AbstractGraph):
"verbose": self.verbose
}
)
generate_answer_node = GenerateAnswerNodeCsv(
generate_answer_node = GenerateAnswerCSVNode(
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
output=["answer"],
node_config={

2
scrapegraphai/nodes/__init__.py
View File

@ -14,4 +14,4 @@ from .search_internet_node import SearchInternetNode
from .generate_scraper_node import GenerateScraperNode
from .search_link_node import SearchLinkNode
from .robots_node import RobotsNode
from .generate_answer_node_csv import GenerateAnswerNodeCsv
from .generate_answer_node_csv import GenerateAnswerCSVNode

2
scrapegraphai/nodes/generate_answer_node_csv.py
View File

@ -14,7 +14,7 @@ from langchain_core.runnables import RunnableParallel
from .base_node import BaseNode
class GenerateAnswerNodeCsv(BaseNode):
class GenerateAnswerCSVNode(BaseNode):
"""
A node that generates an answer using a language model (LLM) based on the user's input
and the content extracted from a webpage. It constructs a prompt from the user's input