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

Update generate_answer_node.py

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-08-17 13:30:44 +02:00
parent c6edeeecb1
commit c2c2e6582f
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
scrapegraphai/nodes/generate_answer_node.py
View File

@ -88,19 +88,22 @@ class GenerateAnswerNode(BaseNode):
# Initialize the output parser
if self.node_config.get("schema", None) is not None:
output_parser = JsonOutputParser(pydantic_object=self.node_config["schema"])
if isinstance(self.llm_model, ChatOpenAI):
self.llm_model = self.llm_model.with_structured_output(self.node_config["schema"])
else:
output_parser = JsonOutputParser()
format_instructions = output_parser.get_format_instructions()
template_no_chunks_prompt = template_no_chunks
template_chunks_prompt = template_chunks
template_merge_prompt = template_merge
if isinstance(self.llm_model, ChatOpenAI) and not self.script_creator or self.force and not self.script_creator or self.is_md_scraper:
template_no_chunks_prompt = template_no_chunks_md
template_chunks_prompt = template_chunks_md
template_merge_prompt = template_merge_md
else:
template_no_chunks_prompt = template_no_chunks
template_chunks_prompt = template_chunks
template_merge_prompt = template_merge
if self.additional_info is not None:
template_no_chunks_prompt = self.additional_info + template_no_chunks_prompt