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

code generator graph fixed

Browse Source
This commit is contained in:
Matteo Vedovati 2024-09-19 18:09:27 +02:00
parent 0f4b011814
commit eb9c77c2d5
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scrapegraphai/graphs/code_generator_graph.py
View File

@ -12,6 +12,7 @@ from ..nodes import (
GenerateAnswerNode,
PromptRefinerNode,
HtmlAnalyzerNode,
GenerateCodeNode,
)
class CodeGeneratorGraph(AbstractGraph):
@ -115,7 +116,7 @@ class CodeGeneratorGraph(AbstractGraph):
generate_code_node = GenerateCodeNode(
input="user_prompt & refined_prompt & html_info & doc & answer",
output=["code"],
output=["generated_code"],
node_config={
"llm_model": self.llm_model,
"additional_info": self.config.get("additional_info"),

3
scrapegraphai/nodes/__init__.py
View File

@ -24,4 +24,5 @@ from .fetch_screen_node import FetchScreenNode
from .generate_answer_from_image_node import GenerateAnswerFromImageNode
from .concat_answers_node import ConcatAnswersNode
from .prompt_refiner_node import PromptRefinerNode
from .html_analyzer_node import HtmlAnalyzerNode
from .html_analyzer_node import HtmlAnalyzerNode
from .generate_code_node import GenerateCodeNode