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

feat: refactoring of an in if

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-06-04 10:01:20 +02:00
parent 55b486503e
commit 244aada2de
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scrapegraphai/nodes/generate_answer_node.py
View File

@ -46,7 +46,7 @@ class GenerateAnswerNode(BaseNode):
super().__init__(node_name, "node", input, output, 2, node_config) super().__init__(node_name, "node", input, output, 2, node_config)
self.llm_model = node_config["llm_model"] self.llm_model = node_config["llm_model"]
if isinstance(node_config["llm_model"], Ollama) or isinstance(node_config["llm_model"], OpenAI): if isinstance(node_config["llm_model"], Ollama):
self.llm_model.format="json" self.llm_model.format="json"
self.verbose = ( self.verbose = (
True if node_config is None else node_config.get("verbose", False) True if node_config is None else node_config.get("verbose", False)