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

fix: fix generate answer node

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-11-09 11:35:04 +01:00
parent 6f1a88f3e8
commit d332e216db
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scrapegraphai/nodes/generate_answer_node.py
View File

@ -8,7 +8,6 @@ from langchain_core.output_parsers import JsonOutputParser
from langchain_core.runnables import RunnableParallel
from langchain_openai import ChatOpenAI, AzureChatOpenAI
from langchain_aws import ChatBedrock
from langchain_mistralai import ChatMistralAI
from langchain_community.chat_models import ChatOllama
from tqdm import tqdm
from .base_node import BaseNode
@ -77,7 +76,7 @@ class GenerateAnswerNode(BaseNode):
doc = input_data[1]
if self.node_config.get("schema", None) is not None:
if isinstance(self.llm_model, (ChatOpenAI, ChatMistralAI)):
if isinstance(self.llm_model, ChatOpenAI):
self.llm_model = self.llm_model.with_structured_output(
schema=self.node_config["schema"]
)