From 244aada2de1f3bc88782fa90e604e8b936b79aa4 Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Tue, 4 Jun 2024 10:01:20 +0200 Subject: [PATCH] feat: refactoring of an in if --- scrapegraphai/nodes/generate_answer_node.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index 4b8c5c36..19b0fd5e 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -46,7 +46,7 @@ class GenerateAnswerNode(BaseNode): super().__init__(node_name, "node", input, output, 2, node_config) 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.verbose = ( True if node_config is None else node_config.get("verbose", False)