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

fix: bug on script_creator_graph.py

Browse Source
This commit is contained in:
VinciGit00 2024-05-02 09:23:46 +02:00
parent 55a49a4381
commit 4a3bc37f2f
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scrapegraphai/graphs/script_creator_graph.py
View File

@ -62,14 +62,16 @@ class ScriptCreatorGraph(AbstractGraph):
input="url | local_dir", input="url | local_dir",
output=["doc"], output=["doc"],
node_config={ node_config={
"headless": True if self.config is None else self.config.get("headless", True), "headless": self.headless,
"verbose": self.verbose} "verbose": self.verbose
}
) )
parse_node = ParseNode( parse_node = ParseNode(
input="doc", input="doc",
output=["parsed_doc"], output=["parsed_doc"],
node_config={"chunk_size": self.model_token, node_config={"chunk_size": self.model_token,
"verbose": self.verbose} "verbose": self.verbose
}
) )
rag_node = RAGNode( rag_node = RAGNode(
input="user_prompt & (parsed_doc | doc)", input="user_prompt & (parsed_doc | doc)",