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: fix an if

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-06-03 21:49:34 +02:00
parent 8de720d379
commit c8d556da4e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scrapegraphai/nodes/generate_answer_node.py
View File

@ -12,7 +12,7 @@ from langchain_core.runnables import RunnableParallel
from tqdm import tqdm
from ..utils.logging import get_logger
from ..models import Ollama, Groq
from ..models import Ollama, Groq, OpenAI
# Imports from the library
from .base_node import BaseNode
from ..helpers import template_chunks, template_no_chunks, template_merge, template_chunks_with_schema, template_no_chunks_with_schema
@ -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):
if isinstance(node_config["llm_model"], Ollama) or isinstance(node_config["llm_model"], OpenAI):
self.llm_model.format="json"
self.verbose = (
True if node_config is None else node_config.get("verbose", False)